Finish port

Author: Sytten

packages/caido-sdk-client/src/caido_sdk_client/__init__.py

@@ -17,6 +17,7 @@
     BaseError,
     InstanceNotReadyError,
     NoViewerInResponseError,
+    RankUserError,
     UnsupportedViewerTypeError,
     from_error,
 )
@@ -194,6 +195,7 @@
     "PATAuthOptions",
     "ProjectSDK",
     "RangeInput",
+    "RankUserError",
     "ReadyOptions",
     "ReplayCollectionSDK",
     "ReplayCollectionsListBuilder",

packages/caido-sdk-client/src/caido_sdk_client/errors/__init__.py

@@ -23,6 +23,7 @@
 from caido_sdk_client.errors.misc import (
     NotFoundUserError,
     OtherUserError,
+    RankUserError,
     ReadOnlyUserError,
 )
 from caido_sdk_client.errors.plugin import (
@@ -58,6 +59,7 @@
     "PluginFunctionCallError",
     "PluginUserError",
     "ProjectUserError",
+    "RankUserError",
     "ReadOnlyUserError",
     "RestRequestError",
     "StoreUserError",

packages/caido-sdk-client/src/caido_sdk_client/errors/all_errors.py

@@ -14,6 +14,7 @@
     PermissionDeniedUserErrorFull,
     PluginUserErrorFull,
     ProjectUserErrorFull,
+    RankUserErrorFull,
     ReadOnlyUserErrorFull,
     StoreUserErrorFull,
     TaskInProgressUserErrorFull,
@@ -34,6 +35,7 @@
     CloudUserErrorFull,
     PluginUserErrorFull,
     StoreUserErrorFull,
+    RankUserErrorFull,
     TaskInProgressUserErrorFull,
     WorkflowUserErrorFull,
 ]

packages/caido-sdk-client/src/caido_sdk_client/errors/from_error.py

@@ -13,13 +13,15 @@
 from caido_sdk_client.errors.misc import (
     NotFoundUserError,
     OtherUserError,
+    RankUserError,
     ReadOnlyUserError,
 )
 from caido_sdk_client.errors.plugin import PluginUserError, StoreUserError
 from caido_sdk_client.errors.project import ProjectUserError
 from caido_sdk_client.errors.tasks import TaskInProgressUserError
 from caido_sdk_client.errors.version import NewerVersionUserError
 from caido_sdk_client.errors.workflow import WorkflowUserError
+from caido_sdk_client.graphql.__generated__.schema import RankErrorReason
 
 
 def from_error(error: object) -> BaseError:
@@ -65,6 +67,19 @@ def from_error(error: object) -> BaseError:
         case "StoreUserError":
             return StoreUserError(error)  # type: ignore[arg-type]
 
+        case "RankUserError":
+            code = getattr(error, "code", "")
+            rank_reason_val = getattr(
+                error, "rankReason", getattr(error, "rank_reason", None)
+            )
+            if isinstance(rank_reason_val, RankErrorReason):
+                rank_reason = rank_reason_val
+            elif rank_reason_val is not None:
+                rank_reason = RankErrorReason(str(rank_reason_val))
+            else:
+                rank_reason = RankErrorReason.NOT_ENABLED  # fallback
+            return RankUserError(code, rank_reason)
+
         case "TaskInProgressUserError":
             return TaskInProgressUserError(getattr(error, "taskId", ""))
 

packages/caido-sdk-client/src/caido_sdk_client/errors/misc.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from caido_sdk_client.errors.base import BaseError
+from caido_sdk_client.graphql.__generated__.schema import RankErrorReason
 
 
 class NotFoundUserError(BaseError):
@@ -21,3 +22,8 @@ def __init__(self, code: str, message: str | None = None) -> None:
             super().__init__(message)
         else:
             super().__init__(f"An unknown user error occured: {code!r}")
+
+
+class RankUserError(BaseError):
+    def __init__(self, code: str, rank_reason: RankErrorReason) -> None:
+        super().__init__(f"Ranking failed: {code}: {rank_reason!s}")

packages/caido-sdk-client/src/caido_sdk_client/graphql/__generated__/schema.py

@@ -61,10 +61,10 @@ fragment StoreUserErrorFull on StoreUserError {
   storeReason: reason
 }
 
-# fragment RankUserErrorFull on RankUserError {
-#   ...UserErrorFull
-#   rankReason: reason
-# }
+fragment RankUserErrorFull on RankUserError {
+  ...UserErrorFull
+  rankReason: reason
+}
 
 fragment TaskInProgressUserErrorFull on TaskInProgressUserError {
   ...UserErrorFull

packages/caido-sdk-client/src/caido_sdk_client/graphql/documents/errors.graphql

@@ -44,7 +44,11 @@ subscription FinishedTask {
     }
     status
     error {
+      __typename
       code
+      ... on RankUserError {
+        ...RankUserErrorFull
+      }
     }
   }
 }