extract.py

import re
import os
import json
import shutil

from py_mini_racer import MiniRacer


PS_CLIENT_DIR = "pokemon-showdown-client"

CLIENT_SRC = [
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/abilities.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/aliases.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/items.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/moves.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/pokedex.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/typechart.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/natures.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/conditions.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/learnsets.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/formats-data.js",
    f"{PS_CLIENT_DIR}/data/pokemon-showdown/dist/data/tags.js",
    f"{PS_CLIENT_DIR}/data/teambuilder-tables.js",
    f"{PS_CLIENT_DIR}/js/battle-scene-stub.js",
    f"{PS_CLIENT_DIR}/js/battle-choices.js",
    f"{PS_CLIENT_DIR}/js/battle-dex.js",
    f"{PS_CLIENT_DIR}/js/battle-dex-data.js",
    f"{PS_CLIENT_DIR}/js/battle-text-parser.js",
    f"{PS_CLIENT_DIR}/js/battle-log.js",
    f"{PS_CLIENT_DIR}/js/battle-tooltips.js",
    f"{PS_CLIENT_DIR}/js/battle.js",
]


DATA_SRC = [
    os.path.join(f"{PS_CLIENT_DIR}/data", file)
    for file in os.listdir(f"{PS_CLIENT_DIR}/data")
    if file.endswith(".js") and file not in ["text-afd.js"]
]


def main():
    for src_path in CLIENT_SRC:
        file = os.path.split(src_path)[-1]
        dst_path = os.path.join("meloetta/js/client", file)
        shutil.copy(src_path, dst_path)

    _ctx = MiniRacer()
    exports = []
    for file in DATA_SRC:
        with open(file, "r", encoding="utf-8") as f:
            file_src = f.read()

        exports += re.findall(r"exports.(\w+) =", file_src)

        file_src = file_src.replace("exports.", "")
        _ctx.eval(file_src)

    for export in exports:
        try:
            string = _ctx.execute("JSON.stringify({})".format(export))
            obj = json.loads(string)
        except:
            print(f"{export} failed to write")
        else:
            with open(f"meloetta/js/data/{export}.json", "w") as f:
                json.dump(obj, f)

    try:
        os.system("npx prettier -w --tab-width 4 meloetta/js")
    except:
        print("[OPTIONAL] run `npx install prettier` to install prettier")
        print("[OPTIONAL] then run `npx prettier -w --tab-width 4 js` to format")


if __name__ == "__main__":
    build_indexes = os.path.join(
        "pokemon-showdown-client", "build-tools", "build-indexes"
    )
    build_learnsets = os.path.join(
        "pokemon-showdown-client", "build-tools", "build-learnsets"
    )
    build = os.path.join("pokemon-showdown-client", "build")
    os.system(f"node {build_indexes}")
    os.system(f"node {build_learnsets}")
    os.system(f"node {build}")
    main()