agentcore-cli 0.31.1__py3-none-any.whl

agentcore/cloudformation/__init__.py

@@ -0,0 +1,84 @@
+"""CloudFormation helpers (port of upstream ``src/cli/cloudformation``)."""
+
+from agentcore.cloudformation.bootstrap import (
+    CDK_TOOLKIT_STACK_NAME,
+    BootstrapStatus,
+    check_bootstrap_status,
+    format_cdk_environment,
+)
+from agentcore.cloudformation.logical_ids import to_pascal_id
+from agentcore.cloudformation.outputs import (
+    StackOutputs,
+    build_deployed_state,
+    get_stack_outputs,
+    get_stack_outputs_by_project,
+    omit_payment_authorization_outputs,
+    parse_agent_outputs,
+    parse_capacity_provider_outputs,
+    parse_config_bundle_outputs,
+    parse_dataset_outputs,
+    parse_evaluator_outputs,
+    parse_gateway_outputs,
+    parse_harness_outputs,
+    parse_knowledge_base_data_source_outputs,
+    parse_knowledge_base_outputs,
+    parse_memory_outputs,
+    parse_online_eval_outputs,
+    parse_payment_outputs,
+    parse_policy_engine_outputs,
+    parse_policy_outputs,
+    parse_runtime_endpoint_outputs,
+)
+from agentcore.cloudformation.stack_discovery import (
+    DiscoveredStack,
+    discover_stacks_by_project,
+    find_stack,
+    get_stack_name,
+)
+from agentcore.cloudformation.stack_status import (
+    BlockingStack,
+    StackStatusResult,
+    check_stack_status,
+    check_stacks_status,
+)
+from agentcore.cloudformation.types import TERMINAL_STACK_STATUSES, failure_like, get_status_color, is_failure_event
+
+__all__ = [
+    "CDK_TOOLKIT_STACK_NAME",
+    "TERMINAL_STACK_STATUSES",
+    "BlockingStack",
+    "BootstrapStatus",
+    "DiscoveredStack",
+    "StackOutputs",
+    "StackStatusResult",
+    "build_deployed_state",
+    "check_bootstrap_status",
+    "check_stack_status",
+    "check_stacks_status",
+    "discover_stacks_by_project",
+    "failure_like",
+    "find_stack",
+    "format_cdk_environment",
+    "get_stack_name",
+    "get_stack_outputs",
+    "get_stack_outputs_by_project",
+    "get_status_color",
+    "is_failure_event",
+    "omit_payment_authorization_outputs",
+    "parse_agent_outputs",
+    "parse_capacity_provider_outputs",
+    "parse_config_bundle_outputs",
+    "parse_dataset_outputs",
+    "parse_evaluator_outputs",
+    "parse_gateway_outputs",
+    "parse_harness_outputs",
+    "parse_knowledge_base_data_source_outputs",
+    "parse_knowledge_base_outputs",
+    "parse_memory_outputs",
+    "parse_online_eval_outputs",
+    "parse_payment_outputs",
+    "parse_policy_engine_outputs",
+    "parse_policy_outputs",
+    "parse_runtime_endpoint_outputs",
+    "to_pascal_id",
+]

agentcore/cloudformation/bootstrap.py

@@ -0,0 +1,64 @@
+"""CDK bootstrap detection (port of ``cloudformation/bootstrap.ts``)."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from agentcore.aws._aws_errors import aws_error_name
+from agentcore.aws.account import create_aws_client
+
+CDK_TOOLKIT_STACK_NAME = "CDKToolkit"
+
+_USABLE_STATUSES = frozenset({"CREATE_COMPLETE", "UPDATE_COMPLETE", "UPDATE_ROLLBACK_COMPLETE"})
+
+
+@dataclass(frozen=True)
+class BootstrapStatus:
+    is_bootstrapped: bool
+    stack_status: str | None = None
+    bootstrap_version: int | None = None
+
+
+def _parse_int(value: str) -> int | None:
+    """``Number.parseInt(value, 10)``: leading digits, else None (NaN)."""
+    digits = ""
+    text = value.strip()
+    sign = ""
+    if text[:1] in "+-":
+        sign, text = text[0], text[1:]
+    for ch in text:
+        if not ch.isdigit():
+            break
+        digits += ch
+    return int(sign + digits) if digits else None
+
+
+def check_bootstrap_status(region: str) -> BootstrapStatus:
+    """Whether the environment has a usable ``CDKToolkit`` stack, and its bootstrap version."""
+    cfn = create_aws_client("cloudformation", region)
+    try:
+        resp = cfn.describe_stacks(StackName=CDK_TOOLKIT_STACK_NAME)
+    except Exception as exc:
+        # Stack doesn't exist - not bootstrapped
+        if aws_error_name(exc) == "ValidationError":
+            return BootstrapStatus(is_bootstrapped=False)
+        raise
+
+    stacks = resp.get("Stacks") or []
+    if not stacks:
+        return BootstrapStatus(is_bootstrapped=False)
+    stack = stacks[0]
+    status = stack.get("StackStatus")
+    version_output = next(
+        (o.get("OutputValue") for o in stack.get("Outputs") or [] if o.get("OutputKey") == "BootstrapVersion"), None
+    )
+    version = _parse_int(version_output) if version_output else None
+    return BootstrapStatus(is_bootstrapped=status in _USABLE_STATUSES, stack_status=status, bootstrap_version=version)
+
+
+def format_cdk_environment(account_id: str, region: str) -> str:
+    """Environment string for ``cdk bootstrap`` (``aws://<account>/<region>``)."""
+    return f"aws://{account_id}/{region}"
+
+
+__all__ = ["CDK_TOOLKIT_STACK_NAME", "BootstrapStatus", "check_bootstrap_status", "format_cdk_environment"]

agentcore/cloudformation/logical_ids.py

@@ -0,0 +1,37 @@
+"""CloudFormation logical-ID helpers (port of ``cloudformation/logical-ids.ts``).
+
+A minimal copy of the logical ID utilities the CLI needs for parsing stack outputs; the full
+implementation lives in ``@aws/agentcore-cdk``.
+"""
+
+from __future__ import annotations
+
+import re
+
+_LOGICAL_ID_PATTERN = re.compile(r"^[a-zA-Z][a-zA-Z0-9]*$")
+MAX_LOGICAL_ID_LENGTH = 255
+
+
+def _assert_logical_id(value: str) -> None:
+    if not value or len(value) > MAX_LOGICAL_ID_LENGTH or not _LOGICAL_ID_PATTERN.match(value):
+        raise ValueError(
+            f'Invalid CloudFormation logical ID: "{value}". Must start with a letter, contain only alphanumerics, '
+            f"and be <= {MAX_LOGICAL_ID_LENGTH} characters."
+        )
+
+
+def _to_logical_id_part(name: str) -> str:
+    """``"my-gateway"`` -> ``"MyGateway"``, ``"my_tool"`` -> ``"MyTool"``."""
+    return "".join(part[:1].upper() + part[1:] for part in re.split(r"[-_]", name))
+
+
+def to_pascal_id(*parts: str) -> str:
+    """Convert dynamic names to a valid CloudFormation logical ID in PascalCase."""
+    if not parts:
+        raise ValueError("toPascalId requires at least one part")
+    value = "".join(_to_logical_id_part(p) for p in parts)
+    _assert_logical_id(value)
+    return value
+
+
+__all__ = ["MAX_LOGICAL_ID_LENGTH", "to_pascal_id"]