from collections import defaultdict
import json

with open("assets/spirv/spirv.core.grammar.json") as f:
    j = json.load(f)

bit_enums = defaultdict(dict)
value_enums = defaultdict(dict)
for operand_kind in j["operand_kinds"]:
    category = operand_kind["category"]
    kind = operand_kind["kind"]

    if category == "BitEnum":
        enumerants = operand_kind["enumerants"]
        for enumerant in enumerants:
            bit_enums[kind][enumerant["enumerant"]] = enumerant["value"]
    elif category == "ValueEnum":
        enumerants = operand_kind["enumerants"]
        for enumerant in enumerants:
            value_enums[kind][enumerant["enumerant"]] = enumerant["value"]

out = []

out += [
    "#![allow(unreachable_patterns)]",
    "use anyhow::{bail, Result};",
    "",
    "pub fn enum_to_str(ety: &str, value: u32) -> Result<String> {",
    "    let out: String = match ety {",
]

# ValueEnum
for kind, enumerants in value_enums.items():
    if len(enumerants) == 0:
        continue

    out += [
        f'        "{kind}" => match value {{',
    ]
    for name, value in enumerants.items():
        out += [
            f'            {value} => "{name}".to_owned(),',
        ]
    out += [
        "            _ => value.to_string(),",
        "        }",
    ]

# BitEnum
for kind, enumerants in bit_enums.items():
    if len(enumerants) == 0:
        continue

    out += [
        f'        "{kind}" => match value {{',
        '            0 => "None".to_owned(),',
        "            _ => value.to_string(),",
        "        }",
    ]

out += [
    '        _ => bail!("unknown enum: {}", ety),',
    "    };",
    "    Ok(out)",
    "}",
    "",
]

with open("spirq-spvasm/src/generated/enum_to_str.rs", "w") as f:
    f.write("\n".join(out))