Remove API server
This commit is contained in:
parent
7dcd623c03
commit
29827e40e1
GPG key ID:
ECBCC9082395383F
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