terraformgraph 1.0.1__py3-none-any.whl

terraformgraph/config/logical_connections.yaml

@@ -0,0 +1,183 @@
+# Logical connections between services for diagram rendering
+# Format: source -> target -> connection properties
+
+connections:
+  - source: cloudfront
+    target: s3
+    label: "Origin"
+    type: data_flow
+
+  - source: cloudfront
+    target: alb
+    label: "Origin"
+    type: data_flow
+
+  - source: waf
+    target: cloudfront
+    label: "Protect"
+    type: default
+
+  - source: waf
+    target: alb
+    label: "Protect"
+    type: default
+
+  - source: route53
+    target: cloudfront
+    label: "DNS"
+    type: default
+
+  - source: route53
+    target: alb
+    label: "DNS"
+    type: default
+
+  - source: acm
+    target: cloudfront
+    label: "TLS"
+    type: default
+
+  - source: acm
+    target: alb
+    label: "TLS"
+    type: default
+
+  - source: alb
+    target: ecs
+    label: "Route"
+    type: data_flow
+
+  - source: alb
+    target: ec2
+    label: "Route"
+    type: data_flow
+
+  - source: alb
+    target: lambda
+    label: "Route"
+    type: data_flow
+
+  - source: ecs
+    target: dynamodb
+    label: "Read/Write"
+    type: data_flow
+
+  - source: ecs
+    target: s3
+    label: "Read/Write"
+    type: data_flow
+
+  - source: ecs
+    target: rds
+    label: "Read/Write"
+    type: data_flow
+
+  - source: ecs
+    target: elasticache
+    label: "Cache"
+    type: data_flow
+
+  - source: ecs
+    target: sqs
+    label: "Queue"
+    type: trigger
+
+  - source: ecs
+    target: sns
+    label: "Publish"
+    type: trigger
+
+  - source: lambda
+    target: dynamodb
+    label: "Read/Write"
+    type: data_flow
+
+  - source: lambda
+    target: s3
+    label: "Read/Write"
+    type: data_flow
+
+  - source: lambda
+    target: sqs
+    label: "Process"
+    type: trigger
+
+  - source: sqs
+    target: lambda
+    label: "Trigger"
+    type: trigger
+
+  - source: sns
+    target: sqs
+    label: "Subscribe"
+    type: trigger
+
+  - source: sns
+    target: lambda
+    label: "Trigger"
+    type: trigger
+
+  - source: eventbridge
+    target: lambda
+    label: "Trigger"
+    type: trigger
+
+  - source: eventbridge
+    target: sqs
+    label: "Route"
+    type: trigger
+
+  - source: kms
+    target: s3
+    label: "Encrypt"
+    type: encrypt
+
+  - source: kms
+    target: dynamodb
+    label: "Encrypt"
+    type: encrypt
+
+  - source: kms
+    target: sqs
+    label: "Encrypt"
+    type: encrypt
+
+  - source: kms
+    target: sns
+    label: "Encrypt"
+    type: encrypt
+
+  - source: kms
+    target: secrets_manager
+    label: "Encrypt"
+    type: encrypt
+
+  - source: secrets_manager
+    target: ecs
+    label: "Inject"
+    type: default
+
+  - source: secrets_manager
+    target: lambda
+    label: "Inject"
+    type: default
+
+  - source: cognito
+    target: api_gateway
+    label: "Auth"
+    type: default
+
+  - source: cognito
+    target: alb
+    label: "Auth"
+    type: default
+
+  - source: api_gateway
+    target: lambda
+    label: "Invoke"
+    type: trigger
+
+  - source: step_functions
+    target: lambda
+    label: "Orchestrate"
+    type: trigger

terraformgraph/config_loader.py

@@ -0,0 +1,55 @@
+"""Configuration loader for Terraform Diagram Generator."""
+
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+import yaml
+
+
+class ConfigLoader:
+    """Loads configuration from YAML files with fallback to defaults."""
+
+    def __init__(
+        self,
+        aggregation_rules_path: Optional[Path] = None,
+        logical_connections_path: Optional[Path] = None
+    ):
+        self._config_dir = Path(__file__).parent / "config"
+        self._aggregation_rules_path = aggregation_rules_path or self._config_dir / "aggregation_rules.yaml"
+        self._logical_connections_path = logical_connections_path or self._config_dir / "logical_connections.yaml"
+
+        self._aggregation_rules: Optional[Dict[str, Any]] = None
+        self._logical_connections: Optional[List[Dict[str, Any]]] = None
+
+    def get_aggregation_rules(self) -> Dict[str, Any]:
+        """Load and return aggregation rules."""
+        if self._aggregation_rules is None:
+            self._aggregation_rules = self._load_yaml(self._aggregation_rules_path)
+        return self._aggregation_rules
+
+    def get_logical_connections(self) -> List[Dict[str, Any]]:
+        """Load and return logical connections."""
+        if self._logical_connections is None:
+            data = self._load_yaml(self._logical_connections_path)
+            self._logical_connections = data.get("connections", [])
+        return self._logical_connections
+
+    def _load_yaml(self, path: Path) -> Dict[str, Any]:
+        """Load YAML file and return parsed content."""
+        if not path.exists():
+            raise FileNotFoundError(f"Configuration file not found: {path}")
+
+        with open(path, "r") as f:
+            return yaml.safe_load(f) or {}
+
+    def get_flat_aggregation_rules(self) -> Dict[str, Dict[str, Any]]:
+        """Return aggregation rules flattened to service_name -> config mapping."""
+        rules = self.get_aggregation_rules()
+        flat = {}
+        for category, services in rules.items():
+            for service_name, config in services.items():
+                flat[service_name] = {
+                    "category": category,
+                    **config
+                }
+        return flat