from fastapi import FastAPI, HTTPException, Request
from fastapi.middleware.cors import CORSMiddleware
import os

app = FastAPI(
    title="Token Extractor API",
    version="1.0.0",
    description="Extract oauth_id_token and oauth_access_token from cookies.",
)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # You may restrict this to certain domains
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


@app.get(
    "/tokens",
    summary="Extract oauth tokens from cookies",
    description="Parse cookies and return oauth_id_token and oauth_access_token.",
)
async def get_oauth_tokens(request: Request):
    cookies = request.cookies
    oauth_id_token = cookies.get("oauth_id_token")

    if oauth_id_token is None:
        raise HTTPException(
            status_code=401,
            detail="Missing oauth_id_token cookie",
        )

    return {
        "oauth_id_token": oauth_id_token,
    }