1
0
mirror of synced 2024-11-24 14:30:11 +01:00
bemaniutils/bemani/utils/jsx.py

66 lines
2.0 KiB
Python

import argparse
import os
from react.jsx import JSXTransformer # type: ignore
from bemani.frontend.app import polyfill_fragments
SCRIPT_PATH: str = os.path.dirname(os.path.realpath(__file__))
def main() -> None:
parser = argparse.ArgumentParser(
description="Prebuild all JSX files so they can be statically served by nginx or similar."
)
parser.add_argument(
"-d",
"--output-directory",
help="Output directory for the JSX files.",
type=str,
default="./build/jsx",
)
args = parser.parse_args()
outdir = os.path.abspath(args.output_directory)
print(f"Compiling all files into {outdir}...")
os.makedirs(outdir, exist_ok=True)
basedir = os.path.abspath(os.path.join(SCRIPT_PATH, "../frontend/static"))
if not basedir.endswith("/"):
basedir += "/"
dirs = [basedir]
files = []
while dirs:
curdir = dirs.pop()
for dirpath, dnames, fnames in os.walk(curdir):
for fname in fnames:
if fname.endswith(".react.js"):
fullpath = os.path.join(dirpath, fname)
if fullpath.startswith(basedir):
files.append(fullpath[len(basedir) :])
for dname in dnames:
if dname == "." or dname == "..":
continue
dirs.append(os.path.join(dirpath, dname))
transformer = JSXTransformer()
for fname in sorted(list(set(files))):
print(f"Building {fname}...")
infile = os.path.join(basedir, fname)
outfile = os.path.join(outdir, fname)
os.makedirs(os.path.dirname(outfile), exist_ok=True)
with open(infile, "rb") as f:
jsx = transformer.transform_string(
polyfill_fragments(f.read().decode("utf-8"))
).encode("utf-8")
print(f"Writing {outfile}...")
with open(outfile, "wb") as f:
f.write(jsx)
if __name__ == "__main__":
main()