hms-licht-ton/backend/app/services/rentman_service.py

"""Rentman API client wrapper for equipment import and request submission."""
import httpx
import logging
import re
from typing import Any

from app.config import get_settings

logger = logging.getLogger(__name__)
settings = get_settings()

RENTMAN_BASE_URL = "https://api.rentman.net"


class RentmanService:
    """Wrapper around the Rentman REST API using httpx."""

    def __init__(self, token: str | None = None) -> None:
        self._token = token or settings.rentman_api_token
        self._base_url = RENTMAN_BASE_URL

    def _headers(self) -> dict[str, str]:
        return {
            "Authorization": f"Bearer {self._token}",
            "Content-Type": "application/json",
        }

    async def get_equipment_page(self, limit: int = 100, offset: int = 0) -> dict[str, Any]:
        """Fetch a single page of equipment from Rentman.

        Returns the raw JSON response dict with 'data' and optional 'itemCount'.
        """
        url = f"{self._base_url}/equipment"
        params = {"limit": limit, "offset": offset}
        async with httpx.AsyncClient(timeout=30.0) as client:
            resp = await client.get(url, headers=self._headers(), params=params)
            resp.raise_for_status()
            return resp.json()

    async def get_all_equipment(self, limit: int = 100) -> list[dict[str, Any]]:
        """Paginate through all equipment pages until data is empty."""
        all_items: list[dict[str, Any]] = []
        offset = 0
        while True:
            page = await self.get_equipment_page(limit=limit, offset=offset)
            data = page.get("data", [])
            if not data:
                break
            all_items.extend(data)
            offset += limit
        return all_items

    async def create_project_request(self, payload: dict[str, Any]) -> dict[str, Any]:
        """POST /projectrequests to create a new project request in Rentman."""
        url = f"{self._base_url}/projectrequests"
        async with httpx.AsyncClient(timeout=30.0) as client:
            resp = await client.post(url, headers=self._headers(), json=payload)
            resp.raise_for_status()
            return resp.json()

    async def add_equipment_to_request(
        self, request_id: str, equipment_payload: dict[str, Any]
    ) -> dict[str, Any]:
        """POST /projectrequests/{id}/projectrequestequipment for a single item."""
        url = f"{self._base_url}/projectrequests/{request_id}/projectrequestequipment"
        async with httpx.AsyncClient(timeout=30.0) as client:
            resp = await client.post(url, headers=self._headers(), json=equipment_payload)
            resp.raise_for_status()
            return resp.json()

    @staticmethod
    def transform_equipment(raw: dict[str, Any]) -> dict[str, Any]:
        """Map a raw Rentman equipment object to equipment_cache schema."""
        images = raw.get("images") or raw.get("files") or []
        if isinstance(images, list):
            image_urls = [
                img.get("url", img.get("filename", "")) if isinstance(img, dict) else str(img)
                for img in images
            ]
        else:
            image_urls = []

        group = raw.get("equipment_group") or {}
        category = group.get("name", "") if isinstance(group, dict) else str(group or "")

        return {
            "rentman_id": str(raw.get("id", "")),
            "name": raw.get("name", ""),
            "number": raw.get("number") or raw.get("code", ""),
            "category": category,
            "subcategory": raw.get("subcategory", ""),
            "description": raw.get("description", ""),
            "specifications": raw.get("specifications", {}),
            "images": image_urls,
            "rental_price": raw.get("rental_price"),
            "brand": raw.get("brand", ""),
            "available": raw.get("available", True),
        }

    @staticmethod
    def build_project_request_payload(data: dict[str, Any]) -> dict[str, Any]:
        """Map frontend rental request data to Rentman POST /projectrequests payload."""
        date_start = data.get("date_start")
        date_end = data.get("date_end")

        contact_name = data.get("contact_name", "")
        parts = contact_name.strip().split(" ", 1)
        first_name = parts[0] if parts else ""
        last_name = parts[1] if len(parts) > 1 else ""

        street = data.get("contact_street", "")
        house_number = ""
        if street:
            match = re.search(r"(\d+[a-zA-Z]*)", street)
            if match:
                house_number = match.group(1)

        return {
            "name": data.get("event_name", ""),
            "planperiod_start": f"{date_start}T08:00:00+02:00" if date_start else None,
            "planperiod_end": f"{date_end}T02:00:00+02:00" if date_end else None,
            "usageperiod_start": f"{date_start}T18:00:00+02:00" if date_start else None,
            "usageperiod_end": f"{date_end}T23:59:00+02:00" if date_end else None,
            "contact_name": data.get("contact_company") or data.get("contact_name", ""),
            "contact_person_first_name": first_name,
            "contact_person_lastname": last_name,
            "contact_person_email": data.get("contact_email", ""),
            "contact_person_phone": data.get("contact_phone", ""),
            "location_name": data.get("location", ""),
            "location_mailing_street": street,
            "location_mailing_number": house_number,
            "location_mailing_postalcode": data.get("contact_postalcode", ""),
            "location_mailing_city": data.get("contact_city") or data.get("location", ""),
            "remark": data.get("message", ""),
        }

    @staticmethod
    def build_equipment_payload(item: dict[str, Any]) -> dict[str, Any]:
        """Map a single rental request item to Rentman projectrequestequipment payload."""
        return {
            "name": item.get("equipment_name", ""),
            "quantity": item.get("quantity", 1),
            "quantity_total": item.get("quantity", 1),
            "unit_price": item.get("unit_price", 0),
            "linked_equipment": f"/equipment/{item.get('rentman_equipment_id', '')}" if item.get("rentman_equipment_id") else None,
        }