# "commands/archive/u8.py" from WiiPy by NinjaCheetah
# https://github.com/NinjaCheetah/WiiPy

import pathlib
import libWiiPy
from modules.core import fatal_error


def handle_u8_pack(args):
    input_path = pathlib.Path(args.input)
    output_path = pathlib.Path(args.output)

    u8_data = None
    try:
        u8_data = libWiiPy.archive.pack_u8(input_path)
    except ValueError:
        fatal_error(f"The specified input file/folder \"{input_path}\" does not exist!")
    output_path.write_bytes(u8_data)

    print("U8 archive packed!")


def handle_u8_unpack(args):
    input_path = pathlib.Path(args.input)
    output_path = pathlib.Path(args.output)

    if not input_path.exists():
        fatal_error(f"The specified input file \"{input_path}\" does not exist!")

    u8_data = input_path.read_bytes()
    # U8 archives are sometimes compressed. In the event that the provided data is LZ77 data, assume it's a compressed
    # U8 archive and decompress it before continuing. Standard checks will then catch it if it was something else.
    if u8_data[0:4] == b'LZ77':
        u8_data = libWiiPy.archive.decompress_lz77(u8_data)

    # Output path is deliberately not checked in any way because libWiiPy already has those checks, and it's easier
    # and cleaner to only have one component doing all the checks.
    libWiiPy.archive.extract_u8(u8_data, str(output_path))

    print("U8 archive unpacked!")