from typing import Any

from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session

from app import crud
from app.api import deps
from app.models.user import User
from app.schemas.collection import (
    Collection,
    CollectionCreate,
    CollectionUpdate,
    CollectionWithNotes,
)
from app.schemas.note import Note

router = APIRouter()


@router.get("/", response_model=list[Collection])
def read_collections(
    db: Session = Depends(deps.get_db),
    skip: int = 0,
    limit: int = 100,
    current_user: User = Depends(deps.get_current_active_user),
) -> Any:
    """
    Retrieve collections for the current user.
    """
    collections = crud.get_collections_by_owner(
        db=db, owner_id=current_user.id, skip=skip, limit=limit
    )
    return collections


@router.post("/", response_model=Collection)
def create_collection(
    *,
    db: Session = Depends(deps.get_db),
    collection_in: CollectionCreate,
    current_user: User = Depends(deps.get_current_active_user),
) -> Any:
    """
    Create new collection.
    """
    collection = crud.create_collection(
        db=db, obj_in=collection_in, owner_id=current_user.id
    )
    return collection


@router.put("/{id}", response_model=Collection)
def update_collection(
    *,
    db: Session = Depends(deps.get_db),
    id: int,
    collection_in: CollectionUpdate,
    current_user: User = Depends(deps.get_current_active_user),
) -> Any:
    """
    Update a collection.
    """
    collection = crud.get_collection(db=db, collection_id=id)
    if not collection:
        raise HTTPException(status_code=404, detail="Collection not found")
    if collection.owner_id != current_user.id:
        raise HTTPException(status_code=403, detail="Not enough permissions")
    collection = crud.update_collection(db=db, db_obj=collection, obj_in=collection_in)
    return collection


@router.get("/{id}", response_model=CollectionWithNotes)
def read_collection(
    *,
    db: Session = Depends(deps.get_db),
    id: int,
    current_user: User = Depends(deps.get_current_active_user),
) -> Any:
    """
    Get collection by ID with all its notes.
    """
    collection = crud.get_collection(db=db, collection_id=id)
    if not collection:
        raise HTTPException(status_code=404, detail="Collection not found")
    if collection.owner_id != current_user.id:
        raise HTTPException(status_code=403, detail="Not enough permissions")
    return collection


@router.get("/{id}/notes", response_model=list[Note])
def read_collection_notes(
    *,
    db: Session = Depends(deps.get_db),
    id: int,
    skip: int = 0,
    limit: int = 100,
    current_user: User = Depends(deps.get_current_active_user),
) -> Any:
    """
    Get notes from a specific collection.
    """
    collection = crud.get_collection(db=db, collection_id=id)
    if not collection:
        raise HTTPException(status_code=404, detail="Collection not found")
    if collection.owner_id != current_user.id:
        raise HTTPException(status_code=403, detail="Not enough permissions")
    notes = crud.get_multi_by_collection(
        db=db, owner_id=current_user.id, collection_id=id, skip=skip, limit=limit
    )
    return notes


@router.delete("/{id}", status_code=204, response_model=None)
def delete_collection(
    *,
    db: Session = Depends(deps.get_db),
    id: int,
    current_user: User = Depends(deps.get_current_active_user),
) -> Any:
    """
    Delete a collection.
    """
    collection = crud.get_collection(db=db, collection_id=id)
    if not collection:
        raise HTTPException(status_code=404, detail="Collection not found")
    if collection.owner_id != current_user.id:
        raise HTTPException(status_code=403, detail="Not enough permissions")
    crud.remove_collection(db=db, collection_id=id, owner_id=current_user.id)
    return None