forked from Hay1tsme/artemis
beerpsi
994fa0d41e
Allows us to extend the logging infrastructure, e.g. by adding Loki/Discord webhooks to the mix.
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
import yaml
|
|
import logging
|
|
from starlette.routing import Route
|
|
from starlette.requests import Request
|
|
from starlette.applications import Starlette
|
|
from starlette.responses import PlainTextResponse
|
|
from os import environ, path, mkdir, W_OK, access
|
|
from typing import List
|
|
|
|
from core import CoreConfig, TitleServlet, MuchaServlet, AllnetServlet, BillingServlet, logger
|
|
from core.frontend import FrontendServlet
|
|
|
|
async def dummy_rt(request: Request):
|
|
return PlainTextResponse("Service OK")
|
|
|
|
cfg_dir = environ.get("ARTEMIS_CFG_DIR", "config")
|
|
cfg: CoreConfig = CoreConfig()
|
|
if path.exists(f"{cfg_dir}/core.yaml"):
|
|
cfg.update(yaml.safe_load(open(f"{cfg_dir}/core.yaml")))
|
|
|
|
if not path.exists(cfg.server.log_dir):
|
|
mkdir(cfg.server.log_dir)
|
|
|
|
if not access(cfg.server.log_dir, W_OK):
|
|
print(
|
|
f"Log directory {cfg.server.log_dir} NOT writable, please check permissions"
|
|
)
|
|
exit(1)
|
|
|
|
logger = logger.create_logger(
|
|
"Core",
|
|
cfg,
|
|
logging.DEBUG if cfg.server.is_develop else logging.INFO,
|
|
)
|
|
|
|
logger.info(f"Artemis starting in {'develop' if cfg.server.is_develop else 'production'} mode")
|
|
|
|
title = TitleServlet(cfg, cfg_dir) # This has to be loaded first to load plugins
|
|
mucha = MuchaServlet(cfg, cfg_dir)
|
|
|
|
route_lst: List[Route] = [
|
|
# Mucha
|
|
Route("/mucha_front/boardauth.do", mucha.handle_boardauth, methods=["POST"]),
|
|
Route("/mucha_front/updatacheck.do", mucha.handle_updatecheck, methods=["POST"]),
|
|
Route("/mucha_front/downloadstate.do", mucha.handle_dlstate, methods=["POST"]),
|
|
# General
|
|
Route("/", dummy_rt),
|
|
Route("/robots.txt", FrontendServlet.robots)
|
|
]
|
|
|
|
if not cfg.billing.standalone:
|
|
billing = BillingServlet(cfg, cfg_dir)
|
|
route_lst += [
|
|
Route("/request", billing.handle_billing_request, methods=["POST"]),
|
|
Route("/request/", billing.handle_billing_request, methods=["POST"]),
|
|
]
|
|
|
|
if not cfg.allnet.standalone:
|
|
allnet = AllnetServlet(cfg, cfg_dir)
|
|
route_lst += [
|
|
Route("/sys/servlet/PowerOn", allnet.handle_poweron, methods=["GET", "POST"]),
|
|
Route("/sys/servlet/DownloadOrder", allnet.handle_dlorder, methods=["GET", "POST"]),
|
|
Route("/sys/servlet/LoaderStateRecorder", allnet.handle_loaderstaterecorder, methods=["GET", "POST"]),
|
|
Route("/sys/servlet/Alive", allnet.handle_alive, methods=["GET", "POST"]),
|
|
Route("/naomitest.html", allnet.handle_naomitest),
|
|
]
|
|
|
|
if cfg.allnet.allow_online_updates:
|
|
route_lst += [
|
|
Route("/report-api/Report", allnet.handle_dlorder_report, methods=["POST"]),
|
|
Route("/dl/ini/{file:str}", allnet.handle_dlorder_ini),
|
|
]
|
|
|
|
for code, game in title.title_registry.items():
|
|
route_lst += game.get_routes()
|
|
|
|
app = Starlette(cfg.server.is_develop, route_lst)
|