import json

path = "/home/quant/.local/share/Steam/steamapps/common/Noita/tools_modding/lua_api_documentation.html"

lines = open(path).readlines()

lines_iter = iter(lines)

parsed = []

def parse_arg(arg_s):
    if "|" in arg_s:
        raise ValueError("multiple argument types not supported")
    other, *default = arg_s.split(" = ", maxsplit=1)
    if default:
        default = default[0]
    else:
        default = None
    name, typ = other.split(":", maxsplit=1)
    return {
        "name": name,
        "typ": typ,
        "default": default,
    }

def parse_ret(ret_s):
    if not ret_s:
        return None
    
    optional = ret_s.endswith("|nil")
    ret_s = ret_s.removesuffix("|nil")
    
    if "|" in ret_s:
        raise ValueError("multiple return types not supported")
    if "{" in ret_s:
        raise ValueError("tables in returns not supported")
    
    typ = other
    name = None
    if ":" in other:
        name, typ = other.split(":", maxsplit=1)

    if name == "entity_id":
        typ = "entity_id"

    return {
        "name": name,
        "typ": typ,
        "optional": optional
    }
    

ignore = {

}
skipped = 0

# 2 lazy 2 parse xml properly
try:
    while True:
        line = next(lines_iter)
        if line.startswith('<th><span class="function">'):
            fn_line = line.strip()
            ret_line = next(lines_iter).strip()
            desc_line = next(lines_iter).strip()
            
            fn_name, other = fn_line.removeprefix('<th><span class="function">').split('</span>(<span class="field_name">', maxsplit=1)
            args = other.removesuffix('</span><span class="func">)</span></th>').strip().split(", ")
            try:
                args = [parse_arg(arg) for arg in args if ":" in arg]
            except ValueError as e:
                skipped += 1
                print(f"Skipping {fn_name}: {e}")
                continue

            ret = ret_line.removeprefix('<th><span class="field_name">').removesuffix('</span></th></th>').strip()
            try:
                ret = parse_ret(ret)
            except ValueError as e:
                print(f"Skipping {fn_name}: {e}")
                skipped += 1
                continue
            desc = desc_line.removeprefix('<th><span class="description">').removesuffix('</span></th></th>').strip().replace("</br>", "\n")
            if not desc:
                desc = "Nolla forgot to include a description :("

            #print(fn_line, ret_line, desc_line)
            
            if fn_name not in ignore:
                #print(fn_name, args, "->", ret)
                parsed.append({
                    "fn_name": fn_name,
                    "args": args,
                    "desc": desc,
                })

except StopIteration:
    pass


print("Total skipped:", skipped)
json.dump(parsed, open("ewext/noita_api_macro/src/lua_api.json", "w"), indent=None)