From a6f660f62764f7dae9fb197aad81f83b62cfc3a4 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Tue, 5 Mar 2024 00:57:07 -0500
Subject: [PATCH] Base WAD extraction code

---
 .gitignore       | 172 +++++++++++++++++++++++++++++++++++++++++++++++
 main.py          |  18 +++++
 modules/wad.py   |  46 +++++++++++++
 requirements.txt |   2 +
 4 files changed, 238 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 main.py
 create mode 100644 modules/wad.py
 create mode 100644 requirements.txt

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..aef39b9
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,172 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+main.build/
+main.dist/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+.idea/
+
+# Allows me to keep TMD files in my repository folder for testing without accidentally publishing them
+*.tmd
+*.wad
+out_prod/
+remakewad.pl
+
+# Also awful macOS files
+*._*
+*.DS_Store
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..ae51e55
--- /dev/null
+++ b/main.py
@@ -0,0 +1,18 @@
+# "main.py" from libWiiPy-cli by NinjaCheetah
+# https://github.com/NinjaCheetah/libWiiPy-cli
+
+import sys
+from modules.wad import *
+
+opts = [opt for opt in sys.argv[1:] if opt.startswith("-")]
+args = [arg for arg in sys.argv[1:] if not arg.startswith("-")]
+
+if __name__ == "__main__":
+    if "WAD" in args:
+        if "-u" in opts:
+            if len(args) == 3:
+                extract_wad_to_folder(args[1], args[2])
+                exit(0)
+        raise SystemExit(f"Usage: {sys.argv[0]} WAD (-u | -p) <input> <output>")
+    else:
+        raise SystemExit(f"Usage: {sys.argv[0]} WAD (-u | -p) <input> <output>")
diff --git a/modules/wad.py b/modules/wad.py
new file mode 100644
index 0000000..42465a6
--- /dev/null
+++ b/modules/wad.py
@@ -0,0 +1,46 @@
+# "wad.py" from libWiiPy-cli by NinjaCheetah
+# https://github.com/NinjaCheetah/libWiiPy-cli
+
+import os
+import pathlib
+import struct
+from libWiiPy import wad, tmd, ticket, content
+
+
+def extract_wad_to_folder(in_file_input: str, out_folder_input: str):
+    if not os.path.isfile(in_file_input):
+        raise FileNotFoundError(in_file_input)
+    out_folder = pathlib.Path(out_folder_input)
+    if not out_folder.is_dir():
+        out_folder.mkdir()
+
+    with open(in_file_input, "rb") as wad_file:
+        wad_data = wad.WAD(wad_file.read())
+
+        tmd_data = tmd.TMD(wad_data.get_tmd_data())
+        ticket_data = ticket.Ticket(wad_data.get_ticket_data())
+        content_data = content.ContentRegion(wad_data.get_content_data(), tmd_data.content_records)
+
+        title_key = ticket_data.get_title_key()
+
+        cert_name = tmd_data.get_title_id() + ".cert"
+        cert_out = open(os.path.join(out_folder, cert_name), "wb")
+        cert_out.write(wad_data.get_cert_data())
+        cert_out.close()
+
+        tmd_name = tmd_data.get_title_id() + ".tmd"
+        tmd_out = open(os.path.join(out_folder, tmd_name), "wb")
+        tmd_out.write(wad_data.get_tmd_data())
+        tmd_out.close()
+
+        ticket_name = tmd_data.get_title_id() + ".tik"
+        ticket_out = open(os.path.join(out_folder, ticket_name), "wb")
+        ticket_out.write(wad_data.get_ticket_data())
+        ticket_out.close()
+
+        for content_file in range(0, tmd_data.get_num_contents()):
+            content_file_name = "000000"
+            content_file_name += str(hex(content_file)) + ".app"
+            content_out = open(os.path.join(out_folder, content_file_name), "wb")
+            content_out.write(content_data.get_content(content_file, title_key))
+            content_out.close()
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..0c02c3d
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,2 @@
+git+https://github.com/NinjaCheetah/libWiiPy
+pycryptodome
\ No newline at end of file