"""Media upload — FastAPI only, stored on S3 (SeaweedFS)."""
from __future__ import annotations

from fastapi import APIRouter, File, Header, HTTPException, UploadFile

from app.services.auth import get_user_by_token
from app.services import s3_storage
from app.services.pb_client import pb

router = APIRouter(tags=["media"])

MAX_IMAGE = 10 * 1024 * 1024
MAX_VIDEO = 200 * 1024 * 1024


def _user(authorization: str | None) -> dict:
  if not authorization or not authorization.startswith("Bearer "):
    raise HTTPException(401, "未登录")
  user = get_user_by_token(authorization[7:])
  if not user:
    raise HTTPException(401, "未登录")
  return user


@router.post("/upload-media")
async def upload_media(
  file: UploadFile = File(...),
  purpose: str = "general",
  authorization: str | None = Header(None),
):
  user = _user(authorization)
  ctype = (file.content_type or "").lower()
  max_size = MAX_VIDEO if ctype.startswith("video/") else MAX_IMAGE
  result = await s3_storage.upload_file(file, purpose=purpose, max_size=max_size)

  if pb.health_ok():
    try:
      pb.create_record(
        "media_assets",
        {
          "userId": user["id"],
          "url": result.url,
          "filename": file.filename or "",
          "contentType": result.content_type,
          "size": result.size,
          "objectKey": result.object_key,
        },
      )
    except Exception:
      pass

  return {
    "success": True,
    "url": result.url,
    "object_key": result.object_key,
    "size": result.size,
    "content_type": result.content_type,
  }