Phase 2 Complete by Claude Code

backend/melodies/router.py

@@ -1 +1,118 @@
-# TODO: CRUD endpoints for melodies
+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)