from datetime import datetime

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

from ..auth import get_current_user
from ..database import get_db
from ..models import Character, User
from ..plan_limits import check_limit
from ..schemas import CharacterCreate, CharacterResponse, CharacterUpdate

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


@router.get("/", response_model=list[CharacterResponse])
def list_characters(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    return (
        db.query(Character)
        .filter(Character.user_id == current_user.id)
        .order_by(Character.created_at.desc())
        .all()
    )


@router.get("/{character_id}", response_model=CharacterResponse)
def get_character(
    character_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    character = (
        db.query(Character)
        .filter(Character.id == character_id, Character.user_id == current_user.id)
        .first()
    )
    if not character:
        raise HTTPException(status_code=404, detail="Character not found")
    return character


@router.post("/", response_model=CharacterResponse, status_code=201)
def create_character(
    data: CharacterCreate,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    # Check plan limit for characters
    count = db.query(Character).filter(Character.user_id == current_user.id).count()
    allowed, msg = check_limit(current_user, "characters_max", count)
    if not allowed:
        raise HTTPException(status_code=403, detail={"message": msg, "upgrade_required": True})

    character = Character(**data.model_dump())
    character.user_id = current_user.id
    db.add(character)
    db.commit()
    db.refresh(character)
    return character


@router.put("/{character_id}", response_model=CharacterResponse)
def update_character(
    character_id: int,
    data: CharacterUpdate,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    character = (
        db.query(Character)
        .filter(Character.id == character_id, Character.user_id == current_user.id)
        .first()
    )
    if not character:
        raise HTTPException(status_code=404, detail="Character not found")
    update_data = data.model_dump(exclude_unset=True)
    for key, value in update_data.items():
        setattr(character, key, value)
    character.updated_at = datetime.utcnow()
    db.commit()
    db.refresh(character)
    return character


@router.delete("/{character_id}", status_code=204)
def delete_character(
    character_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    character = (
        db.query(Character)
        .filter(Character.id == character_id, Character.user_id == current_user.id)
        .first()
    )
    if not character:
        raise HTTPException(status_code=404, detail="Character not found")
    db.delete(character)
    db.commit()