papermario/tools/star_rod_enum_to_decomp.py

#!/usr/bin/env python3

import argparse
import sys
import re
import os
from glob import glob


def create_enum(file_content, prefix, ordering):
    ret = ""
    max_size = 0

    if ordering:
        for key, value in re.findall(r"(\S+)\s+=\s+(\S+)", file_content):
            if len(key) > max_size:
                max_size = len(key)
    else:
        for key, value in re.findall(r"(\S+)\s+=\s+(\S+)", file_content):
            if len(value) > max_size:
                max_size = len(value)

    if prefix is not None:
        max_size += len(prefix)
    else:
        prefix = ""

    for key, value in re.findall(r"(\S+)\s+=\s+(\S+)", file_content):
        if ordering:
            key = (
                "_".join(re.sub(r"([A-Z]{1,2})", r" \1", key).split())
                .replace("N_P_C", "NPC")
                .replace("__", "_")
                .replace("-", "")
            )
            key = prefix.upper() + "_{:<{width}}".format(key, width=max_size + 2).upper()
            ret += "    " + key + " = 0x" + "{:>{fill}{width}}".format(value, fill=0, width=8) + ",\n"
        else:
            value = (
                "_".join(re.sub(r"([A-Z]{1,2})", r" \1", value).split())
                .replace("N_P_C", "NPC")
                .replace("__", "_")
                .replace("-", "")
            )
            value = prefix.upper() + "_{:<{width}}".format(value, width=max_size + 2).upper()
            ret += "    " + value + " = 0x" + "{:>{fill}{width}}".format(key, fill=0, width=8) + ",\n"

    ret += "};\n"

    return ret


def single_translation(file_path):
    ret = ""

    try:
        file = open(file_path)
        file_content = file.read()
        file.close()
    except:
        print("File not found at the given path.")
        return "t", "y"

    enum_namespace = re.search(r"(\w*)\s+%\s+namespace", file_content).group(1)
    enum_name = re.search(r"(\w*)\s+%\s+library name", file_content).group(1)
    reversed_order = re.search(r"(\w*)\s+%\s+reversed", file_content).group(1)
    ret += "enum " + enum_name + " {\n"

    if reversed_order == "true":
        ret += create_enum(file_content, enum_namespace, True)
    else:
        ret += create_enum(file_content, enum_namespace, False)

    return enum_name, ret


def recursive_translation(database_path):
    ret = ""

    stuff = [f for f in glob(database_path + "/**/*.*", recursive=True) if f.endswith(".enum") or f.endswith(".flags")]
    for element in stuff:
        try:
            file = open(element)
            file_content = file.read()
            file.close()
        except:
            continue

        enum_namespace = re.search(r"(\w*)\s+%\s+namespace", file_content).group(1)
        enum_name = re.search(r"(\w*)\s+%\s+library name", file_content).group(1)
        reversed_order = re.search(r"(\w*)\s+%\s+reversed", file_content).group(1)
        ret += "enum " + enum_name + " {\n"

        if reversed_order == "true":
            ret += create_enum(file_content, enum_namespace, True)
        else:
            ret += create_enum(file_content, enum_namespace, False)

        ret += "\n"

    return "enums_and_flags", ret


def main(args):
    ret = ""

    if args.recursive:
        enum_name, result = recursive_translation(args.query)
    else:
        enum_name, result = single_translation(args.query)

    ret += result

    file = open("" + enum_name + ".txt", "w+")
    file.write(ret)
    file.close()


parser = argparse.ArgumentParser(
    description="Convert a StarRod enum into an enum that is in a decomp compatible format"
)
parser.add_argument("query", help="StarRod enum file or folder")
parser.add_argument(
    "-r",
    "--recursive",
    help="recursively convert all files to enums",
    type=bool,
    required=False,
)

args = parser.parse_args()


if __name__ == "__main__":
    main(parser.parse_args())