Remove API server
This commit is contained in:
parent
7dcd623c03
commit
29827e40e1
1 changed files with 0 additions and 143 deletions
|
@ -1,143 +0,0 @@
|
|||
import socketserver
|
||||
import socket
|
||||
import threading
|
||||
import time
|
||||
import pathlib
|
||||
import json
|
||||
|
||||
from .classes.config import MonsterConfig
|
||||
|
||||
|
||||
class ReportMonsterHandler(socketserver.BaseRequestHandler):
|
||||
def requires_login(func, *args, **kwargs) :
|
||||
def inner(self, *args, **kwargs) :
|
||||
if not self.user:
|
||||
return "95 Authentication Required: use login to authenticate first"
|
||||
return func(self, *args, **kwargs)
|
||||
return inner
|
||||
|
||||
def command_login(self, *args):
|
||||
if not len(args) == 2:
|
||||
return "90 Invalid Argument: login takes exactly two arguments"
|
||||
|
||||
for cuser in self.server.config.users:
|
||||
if cuser.username == args[0]:
|
||||
if cuser.validatePasword(args[1]):
|
||||
self.user = cuser
|
||||
return f"10 Login Successful: logged in as {self.user.username}"
|
||||
|
||||
return "99 Authentication Failed: username or password incorrect"
|
||||
|
||||
def command_whoami(self, *args):
|
||||
if args:
|
||||
return "90 Invalid Argument: whoami takes no arguments"
|
||||
if self.user:
|
||||
return "11 Logged In: logged in as {self.user.username}"
|
||||
return "12 Not Logged In"
|
||||
|
||||
@requires_login
|
||||
def command_vessels(self, *args):
|
||||
return "20 OK: " + json.dumps([vessel.name for vessel in self.server.config.vessels])
|
||||
|
||||
@requires_login
|
||||
def command_users(self, *args):
|
||||
if not len(args) == 1:
|
||||
return "90 Invalid Argument: users takes exactly one argument"
|
||||
|
||||
vessels = list(filter(lambda v: v.name == args[0], self.server.config.vessels))
|
||||
|
||||
if not vessels:
|
||||
return f"92 No Such Object: no vessel called {args[0]} found"
|
||||
if len(vessels) > 1:
|
||||
return f"98 Configuration Error: multiple vessels called {args[0]} found"
|
||||
|
||||
vessel = vessels[0]
|
||||
users = vessel.getUsers()
|
||||
|
||||
result = [user for user in users.values()]
|
||||
|
||||
return "20 OK: " + json.dumps(result)
|
||||
|
||||
@requires_login
|
||||
def command_user(self, *args):
|
||||
if not len(args) == 2:
|
||||
return "90 Invalid Argument: user takes exactly two arguments"
|
||||
|
||||
vessels = list(filter(lambda v: v.name == args[0], self.server.config.vessels))
|
||||
|
||||
if not vessels:
|
||||
return f"92 No Such Object: no vessel called {args[0]} found"
|
||||
if len(vessels) > 1:
|
||||
return f"98 Configuration Error: multiple vessels called {args[0]} found"
|
||||
|
||||
vessel = vessels[0]
|
||||
users = vessel.getUsers(username = args[1])
|
||||
|
||||
if not users:
|
||||
return f"92 No Such Object: no user called {args[1]} found on vessel {args[0]}"
|
||||
if len(users) > 1:
|
||||
return f"98 Configuration Error: multiple users called {args[1]} found on vessel {args[0]}"
|
||||
|
||||
return "20 OK: " + json.dumps(list(users.values())[0])
|
||||
|
||||
def handle(self):
|
||||
self.user = False
|
||||
|
||||
self.request.sendall("ReportMonster Server\n(c) Kumi Systems e.U. (https://kumi.systems/)\n\n> ".encode())
|
||||
|
||||
while True:
|
||||
data = self.request.recv(1024)
|
||||
if data:
|
||||
plain = data.decode()
|
||||
if not (parts := plain.split()):
|
||||
response = "00 No Command Received"
|
||||
|
||||
else:
|
||||
parts = plain.split()
|
||||
|
||||
command = parts[0]
|
||||
args = parts[1:]
|
||||
|
||||
COMMANDS = {
|
||||
"login": self.command_login,
|
||||
"user": self.command_user,
|
||||
"users": self.command_users,
|
||||
"vessels": self.command_vessels,
|
||||
"w": self.command_whoami,
|
||||
"whoami": self.command_whoami,
|
||||
}
|
||||
|
||||
if command in COMMANDS.keys():
|
||||
response = COMMANDS[command](*args)
|
||||
else:
|
||||
response = f"91 Invalid Command: don't know command {command}"
|
||||
|
||||
self.request.sendall(response.encode())
|
||||
self.request.sendall("\n> ".encode())
|
||||
|
||||
|
||||
class ReportMonsterServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
||||
allow_reuse_address = True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
HOST = "127.0.0.1"
|
||||
PORT = 6789
|
||||
|
||||
CONFIG = MonsterConfig(pathlib.Path(__file__).resolve().parent / "settings.ini")
|
||||
|
||||
server = ReportMonsterServer((HOST, PORT), ReportMonsterHandler)
|
||||
server.config = CONFIG
|
||||
|
||||
with server:
|
||||
st = threading.Thread(target=server.serve_forever)
|
||||
st.daemon = True
|
||||
st.start()
|
||||
|
||||
try:
|
||||
while st.is_alive():
|
||||
time.sleep(1)
|
||||
except KeyboardInterrupt:
|
||||
print("Received Ctrl+C - stopping server...")
|
||||
server.shutdown()
|
||||
st.join()
|
Loading…
Reference in a new issue