Added simple NUS downloading interface

main.py

@@ -3,6 +3,7 @@
 
 import sys
 from modules.wad import *
+from modules.nus import *
 
 opts = [opt for opt in sys.argv[1:] if opt.startswith("-")]
 args = [arg for arg in sys.argv[1:] if not arg.startswith("-")]
@@ -18,5 +19,14 @@ if __name__ == "__main__":
                 pack_wad_from_folder(args[1], args[2])
                 exit(0)
         raise SystemExit(f"Usage: {sys.argv[0]} WAD (-u | -p) <input> <output>")
+    elif "NUS" in args:
+        if "-d" in opts:
+            if len(args) == 2:
+                download_title(args[1])
+                exit(0)
+            elif len(args) == 3:
+                download_title(args[1], args[2])
+                exit(0)
+        raise SystemExit(f"Usage: {sys.argv[0]} NUS -d <Title ID> <Title Version (Optional)>")
     else:
         raise SystemExit(f"Usage: {sys.argv[0]} WAD (-u | -p) <input> <output>")

modules/nus.py

@@ -0,0 +1,22 @@
+# "nus.py" from libWiiPy-cli by NinjaCheetah
+# https://github.com/NinjaCheetah/libWiiPy-cli
+
+import libWiiPy
+
+
+def download_title(title_id: str, title_version_input: str = None):
+    title_version = None
+    if title_version_input is not None:
+        try:
+            title_version = int(title_version_input)
+        except ValueError:
+            print("Enter a valid integer for the Title Version.")
+            return
+
+    title = libWiiPy.download_title(title_id, title_version)
+
+    file_name = title_id + "-v" + str(title.tmd.title_version) + ".wad"
+
+    wad_file = open(file_name, "wb")
+    wad_file.write(title.dump_wad())
+    wad_file.close()

requirements.txt

@@ -1,2 +1 @@
 git+https://github.com/NinjaCheetah/libWiiPy
-pycryptodome