feat(sync): keycloak as source-of-truth with auto catalog sync and token refresh

app/services/idp_admin_service.py

@@ -29,6 +29,12 @@ class KeycloakDeleteResult:
     user_id: str | None = None
 
 
+@dataclass
+class KeycloakGroupSyncResult:
+    group_id: str
+    action: str
+
+
 class KeycloakAdminService:
     def __init__(self, settings: Settings) -> None:
         self.base_url = settings.keycloak_base_url.rstrip("/")
@@ -97,6 +103,64 @@ class KeycloakAdminService:
             raise HTTPException(status_code=502, detail="idp_lookup_failed")
         return resp.json()
 
+    def _lookup_group_by_id(self, client: httpx.Client, group_id: str) -> dict | None:
+        resp = client.get(f"/admin/realms/{self.realm}/groups/{group_id}")
+        if resp.status_code == 404:
+            return None
+        if resp.status_code >= 400:
+            raise HTTPException(status_code=502, detail="idp_group_lookup_failed")
+        payload = resp.json() if resp.content else {}
+        return payload if isinstance(payload, dict) else None
+
+    def _lookup_group_by_name(self, client: httpx.Client, *, name: str, parent_group_id: str | None) -> dict | None:
+        if parent_group_id:
+            resp = client.get(
+                f"/admin/realms/{self.realm}/groups/{parent_group_id}/children",
+                params={"search": name, "briefRepresentation": "false"},
+            )
+            if resp.status_code >= 400:
+                raise HTTPException(status_code=502, detail="idp_group_lookup_failed")
+            matches = resp.json() if isinstance(resp.json(), list) else []
+            for row in matches:
+                if isinstance(row, dict) and str(row.get("name", "")).strip() == name:
+                    return row
+            return None
+
+        resp = client.get(
+            f"/admin/realms/{self.realm}/groups",
+            params={"search": name, "exact": "true", "briefRepresentation": "false"},
+        )
+        if resp.status_code >= 400:
+            raise HTTPException(status_code=502, detail="idp_group_lookup_failed")
+        matches = resp.json() if isinstance(resp.json(), list) else []
+        for row in matches:
+            if not isinstance(row, dict):
+                continue
+            if str(row.get("name", "")).strip() != name:
+                continue
+            parent_id = row.get("parentId")
+            if parent_group_id:
+                if str(parent_id or "") == parent_group_id:
+                    return row
+            elif not parent_id:
+                return row
+        return None
+
+    @staticmethod
+    def _normalize_group_attributes(attributes: dict[str, str | list[str]] | None) -> dict[str, list[str]]:
+        if not attributes:
+            return {}
+        output: dict[str, list[str]] = {}
+        for key, value in attributes.items():
+            normalized_key = str(key).strip()
+            if not normalized_key:
+                continue
+            if isinstance(value, list):
+                output[normalized_key] = [str(v) for v in value if str(v)]
+            elif value is not None and str(value):
+                output[normalized_key] = [str(value)]
+        return output
+
     def _lookup_user_by_email_or_username(
         self, client: httpx.Client, *, email: str | None, username: str | None
     ) -> dict | None:
@@ -162,6 +226,59 @@ class KeycloakAdminService:
                 raise HTTPException(status_code=502, detail="idp_create_failed")
             return KeycloakSyncResult(user_id=user_id, action="created", user_sub=user_id)
 
