Wip stuff

scripts/parse_lua_api.py

@@ -0,0 +1,102 @@
+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)