from fastapi import APIRouter, Depends, File, Form, UploadFile
from sqlalchemy.orm import Session

from app.common.db.session import get_db
from app.common.responses.api_response import success_response
from app.common.security.deps import get_current_user
from app.models.entities import User
from app.modules.assets.service import AssetsService


router = APIRouter(prefix="/api/v1/assets", tags=["assets"])


@router.post("/upload-token")
def create_upload_token(mediaType: str = "image", db: Session = Depends(get_db)):
    return success_response(AssetsService(db).create_upload_token(mediaType))


@router.post("")
def upload_asset(
    file: UploadFile = File(...),
    mediaType: str = Form("image"),
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db),
):
    return success_response(AssetsService(db).save_asset(current_user.id, file, mediaType))


@router.get("")
def list_assets(
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db),
):
    return success_response(AssetsService(db).list_assets(current_user.id))


@router.delete("/{asset_id}")
def delete_asset(
    asset_id: int,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db),
):
    return success_response(AssetsService(db).delete_asset(current_user.id, asset_id))