From ede33dc503a954b153544dc6d5dfe8d2fc35e49f Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Thu, 16 May 2024 21:24:42 -0400
Subject: [PATCH 1/9] Add highly experimental U8 handling module

---
 .gitignore               |  3 +-
 pyproject.toml           |  2 +-
 src/libWiiPy/__init__.py |  1 +
 src/libWiiPy/types.py    | 40 +++++++++++++++----
 src/libWiiPy/u8.py       | 83 ++++++++++++++++++++++++++++++++++++++++
 src/libWiiPy/wad.py      |  6 +--
 6 files changed, 122 insertions(+), 13 deletions(-)
 create mode 100644 src/libWiiPy/u8.py

diff --git a/.gitignore b/.gitignore
index aef39b9..8522187 100644
--- a/.gitignore
+++ b/.gitignore
@@ -161,9 +161,10 @@ cython_debug/
 #  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
+# Relevant files that are used for testing libWiiPy's features.
 *.tmd
 *.wad
+*.arc
 out_prod/
 remakewad.pl
 
diff --git a/pyproject.toml b/pyproject.toml
index 5205905..08b02c5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "libWiiPy"
-version = "0.2.3"
+version = "0.3.0"
 authors = [
     { name="NinjaCheetah", email="ninjacheetah@ncxprogramming.com" },
     { name="Lillian Skinner", email="lillian@randommeaninglesscharacters.com" }
diff --git a/src/libWiiPy/__init__.py b/src/libWiiPy/__init__.py
index 2df7705..f381d5b 100644
--- a/src/libWiiPy/__init__.py
+++ b/src/libWiiPy/__init__.py
@@ -11,3 +11,4 @@ from .title import *
 from .tmd import *
 from .wad import *
 from .nus import *
+from .u8 import *
diff --git a/src/libWiiPy/types.py b/src/libWiiPy/types.py
index 6080dbc..48d8d91 100644
--- a/src/libWiiPy/types.py
+++ b/src/libWiiPy/types.py
@@ -1,6 +1,6 @@
 # "types.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
-
+from builtins import type
 from dataclasses import dataclass
 
 
@@ -14,21 +14,21 @@ class ContentRecord:
     Attributes
     ----------
     content_id : int
-        ID of the content.
+        The unique ID of the content.
     index : int
-        Index of the content in the list of contents.
+        The index of this content in the content records.
     content_type : int
         The type of the content.
     content_size : int
-        The size of the content.
+        The size of the content when decrypted.
     content_hash
         The SHA-1 hash of the decrypted content.
     """
-    content_id: int  # The unique ID of the content.
-    index: int  # The index of this content in the content record.
+    content_id: int
+    index: int
     content_type: int  # Type of content, possible values of: 0x0001: Normal, 0x4001: DLC, 0x8001: Shared.
-    content_size: int  # Size of the content when decrypted.
-    content_hash: bytes  # SHA-1 hash of the content when decrypted.
+    content_size: int
+    content_hash: bytes
 
 
 @dataclass
@@ -50,3 +50,27 @@ class TitleLimit:
     limit_type: int
     # The maximum value of the limit applied.
     maximum_usage: int
+
+
+@dataclass
+class U8Node:
+    """
+    A U8Node object that contains the data of a single node in a U8 file header. Each node keeps track of whether this
+    node is for a file or directory, the offset of the name of the file/directory, the offset of the data for the file/
+    directory, and the size of the data.
+
+    Attributes
+    ----------
+    type : int
+        Whether this node refers to a file or a directory. Either 0x0000 for files, or 0x0100 for directories.
+    name_offset : int
+        The offset of the name of the file/directory this node refers to.
+    data_offset : int
+        The offset of the data for the file/directory this node refers to.
+    size : int
+        The size of the data for this node.
+    """
+    type: int
+    name_offset: int
+    data_offset: int
+    size: int
diff --git a/src/libWiiPy/u8.py b/src/libWiiPy/u8.py
new file mode 100644
index 0000000..ef16df7
--- /dev/null
+++ b/src/libWiiPy/u8.py
@@ -0,0 +1,83 @@
+# "u8.py" from libWiiPy by NinjaCheetah & Contributors
+# https://github.com/NinjaCheetah/libWiiPy
+#
+# See https://wiibrew.org/wiki/U8_archive for details about the U8 archive format
+
+import io
+import binascii
+from typing import List
+from .types import U8Node
+
+
+class U8Archive:
+    def __init__(self):
+        """
+        A U8 object that allows for extracting and packing U8 archives.
+
+        Attributes
+        ----------
+        """
+        self.u8_magic = b''
+        self.root_node_offset = 0  # Offset of the root node, which will always be 0x20.
+        self.header_size = 0  # The size of the U8 header.
+        self.data_offset = 0  # The offset of the data, which is root_node_offset + header_size, aligned to 0x40.
+        self.header_padding = b''
+        self.root_node = U8Node
+        self.u8_node_list: List[U8Node] = []  # All the nodes in the header of a U8 file.
+        self.file_name_list: List[str] = []
+        self.u8_file_data_list: List[bytes] = []
+        self.u8_file_structure = dict
+
+    def load(self, u8_data: bytes) -> None:
+        """
+        Loads raw U8 data into a new U8 object. This allows for extracting the file and updating its contents.
+
+        Parameters
+        ----------
+        u8_data : bytes
+            The data for the U8 file to load.
+        """
+        with io.BytesIO(u8_data) as u8_data:
+            # Read the first 4 bytes of the file to ensure that it's a U8 archive.
+            u8_data.seek(0x0)
+            self.u8_magic = u8_data.read(4)
+            if self.u8_magic != b'\x55\xAA\x38\x2D':
+                raise TypeError("This is not a valid U8 archive!")
+            self.root_node_offset = int(binascii.hexlify(u8_data.read(4)), 16)
+            self.header_size = int(binascii.hexlify(u8_data.read(4)), 16)
+            self.data_offset = int(binascii.hexlify(u8_data.read(4)), 16)
+            self.header_padding = u8_data.read(16)
+            root_node_type = int.from_bytes(u8_data.read(2))
+            root_node_name_offset = int.from_bytes(u8_data.read(2))
+            root_node_data_offset = int.from_bytes(u8_data.read(4))
+            root_node_size = int.from_bytes(u8_data.read(4))
+            self.root_node = U8Node(root_node_type, root_node_name_offset, root_node_data_offset, root_node_size)
+            self.u8_node_list.append(self.root_node)
+            # Iterate over the number of nodes that the root node lists, minus one since the count includes itself.
+            for node in range(self.root_node.size - 1):
+                node_type = int.from_bytes(u8_data.read(2))
+                node_name_offset = int.from_bytes(u8_data.read(2))
+                node_data_offset = int.from_bytes(u8_data.read(4))
+                node_size = int.from_bytes(u8_data.read(4))
+                self.u8_node_list.append(U8Node(node_type, node_name_offset, node_data_offset, node_size))
+            # Iterate over all loaded nodes and create a list of file names.
+            name_base_offset = u8_data.tell()
+            for node in self.u8_node_list:
+                u8_data.seek(name_base_offset + node.name_offset)
+                name_bin = b''
+                while name_bin[-1:] != b'\x00':
+                    name_bin += u8_data.read(1)
+                name_bin = name_bin[:-1]
+                name = str(name_bin.decode())
+                self.file_name_list.append(name)
+                if node.type == 0:
+                    u8_data.seek(node.data_offset)
+                    self.u8_file_data_list.append(u8_data.read(node.size))
+                else:
+                    self.u8_file_data_list.append(b'')
+            # This does nothing for now.
+            next_dir = 0
+            for node in range(len(self.u8_node_list)):
+                if self.u8_node_list[node].type == 256 and node != 0:
+                    next_dir = self.u8_node_list[node].size
+
diff --git a/src/libWiiPy/wad.py b/src/libWiiPy/wad.py
index 5421784..19d1700 100644
--- a/src/libWiiPy/wad.py
+++ b/src/libWiiPy/wad.py
@@ -49,7 +49,7 @@ class WAD:
         self.wad_content_data: bytes = b''
         self.wad_meta_data: bytes = b''
 
-    def load(self, wad_data) -> None:
+    def load(self, wad_data: bytes) -> None:
         """
         Loads raw WAD data and sets all attributes of the WAD object. This allows for manipulating an already
         existing WAD file.
@@ -57,7 +57,7 @@ class WAD:
         Parameters
         ----------
         wad_data : bytes
-            The data for the WAD you wish to load.
+            The data for the WAD file to load.
         """
         with io.BytesIO(wad_data) as wad_data:
             # Read the first 8 bytes of the file to ensure that it's a WAD. Has two possible valid values for the two
@@ -67,7 +67,7 @@ class WAD:
             wad_magic_hex = binascii.hexlify(wad_magic_bin)
             wad_magic = str(wad_magic_hex.decode())
             if wad_magic != "0000002049730000" and wad_magic != "0000002069620000":
-                raise TypeError("This does not appear to be a valid WAD file.")
+                raise TypeError("This is not a valid WAD file!")
             # ====================================================================================
             # Get the sizes of each data region contained within the WAD.
             # ====================================================================================

From cbaafca0d192fc3e0877fc9aa0506589025d4237 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Wed, 29 May 2024 12:48:44 -0400
Subject: [PATCH 2/9] Working basic code to extract a U8 archive

---
 src/libWiiPy/types.py |  1 -
 src/libWiiPy/u8.py    | 36 +++++++++++++++++++++++++++++++-----
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/src/libWiiPy/types.py b/src/libWiiPy/types.py
index 48d8d91..9dd6813 100644
--- a/src/libWiiPy/types.py
+++ b/src/libWiiPy/types.py
@@ -1,6 +1,5 @@
 # "types.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
-from builtins import type
 from dataclasses import dataclass
 
 
diff --git a/src/libWiiPy/u8.py b/src/libWiiPy/u8.py
index ef16df7..69a5ad4 100644
--- a/src/libWiiPy/u8.py
+++ b/src/libWiiPy/u8.py
@@ -5,6 +5,7 @@
 
 import io
 import binascii
+import os
 from typing import List
 from .types import U8Node
 
@@ -75,9 +76,34 @@ class U8Archive:
                     self.u8_file_data_list.append(u8_data.read(node.size))
                 else:
                     self.u8_file_data_list.append(b'')
-            # This does nothing for now.
-            next_dir = 0
-            for node in range(len(self.u8_node_list)):
-                if self.u8_node_list[node].type == 256 and node != 0:
-                    next_dir = self.u8_node_list[node].size
+
+    def extract_to_folder(self, output_folder) -> None:
+        if os.path.isdir(output_folder):
+            raise ValueError("Output folder already exists!")
+        if self.u8_node_list is []:
+            raise ValueError("No U8 file is loaded!")
+
+        os.mkdir(output_folder)
+
+        current_dir = ""
+        for node in range(len(self.u8_node_list)):
+            if self.u8_node_list[node].name_offset != 0:
+                if self.u8_node_list[node].type == 256:
+                    if self.u8_node_list[node].data_offset == 0:
+                        os.mkdir(os.path.join(output_folder, self.file_name_list[node]))
+                        current_dir = self.file_name_list[node]
+                    elif self.u8_node_list[node].data_offset < node:
+                        lower_path = os.path.join(output_folder, current_dir)
+                        os.mkdir(os.path.join(lower_path, self.file_name_list[node]))
+                        current_dir = os.path.join(current_dir, self.file_name_list[node])
+                elif self.u8_node_list[node].type == 0:
+                    lower_path = os.path.join(output_folder, current_dir)
+                    output_file = open(os.path.join(lower_path, self.file_name_list[node]), "wb")
+                    output_file.write(self.u8_file_data_list[node])
+                    output_file.close()
+
+    def pack_from_folder(self, input_folder) -> None:
+        if not os.path.isdir(input_folder):
+            raise ValueError("Input folder does not exist!")
+
 

From bc9224e40bd0893da3e395791a004aacdb01f739 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Wed, 29 May 2024 20:05:41 -0400
Subject: [PATCH 3/9] Added base for tests

---
 tests/__init.py__.py     | 11 +++++++++++
 tests/test_commonkeys.py | 21 +++++++++++++++++++++
 2 files changed, 32 insertions(+)
 create mode 100644 tests/__init.py__.py
 create mode 100644 tests/test_commonkeys.py

diff --git a/tests/__init.py__.py b/tests/__init.py__.py
new file mode 100644
index 0000000..ef0634e
--- /dev/null
+++ b/tests/__init.py__.py
@@ -0,0 +1,11 @@
+# "__init__.py" from libWiiPy by NinjaCheetah & Contributors
+# https://github.com/NinjaCheetah/libWiiPy
+#
+# Complete set of tests to be run.
+
+import unittest
+
+from test_commonkeys import TestCommonKeys
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/tests/test_commonkeys.py b/tests/test_commonkeys.py
new file mode 100644
index 0000000..0bba6d5
--- /dev/null
+++ b/tests/test_commonkeys.py
@@ -0,0 +1,21 @@
+# "test_commonkeys.py" from libWiiPy by NinjaCheetah & Contributors
+# https://github.com/NinjaCheetah/libWiiPy
+
+import unittest
+
+from libWiiPy import commonkeys
+
+
+class TestCommonKeys(unittest.TestCase):
+    def test_common(self):
+        self.assertEqual(commonkeys.get_common_key(0), b'\xeb\xe4*"^\x85\x93\xe4H\xd9\xc5Es\x81\xaa\xf7')
+
+    def test_korean(self):
+        self.assertEqual(commonkeys.get_common_key(1), b'c\xb8+\xb4\xf4aN.\x13\xf2\xfe\xfb\xbaL\x9b~')
+
+    def test_vwii(self):
+        self.assertEqual(commonkeys.get_common_key(2), b'0\xbf\xc7n|\x19\xaf\xbb#\x1630\xce\xd7\xc2\x8d')
+
+
+if __name__ == '__main__':
+    unittest.main()

From ade4b68394e9abdce2027321ad98cfc725d173f2 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Sun, 2 Jun 2024 22:18:05 -0400
Subject: [PATCH 4/9] Entirely restructured package to make it more of a proper
 Python package THIS IS A BREAKING COMMIT! ALL v0.2.X BASED PROJECTS WILL NEED
 TO BE UPDATED TO SUPPORT v0.3.X!

---
 src/__init__.py                        |   0
 src/libWiiPy/__init__.py               |  15 ++--
 src/libWiiPy/archive/__init__.py       |   4 +
 src/libWiiPy/{ => archive}/u8.py       | 101 ++++++++++++++++++-------
 src/libWiiPy/title/__init__.py         |   8 ++
 src/libWiiPy/{ => title}/commonkeys.py |   2 +-
 src/libWiiPy/{ => title}/content.py    |   6 +-
 src/libWiiPy/{ => title}/crypto.py     |   6 +-
 src/libWiiPy/{ => title}/nus.py        |   8 +-
 src/libWiiPy/{ => title}/ticket.py     |   6 +-
 src/libWiiPy/{ => title}/title.py      |  10 +--
 src/libWiiPy/{ => title}/tmd.py        |   4 +-
 src/libWiiPy/{ => title}/wad.py        |   4 +-
 src/libWiiPy/types.py                  |  24 ------
 14 files changed, 113 insertions(+), 85 deletions(-)
 create mode 100644 src/__init__.py
 create mode 100644 src/libWiiPy/archive/__init__.py
 rename src/libWiiPy/{ => archive}/u8.py (54%)
 create mode 100644 src/libWiiPy/title/__init__.py
 rename src/libWiiPy/{ => title}/commonkeys.py (93%)
 rename src/libWiiPy/{ => title}/content.py (98%)
 rename src/libWiiPy/{ => title}/crypto.py (96%)
 rename src/libWiiPy/{ => title}/nus.py (97%)
 rename src/libWiiPy/{ => title}/ticket.py (98%)
 rename src/libWiiPy/{ => title}/title.py (97%)
 rename src/libWiiPy/{ => title}/tmd.py (99%)
 rename src/libWiiPy/{ => title}/wad.py (99%)

diff --git a/src/__init__.py b/src/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/libWiiPy/__init__.py b/src/libWiiPy/__init__.py
index f381d5b..0d3aa40 100644
--- a/src/libWiiPy/__init__.py
+++ b/src/libWiiPy/__init__.py
@@ -1,14 +1,9 @@
 # "__init__.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
-# These are the essential modules from libWiiPy that you'd probably want imported by default.
+# These are the essential submodules from libWiiPy that you'd probably want imported by default.
 
-from .commonkeys import *
-from .content import *
-from .ticket import *
-from .crypto import *
-from .title import *
-from .tmd import *
-from .wad import *
-from .nus import *
-from .u8 import *
+__all__ = ["archive", "title"]
+
+from . import archive
+from . import title
diff --git a/src/libWiiPy/archive/__init__.py b/src/libWiiPy/archive/__init__.py
new file mode 100644
index 0000000..39144ba
--- /dev/null
+++ b/src/libWiiPy/archive/__init__.py
@@ -0,0 +1,4 @@
+# "archive/__init__.py" from libWiiPy by NinjaCheetah & Contributors
+# https://github.com/NinjaCheetah/libWiiPy
+
+from .u8 import *
diff --git a/src/libWiiPy/u8.py b/src/libWiiPy/archive/u8.py
similarity index 54%
rename from src/libWiiPy/u8.py
rename to src/libWiiPy/archive/u8.py
index 69a5ad4..0dd9847 100644
--- a/src/libWiiPy/u8.py
+++ b/src/libWiiPy/archive/u8.py
@@ -1,4 +1,4 @@
-# "u8.py" from libWiiPy by NinjaCheetah & Contributors
+# "archive/u8.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/U8_archive for details about the U8 archive format
@@ -6,14 +6,38 @@
 import io
 import binascii
 import os
+from dataclasses import dataclass
 from typing import List
-from .types import U8Node
+
+
+@dataclass
+class U8Node:
+    """
+    A U8Node object that contains the data of a single node in a U8 file header. Each node keeps track of whether this
+    node is for a file or directory, the offset of the name of the file/directory, the offset of the data for the file/
+    directory, and the size of the data.
+
+    Attributes
+    ----------
+    type : int
+        Whether this node refers to a file or a directory. Either 0x0000 for files, or 0x0100 for directories.
+    name_offset : int
+        The offset of the name of the file/directory this node refers to.
+    data_offset : int
+        The offset of the data for the file/directory this node refers to.
+    size : int
+        The size of the data for this node.
+    """
+    type: int
+    name_offset: int
+    data_offset: int
+    size: int
 
 
 class U8Archive:
     def __init__(self):
         """
-        A U8 object that allows for extracting and packing U8 archives.
+        A U8 object that allows for managing the contents of a U8 archive.
 
         Attributes
         ----------
@@ -77,33 +101,54 @@ class U8Archive:
                 else:
                     self.u8_file_data_list.append(b'')
 
-    def extract_to_folder(self, output_folder) -> None:
-        if os.path.isdir(output_folder):
-            raise ValueError("Output folder already exists!")
-        if self.u8_node_list is []:
-            raise ValueError("No U8 file is loaded!")
+    def dump(self) -> None:
+        """
+        Dumps the U8Archive object into a U8 file.
+        """
+        u8_data = b''
+        # Magic number.
+        u8_data += b'\x55\xAA\x38\x2D'
+        # Root node offset (this is always 0x20).
+        u8_data += int.to_bytes(0x20, 4)
 
-        os.mkdir(output_folder)
 
-        current_dir = ""
-        for node in range(len(self.u8_node_list)):
-            if self.u8_node_list[node].name_offset != 0:
-                if self.u8_node_list[node].type == 256:
-                    if self.u8_node_list[node].data_offset == 0:
-                        os.mkdir(os.path.join(output_folder, self.file_name_list[node]))
-                        current_dir = self.file_name_list[node]
-                    elif self.u8_node_list[node].data_offset < node:
-                        lower_path = os.path.join(output_folder, current_dir)
-                        os.mkdir(os.path.join(lower_path, self.file_name_list[node]))
-                        current_dir = os.path.join(current_dir, self.file_name_list[node])
-                elif self.u8_node_list[node].type == 0:
+def extract_u8(u8_data, output_folder) -> None:
+    if os.path.isdir(output_folder):
+        raise ValueError("Output folder already exists!")
+
+    os.mkdir(output_folder)
+
+    u8_archive = U8Archive()
+    u8_archive.load(u8_data)
+
+    current_dir = ""
+    for node in range(len(u8_archive.u8_node_list)):
+        if u8_archive.u8_node_list[node].name_offset != 0:
+            if u8_archive.u8_node_list[node].type == 256:
+                if u8_archive.u8_node_list[node].data_offset == 0:
+                    os.mkdir(os.path.join(output_folder, u8_archive.file_name_list[node]))
+                    current_dir = u8_archive.file_name_list[node]
+                elif u8_archive.u8_node_list[node].data_offset < node:
                     lower_path = os.path.join(output_folder, current_dir)
-                    output_file = open(os.path.join(lower_path, self.file_name_list[node]), "wb")
-                    output_file.write(self.u8_file_data_list[node])
-                    output_file.close()
-
-    def pack_from_folder(self, input_folder) -> None:
-        if not os.path.isdir(input_folder):
-            raise ValueError("Input folder does not exist!")
+                    os.mkdir(os.path.join(lower_path, u8_archive.file_name_list[node]))
+                    current_dir = os.path.join(current_dir, u8_archive.file_name_list[node])
+            elif u8_archive.u8_node_list[node].type == 0:
+                lower_path = os.path.join(output_folder, current_dir)
+                output_file = open(os.path.join(lower_path, u8_archive.file_name_list[node]), "wb")
+                output_file.write(u8_archive.u8_file_data_list[node])
+                output_file.close()
+
+
+def pack_u8(input_data) -> None:
+    if os.path.isdir(input_data):
+        raise ValueError("Only single-file packing is currently supported!")
+    elif os.path.isfile(input_data):
+        with open(input_data, "rb") as f:
+            u8_archive = U8Archive()
+
+            file_name = os.path.basename(input_data)
+
+            u8_archive.file_name_list.append(file_name)
+            u8_archive.u8_file_data_list.append(f.read())
 
 
diff --git a/src/libWiiPy/title/__init__.py b/src/libWiiPy/title/__init__.py
new file mode 100644
index 0000000..a54b791
--- /dev/null
+++ b/src/libWiiPy/title/__init__.py
@@ -0,0 +1,8 @@
+# "title/__init__.py" from libWiiPy by NinjaCheetah & Contributors
+# https://github.com/NinjaCheetah/libWiiPy
+
+from .content import *
+from .ticket import *
+from .title import *
+from .tmd import *
+from .wad import *
diff --git a/src/libWiiPy/commonkeys.py b/src/libWiiPy/title/commonkeys.py
similarity index 93%
rename from src/libWiiPy/commonkeys.py
rename to src/libWiiPy/title/commonkeys.py
index 04755e2..9875bff 100644
--- a/src/libWiiPy/commonkeys.py
+++ b/src/libWiiPy/title/commonkeys.py
@@ -1,4 +1,4 @@
-# "commonkeys.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/commonkeys.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 
 import binascii
diff --git a/src/libWiiPy/content.py b/src/libWiiPy/title/content.py
similarity index 98%
rename from src/libWiiPy/content.py
rename to src/libWiiPy/title/content.py
index 0029759..c2a3392 100644
--- a/src/libWiiPy/content.py
+++ b/src/libWiiPy/title/content.py
@@ -1,4 +1,4 @@
-# "content.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/content.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/Title for details about how titles are formatted
@@ -7,8 +7,8 @@ import io
 import sys
 import hashlib
 from typing import List
-from .types import ContentRecord
-from .crypto import decrypt_content, encrypt_content
+from src.libWiiPy.types import ContentRecord
+from src.libWiiPy.title.crypto import decrypt_content, encrypt_content
 
 
 class ContentRegion:
diff --git a/src/libWiiPy/crypto.py b/src/libWiiPy/title/crypto.py
similarity index 96%
rename from src/libWiiPy/crypto.py
rename to src/libWiiPy/title/crypto.py
index 734f0bf..5f7d88c 100644
--- a/src/libWiiPy/crypto.py
+++ b/src/libWiiPy/title/crypto.py
@@ -1,9 +1,9 @@
-# "crypto.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/crypto.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 
 import struct
-from .commonkeys import get_common_key
-from .shared import convert_tid_to_iv
+from src.libWiiPy.title.commonkeys import get_common_key
+from src.libWiiPy.shared import convert_tid_to_iv
 
 from Crypto.Cipher import AES
 
diff --git a/src/libWiiPy/nus.py b/src/libWiiPy/title/nus.py
similarity index 97%
rename from src/libWiiPy/nus.py
rename to src/libWiiPy/title/nus.py
index 4282860..d9085c3 100644
--- a/src/libWiiPy/nus.py
+++ b/src/libWiiPy/title/nus.py
@@ -1,4 +1,4 @@
-# "nus.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/nus.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/NUS for details about the NUS
@@ -6,9 +6,9 @@
 import requests
 import hashlib
 from typing import List
-from .title import Title
-from .tmd import TMD
-from .ticket import Ticket
+from src.libWiiPy.title.title import Title
+from src.libWiiPy.title.tmd import TMD
+from src.libWiiPy.title.ticket import Ticket
 
 nus_endpoint = ["http://nus.cdn.shop.wii.com/ccs/download/", "http://ccs.cdn.wup.shop.nintendo.net/ccs/download/"]
 
diff --git a/src/libWiiPy/ticket.py b/src/libWiiPy/title/ticket.py
similarity index 98%
rename from src/libWiiPy/ticket.py
rename to src/libWiiPy/title/ticket.py
index 8f6a0aa..6e433b0 100644
--- a/src/libWiiPy/ticket.py
+++ b/src/libWiiPy/title/ticket.py
@@ -1,12 +1,12 @@
-# "ticket.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/ticket.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/Ticket for details about the ticket format
 
 import io
 import binascii
-from .crypto import decrypt_title_key
-from .types import TitleLimit
+from src.libWiiPy.title.crypto import decrypt_title_key
+from src.libWiiPy.types import TitleLimit
 from typing import List
 
 
diff --git a/src/libWiiPy/title.py b/src/libWiiPy/title/title.py
similarity index 97%
rename from src/libWiiPy/title.py
rename to src/libWiiPy/title/title.py
index 4478a62..d0d6690 100644
--- a/src/libWiiPy/title.py
+++ b/src/libWiiPy/title/title.py
@@ -1,12 +1,12 @@
-# "title.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/title.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/Title for details about how titles are formatted
 
-from .content import ContentRegion
-from .ticket import Ticket
-from .tmd import TMD
-from .wad import WAD
+from src.libWiiPy.title.content import ContentRegion
+from src.libWiiPy.title.ticket import Ticket
+from src.libWiiPy.title.tmd import TMD
+from src.libWiiPy.title.wad import WAD
 
 
 class Title:
diff --git a/src/libWiiPy/tmd.py b/src/libWiiPy/title/tmd.py
similarity index 99%
rename from src/libWiiPy/tmd.py
rename to src/libWiiPy/title/tmd.py
index 4510cc4..fd3c5cf 100644
--- a/src/libWiiPy/tmd.py
+++ b/src/libWiiPy/title/tmd.py
@@ -1,4 +1,4 @@
-# "tmd.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/tmd.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/Title_metadata for details about the TMD format
@@ -7,7 +7,7 @@ import io
 import binascii
 import struct
 from typing import List
-from .types import ContentRecord
+from src.libWiiPy.types import ContentRecord
 
 
 class TMD:
diff --git a/src/libWiiPy/wad.py b/src/libWiiPy/title/wad.py
similarity index 99%
rename from src/libWiiPy/wad.py
rename to src/libWiiPy/title/wad.py
index 19d1700..12ba493 100644
--- a/src/libWiiPy/wad.py
+++ b/src/libWiiPy/title/wad.py
@@ -1,11 +1,11 @@
-# "wad.py" from libWiiPy by NinjaCheetah & Contributors
+# "title/wad.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
 # See https://wiibrew.org/wiki/WAD_files for details about the WAD format
 
 import io
 import binascii
-from .shared import align_value, pad_bytes
+from src.libWiiPy.shared import align_value, pad_bytes
 
 
 class WAD:
diff --git a/src/libWiiPy/types.py b/src/libWiiPy/types.py
index 9dd6813..155090e 100644
--- a/src/libWiiPy/types.py
+++ b/src/libWiiPy/types.py
@@ -49,27 +49,3 @@ class TitleLimit:
     limit_type: int
     # The maximum value of the limit applied.
     maximum_usage: int
-
-
-@dataclass
-class U8Node:
-    """
-    A U8Node object that contains the data of a single node in a U8 file header. Each node keeps track of whether this
-    node is for a file or directory, the offset of the name of the file/directory, the offset of the data for the file/
-    directory, and the size of the data.
-
-    Attributes
-    ----------
-    type : int
-        Whether this node refers to a file or a directory. Either 0x0000 for files, or 0x0100 for directories.
-    name_offset : int
-        The offset of the name of the file/directory this node refers to.
-    data_offset : int
-        The offset of the data for the file/directory this node refers to.
-    size : int
-        The size of the data for this node.
-    """
-    type: int
-    name_offset: int
-    data_offset: int
-    size: int

From 1d77868cb145a13eeeb455b041127869b69baa54 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Mon, 3 Jun 2024 23:20:34 -0400
Subject: [PATCH 5/9] Reworked U8 module and added initial support for dumping
 U8 archives back to bytes

---
 src/libWiiPy/archive/u8.py | 119 +++++++++++++++++++++++++++----------
 1 file changed, 86 insertions(+), 33 deletions(-)

diff --git a/src/libWiiPy/archive/u8.py b/src/libWiiPy/archive/u8.py
index 0dd9847..c5443ae 100644
--- a/src/libWiiPy/archive/u8.py
+++ b/src/libWiiPy/archive/u8.py
@@ -4,10 +4,10 @@
 # See https://wiibrew.org/wiki/U8_archive for details about the U8 archive format
 
 import io
-import binascii
 import os
 from dataclasses import dataclass
 from typing import List
+from src.libWiiPy.shared import align_value
 
 
 @dataclass
@@ -43,14 +43,9 @@ class U8Archive:
         ----------
         """
         self.u8_magic = b''
-        self.root_node_offset = 0  # Offset of the root node, which will always be 0x20.
-        self.header_size = 0  # The size of the U8 header.
-        self.data_offset = 0  # The offset of the data, which is root_node_offset + header_size, aligned to 0x40.
-        self.header_padding = b''
-        self.root_node = U8Node
         self.u8_node_list: List[U8Node] = []  # All the nodes in the header of a U8 file.
         self.file_name_list: List[str] = []
-        self.u8_file_data_list: List[bytes] = []
+        self.file_data_list: List[bytes] = []
         self.u8_file_structure = dict
 
     def load(self, u8_data: bytes) -> None:
@@ -68,24 +63,30 @@ class U8Archive:
             self.u8_magic = u8_data.read(4)
             if self.u8_magic != b'\x55\xAA\x38\x2D':
                 raise TypeError("This is not a valid U8 archive!")
-            self.root_node_offset = int(binascii.hexlify(u8_data.read(4)), 16)
-            self.header_size = int(binascii.hexlify(u8_data.read(4)), 16)
-            self.data_offset = int(binascii.hexlify(u8_data.read(4)), 16)
-            self.header_padding = u8_data.read(16)
-            root_node_type = int.from_bytes(u8_data.read(2))
-            root_node_name_offset = int.from_bytes(u8_data.read(2))
-            root_node_data_offset = int.from_bytes(u8_data.read(4))
+            # The following code is all skipped because these values really don't matter for extraction. They honestly
+            # really only matter to my code when they get calculated and used for packing.
+
+            # Offset of the root node, which will always be 0x20.
+            # root_node_offset = int(binascii.hexlify(u8_data.read(4)), 16)
+            # The size of the U8 header.
+            # header_size = int(binascii.hexlify(u8_data.read(4)), 16)
+            # The offset of the data, which is root_node_offset + header_size, aligned to 0x10.
+            # data_offset = int(binascii.hexlify(u8_data.read(4)), 16)
+
+            # Seek ahead to the size defined in the root node, because it's the total number of nodes in the file. The
+            # rest of the data in the root node (not that it really matters) will get read when we read the whole list.
+            u8_data.seek(u8_data.tell() + 36)
             root_node_size = int.from_bytes(u8_data.read(4))
-            self.root_node = U8Node(root_node_type, root_node_name_offset, root_node_data_offset, root_node_size)
-            self.u8_node_list.append(self.root_node)
-            # Iterate over the number of nodes that the root node lists, minus one since the count includes itself.
-            for node in range(self.root_node.size - 1):
+            # Seek back before the root node so that it gets read with all the rest.
+            u8_data.seek(u8_data.tell() - 12)
+            # Iterate over the number of nodes that the root node lists.
+            for node in range(root_node_size):
                 node_type = int.from_bytes(u8_data.read(2))
                 node_name_offset = int.from_bytes(u8_data.read(2))
                 node_data_offset = int.from_bytes(u8_data.read(4))
                 node_size = int.from_bytes(u8_data.read(4))
                 self.u8_node_list.append(U8Node(node_type, node_name_offset, node_data_offset, node_size))
-            # Iterate over all loaded nodes and create a list of file names.
+            # Iterate over all loaded nodes and create a list of file names and a list of file data.
             name_base_offset = u8_data.tell()
             for node in self.u8_node_list:
                 u8_data.seek(name_base_offset + node.name_offset)
@@ -97,30 +98,75 @@ class U8Archive:
                 self.file_name_list.append(name)
                 if node.type == 0:
                     u8_data.seek(node.data_offset)
-                    self.u8_file_data_list.append(u8_data.read(node.size))
+                    self.file_data_list.append(u8_data.read(node.size))
                 else:
-                    self.u8_file_data_list.append(b'')
+                    self.file_data_list.append(b'')
 
-    def dump(self) -> None:
+    def dump(self) -> bytes:
         """
-        Dumps the U8Archive object into a U8 file.
+        Dumps the U8Archive object into the raw data of a U8 archive.
+
+        Returns
+        -------
+        bytes
+            The full U8 archive as bytes.
         """
+        # This is 0 because the header size DOES NOT include the initial 32 bytes describing the file.
+        header_size = 0
+        # Add 12 bytes for each node, since that's how many bytes each one is made up of.
+        for node in range(len(self.u8_node_list)):
+            header_size += 12
+        # Add the number of bytes used for each file/folder name in the string table.
+        for file_name in self.file_name_list:
+            header_size += len(file_name) + 1
+        # The initial data offset is equal to the file header (32 bytes) + node data aligned to 16 bytes.
+        data_offset = align_value(header_size + 32, 16)
+        # Adjust all nodes to place file data in the same order as the nodes. Why isn't it already like this?
+        current_data_offset = data_offset
+        for node in range(len(self.u8_node_list)):
+            if self.u8_node_list[node].type == 0:
+                self.u8_node_list[node].data_offset = current_data_offset
+                current_data_offset += self.u8_node_list[node].size
+        # Begin joining all the U8 archive data into one variable.
         u8_data = b''
         # Magic number.
         u8_data += b'\x55\xAA\x38\x2D'
         # Root node offset (this is always 0x20).
         u8_data += int.to_bytes(0x20, 4)
+        # Size of the file header (excluding the first 32 bytes).
+        u8_data += int.to_bytes(header_size, 4)
+        # Offset of the beginning of the data region of the U8 archive.
+        u8_data += int.to_bytes(data_offset, 4)
+        # 16 bytes of zeroes.
+        u8_data += (b'\x00' * 16)
+        # Iterate over all the U8 nodes and dump them.
+        for node in self.u8_node_list:
+            u8_data += int.to_bytes(node.type, 2)
+            u8_data += int.to_bytes(node.name_offset, 2)
+            u8_data += int.to_bytes(node.data_offset, 4)
+            u8_data += int.to_bytes(node.size, 4)
+        # Iterate over all file names and dump them. All file names are suffixed by a \x00 byte.
+        for file_name in self.file_name_list:
+            u8_data += str.encode(file_name) + b'\x00'
+        # Apply the extra padding we calculated earlier by padding to where the data offset begins.
+        while len(u8_data) < data_offset:
+            u8_data += b'\x00'
+        # Iterate all file data and dump it.
+        for file in self.file_data_list:
+            u8_data += file
+        # Return the U8 archive.
+        return u8_data
 
 
 def extract_u8(u8_data, output_folder) -> None:
     if os.path.isdir(output_folder):
         raise ValueError("Output folder already exists!")
-
     os.mkdir(output_folder)
-
+    # Create a new U8Archive object and load the provided U8 file data into it.
     u8_archive = U8Archive()
     u8_archive.load(u8_data)
-
+    # TODO: Comment this
+    # Also TODO: You can go more than two layers! Really should've checked that more before assuming it was the case.
     current_dir = ""
     for node in range(len(u8_archive.u8_node_list)):
         if u8_archive.u8_node_list[node].name_offset != 0:
@@ -135,20 +181,27 @@ def extract_u8(u8_data, output_folder) -> None:
             elif u8_archive.u8_node_list[node].type == 0:
                 lower_path = os.path.join(output_folder, current_dir)
                 output_file = open(os.path.join(lower_path, u8_archive.file_name_list[node]), "wb")
-                output_file.write(u8_archive.u8_file_data_list[node])
+                output_file.write(u8_archive.file_data_list[node])
                 output_file.close()
 
 
-def pack_u8(input_data) -> None:
-    if os.path.isdir(input_data):
+def pack_u8(input_path) -> bytes:
+    if os.path.isdir(input_path):
         raise ValueError("Only single-file packing is currently supported!")
-    elif os.path.isfile(input_data):
-        with open(input_data, "rb") as f:
+    elif os.path.isfile(input_path):
+        with open(input_path, "rb") as f:
             u8_archive = U8Archive()
 
-            file_name = os.path.basename(input_data)
+            file_name = os.path.basename(input_path)
+            file_data = f.read()
 
+            u8_archive.file_name_list.append("")
             u8_archive.file_name_list.append(file_name)
-            u8_archive.u8_file_data_list.append(f.read())
 
+            u8_archive.file_data_list.append(b'')
+            u8_archive.file_data_list.append(file_data)
 
+            u8_archive.u8_node_list.append(U8Node(256, 0, 0, 2))
+            u8_archive.u8_node_list.append(U8Node(0, 1, 0, len(file_data)))
+
+            return u8_archive.dump()

From 2755364472c459e4a93d6f25b4b2be525fe74d6f Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Wed, 5 Jun 2024 22:02:19 -0400
Subject: [PATCH 6/9] Rewrote U8 extraction code entirely, now handles U8 files
 in all cases

---
 src/libWiiPy/archive/u8.py | 75 +++++++++++++++++++++++++++++---------
 1 file changed, 57 insertions(+), 18 deletions(-)

diff --git a/src/libWiiPy/archive/u8.py b/src/libWiiPy/archive/u8.py
index c5443ae..314244d 100644
--- a/src/libWiiPy/archive/u8.py
+++ b/src/libWiiPy/archive/u8.py
@@ -5,6 +5,7 @@
 
 import io
 import os
+import pathlib
 from dataclasses import dataclass
 from typing import List
 from src.libWiiPy.shared import align_value
@@ -159,30 +160,68 @@ class U8Archive:
 
 
 def extract_u8(u8_data, output_folder) -> None:
-    if os.path.isdir(output_folder):
+    """
+    Extracts the provided U8 archive file data into the provided output folder path. Note that the folder must not
+    already exist to ensure that the output can correctly represent the file structure of the original U8 archive.
+
+    Parameters
+    ----------
+    u8_data : bytes
+        The data for the U8 file to extract.
+    output_folder : str
+        The path to a new folder to extract the archive to.
+    """
+    output_folder = pathlib.Path(output_folder)
+    if pathlib.Path.is_dir(output_folder):
         raise ValueError("Output folder already exists!")
     os.mkdir(output_folder)
     # Create a new U8Archive object and load the provided U8 file data into it.
     u8_archive = U8Archive()
     u8_archive.load(u8_data)
-    # TODO: Comment this
-    # Also TODO: You can go more than two layers! Really should've checked that more before assuming it was the case.
-    current_dir = ""
+    # This variable stores the path of the directory we're currently processing.
+    current_dir = output_folder
+    # This variable stores the final nodes for every directory we've entered, and is used to handle the recursion of
+    # those directories to ensure that everything gets where it belongs.
+    directory_recursion = [0]
+    # Iterate over every node and extract the files and folders.
     for node in range(len(u8_archive.u8_node_list)):
-        if u8_archive.u8_node_list[node].name_offset != 0:
-            if u8_archive.u8_node_list[node].type == 256:
-                if u8_archive.u8_node_list[node].data_offset == 0:
-                    os.mkdir(os.path.join(output_folder, u8_archive.file_name_list[node]))
-                    current_dir = u8_archive.file_name_list[node]
-                elif u8_archive.u8_node_list[node].data_offset < node:
-                    lower_path = os.path.join(output_folder, current_dir)
-                    os.mkdir(os.path.join(lower_path, u8_archive.file_name_list[node]))
-                    current_dir = os.path.join(current_dir, u8_archive.file_name_list[node])
-            elif u8_archive.u8_node_list[node].type == 0:
-                lower_path = os.path.join(output_folder, current_dir)
-                output_file = open(os.path.join(lower_path, u8_archive.file_name_list[node]), "wb")
-                output_file.write(u8_archive.file_data_list[node])
-                output_file.close()
+        # Code for a directory node. Second check just ensures we ignore the root node.
+        if u8_archive.u8_node_list[node].type == 256 and u8_archive.u8_node_list[node].name_offset != 0:
+            # The size value for a directory node is the position of the last node in this directory, with the root node
+            # counting as node 1.
+            # If the current node is below the end of the current directory, create this directory inside the previous
+            # current directory and make the current.
+            if node + 1 < max(directory_recursion):
+                current_dir = current_dir.joinpath(u8_archive.file_name_list[node])
+                os.mkdir(current_dir)
+            # If the current node is beyond the end of the current directory, we've followed that path all the way down,
+            # so reset back to the root directory and put our new directory there.
+            elif node + 1 > max(directory_recursion):
+                current_dir = output_folder.joinpath(u8_archive.file_name_list[node])
+                os.mkdir(current_dir)
+            # This check is here just in case a directory ever ends with an empty directory and not a file.
+            elif node + 1 == max(directory_recursion):
+                current_dir = current_dir.parent
+                directory_recursion.pop()
+            # If the last node for the directory we just processed is new (which is always should be), add it to the
+            # recursion array.
+            if u8_archive.u8_node_list[node].size not in directory_recursion:
+                directory_recursion.append(u8_archive.u8_node_list[node].size)
+        # Code for a file node.
+        elif u8_archive.u8_node_list[node].type == 0:
+            # Write out the file to the current directory.
+            output_file = open(current_dir.joinpath(u8_archive.file_name_list[node]), "wb")
+            output_file.write(u8_archive.file_data_list[node])
+            output_file.close()
+            # If this file is the final node for the current directory, pop() the recursion array and set the current
+            # directory to the parent of the previous current.
+            if node + 1 in directory_recursion:
+                current_dir = current_dir.parent
+                directory_recursion.pop()
+        # Code for a totally unrecognized node type, which should not happen.
+        elif u8_archive.u8_node_list[node].type != 0 and u8_archive.u8_node_list[node].type != 256:
+            raise ValueError("A node with an invalid type (" + str(u8_archive.u8_node_list[node].type) + ") was"
+                             "found!")
 
 
 def pack_u8(input_path) -> bytes:

From b30017460b01f56d80a28e898d6a5e5619197238 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Fri, 7 Jun 2024 22:30:34 -0400
Subject: [PATCH 7/9] Completed the U8 module, can now handle full folder
 packing

---
 pyproject.toml             |  2 +-
 src/libWiiPy/archive/u8.py | 83 +++++++++++++++++++++++++++++++++-----
 2 files changed, 73 insertions(+), 12 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 08b02c5..9d9a3c3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,7 @@ dependencies = [
 
 [project.urls]
 Homepage = "https://github.com/NinjaCheetah/libWiiPy"
-Issues = "https://github.com/NinjaCheetah/libWiipy/issues"
+Issues = "https://github.com/NinjaCheetah/libWiiPy/issues"
 
 [build-system]
 requires = ["setuptools>=61.0"]
diff --git a/src/libWiiPy/archive/u8.py b/src/libWiiPy/archive/u8.py
index 314244d..6e79b44 100644
--- a/src/libWiiPy/archive/u8.py
+++ b/src/libWiiPy/archive/u8.py
@@ -1,7 +1,7 @@
 # "archive/u8.py" from libWiiPy by NinjaCheetah & Contributors
 # https://github.com/NinjaCheetah/libWiiPy
 #
-# See https://wiibrew.org/wiki/U8_archive for details about the U8 archive format
+# See https://wiibrew.org/wiki/U8_archive for details about the U8 archive format.
 
 import io
 import os
@@ -191,16 +191,16 @@ def extract_u8(u8_data, output_folder) -> None:
             # counting as node 1.
             # If the current node is below the end of the current directory, create this directory inside the previous
             # current directory and make the current.
-            if node + 1 < max(directory_recursion):
+            if node + 1 < directory_recursion[-1]:
                 current_dir = current_dir.joinpath(u8_archive.file_name_list[node])
                 os.mkdir(current_dir)
             # If the current node is beyond the end of the current directory, we've followed that path all the way down,
             # so reset back to the root directory and put our new directory there.
-            elif node + 1 > max(directory_recursion):
+            elif node + 1 > directory_recursion[-1]:
                 current_dir = output_folder.joinpath(u8_archive.file_name_list[node])
                 os.mkdir(current_dir)
             # This check is here just in case a directory ever ends with an empty directory and not a file.
-            elif node + 1 == max(directory_recursion):
+            elif node + 1 == directory_recursion[-1]:
                 current_dir = current_dir.parent
                 directory_recursion.pop()
             # If the last node for the directory we just processed is new (which is always should be), add it to the
@@ -224,23 +224,84 @@ def extract_u8(u8_data, output_folder) -> None:
                              "found!")
 
 
+def _pack_u8_dir(u8_archive: U8Archive, current_path, node_count, name_offset):
+    # First, get the list of everything in current path.
+    root_list = os.listdir(current_path)
+    file_list = []
+    dir_list = []
+    # Create separate lists of the files and directories in the current directory so that we can handle the files first.
+    for path in root_list:
+        if os.path.isfile(current_path.joinpath(path)):
+            file_list.append(path)
+        elif os.path.isdir(current_path.joinpath(path)):
+            dir_list.append(path)
+    # For files, read their data into the file data list, add their name into the file name list, then calculate the
+    # offset for their file name and create a new U8Node() for them.
+    for file in file_list:
+        node_count += 1
+        u8_archive.file_name_list.append(file)
+        u8_archive.file_data_list.append(open(current_path.joinpath(file), "rb").read())
+        u8_archive.u8_node_list.append(U8Node(0, name_offset, 0, len(u8_archive.file_data_list[-1])))
+        name_offset = name_offset + len(file) + 1  # Add 1 to accommodate the null byte at the end of the name.
+    # For directories, add their name to the file name list, add empty data to the file data list (since they obviously
+    # wouldn't have any), find the total number of files and directories inside the directory to calculate the final
+    # node included in it, then recursively call this function again on that directory to process it.
+    for directory in dir_list:
+        node_count += 1
+        u8_archive.file_name_list.append(directory)
+        u8_archive.file_data_list.append(b'')
+        max_node = node_count + sum(1 for _ in current_path.joinpath(directory).rglob('*'))
+        u8_archive.u8_node_list.append(U8Node(256, name_offset, 0, max_node))
+        name_offset = name_offset + len(directory) + 1  # Add 1 to accommodate the null byte at the end of the name.
+        u8_archive, node_count, name_offset = _pack_u8_dir(u8_archive, current_path.joinpath(directory), node_count,
+                                                           name_offset)
+    # Return the U8Archive object, the current node we're on, and the current name offset.
+    return u8_archive, node_count, name_offset
+
+
 def pack_u8(input_path) -> bytes:
+    """
+    Packs the provided file or folder into a new U8 archive, and returns the raw file data for it.
+
+    Parameters
+    ----------
+    input_path
+        The path to the input file or folder.
+
+    Returns
+    -------
+    u8_archive : bytes
+        The data for the packed U8 archive.
+    """
+    input_path = pathlib.Path(input_path)
     if os.path.isdir(input_path):
-        raise ValueError("Only single-file packing is currently supported!")
+        # Append empty entries at the start for the root node, and then create the root U8Node() object, using rglob()
+        # to read the total count of files and directories that will be packed so that we can add the total node count.
+        u8_archive = U8Archive()
+        u8_archive.file_name_list.append("")
+        u8_archive.file_data_list.append(b'')
+        u8_archive.u8_node_list.append(U8Node(256, 0, 0, sum(1 for _ in input_path.rglob('*')) + 1))
+        # Call the private function _pack_u8_dir() on the root note, which will recursively call itself to pack every
+        # subdirectory and file. Discard node_count and name_offset since we don't care about them here, as they're
+        # really only necessary for the directory recursion.
+        u8_archive, _, _ = _pack_u8_dir(u8_archive, input_path, node_count=1, name_offset=1)
+        # Dump and return the U8 file data.
+        return u8_archive.dump()
     elif os.path.isfile(input_path):
+        # Simple code to handle if a single file is provided as input. Not really sure *why* you'd do this, since the
+        # whole point of a U8 archive is to stitch files together, but it's here nonetheless.
         with open(input_path, "rb") as f:
             u8_archive = U8Archive()
-
-            file_name = os.path.basename(input_path)
+            file_name = input_path.name
             file_data = f.read()
-
+            # Append blank file name for the root node.
             u8_archive.file_name_list.append("")
             u8_archive.file_name_list.append(file_name)
-
+            # Append blank data for the root node.
             u8_archive.file_data_list.append(b'')
             u8_archive.file_data_list.append(file_data)
-
+            # Append generic U8Node for the root, followed by the actual file's node.
             u8_archive.u8_node_list.append(U8Node(256, 0, 0, 2))
             u8_archive.u8_node_list.append(U8Node(0, 1, 0, len(file_data)))
-
+            # Return the processed data.
             return u8_archive.dump()

From 5743ee26951636f9e84f16f96045129d9f4ee93e Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Fri, 7 Jun 2024 23:05:45 -0400
Subject: [PATCH 8/9] Remove some pointless comments I noticed

---
 src/libWiiPy/archive/u8.py    | 2 --
 src/libWiiPy/title/content.py | 1 -
 src/libWiiPy/title/ticket.py  | 1 -
 src/libWiiPy/title/title.py   | 4 +---
 src/libWiiPy/title/tmd.py     | 1 -
 src/libWiiPy/title/wad.py     | 1 -
 6 files changed, 1 insertion(+), 9 deletions(-)

diff --git a/src/libWiiPy/archive/u8.py b/src/libWiiPy/archive/u8.py
index 6e79b44..3b0f812 100644
--- a/src/libWiiPy/archive/u8.py
+++ b/src/libWiiPy/archive/u8.py
@@ -285,7 +285,6 @@ def pack_u8(input_path) -> bytes:
         # subdirectory and file. Discard node_count and name_offset since we don't care about them here, as they're
         # really only necessary for the directory recursion.
         u8_archive, _, _ = _pack_u8_dir(u8_archive, input_path, node_count=1, name_offset=1)
-        # Dump and return the U8 file data.
         return u8_archive.dump()
     elif os.path.isfile(input_path):
         # Simple code to handle if a single file is provided as input. Not really sure *why* you'd do this, since the
@@ -303,5 +302,4 @@ def pack_u8(input_path) -> bytes:
             # Append generic U8Node for the root, followed by the actual file's node.
             u8_archive.u8_node_list.append(U8Node(256, 0, 0, 2))
             u8_archive.u8_node_list.append(U8Node(0, 1, 0, len(file_data)))
-            # Return the processed data.
             return u8_archive.dump()
diff --git a/src/libWiiPy/title/content.py b/src/libWiiPy/title/content.py
index c2a3392..fb7554d 100644
--- a/src/libWiiPy/title/content.py
+++ b/src/libWiiPy/title/content.py
@@ -89,7 +89,6 @@ class ContentRegion:
             content_region_data += content
             if padding_bytes > 0:
                 content_region_data += b'\x00' * padding_bytes
-        # Return the raw ContentRegion for the data contained in the object.
         return content_region_data
 
     def get_enc_content_by_index(self, index: int) -> bytes:
diff --git a/src/libWiiPy/title/ticket.py b/src/libWiiPy/title/ticket.py
index 6e433b0..b81615a 100644
--- a/src/libWiiPy/title/ticket.py
+++ b/src/libWiiPy/title/ticket.py
@@ -200,7 +200,6 @@ class Ticket:
             title_limit_data += int.to_bytes(self.title_limits_list[title_limit].maximum_usage, 4)
             # Write the entry to the ticket.
             ticket_data += title_limit_data
-        # Return the raw TMD for the data contained in the object.
         return ticket_data
 
     def get_title_id(self) -> str:
diff --git a/src/libWiiPy/title/title.py b/src/libWiiPy/title/title.py
index d0d6690..5b5b5e7 100644
--- a/src/libWiiPy/title/title.py
+++ b/src/libWiiPy/title/title.py
@@ -79,9 +79,7 @@ class Title:
         self.wad.set_ticket_data(self.ticket.dump())
         # Dump the ContentRegion and set it in the WAD.
         self.wad.set_content_data(self.content.dump())
-        # Dump the WAD with the new regions back into raw data and return it.
-        wad_data = self.wad.dump()
-        return wad_data
+        return self.wad.dump()
 
     def load_tmd(self, tmd: bytes) -> None:
         """
diff --git a/src/libWiiPy/title/tmd.py b/src/libWiiPy/title/tmd.py
index fd3c5cf..bdc0bfa 100644
--- a/src/libWiiPy/title/tmd.py
+++ b/src/libWiiPy/title/tmd.py
@@ -213,7 +213,6 @@ class TMD:
             content_data += binascii.unhexlify(self.content_records[content_record].content_hash)
             # Write the record to the TMD.
             tmd_data += content_data
-        # Return the raw TMD for the data contained in the object.
         return tmd_data
 
     def get_title_region(self) -> str:
diff --git a/src/libWiiPy/title/wad.py b/src/libWiiPy/title/wad.py
index 12ba493..9f86b57 100644
--- a/src/libWiiPy/title/wad.py
+++ b/src/libWiiPy/title/wad.py
@@ -178,7 +178,6 @@ class WAD:
         # Retrieve the content data and write it out.
         wad_data += self.get_content_data()
         wad_data = pad_bytes(wad_data)
-        # Return the raw WAD file for the data contained in the object.
         return wad_data
 
     def get_wad_type(self) -> str:

From 736a9e5c0c30007fc9880e8c8778517ac8087475 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Fri, 7 Jun 2024 23:11:51 -0400
Subject: [PATCH 9/9] Update README for v0.3.0 release

---
 README.md | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 3432bfd..586b543 100644
--- a/README.md
+++ b/README.md
@@ -1,16 +1,16 @@
 ![libWiiPy](https://github.com/NinjaCheetah/libWiiPy/assets/58050615/80093c68-b86e-4b96-87b7-db3855382ca8)
 # libWiiPy
-libWiiPy is a modern Python 3 library for interacting with and editing files from the Wii. It aims to be simple to use, well maintained, and offer as many features as reasonably possible in one library, so that a newly-written Python program could reasonably do 100% of its Wii-related work with just one library. It also aims to be fully cross-platform, so that any tools written with it can also be cross-platform.
+libWiiPy is a modern Python 3 library for handling the various files and formats found on the Wii. It aims to be simple to use, well maintained, and offer as many features as reasonably possible in one library, so that a newly-written Python program could reasonably do 100% of its Wii-related work with just one library. It also aims to be fully cross-platform, so that any tools written with it can also be cross-platform.
 
-libWiiPy is inspired by [libWiiSharp](https://github.com/TheShadowEevee/libWiiSharp), originally created by `Leathl`, now maintained by [@TheShadowEevee](https://github.com/TheShadowEevee). libWiiSharp is absolutely the way to go if you need a C# library for Wii files.
+libWiiPy is inspired by [libWiiSharp](https://github.com/TheShadowEevee/libWiiSharp), which was originally created by `Leathl` and is now maintained by [@TheShadowEevee](https://github.com/TheShadowEevee). If you're looking for a Wii library that isn't in Python, then go check it out!
 
-**Note:** While libWiiPy is directly inspired by libWiiSharp and aims to have feature parity with it, no code from either libWiiSharp or Wii.py was used in the making of this library. All code is original and is written by [@NinjaCheetah](https://github.com/NinjaCheetah), [@rvtr](https://github.com/rvtr), and any other GitHub contributors.
 
 # Features
 This list will expand as libWiiPy is developed, but these features are currently available:
 - TMD, ticket, and WAD parsing
 - WAD content extraction, decryption, re-encryption, and packing
 - Downloading titles from the NUS
+- Packing and unpacking U8 archives (.app, .arc, .carc, .szs)
 
 # Usage
 A wiki, and in the future a potential documenation site, is being worked on, and can be accessed [here](https://github.com/NinjaCheetah/libWiiPy/wiki). It is currently fairly barebones, but it will be improved in the future.
@@ -45,7 +45,7 @@ And that's all! You'll find your compiled pip package in `dist/`.
 # Special Thanks
 This project wouldn't be possible without the amazing people behind its predecessors and all of the people who have contributed to the documentation of the Wii's inner workings over at [WiiBrew](https://wiibrew.org).
 
-## Special Thanks for the Inspiration and Previous Projects
+## Special Thanks to People Behind Related Projects
 - Xuzz, SquidMan, megazig, Matt_P, Omega and The Lemon Man for creating Wii.py
 - Leathl for creating libWiiSharp
 - TheShadowEevee for maintaining libWiiSharp
@@ -59,3 +59,5 @@ Thank you to all of the contributors to the documentation on the WiiBrew pages t
 ### One additional special thanks to [@DamiDoop](https://github.com/DamiDoop)!
 She made the very cool banner you can see at the top of this README, and has also helped greatly with my sanity throughout debugging this library.
 
+**Note:** While libWiiPy is directly inspired by libWiiSharp and aims to have feature parity with it, no code from either libWiiSharp or Wii.py was used in the making of this library. All code is original and is written by [@NinjaCheetah](https://github.com/NinjaCheetah), [@rvtr](https://github.com/rvtr), and any other GitHub contributors.
+