forked from Hay1tsme/artemis
181 lines
7.1 KiB
Python
181 lines
7.1 KiB
Python
import logging, coloredlogs
|
|
from typing import Any, Dict, List
|
|
from twisted.web import resource
|
|
from twisted.web.util import redirectTo
|
|
from twisted.web.http import Request
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
import jinja2
|
|
import bcrypt
|
|
|
|
from core.config import CoreConfig
|
|
from core.data import Data
|
|
from core.utils import Utils
|
|
|
|
class FrontendServlet(resource.Resource):
|
|
def getChild(self, name: bytes, request: Request):
|
|
self.logger.debug(f"{request.getClientIP()} -> {name.decode()}")
|
|
if name == b'':
|
|
return self
|
|
return resource.Resource.getChild(self, name, request)
|
|
|
|
def __init__(self, cfg: CoreConfig, config_dir: str) -> None:
|
|
self.config = cfg
|
|
log_fmt_str = "[%(asctime)s] Frontend | %(levelname)s | %(message)s"
|
|
log_fmt = logging.Formatter(log_fmt_str)
|
|
self.logger = logging.getLogger("frontend")
|
|
self.environment = jinja2.Environment(loader=jinja2.FileSystemLoader("."))
|
|
self.game_list: List[Dict[str, str]] = []
|
|
self.children: Dict[str, Any] = {}
|
|
|
|
fileHandler = TimedRotatingFileHandler("{0}/{1}.log".format(self.config.server.log_dir, "frontend"), when="d", backupCount=10)
|
|
fileHandler.setFormatter(log_fmt)
|
|
|
|
consoleHandler = logging.StreamHandler()
|
|
consoleHandler.setFormatter(log_fmt)
|
|
|
|
self.logger.addHandler(fileHandler)
|
|
self.logger.addHandler(consoleHandler)
|
|
|
|
self.logger.setLevel(cfg.frontend.loglevel)
|
|
coloredlogs.install(level=cfg.frontend.loglevel, logger=self.logger, fmt=log_fmt_str)
|
|
|
|
fe_game = FE_Game(cfg, self.environment)
|
|
games = Utils.get_all_titles()
|
|
for game_dir, game_mod in games.items():
|
|
if hasattr(game_mod, "frontend"):
|
|
try:
|
|
game_fe = game_mod.frontend(cfg, self.environment, config_dir)
|
|
self.game_list.append({"url": game_dir, "name": game_fe.nav_name})
|
|
fe_game.putChild(game_dir.encode(), game_fe)
|
|
except:
|
|
raise
|
|
|
|
self.environment.globals["game_list"] = self.game_list
|
|
self.putChild(b"gate", FE_Gate(cfg, self.environment))
|
|
self.putChild(b"user", FE_User(cfg, self.environment))
|
|
self.putChild(b"game", fe_game)
|
|
|
|
self.logger.info(f"Ready on port {self.config.frontend.port} serving {len(fe_game.children)} games")
|
|
|
|
def render_GET(self, request):
|
|
self.logger.debug(f"{request.getClientIP()} -> {request.uri.decode()}")
|
|
template = self.environment.get_template("core/frontend/index.jinja")
|
|
return template.render(server_name=self.config.server.name, title=self.config.server.name, game_list=self.game_list).encode("utf-16")
|
|
|
|
class FE_Base(resource.Resource):
|
|
"""
|
|
A Generic skeleton class that all frontend handlers should inherit from
|
|
Initializes the environment, data, logger, config, and sets isLeaf to true
|
|
It is expected that game implementations of this class overwrite many of these
|
|
"""
|
|
isLeaf = True
|
|
def __init__(self, cfg: CoreConfig, environment: jinja2.Environment) -> None:
|
|
self.core_config = cfg
|
|
self.data = Data(cfg)
|
|
self.logger = logging.getLogger('frontend')
|
|
self.environment = environment
|
|
self.nav_name = "nav_name"
|
|
|
|
class FE_Gate(FE_Base):
|
|
def render_GET(self, request: Request):
|
|
self.logger.debug(f"{request.getClientIP()} -> {request.uri.decode()}")
|
|
uri: str = request.uri.decode()
|
|
if uri.startswith("/gate/create"):
|
|
return self.create_user(request)
|
|
|
|
if b'e' in request.args:
|
|
try:
|
|
err = int(request.args[b'e'][0].decode())
|
|
except:
|
|
err = 0
|
|
|
|
else: err = 0
|
|
|
|
template = self.environment.get_template("core/frontend/gate/gate.jinja")
|
|
return template.render(title=f"{self.core_config.server.name} | Login Gate", error=err).encode("utf-16")
|
|
|
|
def render_POST(self, request: Request):
|
|
uri = request.uri.decode()
|
|
ip = request.getClientAddress().host
|
|
|
|
if uri == "/gate/gate.login":
|
|
access_code: str = request.args[b"access_code"][0].decode()
|
|
passwd: str = request.args[b"passwd"][0]
|
|
if passwd == b"":
|
|
passwd = None
|
|
|
|
uid = self.data.card.get_user_id_from_card(access_code)
|
|
if uid is None:
|
|
return redirectTo(b"/gate?e=1", request)
|
|
|
|
if passwd is None:
|
|
sesh = self.data.user.login(uid, ip=ip)
|
|
|
|
if sesh is not None:
|
|
return redirectTo(f"/gate/create?ac={access_code}".encode(), request)
|
|
return redirectTo(b"/gate?e=1", request)
|
|
|
|
salt = bcrypt.gensalt()
|
|
hashed = bcrypt.hashpw(passwd, salt)
|
|
sesh = self.data.user.login(uid, hashed, ip)
|
|
|
|
if sesh is None:
|
|
return redirectTo(b"/gate?e=1", request)
|
|
|
|
request.addCookie('session', sesh)
|
|
return redirectTo(b"/user", request)
|
|
|
|
elif uri == "/gate/gate.create":
|
|
access_code: str = request.args[b"access_code"][0].decode()
|
|
username: str = request.args[b"username"][0]
|
|
email: str = request.args[b"email"][0].decode()
|
|
passwd: str = request.args[b"passwd"][0]
|
|
|
|
uid = self.data.card.get_user_id_from_card(access_code)
|
|
if uid is None:
|
|
return redirectTo(b"/gate?e=1", request)
|
|
|
|
salt = bcrypt.gensalt()
|
|
hashed = bcrypt.hashpw(passwd, salt)
|
|
|
|
result = self.data.user.create_user(uid, username, email, hashed.decode(), 1)
|
|
if result is None:
|
|
return redirectTo(b"/gate?e=3", request)
|
|
|
|
sesh = self.data.user.login(uid, hashed, ip)
|
|
if sesh is None:
|
|
return redirectTo(b"/gate", request)
|
|
request.addCookie('session', sesh)
|
|
|
|
return redirectTo(b"/user", request)
|
|
|
|
else:
|
|
return b""
|
|
|
|
def create_user(self, request: Request):
|
|
if b'ac' not in request.args or len(request.args[b'ac'][0].decode()) != 20:
|
|
return redirectTo(b"/gate?e=2", request)
|
|
|
|
ac = request.args[b'ac'][0].decode()
|
|
|
|
template = self.environment.get_template("core/frontend/gate/create.jinja")
|
|
return template.render(title=f"{self.core_config.server.name} | Create User", code=ac).encode("utf-16")
|
|
|
|
class FE_User(FE_Base):
|
|
def render_GET(self, request: Request):
|
|
template = self.environment.get_template("core/frontend/user/index.jinja")
|
|
return template.render().encode("utf-16")
|
|
if b'session' not in request.cookies:
|
|
return redirectTo(b"/gate", request)
|
|
|
|
class FE_Game(FE_Base):
|
|
isLeaf = False
|
|
children: Dict[str, Any] = {}
|
|
|
|
def getChild(self, name: bytes, request: Request):
|
|
if name == b'':
|
|
return self
|
|
return resource.Resource.getChild(self, name, request)
|
|
|
|
def render_GET(self, request: Request) -> bytes:
|
|
return redirectTo(b"/user", request) |