import uvicorn
from uvicorn.config import LOGGING_CONFIG
from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse, FileResponse
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
from pydantic import BaseModel
import logging

from .core import APIProcess
from .conf import settings, DATA_PATH, LOG_PATH
from .utils import json_config

logger = logging.getLogger(__name__)

app = FastAPI()
api_func = APIProcess()

app.mount("/assets", StaticFiles(directory="templates/assets"), name="assets")
templates = Jinja2Templates(directory="templates")


@app.get("/", response_class=HTMLResponse)
def index(request: Request):
    context = {"request": request}
    return templates.TemplateResponse("index.html", context)


@app.get("/api/v1/data")
def get_data():
    data = json_config.load(DATA_PATH)
    return data


@app.get("/api/v1/log")
async def get_log():
    return FileResponse(LOG_PATH)


@app.get("/api/v1/resetRule")
def reset_rule():
    return api_func.reset_rule()


@app.get("api/v1/removeRule/{bangumi_title}")
def remove_rule(bangumi_title: str):
    return api_func.remove_rule(bangumi_title)


class RssLink(BaseModel):
    rss_link: str


@app.post("/api/v1/collection")
async def collection(link: RssLink):
    return api_func.download_collection(link.rss_link)


@app.post("/api/v1/subscribe")
async def subscribe(link: RssLink):
    return api_func.add_subscribe(link.rss_link)


class AddRule(BaseModel):
    title: str
    season: int


@app.post("/api/v1/addRule")
async def add_rule(info: AddRule):
    return api_func.add_rule(info.title, info.season)


def run():
    LOGGING_CONFIG["formatters"]["default"]["fmt"] = "[%(asctime)s] %(levelprefix)s %(message)s"
    uvicorn.run(app, host="0.0.0.0", port=settings.program.webui_port)


if __name__ == "__main__":
    run()