from fastapi import APIRouter, Depends, HTTPException, Query, Body
from sqlalchemy.orm import Session
from typing import List, Optional
from pydantic import BaseModel
from ..db.database import get_db
from ..models.mood import MoodCategory
from ..schemas.mood import MoodCategoryResponse, MoodPlaylistResponse
from ..services.mood_engine import analyze_song_mood, get_mood_playlist, seed_mood_categories
from ..services.audio import get_stream_path


class SetMoodRequest(BaseModel):
    mood: str

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


@router.get("/categories", response_model=List[MoodCategoryResponse])
def list_mood_categories(db: Session = Depends(get_db)):
    seed_mood_categories(db)
    categories = db.query(MoodCategory).all()
    return [MoodCategoryResponse.model_validate(c) for c in categories]


@router.post("/analyze")
def analyze_mood(song_id: Optional[str] = None, db: Session = Depends(get_db)):
    if song_id:
        return analyze_song_mood(song_id, db)
    
    from ..models.song import Song
    songs = db.query(Song).all()
    results = []
    for song in songs:
        results.append(analyze_song_mood(song.id, db))
    return {"analyzed": len(results), "results": results}


@router.get("/{mood}/playlist", response_model=MoodPlaylistResponse)
def get_mood_playlist_endpoint(mood: str, limit: int = Query(50, ge=1, le=200), db: Session = Depends(get_db)):
    songs = get_mood_playlist(mood, db, limit)
    
    song_responses = []
    for s in songs:
        from ..schemas.song import SongResponse
        song_responses.append(SongResponse.model_validate(s))
    
    return MoodPlaylistResponse(
        mood=mood,
        songs=song_responses,
        total_songs=len(song_responses),
    )


@router.post("/save")
def save_mood_playlist(mood: str, name: Optional[str] = None, db: Session = Depends(get_db)):
    from ..models.playlist import Playlist
    import uuid
    
    playlist_id = str(uuid.uuid4())
    playlist = Playlist(
        id=playlist_id,
        name=name or f"{mood} Playlist",
        mood_category=mood.lower(),
    )
    db.add(playlist)
    
    songs = get_mood_playlist(mood, db, 50)
    from ..models.playlist import PlaylistSong
    for i, song in enumerate(songs):
        ps = PlaylistSong(playlist_id=playlist_id, song_id=song.id, position=i)
        db.add(ps)
    
    db.commit()
    return {"id": playlist_id, "name": playlist.name, "songs": len(songs)}


@router.post("/set")
def set_mood(data: SetMoodRequest, db: Session = Depends(get_db)):
    from ..models.settings import UserSetting
    setting = db.query(UserSetting).filter(UserSetting.key == "default_mood").first()
    if setting:
        setting.value = data.mood
    else:
        setting = UserSetting(key="default_mood", value=data.mood)
        db.add(setting)
    db.commit()
    return {"mood": data.mood}