Enhance graph visualizer to map items, stats, and abilities for a complete content journey

Author: DerekGooding

scripts/export_graph_mermaid.py

@@ -10,60 +10,97 @@ def load_graph():
 
 def generate_mermaid():
     nodes = load_graph()
+    
     lines = ["graph TD"]
 
-    # Define styles
+    # Define styles with better contrast
     lines.append("classDef quest fill:#4b0082,stroke:#f9f,stroke-width:2px,color:#fff;")
     lines.append("classDef location fill:#00008b,stroke:#ccf,stroke-width:2px,color:#fff;")
     lines.append("classDef cadence fill:#006400,stroke:#cfc,stroke-width:2px,color:#fff;")
+    lines.append("classDef item fill:#444,stroke:#fff,stroke-width:1px,color:#fff;")
+    lines.append("classDef stat fill:#8b4513,stroke:#ff8c00,stroke-width:1px,color:#fff;")
+    lines.append("classDef ability fill:#2f4f4f,stroke:#00ced1,stroke-width:1px,color:#fff;")
     lines.append("classDef root fill:#8b8b00,stroke:#ff9,stroke-width:4px,color:#fff;")
 
-    # Filter for significant nodes to avoid clutter
-    # Focus on Quests, Locations, Cadences
-    significant_types = ["Quest", "Location", "Cadence"]
-    sig_nodes = {n["id"]: n for n in nodes if n["type"] in significant_types}
-
-    for nid, node in sig_nodes.items():
-        # Node Label
-        name = node["name"].replace('"', "'")
-        style = ""
+    # We want to show almost everything now
+    # Filter out some very common ones if they make it too noisy, 
+    # but for now let's include all for a "Complete Map"
+    
+    for node in nodes:
+        nid = node["id"]
+        ntype = node["type"]
+        name = node["name"]
 
-        if node["name"] == "Prologue":
-            style = ":::root"
-        elif node["type"] == "Quest":
-            style = ":::quest"
-        elif node["type"] == "Location":
-            style = ":::location"
-        elif node["type"] == "Cadence":
-            style = ":::cadence"
+        # 1. Define Node
+        style = ntype.lower()
+        if nid == "quest_prologue":
+            style = "root"
+            
+        # Shorten some names for the graph
+        display_name = name.replace('"', "'")
+        lines.append(f'{nid}["{display_name}"]:::{style}')
 
-        lines.append(f'{nid}["{name}"]{style}')
+        # 2. In-Edges (Requirements)
+        in_edges = node.get("in_edges", {})
+        
+        # Quest Requirements
+        if "requires_quest" in in_edges:
+            for req_id in in_edges["requires_quest"]:
+                # If Location requires Quest, it's an "Enables" link
+                if ntype == "Location":
+                    lines.append(f"{req_id} -->|Enables| {nid}")
+                else:
+                    lines.append(f"{req_id} -->|Requires| {nid}")
+                    
+        # Ability Requirements (for Refinements usually)
+        if "requires_ability" in in_edges:
+            for req_id in in_edges["requires_ability"]:
+                lines.append(f"{req_id} -.->|Allows| {nid}")
+                
+        # Stat Requirements
+        if "requires_stat" in in_edges:
+            for stat_name, val in in_edges["requires_stat"].items():
+                stat_id = f"stat_{stat_name.lower()}"
+                lines.append(f"{stat_id} -.->|Requires {val}| {nid}")
 
-        # 2. Location Unlocks (Quest -> Location)
-        if "unlocks_location" in node["out_edges"]:
-            for edge in node["out_edges"]["unlocks_location"]:
-                target_id = edge["targetId"]
-                if target_id in sig_nodes:
+        # 3. Out-Edges (Unlocks / Rewards / Productions)
+        out_edges = node.get("out_edges", {})
+        
+        # Cadence/Ability Unlocks
+        for unlock_type in ["unlocks_cadence", "unlocks_ability", "provides_ability"]:
+            if unlock_type in out_edges:
+                for edge in out_edges[unlock_type]:
+                    target_id = edge["targetId"]
                     lines.append(f"{nid} ==>|Unlocks| {target_id}")
+                    
+        # Location Unlocks (explicit)
+        if "unlocks_location" in out_edges:
+            for edge in out_edges["unlocks_location"]:
+                target_id = edge["targetId"]
+                lines.append(f"{nid} ==>|Unlocks| {target_id}")
 
-        # 3. Requirements (Location -> Quest)
-        if "requires_quest" in node["in_edges"]:
-            for req_id in node["in_edges"]["requires_quest"]:
-                if req_id in sig_nodes:
-                    # Avoid duplicate edges with 'requires_quest' which is also on Quest nodes.
-                    if node["type"] == "Location":
-                        lines.append(f"{req_id} -->|Enables| {nid}")
-                    elif node["type"] == "Quest":
-                        lines.append(f"{req_id} -->|Requires| {nid}")
+        # Item Rewards / Production
+        for reward_type in ["rewards", "produces", "rewards_item"]:
+            if reward_type in out_edges:
+                for edge in out_edges[reward_type]:
+                    target_id = edge["targetId"]
+                    qty = edge.get("quantity", 1)
+                    lines.append(f"{nid} --o|Gives {qty}| {target_id}")
 
-        # 4. Cadence Unlocks (Quest -> Cadence)
-        if "unlocks_cadence" in node["out_edges"]:
-            for edge in node["out_edges"]["unlocks_cadence"]:
+        # Item Consumption
+        if "consumes" in out_edges:
+            for edge in out_edges["consumes"]:
                 target_id = edge["targetId"]
-                if target_id in sig_nodes:
-                    lines.append(f"{nid} ==>|Unlocks| {target_id}")
+                qty = edge.get("quantity", 1)
+                # Link Item to the consumer
+                lines.append(f"{target_id} --x|Consumes {qty}| {nid}")
+
+        # Containment (Optional, but useful for locations)
+        if ntype == "Location" and "contains" in out_edges:
+            for edge in out_edges["contains"]:
+                target_id = edge["targetId"]
+                # lines.append(f"{nid} -.->|Home of| {target_id}") # Usually too noisy
 
-    # Output
     return "\n".join(lines)
 
 if __name__ == "__main__":

scripts/visualize_graph.py

@@ -40,6 +40,9 @@ def generate_html(mermaid_code):
             .quest-box {{ background-color: #4b0082; border: 1px solid #f9f; }}
             .location-box {{ background-color: #00008b; border: 1px solid #ccf; }}
             .cadence-box {{ background-color: #006400; border: 1px solid #cfc; }}
+            .item-box {{ background-color: #444; border: 1px solid #fff; }}
+            .stat-box {{ background-color: #8b4513; border: 1px solid #ff8c00; }}
+            .ability-box {{ background-color: #2f4f4f; border: 1px solid #00ced1; }}
             
             #graph-container {{
                 width: 100%;
@@ -64,6 +67,9 @@ def generate_html(mermaid_code):
                 <div class="legend-item"><div class="legend-box quest-box"></div> Quest</div>
                 <div class="legend-item"><div class="legend-box location-box"></div> Location</div>
                 <div class="legend-item"><div class="legend-box cadence-box"></div> Cadence</div>
+                <div class="legend-item"><div class="legend-box item-box"></div> Item</div>
+                <div class="legend-item"><div class="legend-box stat-box"></div> Stat</div>
+                <div class="legend-item"><div class="legend-box ability-box"></div> Ability</div>
             </div>
         </div>
         <div id="graph-container">