from pathlib import Path
from typing import Annotated

import uvicorn
from fastapi import FastAPI, Form
from fastapi.responses import HTMLResponse, PlainTextResponse


app = FastAPI()

SECRET_PATH = Path("./secret/flag.txt")
HTML_PATH = Path("./index.html")


if not SECRET_PATH.exists():
    flag = "konaph{example_flag}"
else:
    flag = SECRET_PATH.read_text()


@app.get("/")
async def _() -> HTMLResponse:
    return HTMLResponse(content=HTML_PATH.read_text())


@app.post("/login")
async def _(username: Annotated[str, Form()], password: Annotated[str, Form()]):
    if username == "im_tsukasa" and password == "ker0r0_k4w4ii":
        return PlainTextResponse(f"登录成功！答案为 {flag}。\n给此方 BOT 私发：提交答案 {flag}")
    return PlainTextResponse("用户名或密码错误！", status_code=401)


if __name__ == "__main__":
    uvicorn.run(
        app=app,
        host="0.0.0.0",
        port=8000,
    )