119 lines
3.5 KiB
Python
119 lines
3.5 KiB
Python
from fastapi import APIRouter, Depends, UploadFile, File, Query, HTTPException
|
|
from typing import Optional
|
|
from auth.models import TokenPayload
|
|
from auth.dependencies import require_melody_access, require_viewer
|
|
from melodies.models import (
|
|
MelodyCreate, MelodyUpdate, MelodyInDB, MelodyListResponse, MelodyInfo,
|
|
)
|
|
from melodies import service
|
|
|
|
router = APIRouter(prefix="/api/melodies", tags=["melodies"])
|
|
|
|
|
|
@router.get("", response_model=MelodyListResponse)
|
|
async def list_melodies(
|
|
search: Optional[str] = Query(None),
|
|
type: Optional[str] = Query(None),
|
|
tone: Optional[str] = Query(None),
|
|
total_notes: Optional[int] = Query(None),
|
|
_user: TokenPayload = Depends(require_viewer),
|
|
):
|
|
melodies = service.list_melodies(
|
|
search=search,
|
|
melody_type=type,
|
|
tone=tone,
|
|
total_notes=total_notes,
|
|
)
|
|
return MelodyListResponse(melodies=melodies, total=len(melodies))
|
|
|
|
|
|
@router.get("/{melody_id}", response_model=MelodyInDB)
|
|
async def get_melody(
|
|
melody_id: str,
|
|
_user: TokenPayload = Depends(require_viewer),
|
|
):
|
|
return service.get_melody(melody_id)
|
|
|
|
|
|
@router.post("", response_model=MelodyInDB, status_code=201)
|
|
async def create_melody(
|
|
body: MelodyCreate,
|
|
_user: TokenPayload = Depends(require_melody_access),
|
|
):
|
|
return service.create_melody(body)
|
|
|
|
|
|
@router.put("/{melody_id}", response_model=MelodyInDB)
|
|
async def update_melody(
|
|
melody_id: str,
|
|
body: MelodyUpdate,
|
|
_user: TokenPayload = Depends(require_melody_access),
|
|
):
|
|
return service.update_melody(melody_id, body)
|
|
|
|
|
|
@router.delete("/{melody_id}", status_code=204)
|
|
async def delete_melody(
|
|
melody_id: str,
|
|
_user: TokenPayload = Depends(require_melody_access),
|
|
):
|
|
service.delete_melody(melody_id)
|
|
|
|
|
|
@router.post("/{melody_id}/upload/{file_type}")
|
|
async def upload_file(
|
|
melody_id: str,
|
|
file_type: str,
|
|
file: UploadFile = File(...),
|
|
_user: TokenPayload = Depends(require_melody_access),
|
|
):
|
|
"""Upload a binary or preview file. file_type must be 'binary' or 'preview'."""
|
|
if file_type not in ("binary", "preview"):
|
|
raise HTTPException(status_code=400, detail="file_type must be 'binary' or 'preview'")
|
|
|
|
# Verify melody exists
|
|
melody = service.get_melody(melody_id)
|
|
|
|
contents = await file.read()
|
|
content_type = file.content_type or "application/octet-stream"
|
|
|
|
if file_type == "binary":
|
|
content_type = "application/octet-stream"
|
|
|
|
url = service.upload_file(melody_id, contents, file.filename, content_type)
|
|
|
|
# Update the melody document with the new URL if it's a preview
|
|
if file_type == "preview":
|
|
service.update_melody(melody_id, MelodyUpdate(
|
|
information=MelodyInfo(
|
|
name=melody.information.name,
|
|
previewURL=url,
|
|
)
|
|
))
|
|
|
|
return {"url": url, "file_type": file_type}
|
|
|
|
|
|
@router.delete("/{melody_id}/files/{file_type}", status_code=204)
|
|
async def delete_file(
|
|
melody_id: str,
|
|
file_type: str,
|
|
_user: TokenPayload = Depends(require_melody_access),
|
|
):
|
|
"""Delete a binary or preview file. file_type must be 'binary' or 'preview'."""
|
|
if file_type not in ("binary", "preview"):
|
|
raise HTTPException(status_code=400, detail="file_type must be 'binary' or 'preview'")
|
|
|
|
service.get_melody(melody_id)
|
|
service.delete_file(melody_id, file_type)
|
|
|
|
|
|
@router.get("/{melody_id}/files")
|
|
async def get_files(
|
|
melody_id: str,
|
|
_user: TokenPayload = Depends(require_viewer),
|
|
):
|
|
"""Get storage file URLs for a melody."""
|
|
service.get_melody(melody_id)
|
|
return service.get_storage_files(melody_id)
|