refactor(BA-5046): migrate artifact registry action classes to RBAC base classes

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: fregataa

src/ai/backend/manager/services/artifact_registry/actions/base.py

@@ -1,4 +1,3 @@
-from dataclasses import dataclass
 from typing import override
 
 from ai.backend.common.data.permission.types import EntityType
@@ -11,15 +10,13 @@
 from ai.backend.manager.actions.action.types import FieldData
 
 
-@dataclass
 class ArtifactRegistryAction(BaseAction):
     @override
     @classmethod
     def entity_type(cls) -> EntityType:
         return EntityType.ARTIFACT_REGISTRY
 
 
-@dataclass
 class ArtifactBatchRegistryAction(BaseBatchAction):
     @override
     @classmethod

src/ai/backend/manager/services/artifact_registry/actions/common/get_meta.py

@@ -2,31 +2,41 @@
 from dataclasses import dataclass
 from typing import override
 
-from ai.backend.manager.actions.action import BaseActionResult
+from ai.backend.common.data.permission.types import RBACElementType
 from ai.backend.manager.actions.types import ActionOperationType
 from ai.backend.manager.data.artifact_registries.types import ArtifactRegistryData
-from ai.backend.manager.services.artifact_registry.actions.base import ArtifactRegistryAction
+from ai.backend.manager.data.permission.types import RBACElementRef
+from ai.backend.manager.services.artifact_registry.actions.base import (
+    ArtifactRegistrySingleEntityAction,
+    ArtifactRegistrySingleEntityActionResult,
+)
 
 
 @dataclass
-class GetArtifactRegistryMetaAction(ArtifactRegistryAction):
+class GetArtifactRegistryMetaAction(ArtifactRegistrySingleEntityAction):
     registry_id: uuid.UUID | None = None
     registry_name: str | None = None
 
-    @override
-    def entity_id(self) -> str | None:
-        return str(self.registry_id)
-
     @override
     @classmethod
     def operation_type(cls) -> ActionOperationType:
         return ActionOperationType.GET
 
+    @override
+    def target_entity_id(self) -> str:
+        if self.registry_id:
+            return str(self.registry_id)
+        return self.registry_name or ""
+
+    @override
+    def target_element(self) -> RBACElementRef:
+        return RBACElementRef(RBACElementType.ARTIFACT_REGISTRY, self.target_entity_id())
+
 
 @dataclass
-class GetArtifactRegistryMetaActionResult(BaseActionResult):
+class GetArtifactRegistryMetaActionResult(ArtifactRegistrySingleEntityActionResult):
     result: ArtifactRegistryData
 
     @override
-    def entity_id(self) -> str | None:
+    def target_entity_id(self) -> str:
         return str(self.result.id)

src/ai/backend/manager/services/artifact_registry/processors.py

@@ -121,7 +121,7 @@ class ArtifactRegistryProcessors(AbstractProcessorPackage):
     search_reservoir_registries: ActionProcessor[
         SearchReservoirRegistriesAction, SearchReservoirRegistriesActionResult
     ]
-    get_registry_meta: ActionProcessor[
+    get_registry_meta: SingleEntityActionProcessor[
         GetArtifactRegistryMetaAction, GetArtifactRegistryMetaActionResult
     ]
     get_registry_metas: ActionProcessor[
@@ -192,14 +192,19 @@ def __init__(
             validators=[validators.rbac.single_entity],
         )
 
+        self.get_registry_meta = SingleEntityActionProcessor(
+            service.get_registry_meta,
+            action_monitors,
+            validators=[validators.rbac.single_entity],
+        )
+
         # Internal/batch actions without RBAC
         self.get_huggingface_registries = ActionProcessor(
             service.get_huggingface_registries, action_monitors
         )
         self.get_reservoir_registries = ActionProcessor(
             service.get_reservoir_registries, action_monitors
         )
-        self.get_registry_meta = ActionProcessor(service.get_registry_meta, action_monitors)
         self.get_registry_metas = ActionProcessor(service.get_registry_metas, action_monitors)
 
     @override