from SecureAuthentificateXMLRPCServer import  SecureAuthentificateXMLRPCRequestHandler
from SecureXMLRPCServer import SecureXMLRPCServer
import os, hashlib

class UserDB:
    '''
    class for managing all xmlrpc users
    - each user is indyfied via a hash value, which is created out of the username + password
    '''
    def __init__(self,userClass, userlist,jobqueue):
        self.salt=os.urandom(20)
        self.jobqueue=jobqueue
        self.userClass=userClass
        self.userlist={}
        for user in userlist:
            self.createUser(user)

    def createHash(self,user):
        """
        returns a hash out of username and the password and self.salt
        user is a directory with two keys: username and password
        """
        m=hashlib.sha512()
        m.update(user["name"])
        m.update(self.salt)
        m.update(user["password"])
        return m.hexdigest()

    def createUser(self,user):
        self.userlist[self.createHash(user)]=self.userClass(user["name"],self.jobqueue)

    def __getitem__(self,key):
        return self.userlist[key]

        
class SecureAuthentificateXMLRPCRequestHandler2(SecureAuthentificateXMLRPCRequestHandler):
    def testUser(self,username,password):
        """
        Function for testing authentification
        """
        return self.server.activateUser(username,password)

class SecureUserDBXMLRPCServer(SecureXMLRPCServer):
    def __init__(self, addr, userdb, 
                 requestHandler=SecureAuthentificateXMLRPCRequestHandler2,
                 certificate="server.cert", privatekey="server.pem",
                 logRequests=1):
        SecureXMLRPCServer.__init__(self, addr, requestHandler, certificate, privatekey, logRequests)
        self.relam="xmlrpc"
        self.userdb=userdb

    def activateUser(self,username,password):
        """
        Function is executed, if someone ant to login
        -searches for a regular user in the userdb and then put all methods of the user as XMLRPC
        returns weather a user was found or not
        """
        try:
            user = self.userdb[self.userdb.createHash({"name":username,"password":password})]
            self.register_instance(user)
            return True
        except KeyError:
            return False