from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from typing import List, Optional
from database import get_db
from models import Contractor
from security import get_current_active_user
import schemas

router = APIRouter(
    prefix="/contractors",
    tags=["Contractors"],
    dependencies=[Depends(get_current_active_user)]
)

@router.post("/", response_model=schemas.Contractor)
def create_contractor(contractor: schemas.ContractorCreate, db: Session = Depends(get_db)):
    db_contractor = Contractor(**contractor.dict())
    db.add(db_contractor)
    db.commit()
    db.refresh(db_contractor)
    return db_contractor

@router.get("/", response_model=List[schemas.Contractor])
def read_contractors(
    parent_id: Optional[int] = None, 
    only_parents: bool = False,
    is_active: Optional[bool] = None,
    db: Session = Depends(get_db)
):
    query = db.query(Contractor)
    if only_parents:
        query = query.filter(Contractor.parent_id == None)
    elif parent_id is not None:
        query = query.filter(Contractor.parent_id == parent_id)
    
    if is_active is not None:
        query = query.filter(Contractor.is_active == is_active)
    return query.all()

@router.get("/{contractor_id}", response_model=schemas.Contractor)
def read_contractor(contractor_id: int, db: Session = Depends(get_db)):
    db_contractor = db.query(Contractor).filter(Contractor.id == contractor_id).first()
    if not db_contractor:
        raise HTTPException(status_code=404, detail="Contractor not found")
    return db_contractor

@router.put("/{contractor_id}", response_model=schemas.Contractor)
@router.patch("/{contractor_id}", response_model=schemas.Contractor)
def update_contractor(
    contractor_id: int, 
    contractor: schemas.ContractorUpdate, 
    db: Session = Depends(get_db)
):
    db_contractor = db.query(Contractor).filter(Contractor.id == contractor_id).first()
    if not db_contractor:
        raise HTTPException(status_code=404, detail="Contractor not found")
    
    update_data = contractor.dict(exclude_unset=True)
    for key, value in update_data.items():
        setattr(db_contractor, key, value)
    
    db.commit()
    db.refresh(db_contractor)
    return db_contractor

@router.delete("/{contractor_id}", status_code=status.HTTP_204_NO_CONTENT)
def delete_contractor(contractor_id: int, db: Session = Depends(get_db)):
    db_contractor = db.query(Contractor).filter(Contractor.id == contractor_id).first()
    if not db_contractor:
        raise HTTPException(status_code=404, detail="Contractor not found")
    
    # Optional: instead of hard delete, maybe just deactivate
    db.delete(db_contractor)
    db.commit()
    return None