trustgraph-flow 2.5.3__tar.gz → 2.5.4__tar.gz

{trustgraph_flow-2.5.3 → trustgraph_flow-2.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-flow
-Version: 2.5.3
+Version: 2.5.4
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_flow-2.5.4/trustgraph/flow_version.py

@@ -0,0 +1 @@
+__version__ = "2.5.4"

{trustgraph_flow-2.5.3 → trustgraph_flow-2.5.4}/trustgraph/gateway/registry.py

@@ -309,6 +309,13 @@ register(Operation(
     extract_resource=_empty_resource,
     extract_parameters=_no_parameters,
 ))
+register(Operation(
+    name="list-my-workspaces",
+    capability="workspaces:list-own",
+    resource_level=ResourceLevel.SYSTEM,
+    extract_resource=_empty_resource,
+    extract_parameters=_no_parameters,
+))
 register(Operation(
     name="list-workspaces",
     capability="workspaces:admin",

{trustgraph_flow-2.5.3 → trustgraph_flow-2.5.4}/trustgraph/iam/noauth/handler.py

@@ -106,7 +106,7 @@ class NoAuthHandler:
             ):
                 return IamResponse()
 
-            if op == "list-workspaces":
+            if op in ("list-workspaces", "list-my-workspaces"):
                 return IamResponse()
 
             if op in ("create-api-key", "list-api-keys", "revoke-api-key"):

{trustgraph_flow-2.5.3 → trustgraph_flow-2.5.4}/trustgraph/iam/service/iam.py

@@ -68,6 +68,7 @@ _READER_CAPS = {
     "collections:read",
     "knowledge:read",
     "keys:self",
+    "workspaces:list-own",
 }
 
 _WRITER_CAPS = _READER_CAPS | {
@@ -328,6 +329,8 @@ class IamService:
                 return await self.handle_delete_user(v)
             if op == "create-workspace":
                 return await self.handle_create_workspace(v)
+            if op == "list-my-workspaces":
+                return await self.handle_list_my_workspaces(v)
             if op == "list-workspaces":
                 return await self.handle_list_workspaces(v)
             if op == "get-workspace":
@@ -915,6 +918,30 @@ class IamService:
         row = await self.table_store.get_workspace(v.workspace_record.id)
         return IamResponse(workspace=self._row_to_workspace_record(row))
 
+    async def handle_list_my_workspaces(self, v):
+        if not v.actor:
+            return _err("invalid-argument", "actor required")
+
+        user_row = await self.table_store.get_user(v.actor)
+        if user_row is None:
+            return _err("not-found", "user not found")
+
+        user_roles = user_row[6] or []
+        is_admin = "admin" in user_roles
+
+        if is_admin:
+            rows = await self.table_store.list_workspaces()
+        else:
+            user_workspace = user_row[1]
+            row = await self.table_store.get_workspace(user_workspace)
+            rows = [row] if row else []
+
+        return IamResponse(
+            workspaces=[
+                self._row_to_workspace_record(r) for r in rows
+            ],
+        )
+
     async def handle_list_workspaces(self, v):
         rows = await self.table_store.list_workspaces()
         return IamResponse(

{trustgraph_flow-2.5.3 → trustgraph_flow-2.5.4}/trustgraph_flow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-flow
-Version: 2.5.3
+Version: 2.5.4
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_flow-2.5.3/trustgraph/flow_version.py

@@ -1 +0,0 @@
-__version__ = "2.5.3"