server/server/server.py

#!/usr/bin/env python3
from authentication import Authetication
from executor import Executor
from netwrapper import NetWrapper


class Server:

    def __init__(self, homeDirectory: str = "", sessionTimeout: int = 120, availableServer: bool = True):
        self.isAuthenticated = False
        self.homeDirectory = homeDirectory
        self.sessionTimeout = sessionTimeout
        self.availableServer = availableServer
        self.executor = Executor(homeDirectory)
        self.auth=Authetication()
        self.networkInstance = NetWrapper(self.auth.loadUserPublicKeys(),self.auth.loadServerPrivateKey(),self)

    def login(self, username: str, password: str) -> bool:
        self.isAuthenticated = True
        home_directory = self.auth.login(username, password)
        self.executor.baseDir=Executor(home_directory)
        if not home_directory:
            return False
        else:
            return True

    def logout(self) -> None:
        self.isAuthenticated = False
        self.availableServer = False
        self.homeDirectory = ""
        self.executor.baseDir=Executor(self.homeDirectory)

    def parseCommand(self, command: str) -> str:
        if command == "LINOK":
            return "LINOK"
        elif command == "LINERROR":
            return "LINERROR"
        parsedCommand = command.split(" ")
        if len(parsedCommand) > 3:
            return "ERROR"
        # TODO: Ez igy gecironda, ezt meg kene csinalni rendesen.
        elif len(parsedCommand) == 1:
            return self.execute(parsedCommand[0])
        elif len(parsedCommand) == 2:
            return self.execute(parsedCommand[0], parsedCommand[1])
        elif len(parsedCommand) == 3:
            return self.execute(parsedCommand[0], parsedCommand[1], parsedCommand[2])

    def execute(self, command: str, firstParam: str = "", secondParam: str = "") -> str:
        if self.homeDirectory == "" or self.executor.currentDirectory == "" or self.executor.baseDir == "":
            raise Exception("Directories must not be empty string. Did the user log in?")
        if command == "MKD":
            if secondParam != "":
                return "ERROR"
            else:
                return self.executor.createDirectory(firstParam)
        elif command == "RMD":
            if secondParam != "":
                return "ERROR"
            else:
                return self.executor.removeDirectory(firstParam)
        elif command == "GWD":
            if secondParam != "" or firstParam != "":
                return "ERROR"
            else:
                return self.executor.getCurrentDirectory()
        elif command == "CWD":
            if secondParam != "":
                return "ERROR"
            else:
                return self.executor.setCurrentDirectory(firstParam)
        elif command == "LST":
            if secondParam != "" or firstParam != "":
                return "ERROR"
            else:
                return self.executor.listCurrentDirectoryContent()
        elif command == "RMF":
            if secondParam != "":
                return "ERROR"
            else:
                return self.executor.removeFileInCurrentDirectory(firstParam)
        elif command == "UPL":
            if secondParam != "":
                return "ERROR"
            else:
                # TODO
                # Megkapod a filenevet argumentumneknt
                # Fogadni kell egy uzenetet, ami a fajl tartalma
                # Fogadni kell egy uzenetet, ami "EOF"
                # Mindig minden legyen UTF-8-kent kodolva, kiveve a falj, az marard
                # Ha sikeres, OK kuldese, kulonben ERROR kuldese
                pass
        elif command == "DNL":
            if secondParam != "":
                return "ERROR"
            else:
                # TODO
                # Megkapod a filenevet arguemntumkent
                # Kuldeni kell egy uzenetet, ami a falj tartalma
                # Kuldeni kell egy uzenetet, ami EOF
                # Mindig minden legyen UTF-8-kent kodolva, kiveve a falj, az marard
                # Ha sikeres, OK valaszt megkapod, kulonben ERROR valaszt megkapod
                pass
        else:
            return "ERROR"

    def startServer(self):
        while True:
            message = self.networkInstance.recieveMessage().decode('UTF-8')
            self.parseCommand(message)