from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from typing import List
from app.db.session import get_db
from app.models.trip import Trip
from app.models.user import User
from app.models.destination import Destination
from app.schemas.trip import Trip as TripSchema, TripCreate, TripUpdate

router = APIRouter()


@router.get("/", response_model=List[TripSchema])
def get_trips(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
    trips = db.query(Trip).offset(skip).limit(limit).all()
    return trips


@router.get("/{trip_id}", response_model=TripSchema)
def get_trip(trip_id: int, db: Session = Depends(get_db)):
    trip = db.query(Trip).filter(Trip.id == trip_id).first()
    if trip is None:
        raise HTTPException(status_code=404, detail="Trip not found")
    return trip


@router.get("/user/{user_id}", response_model=List[TripSchema])
def get_user_trips(
    user_id: int, skip: int = 0, limit: int = 100, db: Session = Depends(get_db)
):
    user = db.query(User).filter(User.id == user_id).first()
    if user is None:
        raise HTTPException(status_code=404, detail="User not found")

    trips = (
        db.query(Trip).filter(Trip.user_id == user_id).offset(skip).limit(limit).all()
    )
    return trips


@router.post("/", response_model=TripSchema)
def create_trip(trip: TripCreate, user_id: int, db: Session = Depends(get_db)):
    user = db.query(User).filter(User.id == user_id).first()
    if user is None:
        raise HTTPException(status_code=404, detail="User not found")

    destination = (
        db.query(Destination).filter(Destination.id == trip.destination_id).first()
    )
    if destination is None:
        raise HTTPException(status_code=404, detail="Destination not found")

    db_trip = Trip(**trip.dict(), user_id=user_id)
    db.add(db_trip)
    db.commit()
    db.refresh(db_trip)
    return db_trip


@router.put("/{trip_id}", response_model=TripSchema)
def update_trip(trip_id: int, trip: TripUpdate, db: Session = Depends(get_db)):
    db_trip = db.query(Trip).filter(Trip.id == trip_id).first()
    if db_trip is None:
        raise HTTPException(status_code=404, detail="Trip not found")

    update_data = trip.dict(exclude_unset=True)
    if "destination_id" in update_data:
        destination = (
            db.query(Destination)
            .filter(Destination.id == update_data["destination_id"])
            .first()
        )
        if destination is None:
            raise HTTPException(status_code=404, detail="Destination not found")

    for field, value in update_data.items():
        setattr(db_trip, field, value)

    db.commit()
    db.refresh(db_trip)
    return db_trip


@router.delete("/{trip_id}")
def delete_trip(trip_id: int, db: Session = Depends(get_db)):
    db_trip = db.query(Trip).filter(Trip.id == trip_id).first()
    if db_trip is None:
        raise HTTPException(status_code=404, detail="Trip not found")

    db.delete(db_trip)
    db.commit()
    return {"message": "Trip deleted successfully"}