Remove API server

This commit is contained in:
Kumi 2022-09-16 11:53:03 +00:00
parent 7dcd623c03
commit 29827e40e1
Signed by: kumi
GPG key ID: ECBCC9082395383F

View file

@ -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()