Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import json
 from jsonschema import validate, ValidationError
 import re
+import networkx as nx
+import matplotlib.pyplot as plt
 
 # Mock existing resources (customizable)
 existing_resources = {
@@ -32,7 +34,8 @@ arm_schema = {
                     "type": {"type": "string"},
                     "name": {"type": "string"},
                     "location": {"type": "string"},
-                    "properties": {"type": "object"}
+                    "properties": {"type": "object"},
+                    "dependsOn": {"type": "array", "items": {"type": "string"}}
                 }
             }
         },
@@ -62,6 +65,68 @@ def resolve_template_expressions(template):
     resolved_str = re.sub(r'\[(\w+\.\w+)\]', resolve_expression, template_str)
     return json.loads(resolved_str)
 
+def estimate_cost(resources):
+    # This is a simplified cost estimation. In a real scenario, you'd need to
+    # integrate with Azure's pricing API or use a more comprehensive pricing model.
+    cost_estimates = {
+        "Microsoft.Storage/storageAccounts": 10,  # $10 per month
+        "Microsoft.Compute/virtualMachines": 50,  # $50 per month
+        "Microsoft.Web/sites": 30,  # $30 per month
+        "Microsoft.Sql/servers": 40,  # $40 per month
+        # Add more resource types and their estimated costs
+    }
+    
+    total_cost = 0
+    for resource in resources:
+        resource_type = resource.get('type')
+        if resource_type in cost_estimates:
+            total_cost += cost_estimates[resource_type]
+        else:
+            total_cost += 5  # Default cost for unknown resource types
+    
+    return total_cost
+
+def create_dependency_graph(resources):
+    G = nx.DiGraph()
+    for resource in resources:
+        resource_name = resource.get('name')
+        G.add_node(resource_name)
+        dependencies = resource.get('dependsOn', [])
+        for dep in dependencies:
+            G.add_edge(dep, resource_name)
+    return G
+
+def plot_dependency_graph(G):
+    plt.figure(figsize=(12, 8))
+    pos = nx.spring_layout(G)
+    nx.draw(G, pos, with_labels=True, node_color='lightblue', 
+            node_size=3000, font_size=8, font_weight='bold')
+    edge_labels = nx.get_edge_attributes(G, 'weight')
+    nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels)
+    plt.title("Resource Dependency Graph")
+    return plt
+
+def lint_template(template):
+    lint_results = []
+    
+    # Check 1: Ensure all resources have tags
+    for resource in template.get('resources', []):
+        if 'tags' not in resource:
+            lint_results.append(f"Warning: Resource '{resource.get('name')}' does not have any tags.")
+    
+    # Check 2: Ensure parameters have descriptions
+    for param_name, param_value in template.get('parameters', {}).items():
+        if 'metadata' not in param_value or 'description' not in param_value['metadata']:
+            lint_results.append(f"Warning: Parameter '{param_name}' does not have a description.")
+    
+    # Check 3: Warn about hardcoded values in resource properties
+    for resource in template.get('resources', []):
+        for prop, value in resource.get('properties', {}).items():
+            if isinstance(value, (str, int, float)) and not isinstance(value, bool):
+                lint_results.append(f"Info: Consider using a parameter for the hardcoded value in resource '{resource.get('name')}', property '{prop}'.")
+    
+    return lint_results
+
 def simulate_deployment(arm_template):
     try:
         template = json.loads(arm_template)
@@ -69,7 +134,7 @@ def simulate_deployment(arm_template):
         # Validate template
         is_valid, validation_message = validate_template(template)
         if not is_valid:
-            return [], [], [], [], validation_message
+            return [], [], [], [], 0, [], validation_message
 
         # Resolve expressions
         resolved_template = resolve_template_expressions(template)
@@ -107,12 +172,21 @@ def simulate_deployment(arm_template):
             if resource_name not in [r.get('name') for r in resolved_template.get('resources', [])]:
                 resources_to_delete.append(resource_name)
 
-        return resources_to_create, resources_to_update, resources_to_delete, resource_details, "Simulation completed successfully."
+        # Estimate cost
+        estimated_cost = estimate_cost(resource_details)
+
+        # Create dependency graph
+        dependency_graph = create_dependency_graph(resolved_template.get('resources', []))
+
+        # Lint template
+        lint_results = lint_template(template)
+
+        return resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, "Simulation completed successfully."
     except json.JSONDecodeError:
-        return [], [], [], [], "Invalid JSON format. Please check your ARM template."
+        return [], [], [], [], 0, None, [], "Invalid JSON format. Please check your ARM template."
 
 # Streamlit UI
-st.title("Enhanced ARM Template Simulator")
+st.title("Comprehensive ARM Template Simulator")
 st.subheader("Paste your ARM template below to simulate its deployment.")
 
 # Input box for ARM template
@@ -121,7 +195,7 @@ template_input = st.text_area("Paste ARM Template JSON here:", height=300)
 # Button to simulate the evaluation
 if st.button("Simulate Template"):
     if template_input:
-        resources_to_create, resources_to_update, resources_to_delete, resource_details, message = simulate_deployment(template_input)
+        resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, message = simulate_deployment(template_input)
         
         st.subheader("Simulation Results:")
         st.write(message)
@@ -153,6 +227,22 @@ if st.button("Simulate Template"):
                 st.write("**Properties:**")
                 st.json(resource['properties'])
                 st.write("---")
+
+            st.write(f"### Estimated Monthly Cost: ${estimated_cost}")
+
+            st.write("### Resource Dependency Graph:")
+            if dependency_graph:
+                fig = plot_dependency_graph(dependency_graph)
+                st.pyplot(fig)
+            else:
+                st.write("No dependencies found between resources.")
+
+            st.write("### Template Lint Results:")
+            if lint_results:
+                for result in lint_results:
+                    st.write(result)
+            else:
+                st.write("No linting issues found.")
         else:
             st.write("No changes detected in the simulation.")
     else: