workspaces-euc-mcp-server 0.1.1__py3-none-any.whl

workspaces_euc_mcp_server/tools/inventory.py

@@ -0,0 +1,158 @@
+# Copyright bengroeneveldsg. Licensed under the Apache License, Version 2.0 (the "License").
+# You may not use this file except in compliance with the License.
+# A copy of the License is located at http://www.apache.org/licenses/LICENSE-2.0
+"""Inventory & discovery tools (read-only, IAM Tier 0).
+
+These are *workflow* tools: each one fans out across several EUC services and synthesizes a single
+result, rather than mirroring one API call. Collection is best-effort — a failure for one service
+(missing permission, unsupported region) is captured in ``errors`` and does not abort the summary.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from .. import consts
+from ..clients import ClientFactory
+from ..models import EucInventorySummary, InventoryError, ServiceInventory
+from ._common import count_by, paginate, read_only, try_call
+
+
+def collect_inventory(factory: ClientFactory, region: str | None) -> EucInventorySummary:
+    """Collect a cross-service EUC inventory for one region. Pure/testable core."""
+    errors: list[InventoryError] = []
+    services: list[ServiceInventory] = []
+
+    workspaces = factory.client(consts.WORKSPACES_API, region=region)
+
+    personal = try_call(
+        errors,
+        consts.PRODUCT_WORKSPACES_PERSONAL,
+        "DescribeWorkspaces",
+        lambda: paginate(workspaces.describe_workspaces, "Workspaces"),
+    )
+    if personal is not None:
+        services.append(
+            ServiceInventory(
+                service=consts.PRODUCT_WORKSPACES_PERSONAL,
+                resource_type="WorkSpace",
+                count=len(personal),
+                by_state=count_by(personal, "State"),
+            )
+        )
+
+    pools = try_call(
+        errors,
+        consts.PRODUCT_WORKSPACES_POOLS,
+        "DescribeWorkspacesPools",
+        lambda: paginate(workspaces.describe_workspaces_pools, "WorkspacesPools"),
+    )
+    if pools is not None:
+        services.append(
+            ServiceInventory(
+                service=consts.PRODUCT_WORKSPACES_POOLS,
+                resource_type="WorkSpacesPool",
+                count=len(pools),
+                by_state=count_by(pools, "State"),
+            )
+        )
+
+    appstream = factory.client(consts.APPSTREAM_API, region=region)
+    fleets = try_call(
+        errors,
+        consts.PRODUCT_WORKSPACES_APPLICATIONS,
+        "DescribeFleets",
+        lambda: paginate(appstream.describe_fleets, "Fleets"),
+    )
+    if fleets is not None:
+        services.append(
+            ServiceInventory(
+                service=consts.PRODUCT_WORKSPACES_APPLICATIONS,
+                resource_type="Fleet",
+                count=len(fleets),
+                by_state=count_by(fleets, "State"),
+            )
+        )
+
+    stacks = try_call(
+        errors,
+        consts.PRODUCT_WORKSPACES_APPLICATIONS,
+        "DescribeStacks",
+        lambda: paginate(appstream.describe_stacks, "Stacks"),
+    )
+    if stacks is not None:
+        services.append(
+            ServiceInventory(
+                service=consts.PRODUCT_WORKSPACES_APPLICATIONS,
+                resource_type="Stack",
+                count=len(stacks),
+            )
+        )
+
+    secure_browser = factory.client(consts.SECURE_BROWSER_API, region=region)
+    portals = try_call(
+        errors,
+        consts.PRODUCT_SECURE_BROWSER,
+        "ListPortals",
+        lambda: paginate(
+            secure_browser.list_portals,
+            "portals",
+            pagination_in="nextToken",
+            pagination_out="nextToken",
+        ),
+    )
+    if portals is not None:
+        services.append(
+            ServiceInventory(
+                service=consts.PRODUCT_SECURE_BROWSER,
+                resource_type="Portal",
+                count=len(portals),
+                by_state=count_by(portals, "portalStatus"),
+            )
+        )
+
+    instances_client = factory.client(consts.WORKSPACES_INSTANCES_API, region=region)
+    instances = try_call(
+        errors,
+        consts.PRODUCT_WORKSPACES_CORE_INSTANCES,
+        "ListWorkspaceInstances",
+        lambda: paginate(instances_client.list_workspace_instances, "WorkspaceInstances"),
+    )
+    if instances is not None:
+        services.append(
+            ServiceInventory(
+                service=consts.PRODUCT_WORKSPACES_CORE_INSTANCES,
+                resource_type="ManagedInstance",
+                count=len(instances),
+                by_state=count_by(instances, "ProvisionState"),
+            )
+        )
+
+    total = sum(s.count for s in services)
+    return EucInventorySummary(
+        region=region,
+        total_resources=total,
+        services=services,
+        errors=errors,
+    )
+
+
+def register(mcp: Any, factory: ClientFactory) -> None:
+    """Register inventory tools on the FastMCP app."""
+
+    async def get_euc_inventory_summary(region: str | None = None) -> dict[str, Any]:
+        """Summarize all Amazon WorkSpaces EUC resources in an AWS region.
+
+        Fans out across WorkSpaces Personal, WorkSpaces Pools, WorkSpaces Applications (formerly
+        AppStream 2.0), and WorkSpaces Secure Browser (formerly WorkSpaces Web), returning
+        per-service counts broken down by state, the grand total, and any per-service collection
+        errors (e.g. missing permissions). Read-only.
+
+        Args:
+            region: AWS region to inventory. Defaults to the server's configured region.
+        """
+        target_region = region or factory.region
+        summary = collect_inventory(factory, target_region)
+        return summary.model_dump()
+
+    mcp.add_tool(get_euc_inventory_summary, annotations=read_only("EUC inventory summary"))