Added Draft Melodies. Further improvements to the UI

backend/melodies/router.py

@@ -16,13 +16,15 @@ async def list_melodies(
     type: Optional[str] = Query(None),
     tone: Optional[str] = Query(None),
     total_notes: Optional[int] = Query(None),
+    status: Optional[str] = Query(None),
     _user: TokenPayload = Depends(require_permission("melodies", "view")),
 ):
-    melodies = service.list_melodies(
+    melodies = await service.list_melodies(
         search=search,
         melody_type=type,
         tone=tone,
         total_notes=total_notes,
+        status=status,
     )
     return MelodyListResponse(melodies=melodies, total=len(melodies))
 
@@ -32,15 +34,16 @@ async def get_melody(
     melody_id: str,
     _user: TokenPayload = Depends(require_permission("melodies", "view")),
 ):
-    return service.get_melody(melody_id)
+    return await service.get_melody(melody_id)
 
 
 @router.post("", response_model=MelodyInDB, status_code=201)
 async def create_melody(
     body: MelodyCreate,
+    publish: bool = Query(False),
     _user: TokenPayload = Depends(require_permission("melodies", "add")),
 ):
-    return service.create_melody(body)
+    return await service.create_melody(body, publish=publish)
 
 
 @router.put("/{melody_id}", response_model=MelodyInDB)
@@ -49,7 +52,7 @@ async def update_melody(
     body: MelodyUpdate,
     _user: TokenPayload = Depends(require_permission("melodies", "edit")),
 ):
-    return service.update_melody(melody_id, body)
+    return await service.update_melody(melody_id, body)
 
 
 @router.delete("/{melody_id}", status_code=204)
@@ -57,7 +60,23 @@ async def delete_melody(
     melody_id: str,
     _user: TokenPayload = Depends(require_permission("melodies", "delete")),
 ):
-    service.delete_melody(melody_id)
+    await service.delete_melody(melody_id)
+
+
+@router.post("/{melody_id}/publish", response_model=MelodyInDB)
+async def publish_melody(
+    melody_id: str,
+    _user: TokenPayload = Depends(require_permission("melodies", "edit")),
+):
+    return await service.publish_melody(melody_id)
+
+
+@router.post("/{melody_id}/unpublish", response_model=MelodyInDB)
+async def unpublish_melody(
+    melody_id: str,
+    _user: TokenPayload = Depends(require_permission("melodies", "edit")),
+):
+    return await service.unpublish_melody(melody_id)
 
 
 @router.post("/{melody_id}/upload/{file_type}")
@@ -72,7 +91,7 @@ async def upload_file(
         raise HTTPException(status_code=400, detail="file_type must be 'binary' or 'preview'")
 
     # Verify melody exists
-    melody = service.get_melody(melody_id)
+    melody = await service.get_melody(melody_id)
 
     contents = await file.read()
     content_type = file.content_type or "application/octet-stream"
@@ -84,14 +103,14 @@ async def upload_file(
 
     # Update the melody document with the file URL
     if file_type == "preview":
-        service.update_melody(melody_id, MelodyUpdate(
+        await service.update_melody(melody_id, MelodyUpdate(
             information=MelodyInfo(
                 name=melody.information.name,
                 previewURL=url,
             )
         ))
     elif file_type == "binary":
-        service.update_melody(melody_id, MelodyUpdate(url=url))
+        await service.update_melody(melody_id, MelodyUpdate(url=url))
 
     return {"url": url, "file_type": file_type}
 
@@ -106,7 +125,7 @@ async def delete_file(
     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)
+    await service.get_melody(melody_id)
     service.delete_file(melody_id, file_type)
 
 
@@ -116,5 +135,5 @@ async def get_files(
     _user: TokenPayload = Depends(require_permission("melodies", "view")),
 ):
     """Get storage file URLs for a melody."""
-    service.get_melody(melody_id)
+    await service.get_melody(melody_id)
     return service.get_storage_files(melody_id)