import json
import os

commands_dir = "./commands"
files = os.listdir(commands_dir)
table = {}
container = set()
for file in files:
    j = json.load(open(f"{commands_dir}/{file}"))
    cmd_name = list(j.keys())[0]
    j = j[cmd_name]

    print(cmd_name)
    if cmd_name == "SORT" or cmd_name == "MIGRATE":
        continue
    if "command_flags" not in j:
        print(f"{file} No command_flags.")
        continue

    flags = j["command_flags"]
    group = j["group"]
    if (("WRITE" in flags or "MAY_REPLICATE" in flags) and "BLOCKING" not in flags) or cmd_name in ("PING", "SELECT"):
        key_specs = []
        if "key_specs" in j:
            for key_spec in j["key_specs"]:
                begin_search = key_spec["begin_search"]
                find_keys = key_spec["find_keys"]
                key_specs.append({
                    "begin_search": begin_search,
                    "find_keys": find_keys
                })
        if "container" in j:
            cmd_name = j["container"] + "-" + cmd_name
            container.add(j["container"])
        print(f"group: {group}")
        print(f"flags: {flags}")
        if group not in table:
            table[group] = {}
        table[group][cmd_name] = key_specs

with open("table.json", "w") as f:
    json.dump({
        "table": table,
        "container": list(container)
    }, f, indent=4)