1 files changed, 160 insertions, 0 deletions
diff --git a/jb/managers/gr_api.py b/jb/managers/gr_api.py
new file mode 100644
index 0000000..f0be1a7
--- /dev/null
+++ b/jb/managers/gr_api.py
@@ -0,0 +1,160 @@
+"""Client for General Research's product-user API."""
+
+import logging
+from typing import Any
+
+import requests
+
+from jb.models.auth import User
+
+logger = logging.getLogger("amtjb")
+
+
+class GRApiError(RuntimeError):
+    """The General Research API could not satisfy a request."""
+
+
+class GRApiNotFoundError(GRApiError):
+    """The requested General Research API resource does not exist."""
+
+
+class GRApiManager:
+    def __init__(
+        self,
+        base_url: str,
+        token: str,
+        product_id: str,
+        timeout: float = 5.0,
+        session: requests.Session | None = None,
+    ) -> None:
+        if not token:
+            raise ValueError("General Research API token must not be empty")
+
+        self.base_url = base_url.rstrip("/")
+        self.product_id = product_id
+        self.timeout = timeout
+        self.session = session or requests.Session()
+        self.session.headers.update(
+            {
+                "Authorization": token,
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+            }
+        )
+
+    def _request(self, method: str, url: str, **kwargs: Any) -> requests.Response:
+        try:
+            response = self.session.request(method, url, timeout=self.timeout, **kwargs)
+            response.raise_for_status()
+            return response
+        except requests.HTTPError as exc:
+            if exc.response is not None and exc.response.status_code == 404:
+                raise GRApiNotFoundError(
+                    f"General Research API resource not found: {method} {url}"
+                ) from exc
+            raise GRApiError(
+                f"General Research API request failed: {method} {url}"
+            ) from exc
+        except requests.RequestException as exc:
+            raise GRApiError(
+                f"General Research API request failed: {method} {url}"
+            ) from exc
+
+    def _parse_user_response(self, res: dict[str, Any]) -> User:
+        return User.model_validate(
+            {
+                "product_user_id": res["product_user_id"],
+                "email": res["metadata"].get("email_address"),
+                "display_name": res["metadata"].get("display_name"),
+                "blocked": res["blocked"],
+            }
+        )
+
+    def ensure_user_exists(self, user: User) -> User:
+        """Idempotently create the product user if it does not already exist."""
+        url = f"{self.base_url}/{self.product_id}/user/{user.product_user_id}/"
+        res = self._request("PUT", url).json()
+        if res["metadata"].get("email_address") is None:
+            self.set_user_email(user)
+            return self.get_user(user.product_user_id)
+        user_thl = self._parse_user_response(res)
+        if user_thl.email != user.email:
+            raise ValueError(
+                f"user {user.product_user_id} already exists with email {user_thl.email}"
+            )
+        return user_thl
+
+    def get_user(self, product_user_id: str) -> User:
+        """Retrieve the user's email address and display name."""
+        url = f"{self.base_url}/{self.product_id}/user/{product_user_id}/"
+        res = self._request("GET", url).json()
+        return self._parse_user_response(res)
+
+    def get_user_by_email(self, email: str) -> User:
+        user = User.model_validate({"email": email})
+        return self.get_user(user.product_user_id)
+
+    def set_user_email(self, user: User) -> None:
+        """This should only be called once per user upon account creation.
+        A user cannot change their email address."""
+        url = f"{self.base_url}/{self.product_id}/user/{user.product_user_id}/metadata/"
+        _ = self._request(
+            "PATCH",
+            url,
+            json={"email_address": str(user.email)},
+        ).json()
+
+    def set_user_display_name(self, user: User) -> User:
+        """Can be called as many times as needed. Multiple
+        users can have the same display name."""
+        if user.display_name is None:
+            return user
+        url = f"{self.base_url}/{self.product_id}/user/{user.product_user_id}/metadata/"
+        _ = self._request(
+            "PATCH",
+            url,
+            json={"display_name": user.display_name},
+        ).json()
+        return self.get_user(user.product_user_id)
+
+    def transition_user_from_amt(self, user: User, amt_worker_id: str) -> User:
+        """This should only be called once upon transition from an
+        AMT account to a General Research account."""
+        url = f"{self.base_url}/{self.product_id}/user/{amt_worker_id}/"
+        try:
+            self._request("PATCH", url, json={"product_user_id": user.product_user_id})
+        except GRApiNotFoundError as exc:
+            raise ValueError(f"User {amt_worker_id} does not exist") from exc
+        except GRApiError as exc:
+            http_error = exc.__cause__
+            response = (
+                http_error.response
+                if isinstance(http_error, requests.HTTPError)
+                else None
+            )
+            if response is not None and response.status_code == 400:
+                try:
+                    detail = response.json().get("detail")
+                except (ValueError, AttributeError):
+                    detail = None
+                if detail == "Unable to update User":
+                    raise ValueError(
+                        "unable to update user, probably another user already "
+                        "exists with this email"
+                    ) from exc
+            raise
+        self.set_user_email(user)
+        transitioned_user = self.get_user(user.product_user_id)
+        logger.warning(
+            "Transitioned product user from AMT worker %s to %s with email %s",
+            amt_worker_id,
+            transitioned_user.product_user_id,
+            transitioned_user.email,
+            extra={
+                "event": "transition_user_from_amt",
+                "amt_worker_id": amt_worker_id,
+                "product_user_id": transitioned_user.product_user_id,
+                "email": str(transitioned_user.email),
+            },
+        )
+        return transitioned_user