write metadata result

Author: László Vaskó

unblob/cli.py

@@ -192,8 +192,11 @@ def cli(
     )
 
     logger.info("Start processing file", file=file)
-    all_reports = process_file(config, file)
-    return all_reports
+    results = process_file(config, file)
+
+    with file.with_suffix(".result.json").open("w") as fd:
+        fd.write(results.to_json())
+    return results
 
 
 cli.context_class = UnblobContext

unblob/models.py

@@ -1,5 +1,7 @@
 import abc
 import itertools
+import json
+from enum import Enum
 from pathlib import Path
 from typing import List, Optional, Tuple, Type
 
@@ -128,6 +130,25 @@ def errors(self) -> List[ErrorReport]:
     def register(self, result: TaskResult):
         self.results.append(result)
 
+    def to_json(self, indent="  "):
+        return json.dumps(self.results, cls=_JSONEncoder, indent=indent)
+
+
+class _JSONEncoder(json.JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, Enum):
+            return obj.name
+
+        if attr.has(type(obj)):
+            extend_attr_output = True
+            attr_output = attr.asdict(obj, recurse=not extend_attr_output)
+            attr_output["__typename__"] = obj.__class__.__name__
+            return attr_output
+
+        if isinstance(obj, Path):
+            return str(obj)
+        return json.JSONEncoder.default(self, obj)
+
 
 class ExtractError(Exception):
     """There was an error during extraction"""

vulture_whitelist.py

@@ -3,6 +3,7 @@
 import unblob.plugins
 from unblob import cli
 from unblob.file_utils import File, iterbits, round_down
+from unblob.models import _JSONEncoder
 from unblob.parser import _HexStringToRegex
 
 _HexStringToRegex.literal
@@ -11,6 +12,8 @@
 _HexStringToRegex.range_jump
 _HexStringToRegex.alternative
 
+_JSONEncoder.default
+
 sys.breakpointhook
 cli.cli.context_class