80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import asyncio
|
|
from fastapi import FastAPI
|
|
from starlette.staticfiles import StaticFiles
|
|
|
|
from config import CONFIGS # Loads the configuration from .env
|
|
from middleware import AuthMiddleware # Defines authentication middleware
|
|
from middleware import AfterRequestMiddleware # Defines after request middleware
|
|
from dependency import get_session_manager # Defines dependencies across routers
|
|
from openapi import setup_openapi_schema # Adds authorization header to openapi schema
|
|
from exception_handler import setup_exception_handler # Defines exception handlers
|
|
|
|
# Append directory of cli_api.py to be able to import it
|
|
HYSTERIA_CORE_DIR = '/etc/hysteria/core/'
|
|
sys.path.append(HYSTERIA_CORE_DIR)
|
|
|
|
import routers # noqa: This import should be after the sys.path modification, because it imports cli_api
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
'''
|
|
Create FastAPI app.
|
|
'''
|
|
|
|
# Set up FastAPI
|
|
app = FastAPI(
|
|
title='Hysteria Webpanel',
|
|
description='Webpanel for Hysteria',
|
|
version='0.1.0',
|
|
contact={
|
|
'github': 'https://github.com/ReturnFI/Hysteria2'
|
|
},
|
|
debug=CONFIGS.DEBUG,
|
|
root_path=f'/{CONFIGS.ROOT_PATH}',
|
|
)
|
|
|
|
# Set up static files
|
|
app.mount('/assets', StaticFiles(directory='assets'), name='assets')
|
|
|
|
# Set up exception handlers
|
|
setup_exception_handler(app)
|
|
|
|
# Set up authentication middleware
|
|
|
|
app.add_middleware(AuthMiddleware, session_manager=get_session_manager(), api_token=CONFIGS.API_TOKEN)
|
|
# Set up after request middleware
|
|
app.add_middleware(AfterRequestMiddleware)
|
|
|
|
# Set up Routers
|
|
app.include_router(routers.basic.router, prefix='', tags=['Basic Routes[Web]']) # Add basic router
|
|
app.include_router(routers.login.router, prefix='', tags=['Authentication[Web]']) # Add authentication router
|
|
app.include_router(routers.user.router, prefix='/users', tags=['User Management[Web]']) # Add user router
|
|
app.include_router(routers.api.v1.api_v1_router, prefix='/api/v1', tags=['API Version 1']) # Add API version 1 router # type: ignore
|
|
|
|
# Document that the API requires an API key
|
|
setup_openapi_schema(app)
|
|
|
|
return app
|
|
|
|
|
|
app: FastAPI = create_app()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from hypercorn.config import Config
|
|
from hypercorn.asyncio import serve # type: ignore
|
|
from hypercorn.middleware import ProxyFixMiddleware
|
|
|
|
config = Config()
|
|
config.debug = CONFIGS.DEBUG
|
|
config.bind = ['127.0.0.1:8080']
|
|
config.accesslog = '-'
|
|
config.errorlog = '-'
|
|
|
|
# Fix proxy headers
|
|
app = ProxyFixMiddleware(app, 'legacy') # type: ignore
|
|
asyncio.run(serve(app, config)) # type: ignore
|