from datetime import datetime

from fastapi import HTTPException, status
from sqlalchemy.orm import Session

from app.models.subscription import Subscription
from app.models.tenant import Tenant
from app.models.user import User
from app.schemas.tenant import TenantCreate, TenantUpdate


def get_tenant_list(db: Session):
    return db.query(Tenant).filter(Tenant.deleted_at.is_(None)).all()


def get_tenant_by_id(db: Session, id: str):
    tenant = db.query(Tenant).filter(Tenant.id == id, Tenant.deleted_at.is_(None)).first()
    if not tenant:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "The item does not exist"},
        )
    return tenant


def create_tenant(db: Session, data: TenantCreate):
    existing = db.query(Tenant).filter(Tenant.slug == data.slug).first()
    if existing:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT,
            detail={"message": "Slug already exists"},
        )

    try:
        tenant = Tenant(
            name=data.name,
            slug=data.slug,
            business_type_id=data.business_type_id,
            phone=data.phone,
            address=data.address,
            email=data.email,
        )
        db.add(tenant)
        db.flush()

        subscription = Subscription(
            tenant_id=tenant.id,
            plan_id=data.plan_id,
            limits=data.limits,
        )
        db.add(subscription)
        db.commit()
    except:
        db.rollback()
        raise

    db.refresh(tenant)
    return tenant


def update_tenant(db: Session, id: str, data: TenantUpdate):
    tenant = db.query(Tenant).filter(Tenant.id == id, Tenant.deleted_at.is_(None)).first()
    if not tenant:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "The item does not exist"},
        )

    update_data = data.model_dump(exclude_unset=True)

    if "slug" in update_data:
        dup = db.query(Tenant).filter(Tenant.slug == update_data["slug"], Tenant.id != id).first()
        if dup:
            raise HTTPException(
                status_code=status.HTTP_409_CONFLICT,
                detail={"message": "Slug already exists"},
            )

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

    db.commit()
    db.refresh(tenant)
    return tenant


def delete_tenant(db: Session, id: str):
    tenant = db.query(Tenant).filter(Tenant.id == id, Tenant.deleted_at.is_(None)).first()
    if not tenant:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "The item does not exist"},
        )

    tenant.deleted_at = datetime.now()
    db.commit()


def assign_user_to_tenant(db: Session, tenant_id: str, user_id: str):
    tenant = db.query(Tenant).filter(Tenant.id == tenant_id).first()
    if not tenant:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "Tenant not found"},
        )

    user = db.query(User).filter(User.id == user_id).first()
    if not user:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "User not found"},
        )

    user.tenant_id = tenant_id
    user.status = "ACTIVE"
    db.commit()

    return {"message": "User assigned to tenant"}