TRX/tools/generate_init

#!/usr/bin/env python3
import argparse
import json
from pathlib import Path

from shared.versioning import generate_version

try:
    import pyjson5
except ImportError:
    import json5 as pyjson5

REPO_DIR = Path(__file__).parent.parent
SHIP_DIR = REPO_DIR / "data/ship"
SRC_DIR = REPO_DIR / "src"
CONFIG_DIR = SHIP_DIR / "cfg"
GAMEFLOW_PATH = CONFIG_DIR / "TR1X_gameflow.json5"


TEMPLATE = """
#include "game/gameflow.h"
#include "global/types.h"

#include <stddef.h>

const char *g_TR1XVersion = "{version}";

GAMEFLOW_DEFAULT_STRING g_GameFlowDefaultStrings[] = {{
{strings}
    {{ 0, NULL }},
}};
"""


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser()
    parser.add_argument("-o", "--output", type=Path)
    return parser.parse_args()


def get_init_c() -> str:
    gameflow = pyjson5.loads(GAMEFLOW_PATH.read_text(encoding="utf-8"))
    return TEMPLATE.format(
        version=generate_version(),
        strings="\n".join(
            f"    {{ GS_{key}, {json.dumps(value)} }},"
            for key, value in gameflow["strings"].items()
        ),
    )


def update_init_c(output_path: Path) -> None:
    new_text = get_init_c()
    if not output_path.exists() or output_path.read_text() != new_text:
        output_path.write_text(new_text)


def main() -> None:
    args = parse_args()
    if args.output:
        update_init_c(output_path=args.output)
    else:
        print(args.outptu)


if __name__ == "__main__":
    main()