import os
import sys

sys.path.insert(0, ".")
from src.assistant_tool.atlas.atlas_tree_planner import AtlasTreePlanner

planner = AtlasTreePlanner()
os.makedirs("config", exist_ok=True)
with open("config/atlas_monk_ci_tree.toml", "w", encoding="utf-8") as f:
    f.write("# AutoPOE2 - Atlas Passive Tree Blueprint: Monk CI Archetype (Doc 65 / Doc 70)\n")
    f.write("# 115 Points DAG Optimization: Delirium, Beyond, Ritual, Waystone Sustain T16\n\n")
    f.write('archetype = "Monk_CI"\n')
    f.write('description = "Max Profit Atlas Tree optimized for Monk CI Chaos Immune High AOE Pack Clear"\n')
    f.write("total_points = 115\n\n")
    for nid in planner._allocation_sequence:
        node = planner._nodes[nid]
        f.write("[[nodes]]\n")
        f.write(f'node_id = "{node.node_id}"\n')
        f.write(f'name = "{node.name}"\n')
        f.write(f'phase = "{node.phase.value}"\n')
        f.write(f"priority_order = {node.priority_order}\n")
        f.write(f'category = "{node.category}"\n')
        f.write(f'stat_description = "{node.stat_description}"\n')
        prereqs_str = ", ".join(f'"{p}"' for p in node.prerequisites)
        f.write(f"prerequisites = [{prereqs_str}]\n\n")

print(f"Wrote {len(planner._allocation_sequence)} nodes to config/atlas_monk_ci_tree.toml")