initial commit

backend/app/routes/playlists.py

@@ -0,0 +1,112 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from typing import List
+from pydantic import BaseModel
+from app.database import get_db
+from app.models.models import Playlist, Song, User
+from app.auth import get_current_user
+
+router = APIRouter()
+
+class PlaylistCreate(BaseModel):
+    name: str
+    description: str | None = None
+    is_public: bool = False
+
+class PlaylistResponse(BaseModel):
+    id: int
+    name: str
+    description: str | None
+    is_public: bool
+    owner_id: int
+    songs: List[dict]
+    
+    class Config:
+        from_attributes = True
+
+@router.post("/create")
+def create_playlist(
+    playlist: PlaylistCreate,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    db_playlist = Playlist(
+        name=playlist.name,
+        description=playlist.description,
+        is_public=playlist.is_public,
+        owner_id=current_user.id
+    )
+    db.add(db_playlist)
+    db.commit()
+    db.refresh(db_playlist)
+    return db_playlist
+
+@router.get("/my-playlists")
+def get_my_playlists(current_user: User = Depends(get_current_user), db: Session = Depends(get_db)):
+    return db.query(Playlist).filter(Playlist.owner_id == current_user.id).all()
+
+@router.get("/public")
+def get_public_playlists(db: Session = Depends(get_db)):
+    return db.query(Playlist).filter(Playlist.is_public == True).all()
+
+@router.post("/{playlist_id}/add-song/{song_id}")
+def add_song_to_playlist(
+    playlist_id: int,
+    song_id: int,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    playlist = db.query(Playlist).filter(Playlist.id == playlist_id).first()
+    if not playlist or playlist.owner_id != current_user.id:
+        raise HTTPException(status_code=403, detail="Not authorized")
+    
+    song = db.query(Song).filter(Song.id == song_id).first()
+    if not song:
+        raise HTTPException(status_code=404, detail="Song not found")
+    
+    playlist.songs.append(song)
+    db.commit()
+    return {"message": "Song added to playlist"}
+
+@router.get("/{playlist_id}")
+def get_playlist(playlist_id: int, db: Session = Depends(get_db)):
+    playlist = db.query(Playlist).filter(Playlist.id == playlist_id).first()
+    if not playlist:
+        raise HTTPException(status_code=404, detail="Playlist not found")
+    
+    songs_data = [{
+        "id": song.id,
+        "title": song.title,
+        "artist": song.artist,
+        "file_path": song.file_path,
+        "cover_path": song.cover_path,
+        "duration": song.duration,
+        "owner_id": song.owner_id
+    } for song in playlist.songs]
+    
+    return {
+        "id": playlist.id,
+        "name": playlist.name,
+        "description": playlist.description,
+        "is_public": playlist.is_public,
+        "owner_id": playlist.owner_id,
+        "songs": songs_data
+    }
+
+@router.delete("/{playlist_id}/remove-song/{song_id}")
+def remove_song_from_playlist(
+    playlist_id: int,
+    song_id: int,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    playlist = db.query(Playlist).filter(Playlist.id == playlist_id).first()
+    if not playlist or playlist.owner_id != current_user.id:
+        raise HTTPException(status_code=403, detail="Not authorized")
+    
+    song = db.query(Song).filter(Song.id == song_id).first()
+    if song in playlist.songs:
+        playlist.songs.remove(song)
+        db.commit()
+    
+    return {"message": "Song removed from playlist"}