# topology_api/seed/seed_layout.py
"""Seed diagram layout and node positions for the engineering overview SVG."""
import json
from topology_api.db import write_cursor


def seed():
    with write_cursor() as cur:
        # Get base
        cur.execute("SELECT id FROM bases WHERE code = 'guanghan'")
        base_id = cur.fetchone()["id"]

        # Upsert layout
        cur.execute("""
            INSERT INTO diagram_layouts (id, base_id, layout_key, name,
                canvas_width, canvas_height, is_default)
            VALUES (gen_random_uuid(), %(base)s, 'engineering_overview_v1',
                '工程总图 v1', 1500, 1900, true)
            ON CONFLICT (base_id, layout_key) DO UPDATE
                SET name = EXCLUDED.name, is_default = true
            RETURNING id
        """, {"base": base_id})
        layout_id = cur.fetchone()["id"]
        print(f"Layout: {layout_id}")

        # Component key → component_id mapping
        cur.execute("SELECT component_key, id FROM components WHERE base_id = %(base)s",
                    {"base": base_id})
        comp_map = {r["component_key"]: r["id"] for r in cur.fetchall()}

        # Layout and concise engineering labels. All visual semantics live here,
        # not in the Wiki page.
        nodes = [
            # Core cross
            ("core_cabin", 630, 470, 240, 200, "core-card", None, 10,
             "四向安装节点", "北 / 东 / 南 / 西", {}, ["core_cabin"]),
            ("habitat_2", 630, 220, 240, 140, "node-card", "居住舱#2", 5,
             "核心舱北部安装口", "4铺位 + 2睡袋", {}, ["habitat_2"]),
            ("habitat_1", 630, 760, 240, 120, "node-card", "居住舱#1", 5,
             "核心舱南部安装口", None, {}, ["habitat_1"]),
            ("greenhouse", 330, 490, 220, 160, "node-card", "温室模块", 5,
             "核心舱西部安装口", "远端连接吴刚二号", {}, ["greenhouse"]),
            ("lab_module", 950, 490, 220, 160, "node-card", "实验舱", 5,
             "核心舱东部安装口", "三温区实验舱", {}, ["lab_module"]),
            ("wugang_2", 40, 495, 210, 150, "vehicle-card", None, 3,
             "Block 2 载人着陆器", "温室远端接口", {}, ["wugang_2"]),

            # Power cross
            ("power_module_hub", 650, 1260, 200, 180, "power-node-card",
             "电力模块", 8, "四向安装节点", "核电与资源中枢", {},
             ["power_module_hub"]),
            ("power_storage_rack", 600, 990, 300, 160, "power-card",
             "北部模块", 5, "电力模块存储支架", "加压走道端",
             {"radiator": True}, ["power_storage_rack"]),
            ("nuclear_reactor", 620, 1540, 260, 150, "power-card",
             "南部模块", 5, "核反应堆", "30%功率",
             {"radiator": True}, ["nuclear_reactor"]),

            # Power west/east groups (multiple DB components per SVG node)
            ("power_west_group", 280, 1250, 260, 200, "power-card",
             "西部模块", 5, "ISRU · 采矿", "矿石 / 燃料存储",
             {"radiator": True},
             ["isru_module", "mining_module", "ore_fuel_storage"]),
            ("power_east_group", 960, 1250, 260, 200, "power-card",
             "东部模块", 5, "核燃料存储 · 离心机", None,
             {"radiator": True},
             ["fuel_storage_mod", "centrifuge_mod"]),

            # Bingluns
            ("binglun_1", 30, 1280, 210, 140, "vehicle-card", None, 2,
             "电力模块西侧", "燃料运输", {}, ["binglun_1"]),
            ("binglun_2", 645, 1740, 210, 120, "vehicle-card", None, 2,
             "电力模块南侧", None, {}, ["binglun_2"]),
            ("binglun_3", 1260, 1280, 210, 140, "vehicle-card", None, 2,
             "电力模块东侧", "Block 2", {}, ["binglun_3"]),
        ]

        # Delete existing nodes and re-insert
        cur.execute("DELETE FROM diagram_nodes WHERE layout_id = %(lid)s",
                    {"lid": layout_id})

        for (
            node_key, x, y, w, h, style, label, z, display_meta,
            display_hint, decorations, member_keys,
        ) in nodes:
            primary_component_id = (
                comp_map[member_keys[0]] if len(member_keys) == 1 else None
            )
            cur.execute("""
                INSERT INTO diagram_nodes
                    (id, layout_id, component_id, node_key, x, y,
                     width, height, style_key, label_override, z_index,
                     display_meta, display_hint, decorations)
                VALUES (gen_random_uuid(), %(lid)s, %(cid)s, %(nkey)s,
                        %(x)s, %(y)s, %(w)s, %(h)s,
                        %(style)s, %(label)s, %(z)s,
                        %(meta)s, %(hint)s, %(decorations)s)
                RETURNING id
            """, {
                "lid": layout_id, "cid": primary_component_id,
                "nkey": node_key,
                "x": x, "y": y, "w": w, "h": h,
                "style": style, "label": label, "z": z,
                "meta": display_meta, "hint": display_hint,
                "decorations": json.dumps(decorations),
            })
            node_id = cur.fetchone()["id"]
            for display_order, member_key in enumerate(member_keys):
                cur.execute("""
                    INSERT INTO diagram_node_members
                        (diagram_node_id, component_id, display_order)
                    VALUES (%(node)s, %(component)s, %(display_order)s)
                """, {
                    "node": node_id,
                    "component": comp_map[member_key],
                    "display_order": display_order,
                })

        print(f"Nodes inserted: {len(nodes)}")


if __name__ == "__main__":
    seed()