from fastapi import FastAPI, Form, Request
from fastapi.responses import HTMLResponse, JSONResponse
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
import httpx
import uuid
import string
import random
import xml.etree.ElementTree as ET

app = FastAPI()

templates = Jinja2Templates(directory="app/templates")

ascii_list = list(string.ascii_letters + string.digits)
DESKTOP_NAME = "DESKTOP-" + "".join(random.sample(ascii_list, 7)).upper()
MACHINE_ID = uuid.uuid4()
USERNAME = "".join(random.sample(ascii_list, 7)).upper()
VERSION = 2024100
BUILD_NUMBER = "2024.1.4+Build+CL-241.18034.45"

# Configure static files
app.mount("/static", StaticFiles(directory="app/static"), name="static")


# Parse xml response
def get_xml_data(server: str, ticket_path: str):
    client = httpx.Client(verify=False)
    try:
        response = client.get(server + ticket_path, timeout=3)  # timeout 3 sec
        response.raise_for_status()  # check status
        root = ET.fromstring(response.text)

        return {
            "action": root.find("action").text,
            "confirmation_stamp": root.find("confirmationStamp").text,
            "lease_signature": root.find("leaseSignature").text,
            "response_code": root.find("responseCode").text,
            "salt": root.find("salt").text,
            "server_lease": root.find("serverLease").text,
            "server_uid": root.find("serverUid").text,
            "validation_deadline_period": root.find("validationDeadlinePeriod").text,
            "validation_period": root.find("validationPeriod").text,
            "message": root.find("message").text,
        }
    except httpx.ReadTimeout:
        return {"error": "Timeout: Please try again later."}
    except Exception as e:
        return {
            "error": f"An unexpected error occurred: {str(e)}. It seems that this server is not avaliable or you can try again later."
        }


@app.get("/", response_class=HTMLResponse)
async def index(request: Request):
    # 提供默认的 release_data 和 obtain_data
    return templates.TemplateResponse(
        "index.html",
        {
            "request": request,
            "release_data": None,  # 默认值
            "obtain_data": None,  # 默认值
            "server": None,  # 默认服务器地址为空
        },
    )


@app.post("/fetch", response_class=HTMLResponse)
async def fetch_data(request: Request, server: str = Form(...)):
    RELEASE_TICKET_PATH = f"/rpc/releaseTicket.action?buildNumber={BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973268812&secure=false&ticketId=rsoaehfbbg&userName={USERNAME}"
    OBTAIN_TICKET_PATH = f"/rpc/obtainTicket.action?buildDate=20240409&{BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973269845&secure=false&userName={USERNAME}&version={VERSION}&versionNumber={VERSION}"

    release_ticket_data = get_xml_data(server, RELEASE_TICKET_PATH)
    obtain_ticket_data = get_xml_data(server, OBTAIN_TICKET_PATH)

    return templates.TemplateResponse(
        "index.html",
        {
            "request": request,
            "release_data": release_ticket_data,
            "obtain_data": obtain_ticket_data,
            "server": server,
        },
    )


@app.get("/api/fetch")
@app.post("/api/fetch")
async def api_fetch(server: str = Form(...)):
    return await handle_fetch(server)


async def handle_fetch(server: str):
    RELEASE_TICKET_PATH = f"/rpc/releaseTicket.action?buildNumber={BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973268812&secure=false&ticketId=rsoaehfbbg&userName={USERNAME}"
    OBTAIN_TICKET_PATH = f"/rpc/obtainTicket.action?buildDate=20240409&{BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973269845&secure=false&userName={USERNAME}&version={VERSION}&versionNumber={VERSION}"

    release_ticket_data = get_xml_data(server, RELEASE_TICKET_PATH)
    obtain_ticket_data = get_xml_data(server, OBTAIN_TICKET_PATH)

    available = False
    if obtain_ticket_data.get("response_code") == "OK":
        available = True
    elif obtain_ticket_data.get("response_code") == "Error":
        available = False

    return JSONResponse(
        content={
            "available": available,
            "message": obtain_ticket_data.get("message", ""),
            "data": {
                "release_ticket_data": release_ticket_data,
                "obtain_ticket_data": obtain_ticket_data,
            },
        }
    )