# "wiipy.py" from WiiPy by NinjaCheetah
# https://github.com/NinjaCheetah/WiiPy

import sys
from modules.wad import *
from modules.nus import *
from modules.u8 import *
from modules.ash 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])
                print("Success!")
                sys.exit(0)
        if "-p" in opts:
            if len(args) == 3:
                pack_wad_from_folder(args[1], args[2])
                print("Success!")
                sys.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])
                print("Success!")
                sys.exit(0)
            elif len(args) == 3:
                download_title(args[1], args[2])
                print("Success!")
                sys.exit(0)
        raise SystemExit(f"Usage: {sys.argv[0]} NUS -d <Title ID> <Title Version (Optional)>")
    elif "U8" in args:
        if "-u" in opts:
            if len(args) == 3:
                extract_u8_to_folder(args[1], args[2])
                print("Success!")
                sys.exit(0)
        elif "-p" in opts:
            if len(args) == 3:
                pack_u8_from_folder(args[1], args[2])
                print("Success!")
                sys.exit(0)
        raise SystemExit(f"Usage: {sys.argv[0]} U8 (-u | -p) <input> <output>")
    elif "ASH" in args:
        if "-d" in opts:
            if len(args) == 2:
                decompress_ash(args[1])
                print("Success!")
                sys.exit(0)
            elif len(args) == 3:
                decompress_ash(args[1], args[2])
                print("Success!")
                sys.exit(0)
        raise SystemExit(f"Usage: {sys.argv[0]} ASH -d <input> <output (Optional)>")
    else:
        raise SystemExit(f"Usage: {sys.argv[0]} WAD (-u | -p) <input> <output>")