soc/soc-integrator/app/adapters/iris.py

from typing import Any

import httpx


class IrisAdapter:
    def __init__(self, base_url: str, api_key: str) -> None:
        self.base_url = base_url.rstrip("/")
        self.api_key = api_key

    def _headers(self) -> dict[str, str]:
        return {"Authorization": f"Bearer {self.api_key}"} if self.api_key else {}

    async def create_case(self, payload: dict[str, Any]) -> dict[str, Any]:
        headers = self._headers()
        async with httpx.AsyncClient(verify=False, timeout=20.0) as client:
            v2_url = f"{self.base_url}/api/v2/cases"
            response = await client.post(v2_url, json=payload, headers=headers)

            if response.status_code == 404:
                legacy_url = f"{self.base_url}/manage/cases/add"
                response = await client.post(legacy_url, json=payload, headers=headers)
                active_url = legacy_url
            else:
                active_url = v2_url

            try:
                response.raise_for_status()
            except httpx.HTTPStatusError as exc:
                detail = response.text.strip()
                raise RuntimeError(
                    f"IRIS returned {response.status_code} for {active_url}. Response: {detail}"
                ) from exc
            return response.json() if response.content else {"status_code": response.status_code}

    async def update_case(self, case_id: str, payload: dict[str, Any]) -> dict[str, Any]:
        headers = self._headers()
        async with httpx.AsyncClient(verify=False, timeout=20.0) as client:
            v2_url = f"{self.base_url}/api/v2/cases/{case_id}"
            response = await client.put(v2_url, json=payload, headers=headers)

            if response.status_code == 404:
                legacy_url = f"{self.base_url}/manage/cases/update/{case_id}"
                response = await client.post(legacy_url, json=payload, headers=headers)
                active_url = legacy_url
            else:
                active_url = v2_url

            try:
                response.raise_for_status()
            except httpx.HTTPStatusError as exc:
                detail = response.text.strip()
                raise RuntimeError(
                    f"IRIS returned {response.status_code} for {active_url}. Response: {detail}"
                ) from exc
            return response.json() if response.content else {"status_code": response.status_code}

    async def whoami(self) -> dict[str, Any]:
        headers = self._headers()
        url = f"{self.base_url}/user/whoami"
        async with httpx.AsyncClient(verify=False, timeout=20.0) as client:
            response = await client.get(url, headers=headers)
            response.raise_for_status()
            return response.json() if response.content else {"status_code": response.status_code}

    async def list_cases(self, limit: int = 50, offset: int = 0) -> dict[str, Any]:
        headers = self._headers()
        safe_limit = max(1, limit)
        safe_offset = max(0, offset)
        v2_url = f"{self.base_url}/api/v2/cases"
        params = {"limit": safe_limit, "offset": safe_offset}
        async with httpx.AsyncClient(verify=False, timeout=20.0) as client:
            response = await client.get(v2_url, params=params, headers=headers)
            active_url = v2_url

            # Older IRIS releases may not expose /api/v2/cases for listing.
            if response.status_code == 404:
                legacy_url = f"{self.base_url}/manage/cases/list"
                legacy_attempts: list[tuple[str, dict[str, Any] | None, dict[str, Any] | None]] = [
                    ("post", None, {"start": safe_offset, "length": safe_limit, "draw": 1}),
                    ("post", None, {"page": (safe_offset // safe_limit) + 1, "per_page": safe_limit}),
                    ("get", {"limit": safe_limit, "offset": safe_offset}, None),
                ]

                for method, legacy_params, legacy_payload in legacy_attempts:
                    if method == "post":
                        response = await client.post(
                            legacy_url,
                            params=legacy_params,
                            json=legacy_payload,
                            headers=headers,
                        )
                    else:
                        response = await client.get(
                            legacy_url,
                            params=legacy_params,
                            headers=headers,
                        )
                    active_url = legacy_url
                    if response.status_code not in {404, 405, 415}:
                        break

            try:
                response.raise_for_status()
            except httpx.HTTPStatusError as exc:
                detail = response.text.strip()
                raise RuntimeError(
                    f"IRIS returned {response.status_code} for {active_url}. Response: {detail}"
                ) from exc

            if not response.content:
                return {"status_code": response.status_code}

            result = response.json()
            if isinstance(result, dict) and isinstance(result.get("data"), list):
                total = result.get("recordsTotal")
                if not isinstance(total, int):
                    total = len(result["data"])
                return {
                    "items": result["data"],
                    "total": total,
                    "limit": safe_limit,
                    "offset": safe_offset,
                    "source": "legacy_manage_cases_list",
                    "raw": result,
                }
            return result