from fastapi import APIRouter, Depends, HTTPException, UploadFile, File, status, Form
from sqlalchemy.orm import Session
import os
import shutil

from ..db import get_db
from ..models import Grave, GraveStatus, Cemetery, GraveHistory
from ..schemas import GraveCreate, GraveOut, GraveUpdate
from ..dependencies import get_current_user, require_admin
from ..config import settings


router = APIRouter()


@router.post("/", response_model=GraveOut)
def create_grave(
    payload: GraveCreate,
    db: Session = Depends(get_db),
    user=Depends(get_current_user),
):
    cemetery = db.query(Cemetery).get(payload.cemetery_id)
    if not cemetery:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Cemetery not found")
    grave = Grave(**payload.dict(), created_by=user.id)
    db.add(grave)
    db.commit()
    db.refresh(grave)
    db.add(GraveHistory(grave_id=grave.id, action="created", actor_id=user.id))
    db.commit()
    return grave


@router.get("/", response_model=list[GraveOut])
def list_graves(cemetery_id: int | None = None, db: Session = Depends(get_db)):
    q = db.query(Grave)
    if cemetery_id:
        q = q.filter(Grave.cemetery_id == cemetery_id)
    return q.order_by(Grave.id.desc()).all()


@router.get("/{grave_id}", response_model=GraveOut)
def get_grave(grave_id: int, db: Session = Depends(get_db)):
    grave = db.query(Grave).get(grave_id)
    if not grave:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
    return grave


@router.patch("/{grave_id}", response_model=GraveOut, dependencies=[Depends(require_admin)])
def update_grave(grave_id: int, payload: GraveUpdate, db: Session = Depends(get_db), user=Depends(get_current_user)):
    grave = db.query(Grave).get(grave_id)
    if not grave:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
    before_status = grave.status
    for k, v in payload.dict(exclude_unset=True).items():
        setattr(grave, k, v)
    db.commit()
    db.refresh(grave)
    action = "updated"
    if before_status != grave.status:
        action = grave.status.value
    db.add(GraveHistory(grave_id=grave.id, action=action, actor_id=user.id))
    db.commit()
    return grave


@router.delete("/{grave_id}", status_code=204, dependencies=[Depends(require_admin)])
def delete_grave(grave_id: int, db: Session = Depends(get_db), user=Depends(get_current_user)):
    grave = db.query(Grave).get(grave_id)
    if not grave:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
    db.add(GraveHistory(grave_id=grave.id, action="deleted", actor_id=user.id))
    db.delete(grave)
    db.commit()
    return None


@router.post("/{grave_id}/upload", response_model=GraveOut)
def upload_temp_photo(grave_id: int, file: UploadFile = File(...), db: Session = Depends(get_db), user=Depends(get_current_user)):
    grave = db.query(Grave).get(grave_id)
    if not grave:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
    os.makedirs(settings.uploads_temp_dir, exist_ok=True)
    file_path = os.path.join(settings.uploads_temp_dir, f"grave_{grave_id}_{file.filename}")
    with open(file_path, "wb") as buffer:
        shutil.copyfileobj(file.file, buffer)
    grave.temp_photo_path = file_path
    db.commit()
    db.refresh(grave)
    return grave