import_to_call.py

#!/bin/python3.9
import sys
from typing import List

def get_args(sig: str):
    start_index = sig.index("(", sig.index("fn ") + 3) + 1
    end_index = sig.index(")", start_index)
    arg_list = sig[start_index:end_index]
    out_args: List[str] = []
    arg_count = arg_list.count(": ")
    current_arg = 0
    while current_arg < arg_count:
        end_index = arg_list.index(": ")
        out_args.append(arg_list[:end_index])
        if current_arg + 1 != arg_count:
            start_index = arg_list.index(", ") + 2
            arg_list = arg_list[start_index:]
        current_arg += 1
    return out_args

def get_name(sig: str):
    start_index = sig.index("fn ") + 3
    end_index = sig.index("(", start_index)
    return sig[start_index:end_index]

if len(sys.argv) != 3:
    print("./import_to_call.py <infile> <outfile>")
    sys.exit(2)

in_file = open(sys.argv[1], 'r')
out_file = open(sys.argv[2], 'w')

lines = in_file.readlines()
in_file.close()
out_file.write("use crate::*;\n\nmod impl_ {\n    use crate::*;\n\n    extern \"C\" {\n")
filtered_lines: List[str] = []
for line in lines:
    out_file.write("    " + line)
    line = line.strip()
    if not line.startswith("pub(super) fn"):
        continue
    filtered_lines.append(line)

out_file.write("    }\n}\n\n")

for line in filtered_lines:
    params = get_args(line)
    func_name = get_name(line)
    func_call = "pub fn" + line.removeprefix("pub(super) fn")
    func_call = func_call.removesuffix(";") + " {"
    returns_vec = func_call.endswith("-> cpp::simd::Vector2 {") or func_call.endswith("-> cpp::simd::Vector3 {") or func_call.endswith("-> cpp::simd::Vector4 {")
    func_call = func_call.replace("-> cpp::simd::Vector", "-> phx::Vec").replace("*const ", "&").replace("*mut", "&mut").replace("&mut lua_State", "*mut lua_State")
    func_call += "\n    unsafe {\n        impl_::" + func_name + "("
    for idx, param in enumerate(params):
        func_call += param
        if idx + 1 != len(params):
            func_call += ", "
    func_call += ")"
    if returns_vec:
        func_call += ".into()"
    func_call += "\n    }\n}"
    out_file.write(func_call + "\n\n")