+    def ensure_group(
+        self,
+        *,
+        name: str,
+        group_id: str | None = None,
+        parent_group_id: str | None = None,
+        attributes: dict[str, str | list[str]] | None = None,
+    ) -> KeycloakGroupSyncResult:
+        if not name:
+            raise HTTPException(status_code=400, detail="idp_group_name_required")
+        normalized_attrs = self._normalize_group_attributes(attributes)
+
+        with self._client() as client:
+            existing = self._lookup_group_by_id(client, group_id) if group_id else None
+            if existing is None:
+                existing = self._lookup_group_by_name(client, name=name, parent_group_id=parent_group_id)
+
+            if existing and existing.get("id"):
+                resolved_id = str(existing["id"])
+                payload = {"name": name, "attributes": normalized_attrs}
+                put_resp = client.put(f"/admin/realms/{self.realm}/groups/{resolved_id}", json=payload)
+                if put_resp.status_code >= 400:
+                    raise HTTPException(status_code=502, detail="idp_group_update_failed")
+                return KeycloakGroupSyncResult(group_id=resolved_id, action="updated")
+
+            payload = {"name": name, "attributes": normalized_attrs}
+            if parent_group_id:
+                create_resp = client.post(f"/admin/realms/{self.realm}/groups/{parent_group_id}/children", json=payload)
+            else:
+                create_resp = client.post(f"/admin/realms/{self.realm}/groups", json=payload)
+            if create_resp.status_code >= 400:
+                raise HTTPException(status_code=502, detail="idp_group_create_failed")
+
+            location = create_resp.headers.get("Location", "")
+            resolved_id = location.rstrip("/").split("/")[-1] if location and "/" in location else ""
+            if not resolved_id:
+                found = self._lookup_group_by_name(client, name=name, parent_group_id=parent_group_id)
+                resolved_id = str(found.get("id")) if found and found.get("id") else ""
+            if not resolved_id:
+                raise HTTPException(status_code=502, detail="idp_group_create_failed")
+            return KeycloakGroupSyncResult(group_id=resolved_id, action="created")
+
+    def delete_group(self, *, group_id: str | None) -> KeycloakDeleteResult:
+        if not group_id:
+            return KeycloakDeleteResult(action="not_found")
+        with self._client() as client:
+            resp = client.delete(f"/admin/realms/{self.realm}/groups/{group_id}")
+            if resp.status_code in {204, 404}:
+                return KeycloakDeleteResult(action="deleted" if resp.status_code == 204 else "not_found")
+            if resp.status_code >= 400:
+                raise HTTPException(status_code=502, detail="idp_group_delete_failed")
+            return KeycloakDeleteResult(action="deleted")
+
     def reset_password(
         self,
         *,
@@ -207,3 +324,63 @@ class KeycloakAdminService:
             if resp.status_code >= 400:
                 raise HTTPException(status_code=502, detail="idp_delete_failed")
             return KeycloakDeleteResult(action="deleted", user_id=user_id)
+
+    def list_groups_tree(self) -> list[dict]:
+        with self._client() as client:
+            resp = client.get(
+                f"/admin/realms/{self.realm}/groups",
+                params={"first": 0, "max": 5000, "briefRepresentation": "false"},
+            )
+            if resp.status_code >= 400:
+                raise HTTPException(status_code=502, detail="idp_group_lookup_failed")
+            payload = resp.json() if resp.content else []
+            return payload if isinstance(payload, list) else []
+
+    def list_users(self) -> list[dict]:
+        users: list[dict] = []
+        first = 0
+        page_size = 200
+        with self._client() as client:
+            while True:
+                resp = client.get(
+                    f"/admin/realms/{self.realm}/users",
+                    params={"first": first, "max": page_size},
+                )
+                if resp.status_code >= 400:
+                    raise HTTPException(status_code=502, detail="idp_lookup_failed")
+                batch = resp.json() if isinstance(resp.json(), list) else []
+                users.extend([row for row in batch if isinstance(row, dict)])
+                if len(batch) < page_size:
+                    break
+                first += page_size
+        return users
+
+    def list_clients(self) -> list[dict]:
+        clients: list[dict] = []
+        first = 0
+        page_size = 200
+        with self._client() as client:
+            while True:
+                resp = client.get(
+                    f"/admin/realms/{self.realm}/clients",
+                    params={"first": first, "max": page_size},
+                )
+                if resp.status_code >= 400:
+                    raise HTTPException(status_code=502, detail="idp_lookup_failed")
+                batch = resp.json() if isinstance(resp.json(), list) else []
+                clients.extend([row for row in batch if isinstance(row, dict)])
+                if len(batch) < page_size:
+                    break
+                first += page_size
+        return clients
+
+    def list_client_roles(self, client_uuid: str) -> list[dict]:
+        with self._client() as client:
+            resp = client.get(
+                f"/admin/realms/{self.realm}/clients/{client_uuid}/roles",
+                params={"first": 0, "max": 5000},
+            )
+            if resp.status_code >= 400:
+                raise HTTPException(status_code=502, detail="idp_lookup_failed")
+            payload = resp.json() if resp.content else []
+            return payload if isinstance(payload, list) else []