| 1 |
f7af2b3 |
Dawsh |
2y ago |
import asyncio |
| 2 |
f7af2b3 |
Dawsh |
2y ago |
import logging |
| 3 |
f7af2b3 |
Dawsh |
2y ago |
from datetime import datetime as dt |
| 4 |
f7af2b3 |
Dawsh |
2y ago |
from datetime import timedelta as td |
| 5 |
f7af2b3 |
Dawsh |
2y ago |
|
| 6 |
f7af2b3 |
Dawsh |
2y ago |
from apscheduler.schedulers.asyncio import AsyncIOScheduler |
| 7 |
f7af2b3 |
Dawsh |
2y ago |
from fastapi import FastAPI, Request, status |
| 8 |
f7af2b3 |
Dawsh |
2y ago |
from fastapi.encoders import jsonable_encoder |
| 9 |
f7af2b3 |
Dawsh |
2y ago |
from fastapi.exceptions import RequestValidationError |
| 10 |
f7af2b3 |
Dawsh |
2y ago |
from fastapi.middleware.cors import CORSMiddleware |
| 11 |
eafb4a4 |
Mrdn |
2y ago |
from fastapi.responses import HTMLResponse, JSONResponse |
| 12 |
d4ec019 |
Dawsh |
2y ago |
from fastapi_pagination import add_pagination |
| 13 |
f7af2b3 |
Dawsh |
2y ago |
from starlette.staticfiles import StaticFiles |
| 14 |
eafb4a4 |
Mrdn |
2y ago |
from uvicorn import Config, Server |
| 15 |
f7af2b3 |
Dawsh |
2y ago |
|
| 16 |
f7af2b3 |
Dawsh |
2y ago |
import config |
| 17 |
f7af2b3 |
Dawsh |
2y ago |
from app.templates import render_template |
| 18 |
01cdf2e |
Dawsh |
2y ago |
from config import ( |
| 19 |
01cdf2e |
Dawsh |
2y ago |
DEBUG, |
| 20 |
01cdf2e |
Dawsh |
2y ago |
DOCS, |
| 21 |
01cdf2e |
Dawsh |
2y ago |
HOME_PAGE_TEMPLATE, |
| 22 |
01cdf2e |
Dawsh |
2y ago |
UVICORN_HOST, |
| 23 |
01cdf2e |
Dawsh |
2y ago |
UVICORN_PORT, |
| 24 |
01cdf2e |
Dawsh |
2y ago |
UVICORN_SSL_CERTFILE, |
| 25 |
01cdf2e |
Dawsh |
2y ago |
UVICORN_SSL_KEYFILE, |
| 26 |
01cdf2e |
Dawsh |
2y ago |
UVICORN_UDS, |
| 27 |
01cdf2e |
Dawsh |
2y ago |
) |
| 28 |
4466347 |
Mrdn |
2y ago |
|
| 29 |
f7af2b3 |
Dawsh |
2y ago |
from . import __version__, telegram |
| 30 |
2de9d72 |
Dawsh |
2y ago |
from .routes import api_router |
| 31 |
01cdf2e |
Dawsh |
2y ago |
from .tasks import ( |
| 32 |
01cdf2e |
Dawsh |
2y ago |
delete_expired_reminders, |
| 33 |
01cdf2e |
Dawsh |
2y ago |
nodes_startup, |
| 34 |
01cdf2e |
Dawsh |
2y ago |
record_realtime_bandwidth, |
| 35 |
01cdf2e |
Dawsh |
2y ago |
record_user_usages, |
| 36 |
01cdf2e |
Dawsh |
2y ago |
reset_user_data_usage, |
| 37 |
01cdf2e |
Dawsh |
2y ago |
review_users, |
| 38 |
01cdf2e |
Dawsh |
2y ago |
send_notifications, |
| 39 |
01cdf2e |
Dawsh |
2y ago |
) |
| 40 |
f7af2b3 |
Dawsh |
2y ago |
|
| 41 |
f7af2b3 |
Dawsh |
2y ago |
logger = logging.getLogger(__name__) |
| 42 |
f7af2b3 |
Dawsh |
2y ago |
dashboard_path = "/dashboard/" |
| 43 |
f7af2b3 |
Dawsh |
2y ago |
|
| 44 |
f7af2b3 |
Dawsh |
2y ago |
app = FastAPI( |
| 45 |
f7af2b3 |
Dawsh |
2y ago |
title="MarzneshinAPI", |
| 46 |
f7af2b3 |
Dawsh |
2y ago |
description="Unified GUI Censorship Resistant Solution Powered by Xray", |
| 47 |
f7af2b3 |
Dawsh |
2y ago |
version=__version__, |
| 48 |
f7af2b3 |
Dawsh |
2y ago |
docs_url="/docs" if DOCS else None, |
| 49 |
f7af2b3 |
Dawsh |
2y ago |
redoc_url="/redoc" if DOCS else None, |
| 50 |
f7af2b3 |
Dawsh |
2y ago |
) |
| 51 |
f7af2b3 |
Dawsh |
2y ago |
|
| 52 |
2de9d72 |
Dawsh |
2y ago |
app.include_router(api_router) |
| 53 |
d4ec019 |
Dawsh |
2y ago |
add_pagination(app) |
| 54 |
f7af2b3 |
Dawsh |
2y ago |
|
| 55 |
f7af2b3 |
Dawsh |
2y ago |
|
| 56 |
f7af2b3 |
Dawsh |
2y ago |
@app.get("/", response_class=HTMLResponse) |
| 57 |
2de9d72 |
Dawsh |
2y ago |
def home_page(): |
| 58 |
f7af2b3 |
Dawsh |
2y ago |
return render_template(HOME_PAGE_TEMPLATE) |
| 59 |
f7af2b3 |
Dawsh |
2y ago |
|
| 60 |
f7af2b3 |
Dawsh |
2y ago |
|
| 61 |
f7af2b3 |
Dawsh |
2y ago |
app.add_middleware( |
| 62 |
f7af2b3 |
Dawsh |
2y ago |
CORSMiddleware, |
| 63 |
f7af2b3 |
Dawsh |
2y ago |
allow_origins=["*"], |
| 64 |
f7af2b3 |
Dawsh |
2y ago |
allow_credentials=True, |
| 65 |
f7af2b3 |
Dawsh |
2y ago |
allow_methods=["*"], |
| 66 |
f7af2b3 |
Dawsh |
2y ago |
allow_headers=["*"], |
| 67 |
f7af2b3 |
Dawsh |
2y ago |
) |
| 68 |
f7af2b3 |
Dawsh |
2y ago |
|
| 69 |
f7af2b3 |
Dawsh |
2y ago |
scheduler = AsyncIOScheduler(timezone="UTC") |
| 70 |
eafb4a4 |
Mrdn |
2y ago |
scheduler.add_job(record_user_usages, "interval", coalesce=True, seconds=30) |
| 71 |
eafb4a4 |
Mrdn |
2y ago |
scheduler.add_job(review_users, "interval", seconds=600, coalesce=True, max_instances=1) |
| 72 |
eafb4a4 |
Mrdn |
2y ago |
scheduler.add_job(reset_user_data_usage, "interval", coalesce=True, hours=1) |
| 73 |
eafb4a4 |
Mrdn |
2y ago |
scheduler.add_job( |
| 74 |
eafb4a4 |
Mrdn |
2y ago |
record_realtime_bandwidth, "interval", seconds=1, coalesce=True, max_instances=1 |
| 75 |
eafb4a4 |
Mrdn |
2y ago |
) |
| 76 |
f7af2b3 |
Dawsh |
2y ago |
|
| 77 |
f7af2b3 |
Dawsh |
2y ago |
if config.WEBHOOK_ADDRESS: |
| 78 |
f7af2b3 |
Dawsh |
2y ago |
scheduler.add_job(send_notifications, "interval", seconds=30, replace_existing=True) |
| 79 |
eafb4a4 |
Mrdn |
2y ago |
scheduler.add_job( |
| 80 |
eafb4a4 |
Mrdn |
2y ago |
delete_expired_reminders, |
| 81 |
eafb4a4 |
Mrdn |
2y ago |
"interval", |
| 82 |
eafb4a4 |
Mrdn |
2y ago |
hours=2, |
| 83 |
eafb4a4 |
Mrdn |
2y ago |
start_date=dt.utcnow() + td(minutes=1), |
| 84 |
eafb4a4 |
Mrdn |
2y ago |
) |
| 85 |
f7af2b3 |
Dawsh |
2y ago |
|
| 86 |
f7af2b3 |
Dawsh |
2y ago |
|
| 87 |
f6a32ea |
Dawsh |
2y ago |
@app.on_event("startup") |
| 88 |
f6a32ea |
Dawsh |
2y ago |
async def on_start(): |
| 89 |
f6a32ea |
Dawsh |
2y ago |
await nodes_startup() |
| 90 |
f6a32ea |
Dawsh |
2y ago |
|
| 91 |
f6a32ea |
Dawsh |
2y ago |
|
| 92 |
f7af2b3 |
Dawsh |
2y ago |
@app.on_event("shutdown") |
| 93 |
f7af2b3 |
Dawsh |
2y ago |
async def on_shutdown(): |
| 94 |
f7af2b3 |
Dawsh |
2y ago |
scheduler.shutdown() |
| 95 |
f7af2b3 |
Dawsh |
2y ago |
logger.info("Sending pending notifications before shutdown...") |
| 96 |
f7af2b3 |
Dawsh |
2y ago |
await send_notifications() |
| 97 |
f7af2b3 |
Dawsh |
2y ago |
|
| 98 |
f7af2b3 |
Dawsh |
2y ago |
|
| 99 |
f7af2b3 |
Dawsh |
2y ago |
@app.exception_handler(RequestValidationError) |
| 100 |
f7af2b3 |
Dawsh |
2y ago |
def validation_exception_handler(request: Request, exc: RequestValidationError): |
| 101 |
f7af2b3 |
Dawsh |
2y ago |
details = {} |
| 102 |
f7af2b3 |
Dawsh |
2y ago |
for error in exc.errors(): |
| 103 |
f7af2b3 |
Dawsh |
2y ago |
details[error["loc"][-1]] = error.get("msg") |
| 104 |
f7af2b3 |
Dawsh |
2y ago |
return JSONResponse( |
| 105 |
f7af2b3 |
Dawsh |
2y ago |
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, |
| 106 |
f7af2b3 |
Dawsh |
2y ago |
content=jsonable_encoder({"detail": details}), |
| 107 |
f7af2b3 |
Dawsh |
2y ago |
) |
| 108 |
f7af2b3 |
Dawsh |
2y ago |
|
| 109 |
f7af2b3 |
Dawsh |
2y ago |
|
| 110 |
f7af2b3 |
Dawsh |
2y ago |
async def main(): |
| 111 |
eafb4a4 |
Mrdn |
2y ago |
if not DEBUG: |
| 112 |
f7af2b3 |
Dawsh |
2y ago |
app.mount( |
| 113 |
f7af2b3 |
Dawsh |
2y ago |
"/dashboard/", |
| 114 |
4466347 |
Mrdn |
2y ago |
StaticFiles(directory="dashboard/dist", html=True), |
| 115 |
4466347 |
Mrdn |
2y ago |
name="dashboard", |
| 116 |
f7af2b3 |
Dawsh |
2y ago |
) |
| 117 |
f7af2b3 |
Dawsh |
2y ago |
scheduler.start() |
| 118 |
f7af2b3 |
Dawsh |
2y ago |
asyncio.create_task(telegram.start_bot()) |
| 119 |
f7af2b3 |
Dawsh |
2y ago |
cfg = Config( |
| 120 |
f7af2b3 |
Dawsh |
2y ago |
app=app, |
| 121 |
f7af2b3 |
Dawsh |
2y ago |
host=UVICORN_HOST, |
| 122 |
f7af2b3 |
Dawsh |
2y ago |
port=UVICORN_PORT, |
| 123 |
f7af2b3 |
Dawsh |
2y ago |
uds=(None if DEBUG else UVICORN_UDS), |
| 124 |
f7af2b3 |
Dawsh |
2y ago |
ssl_certfile=UVICORN_SSL_CERTFILE, |
| 125 |
f7af2b3 |
Dawsh |
2y ago |
ssl_keyfile=UVICORN_SSL_KEYFILE, |
| 126 |
f7af2b3 |
Dawsh |
2y ago |
workers=1, |
| 127 |
f7af2b3 |
Dawsh |
2y ago |
reload=DEBUG, |
| 128 |
f7af2b3 |
Dawsh |
2y ago |
log_level=logging.DEBUG if DEBUG else logging.INFO, |
| 129 |
f7af2b3 |
Dawsh |
2y ago |
) |
| 130 |
f7af2b3 |
Dawsh |
2y ago |
server = Server(cfg) |
| 131 |
f7af2b3 |
Dawsh |
2y ago |
await server.serve() |