import os
from fastapi import FastAPI, Header, HTTPException
from fastapi.middleware.cors import CORSMiddleware
import boto3

# -----------------------------
# CONFIG
# -----------------------------
DO_SPACES_REGION = "blr1"
DO_SPACES_BUCKET = "milestone"
DO_ACCESS_KEY = os.getenv("DO_SPACES_KEY")
DO_SECRET_KEY = os.getenv("DO_SPACES_SECRET")

BASE_URL = f"https://{DO_SPACES_BUCKET}.{DO_SPACES_REGION}.digitaloceanspaces.com"
FOLDER_PREFIX = "bikini-theme/target/"

AUTH_TOKEN = "logicgo_admin@123"

# -----------------------------
# FASTAPI INIT
# -----------------------------
app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# -----------------------------
# SPACES CLIENT
# -----------------------------
def get_spaces_client():
    endpoint = f"{DO_SPACES_REGION}.digitaloceanspaces.com"
    return boto3.client(
        "s3",
        region_name=DO_SPACES_REGION,
        endpoint_url=f"https://{endpoint}",
        aws_access_key_id=DO_ACCESS_KEY,
        aws_secret_access_key=DO_SECRET_KEY,
    )

# -----------------------------
# AUTH MIDDLEWARE
# -----------------------------
def check_auth(authorization: str = Header(None)):
    if not authorization:
        raise HTTPException(status_code=401, detail="Missing Authorization header")

    token = authorization.replace("Bearer ", "").strip()
    if token != AUTH_TOKEN:
        raise HTTPException(status_code=403, detail="Invalid authorization token")


# -----------------------------
# PROTECTED ROUTE
# -----------------------------
@app.get("/bikini-theme")
def list_bikini_theme_images(authorization: str = Header(None)):
    check_auth(authorization)

    client = get_spaces_client()

    # List objects
    resp = client.list_objects_v2(
        Bucket=DO_SPACES_BUCKET,
        Prefix=FOLDER_PREFIX
    )

    contents = resp.get("Contents", [])

    image_files = [
        obj["Key"]
        for obj in contents
        if obj["Key"].lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
    ]

    images = {}
    for file_key in image_files:
        file_name = file_key.split("/")[-1]
        name_without_ext = file_name.rsplit(".", 1)[0]

        images[name_without_ext] = f"{BASE_URL}/{file_key}"

    return {"images": images}


@app.get("/")
def home():
    return {"message": "FastAPI DO Spaces bikini-theme API running on HuggingFace"}