mirror of
https://github.com/encounter/ac-decomp.git
synced 2026-03-30 10:57:04 -07:00
Cuyler36
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
import pyjkernel
|
|
import os
|
|
import argparse
|
|
|
|
def unpack_dir(archive: pyjkernel.JKRArchive, dir: str, verbose=False):
|
|
if verbose:
|
|
print('Dumping dir: ' + dir)
|
|
# create all files
|
|
for file in archive.list_files(dir):
|
|
if verbose:
|
|
print('Dumping file: ' + file.name)
|
|
with open(os.path.join(dir, file.name), "wb") as f:
|
|
f.write(archive.get_file(dir + os.sep + file.name).data)
|
|
|
|
# create all subdirectories and recurse through them
|
|
for subdir in archive.list_folders(dir):
|
|
if not os.path.exists(dir + os.sep + subdir):
|
|
os.mkdir(dir + os.sep + subdir)
|
|
unpack_dir(archive, dir + os.sep + subdir, verbose)
|
|
|
|
def unpack_archive(path: str, out_path: str, verbose=False):
|
|
archive = pyjkernel.from_archive_file(path, True)
|
|
orig_dir = os.path.abspath(os.curdir)
|
|
os.chdir(out_path)
|
|
if not os.path.exists(archive.root_name):
|
|
os.mkdir(archive.root_name)
|
|
unpack_dir(archive, archive.root_name, verbose)
|
|
os.chdir(orig_dir)
|
|
|
|
def pack_dir(archive: pyjkernel.JKRArchive, path: str, verbose=False):
|
|
local_path = os.path.dirname(path)
|
|
orig_dir = os.path.abspath(os.curdir)
|
|
if local_path != "":
|
|
os.chdir(local_path)
|
|
local_root = os.path.basename(os.path.normpath(path))
|
|
|
|
for root, dirs, files in os.walk(local_root):
|
|
files.sort(key=lambda item: (item.lower(), item))
|
|
for dir in dirs:
|
|
archive.create_folder(root + os.sep + dir)
|
|
|
|
for file in files:
|
|
if verbose:
|
|
print('Packing file: ' + root + os.sep + file)
|
|
with open(root + os.sep + file, "rb") as f:
|
|
archive.create_file(root + os.sep + file, bytearray(f.read()), pyjkernel.JKRPreloadType.ARAM)
|
|
os.chdir(orig_dir)
|
|
|
|
def pack_archive(root_path: str, out_path: str, verbose=False):
|
|
root_name = os.path.basename(os.path.normpath(root_path))
|
|
archive = pyjkernel.create_new_archive(root_name)
|
|
pack_dir(archive, root_path, verbose)
|
|
pyjkernel.write_archive_file(archive, out_path, True, pyjkernel.jkrcomp.JKRCompression.NONE, 0)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Pack or unpack JSystem JKernel archives.')
|
|
parser.add_argument('-v', help='Enable verbose logging.', required=False, action='store_true')
|
|
parser.add_argument('path', help='The path of the folder to pack or archive file to unpack.')
|
|
parser.add_argument('out', help='The path of the destination folder or file.')
|
|
|
|
args = parser.parse_args()
|
|
if os.path.isfile(args.path):
|
|
unpack_archive(args.path, args.out, args.v)
|
|
elif os.path.isdir(args.path):
|
|
pack_archive(args.path, args.out, args.v)
|
|
else:
|
|
raise Exception('path is not a valid file or directory!')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|