|
|
from fastapi import FastAPI, Depends
|
|
|
from starlette.requests import Request
|
|
|
|
|
|
from .server_data import ServerData
|
|
|
from .database.db import database
|
|
|
|
|
|
# from .utils.dependencies import right_checkers, get_current_user
|
|
|
from .utils.dependencies import get_current_user
|
|
|
from .utils.responses import (
|
|
|
ResponseStructure,
|
|
|
get_base_url,
|
|
|
validate_response,
|
|
|
)
|
|
|
from .utils.users import check_user_rights
|
|
|
|
|
|
from .schemas.responses import GetRootResponse
|
|
|
from .schemas.users import User
|
|
|
|
|
|
from .routers.users import router as users_router
|
|
|
from .routers.commands import router as commands_router
|
|
|
|
|
|
|
|
|
# TODO
|
|
|
# 1. Добавить список отклоняемых токенов и их обработку.
|
|
|
# 2. Добавить предварительный вариант генерации access_tokens.
|
|
|
# 3. Продумать, наконец, концепцию cid и wid, в ключе их применения при
|
|
|
# настройке команд и запуске воркеров.
|
|
|
# 4. Решить сколько команд может запускать пользователь единовременно.
|
|
|
# 5. Добавить способ указания принадлежности воркеров и команд пользователям.
|
|
|
# 6. Разобраться с местом раcположения базы данных.
|
|
|
|
|
|
|
|
|
data = ServerData()
|
|
|
app = FastAPI()
|
|
|
|
|
|
|
|
|
@app.on_event("startup")
|
|
|
async def startup():
|
|
|
await database.connect()
|
|
|
|
|
|
|
|
|
@app.on_event("shutdown")
|
|
|
async def shutdown():
|
|
|
await database.disconnect()
|
|
|
|
|
|
|
|
|
@app.get("/", response_model=GetRootResponse, tags=["Home"])
|
|
|
async def get_primary_commands(request: Request,
|
|
|
user: User = Depends(get_current_user)):
|
|
|
check_user_rights(user, "read")
|
|
|
|
|
|
resp = ResponseStructure(get_base_url(request))
|
|
|
resp.add_data(name="Calculate Server", version="0.1")
|
|
|
resp.add_link("commands", "/commands?gui={is_gui}", templated=True)
|
|
|
resp.add_link("find_command", "/commands/{console_command}?gui={is_gui}",
|
|
|
templated=True)
|
|
|
resp.add_link("workers", "/workers")
|
|
|
|
|
|
return validate_response(resp.get_dict(), GetRootResponse,
|
|
|
media_type="application/hal+json")
|
|
|
|
|
|
|
|
|
# @app.get("/workers/{wid}", tags=["Workers management"],
|
|
|
# dependencies=[Depends(right_checkers["write"])])
|
|
|
# async def get_worker(wid: int):
|
|
|
# '''Тестовый обработчик.'''
|
|
|
# worker = data.get_worker_object(wid=wid)
|
|
|
# worker.run()
|
|
|
# print(f"worker: {type(worker)} object {worker}")
|
|
|
|
|
|
# await worker.send({"text": "INFO"})
|
|
|
# worker_data = await worker.get()
|
|
|
|
|
|
# if worker_data['type'] == 'log':
|
|
|
# data.log_message[data['level']](data['msg'])
|
|
|
# return worker_data
|
|
|
|
|
|
# Authentification and users management.
|
|
|
app.include_router(users_router)
|
|
|
|
|
|
# Commands running and management.
|
|
|
app.include_router(commands_router)
|
