diagram-to-iac 0.7.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

diagram_to_iac/agents/supervisor_langgraph/demonstrator.py

@@ -0,0 +1,22 @@
+class DryRunDemonstrator:
+    """Helper to display dry-run information and collect user feedback."""
+
+    def show_issue(self, title: str, body: str) -> bool:
+        print("=== DRY RUN: GitHub issue would be created ===")
+        print(f"Title: {title}")
+        print(f"Body:\n{body}")
+        try:
+            answer = input("Proceed after reviewing? [y/N]: ").strip().lower()
+        except (KeyboardInterrupt, EOFError):
+            answer = ""
+        
+        # Return True only if user explicitly chooses to proceed
+        should_proceed = answer in ['y', 'yes']
+        
+        if should_proceed:
+            print("✅ User chose to proceed with issue creation")
+        else:
+            print("❌ User chose not to proceed, dry-run completed")
+            
+        return should_proceed
+

diagram_to_iac/agents/supervisor_langgraph/guards.py

@@ -0,0 +1,23 @@
+"""Utility guards for SupervisorAgent."""
+
+from __future__ import annotations
+
+import os
+import time
+from typing import Iterable
+
+from diagram_to_iac.core.errors import MissingSecretError
+
+
+def check_required_secrets(required: Iterable[str] | None = None) -> None:
+    """Ensure required secrets exist and are not empty."""
+    start = time.perf_counter()
+    secrets = list(required) if required is not None else ["GITHUB_TOKEN"]
+    missing = [s for s in secrets if not os.environ.get(s, "").strip()]
+    if missing:
+        raise MissingSecretError(
+            "Missing required secret(s): " + ", ".join(missing)
+        )
+    # Function should be extremely fast; return quickly
+    _ = time.perf_counter() - start
+

diagram_to_iac/agents/supervisor_langgraph/pat_loop.py

@@ -0,0 +1,49 @@
+import os
+import time
+import logging
+
+from . import __name__ as pkg_name
+from diagram_to_iac.agents.git_langgraph import GitAgentInput
+
+
+def request_and_wait_for_pat(repo_url: str, git_agent, poll_interval: int = 30, timeout: int = 600) -> bool:
+    """Request TF_TOKEN_APP_TERRAFORM_IO from the user and wait until provided.
+
+    Parameters
+    ----------
+    repo_url: str
+        Repository URL used for context in the GitHub issue comment.
+    git_agent: GitAgent
+        Agent used to post a comment requesting the token.
+    poll_interval: int, optional
+        How often to check for the environment variable, in seconds.
+    timeout: int, optional
+        Maximum time to wait before giving up, in seconds.
+
+    Returns
+    -------
+    bool
+        ``True`` if the token was detected before timeout, ``False`` otherwise.
+    """
+    logger = logging.getLogger(f"{pkg_name}.pat_loop")
+    message = (
+        "Please add the Terraform Cloud token `TF_TOKEN_APP_TERRAFORM_IO` as an "
+        "environment variable and reply 'retry init' once done."
+    )
+
+    try:
+        git_agent.run(
+            GitAgentInput(
+                query=f"open issue {message} for repository {repo_url}",
+            )
+        )
+    except Exception as e:  # pragma: no cover - log but continue
+        logger.error(f"Failed to post PAT request comment: {e}")
+
+    env_name = "TF_TOKEN_APP_TERRAFORM_IO"
+    end = time.time() + timeout
+    while time.time() < end:
+        if os.environ.get(env_name):
+            return True
+        time.sleep(poll_interval)
+    return False

diagram_to_iac/agents/supervisor_langgraph/router.py

@@ -0,0 +1,9 @@
+from typing import Dict
+
+STACK_SUPPORT_THRESHOLD = 0.7
+
+
+def route_on_stack(histogram: Dict[str, float]) -> bool:
+    """Return True if any stack falls below the configured support threshold."""
+    return any(value < STACK_SUPPORT_THRESHOLD for value in histogram.values())
+

diagram_to_iac/agents/terraform_langgraph/__init__.py

@@ -0,0 +1,15 @@
+"""
+Terraform LangGraph Agent Package
+
+This package contains the TerraformAgent implementation with its dedicated tools and configuration.
+"""
+
+from .agent import TerraformAgent, TerraformAgentInput, TerraformAgentOutput
+from .parser import classify_terraform_error
+
+__all__ = [
+    "TerraformAgent",
+    "TerraformAgentInput",
+    "TerraformAgentOutput",
+    "classify_terraform_error",
+]