From 8338cb62f7e294877fdb56ab657f19b103f076cc Mon Sep 17 00:00:00 2001
From: Adrian Vollmer <computerfluesterer@protonmail.com>
Date: Mon, 29 Apr 2024 09:35:11 +0200
Subject: [PATCH] Fix extraction

---
 src/zundler/__main__.py | 2 +-
 src/zundler/args.py     | 7 +++----
 src/zundler/embed.py    | 2 +-
 3 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/zundler/__main__.py b/src/zundler/__main__.py
index 0d2326b..675c639 100644
--- a/src/zundler/__main__.py
+++ b/src/zundler/__main__.py
@@ -5,7 +5,7 @@ def main():
 
     from .embed import embed_assets, extract_assets
 
-    if args.revert:
+    if args.extract:
         extract_assets(
             args.input_path,
             output_path=args.output_path,
diff --git a/src/zundler/args.py b/src/zundler/args.py
index dda507a..85227a2 100644
--- a/src/zundler/args.py
+++ b/src/zundler/args.py
@@ -30,12 +30,11 @@ parser.add_argument(
 )
 
 parser.add_argument(
-    "-R",
-    "--revert",
+    "-x",
+    "--extract",
     default=False,
     action="store_true",
-    help="set this flag for the reverse operation: deflate a Zundler file"
-    ' into its components (ouput_path must be a directory; "." by default)',
+    help='extract files from a Zundler file (ouput_path must be a directory; "." by default)',
 )
 
 parser.add_argument(
diff --git a/src/zundler/embed.py b/src/zundler/embed.py
index 535ae89..ef26356 100644
--- a/src/zundler/embed.py
+++ b/src/zundler/embed.py
@@ -324,7 +324,7 @@ def extract_assets(input_path, output_path=None):
         blob = base64.b64decode(blob)
         blob = zlib.decompress(blob).decode()
         blob = json.loads(blob)
-        file_tree = blob["file_tree"]
+        file_tree = blob["fileTree"]
     except Exception as e:
         logger.error(str(e))
         logger.error("Does not look like a Zundler output file: %s" % input_path)