feat(cli): add simple cli interface for updating the tpk file

Author: K0lb3

UnityPy/__main__.py

@@ -0,0 +1,4 @@
+if __name__ == "__main__":
+    from UnityPy.cli import main
+
+    main()

UnityPy/cli/__init__.py

@@ -0,0 +1,28 @@
+from argparse import ArgumentParser
+
+from UnityPy.cli.update_tpk import update_tpk
+
+
+def main():
+    parser = ArgumentParser(
+        prog="UnityPy",
+        description="UnityPy cli utility",
+    )
+
+    subparsers = parser.add_subparsers(title="utils")
+
+    update_tpk_parser = subparsers.add_parser("update_tpk", help="Updates TPK (typetree dump) file")
+    update_tpk_parser.set_defaults(func=update_tpk)
+
+    args = parser.parse_args()
+
+    if not hasattr(args, "func"):
+        parser.print_help()
+        return
+
+    operands = getattr(args, "operands", [])
+    args.func(*operands)
+
+
+if __name__ == "__main__":
+    main()

UnityPy/cli/update_tpk.py

@@ -0,0 +1,21 @@
+import os
+from io import BytesIO
+from urllib.request import urlopen
+from zipfile import ZipFile
+
+URL = "https://nightly.link/AssetRipper/Tpk/workflows/type_tree_tpk/master/uncompressed_file.zip"
+RESOURCE_PATH = os.path.join(os.path.dirname(__file__), "..", "resources")
+
+
+def update_tpk():
+    print("Updating TPK file...")
+    print("\tDownloading...")
+    with urlopen(URL) as response:
+        zip_data = response.read()
+    print("\tExtracting...")
+    with ZipFile(BytesIO(zip_data)) as zip_file:
+        zip_file.extract("uncompressed.tpk", path=RESOURCE_PATH)
+    print("\tDone.")
+
+
+__all__ = ["update_tpk"]

pyproject.toml

@@ -54,6 +54,9 @@ dependencies = [
 ]
 dynamic = ["version"]
 
+[project.scripts]
+UnityPy = "UnityPy.cli:main"
+
 [project.optional-dependencies]
 # optional dependencies must be lowercase/normalized
 ttgen = ["typetreegeneratorapi>=0.0.5"]