hesabixArc/hesabixAPI/app/services/warehouse_service.py

from __future__ import annotations

from typing import Dict, Any, Optional
from sqlalchemy.orm import Session
from sqlalchemy import and_

from app.core.responses import ApiError
from adapters.db.models.warehouse import Warehouse
from adapters.db.repositories.warehouse_repository import WarehouseRepository
from adapters.api.v1.schema_models.warehouse import WarehouseCreateRequest, WarehouseUpdateRequest
from adapters.api.v1.schemas import QueryInfo, FilterItem
from app.services.query_service import QueryService


def _to_dict(obj: Warehouse) -> Dict[str, Any]:
    return {
        "id": obj.id,
        "business_id": obj.business_id,
        "code": obj.code,
        "name": obj.name,
        "description": obj.description,
        "is_default": obj.is_default,
        "created_at": obj.created_at,
        "updated_at": obj.updated_at,
    }


def create_warehouse(db: Session, business_id: int, payload: WarehouseCreateRequest) -> Dict[str, Any]:
    code = payload.code.strip()
    dup = db.query(Warehouse).filter(and_(Warehouse.business_id == business_id, Warehouse.code == code)).first()
    if dup:
        raise ApiError("DUPLICATE_WAREHOUSE_CODE", "کد انبار تکراری است", http_status=400)
    repo = WarehouseRepository(db)
    obj = repo.create(
        business_id=business_id,
        code=code,
        name=payload.name.strip(),
        description=payload.description,
        is_default=bool(payload.is_default),
    )
    if obj.is_default:
        db.query(Warehouse).filter(and_(Warehouse.business_id == business_id, Warehouse.id != obj.id)).update({Warehouse.is_default: False})
        db.commit()
    return {"message": "WAREHOUSE_CREATED", "data": _to_dict(obj)}


def list_warehouses(db: Session, business_id: int) -> Dict[str, Any]:
    repo = WarehouseRepository(db)
    rows = repo.list(business_id)
    return {"items": [_to_dict(w) for w in rows]}


def get_warehouse(db: Session, business_id: int, warehouse_id: int) -> Optional[Dict[str, Any]]:
    obj = db.get(Warehouse, warehouse_id)
    if not obj or obj.business_id != business_id:
        return None
    return _to_dict(obj)


def update_warehouse(db: Session, business_id: int, warehouse_id: int, payload: WarehouseUpdateRequest) -> Optional[Dict[str, Any]]:
    repo = WarehouseRepository(db)
    obj = db.get(Warehouse, warehouse_id)
    if not obj or obj.business_id != business_id:
        return None
    if payload.code and payload.code.strip() != obj.code:
        dup = db.query(Warehouse).filter(and_(Warehouse.business_id == business_id, Warehouse.code == payload.code.strip(), Warehouse.id != warehouse_id)).first()
        if dup:
            raise ApiError("DUPLICATE_WAREHOUSE_CODE", "کد انبار تکراری است", http_status=400)

    updated = repo.update(
        warehouse_id,
        code=payload.code.strip() if isinstance(payload.code, str) else None,
        name=payload.name.strip() if isinstance(payload.name, str) else None,
        description=payload.description,
        is_default=payload.is_default if payload.is_default is not None else None,
    )
    if not updated:
        return None
    if updated.is_default:
        db.query(Warehouse).filter(and_(Warehouse.business_id == business_id, Warehouse.id != updated.id)).update({Warehouse.is_default: False})
        db.commit()
    return {"message": "WAREHOUSE_UPDATED", "data": _to_dict(updated)}


def delete_warehouse(db: Session, business_id: int, warehouse_id: int) -> bool:
    obj = db.get(Warehouse, warehouse_id)
    if not obj or obj.business_id != business_id:
        return False
    repo = WarehouseRepository(db)
    return repo.delete(warehouse_id)



def query_warehouses(db: Session, business_id: int, query_info: QueryInfo) -> Dict[str, Any]:
    """Query warehouses with filters/search/sorting/pagination scoped to business."""
    # Ensure business scoping via filters
    base_filter = FilterItem(property="business_id", operator="=", value=business_id)
    merged_filters = [base_filter]
    if query_info.filters:
        merged_filters.extend(query_info.filters)

    effective_query = QueryInfo(
        sort_by=query_info.sort_by,
        sort_desc=query_info.sort_desc,
        take=query_info.take,
        skip=query_info.skip,
        search=query_info.search,
        search_fields=query_info.search_fields,
        filters=merged_filters,
    )

    results, total = QueryService.query_with_filters(Warehouse, db, effective_query)
    items = [_to_dict(w) for w in results]
    limit = max(1, effective_query.take)
    page = (effective_query.skip // limit) + 1
    total_pages = (total + limit - 1) // limit

    return {
        "items": items,
        "total": total,
        "page": page,
        "limit": limit,
        "total_pages": total_pages,
    }