Merge pull request #119 from phj-loopy/master

feat: add Functions and Container Registries inventory collection

Author: lhhyung

pkg/pip_requirements.txt

@@ -16,4 +16,6 @@ azure-mgmt-containerinstance
 azure-mgmt-webpubsub
 azure-mgmt-resource
 azure-mgmt-advisor==10.0.0b1
-azure-mgmt-cognitiveservices
+azure-mgmt-cognitiveservices
+
+azure-mgmt-containerregistry

src/plugin/connector/base.py

@@ -8,6 +8,8 @@
 from azure.mgmt.cognitiveservices import CognitiveServicesManagementClient
 from azure.mgmt.compute import ComputeManagementClient
 from azure.mgmt.containerinstance import ContainerInstanceManagementClient
+# Container Registry 클라이언트 import 추가
+from azure.mgmt.containerregistry import ContainerRegistryManagementClient
 from azure.mgmt.cosmosdb import CosmosDBManagementClient
 from azure.mgmt.keyvault import KeyVaultManagementClient
 from azure.mgmt.monitor import MonitorManagementClient
@@ -41,6 +43,7 @@ def __init__(self, *args, **kwargs):
         self.sql_client = None
         self.monitor_client = None
         self.container_instance_client = None
+        self.container_registry_client = None
         self.resource_client = None
         self.resource_health_client = None
         self.storage_client = None
@@ -81,6 +84,10 @@ def set_connect(self, secret_data: dict):
         self.container_instance_client = ContainerInstanceManagementClient(
             credential=self.credential, subscription_id=subscription_id
         )
+        self.container_registry_client = ContainerRegistryManagementClient(
+            credential=self.credential, subscription_id=subscription_id
+        )
+
         self.resource_client = ResourceManagementClient(
             credential=self.credential, subscription_id=subscription_id
         )

src/plugin/connector/container_registries/__init__.py

@@ -0,0 +1,105 @@
+import logging
+
+from plugin.connector.base import AzureBaseConnector
+
+_LOGGER = logging.getLogger("spaceone")
+
+
+class ContainerRegistriesConnector(AzureBaseConnector):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.set_connect(kwargs.get("secret_data"))
+
+    def list_registries(self):
+        return self.container_registry_client.registries.list()
+
+    def get_registry(self, resource_group_name, registry_name):
+        return self.container_registry_client.registries.get(
+            resource_group_name=resource_group_name, registry_name=registry_name
+        )
+
+    def list_webhooks(self, resource_group_name, registry_name):
+        try:
+            return self.container_registry_client.webhooks.list(
+                resource_group_name=resource_group_name, registry_name=registry_name
+            )
+        except Exception as e:
+            _LOGGER.error(f"[list_webhooks] Error: {e}")
+            return []
+
+    def list_replications(self, resource_group_name, registry_name):
+        try:
+            return self.container_registry_client.replications.list(
+                resource_group_name=resource_group_name, registry_name=registry_name
+            )
+        except Exception as e:
+            _LOGGER.error(f"[list_replications] Error: {e}")
+            return []
+
+    def list_usages(self, resource_group_name, registry_name):
+        try:
+            return self.container_registry_client.registries.list_usages(
+                resource_group_name=resource_group_name, registry_name=registry_name
+            )
+        except Exception as e:
+            _LOGGER.error(f"[list_usages] Error: {e}")
+            return None
+
+    def list_tokens(self, resource_group_name, registry_name):
+        try:
+            tokens_ops = getattr(self.container_registry_client, "tokens", None)
+            if tokens_ops:
+                return tokens_ops.list(
+                    resource_group_name=resource_group_name, registry_name=registry_name
+                )
+        except Exception as e:
+            _LOGGER.error(f"[list_tokens] Error: {e}")
+        return []
+
+    def list_scope_maps(self, resource_group_name, registry_name):
+        try:
+            scope_maps_ops = getattr(self.container_registry_client, "scope_maps", None)
+            if scope_maps_ops:
+                return scope_maps_ops.list(
+                    resource_group_name=resource_group_name, registry_name=registry_name
+                )
+        except Exception as e:
+            _LOGGER.error(f"[list_scope_maps] Error: {e}")
+        return []
+
+    def list_cache_rules(self, resource_group_name, registry_name):
+        try:
+            cache_rules_ops = getattr(
+                self.container_registry_client, "cache_rules", None
+            )
+            if cache_rules_ops:
+                return cache_rules_ops.list(
+                    resource_group_name=resource_group_name, registry_name=registry_name
+                )
+        except Exception as e:
+            _LOGGER.error(f"[list_cache_rules] Error: {e}")
+        return []
+
+    def list_connected_registries(self, resource_group_name, registry_name):
+        try:
+            connected_ops = getattr(
+                self.container_registry_client, "connected_registries", None
+            )
+            if connected_ops:
+                return connected_ops.list(
+                    resource_group_name=resource_group_name, registry_name=registry_name
+                )
+        except Exception as e:
+            _LOGGER.error(f"[list_connected_registries] Error: {e}")
+        return []
+
+    def list_tasks(self, resource_group_name, registry_name):
+        try:
+            tasks_ops = getattr(self.container_registry_client, "tasks", None)
+            if tasks_ops:
+                return tasks_ops.list(
+                    resource_group_name=resource_group_name, registry_name=registry_name
+                )
+        except Exception as e:
+            _LOGGER.error(f"[list_tasks] Error: {e}")
+        return []

