Genesis commit

main.py

@@ -0,0 +1,67 @@
+from fastapi.responses import StreamingResponse
+from fastapi import HTTPException
+from mcstatus import JavaServer
+from dotenv import load_dotenv
+from fastapi import FastAPI
+from mcrcon import MCRcon
+from os import environ
+
+from lib import CommandModel, LogsModel, sse_from_iterator
+from handler import ServerHandler
+
+load_dotenv()
+MCSMGR_PASSWORD = environ.get("MCSMGR_PASSWORD", "SuperSecretPassword")
+MINECRAFT_SERVER_COMMAND = environ.get(
+    "MINECRAFT_SERVER_COMMAND", "java -jar fabric.jar nogui")
+MINECRAFT_SERVER_ADDRESS = environ.get("MINECRAFT_SERVER_ADDRESS", "localhost")
+MINECRAFT_SERVER_RCON_PASSWORD = environ.get(
+    "MINECRAFT_SERVER_RCON_PASSWORD", "SuperSecretPassword")
+
+app = FastAPI()
+handler = ServerHandler(MINECRAFT_SERVER_COMMAND.split())
+server = JavaServer(MINECRAFT_SERVER_ADDRESS)
+
+
+@app.get("/status")
+async def status():
+    try:
+        info = server.status()
+    except:
+        return {"online": False}
+
+    return {
+        "online": True,
+        "latency": info.latency,
+        "motd": info.motd,
+        "players": {
+            "list": info.players.sample,
+            "online": info.players.online,
+            "max": info.players.max,
+        },
+    }
+
+
+@app.post("/command")
+async def command(data: CommandModel):
+    if data.password != MCSMGR_PASSWORD:
+        raise HTTPException(403, "Invalid password")
+
+    try:
+        with MCRcon(MINECRAFT_SERVER_ADDRESS, MINECRAFT_SERVER_RCON_PASSWORD) as mcr:
+            return mcr.command(data.cmd)
+    except:
+        raise HTTPException(500, "Unable to reach Server")
+
+
+@app.post("/start")
+async def start():
+    await handler.start()
+
+
+@app.get("/logs")
+async def logs(data: LogsModel):
+    if data.password != MCSMGR_PASSWORD:
+        raise HTTPException(401, "Invalid password")
+
+    generator = sse_from_iterator(handler.reader())
+    return StreamingResponse(generator, media_type="text/event-stream")