80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
# Copyright (C) 2020 by Thomas Lindner <tom@dl6tom.de>
|
|
# Copyright (C) 2020 by Cathy Hu <cathy.hu@fau.de>
|
|
#
|
|
# SPDX-License-Identifier: 0BSD
|
|
|
|
""" Entrypoint of Kibicara. """
|
|
|
|
from asyncio import run as asyncio_run
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from hypercorn.config import Config
|
|
from hypercorn.asyncio import serve
|
|
from kibicara.config import args, config
|
|
from kibicara.model import Mapping
|
|
from kibicara.platformapi import Spawner
|
|
from kibicara.webapi import router
|
|
from logging import basicConfig, DEBUG, getLogger, INFO, WARNING
|
|
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
class Main:
|
|
"""Entrypoint for Kibicara.
|
|
|
|
Initializes the platform bots and starts the hypercorn webserver serving the
|
|
Kibicara application and the specified frontend on port 8000.
|
|
"""
|
|
|
|
def __init__(self):
|
|
asyncio_run(self.__run())
|
|
|
|
async def __run(self):
|
|
LOGLEVELS = {
|
|
None: WARNING,
|
|
1: INFO,
|
|
2: DEBUG,
|
|
}
|
|
basicConfig(
|
|
level=LOGLEVELS.get(args.verbose, DEBUG),
|
|
format="%(asctime)s %(name)s %(message)s",
|
|
)
|
|
getLogger('aiosqlite').setLevel(WARNING)
|
|
Mapping.create_all()
|
|
await Spawner.init_all()
|
|
await self.__start_webserver()
|
|
|
|
async def __start_webserver(self):
|
|
class SinglePageApplication(StaticFiles):
|
|
async def get_response(self, path, scope):
|
|
response = await super().get_response(path, scope)
|
|
if response.status_code == 404:
|
|
response = await super().get_response('.', scope)
|
|
return response
|
|
|
|
app = FastAPI()
|
|
server_config = Config()
|
|
server_config.accesslog = '-'
|
|
server_config.behind_proxy = config['behind_proxy']
|
|
if config['production']:
|
|
server_config.bind = ['0.0.0.0:8000', '[::]:8000']
|
|
api = FastAPI()
|
|
api.include_router(router)
|
|
app.mount('/api', api)
|
|
if not config['production'] and config['cors_allow_origin']:
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=config['cors_allow_origin'],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
if config['frontend_path'] is not None:
|
|
app.mount(
|
|
'/',
|
|
app=SinglePageApplication(directory=config['frontend_path'], html=True),
|
|
)
|
|
await serve(app, server_config)
|