src/plugin/connector/container_registries/container_registries_connector.py

@@ -0,0 +1,41 @@
+import logging
+from plugin.connector.base import AzureBaseConnector
+
+_LOGGER = logging.getLogger("spaceone")
+
+MANAGEMENT_BASE_URL = "https://management.azure.com"
+WEB_API_VERSION = "2025-03-01"
+
+
+class FunctionsConnector(AzureBaseConnector):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        secret_data = kwargs.get("secret_data", {})
+        self.set_connect(secret_data)
+        self.subscription_id = secret_data.get("subscription_id")
+
+    def list_function_apps(self):
+        function_apps = []
+
+        for res in self.resource_client.resources.list():
+            if res.type == "Microsoft.Web/sites":
+                kind = (getattr(res, "kind", "") or "").lower()
+                if "functionapp" in kind:
+                    function_apps.append(res)
+
+        _LOGGER.debug(f"[FunctionsConnector] Found {len(function_apps)} function apps")
+        return function_apps
+
+    def get_function_app_detail(self, resource_id: str) -> dict:
+        if not resource_id:
+            return {}
+
+        url = f"{MANAGEMENT_BASE_URL}{resource_id}?api-version={WEB_API_VERSION}"
+        return self.request_azure_api(url, method="GET") or {}
+
+    def get_app_service_plan_detail(self, server_farm_id: str) -> dict:
+        if not server_farm_id:
+            return {}
+
+        url = f"{MANAGEMENT_BASE_URL}{server_farm_id}?api-version={WEB_API_VERSION}"
+        return self.request_azure_api(url, method="GET") or {}

src/plugin/connector/functions/__init__.py

@@ -19,7 +19,6 @@
     # "inventory.Region",
 ]
 
-
 @app.route("Collector.init")
 def collector_init(params: dict) -> dict:
     return _create_init_metadata()

src/plugin/connector/functions/functions_connector.py

@@ -2,6 +2,7 @@
 from .application_gatways import *
 from .cognitive_services import *
 from .container_instances import *
+from .container_registries import *
 from .cosmos_db import *
 from .disks import *
 from .key_vaults import *
@@ -23,3 +24,4 @@
 from .virtual_networks import *
 from .vm_scale_sets import *
 from .web_pub_sub_service import *
+from .functions import *

src/plugin/main.py

@@ -1 +1 @@
-from .instance_manager import CognitiveServicesManager
+from .instance_manager import CognitiveServicesManager

src/plugin/manager/__init__.py

@@ -0,0 +1 @@
+from .registry_manager import ContainerRegistriesManager