Formatting and Documentation

pyjsclear/__init__.py

@@ -5,14 +5,18 @@
 Python package.
 """
 
+from pathlib import Path
+
 from .deobfuscator import Deobfuscator
 
 
+__all__ = ['Deobfuscator', 'deobfuscate', 'deobfuscate_file']
+
 __version__ = '0.1.4'
 
 
 def deobfuscate(code: str, max_iterations: int = 50) -> str:
-    """Deobfuscate JavaScript code. Returns cleaned source.
+    """Deobfuscate JavaScript code and return cleaned source.
 
     Args:
         code: JavaScript source code string.
@@ -24,7 +28,17 @@ def deobfuscate(code: str, max_iterations: int = 50) -> str:
     return Deobfuscator(code, max_iterations=max_iterations).execute()
 
 
-def deobfuscate_file(input_path: str, output_path: str | None = None, max_iterations: int = 50) -> str | bool:
+def _write_output(output_path: str | Path, content: str) -> None:
+    """Write deobfuscated content to the given file path."""
+    with open(output_path, 'w') as output_file:
+        output_file.write(content)
+
+
+def deobfuscate_file(
+    input_path: str | Path,
+    output_path: str | Path | None = None,
+    max_iterations: int = 50,
+) -> str | bool:
     """Deobfuscate a JavaScript file.
 
     Args:
@@ -40,8 +54,8 @@ def deobfuscate_file(input_path: str, output_path: str | None = None, max_iterat
 
     result = deobfuscate(code, max_iterations=max_iterations)
 
-    if output_path:
-        with open(output_path, 'w') as output_file:
-            output_file.write(result)
-        return result != code
-    return result
+    if not output_path:
+        return result
+
+    _write_output(output_path, result)
+    return result != code

pyjsclear/__main__.py

@@ -6,29 +6,38 @@
 from . import deobfuscate
 
 
+def _read_input(source_path: str) -> str:
+    """Read JavaScript source from stdin or a file path."""
+    if source_path == '-':
+        return sys.stdin.read()
+    with open(source_path, 'r', errors='replace') as input_file:
+        return input_file.read()
+
+
+def _write_output(destination_path: str, content: str) -> None:
+    """Write deobfuscated content to the given file path."""
+    with open(destination_path, 'w') as output_file:
+        output_file.write(content)
+
+
 def main() -> None:
-    parser = argparse.ArgumentParser(description='Deobfuscate JavaScript files.')
-    parser.add_argument('input', help='Input JS file (use - for stdin)')
-    parser.add_argument('-o', '--output', help='Output file (default: stdout)')
-    parser.add_argument(
+    """Parse CLI arguments and run the deobfuscator."""
+    argument_parser = argparse.ArgumentParser(description='Deobfuscate JavaScript files.')
+    argument_parser.add_argument('input', help='Input JS file (use - for stdin)')
+    argument_parser.add_argument('-o', '--output', help='Output file (default: stdout)')
+    argument_parser.add_argument(
         '--max-iterations',
         type=int,
         default=50,
         help='Maximum transform passes (default: 50)',
     )
-    args = parser.parse_args()
-
-    if args.input == '-':
-        code = sys.stdin.read()
-    else:
-        with open(args.input, 'r', errors='replace') as input_file:
-            code = input_file.read()
+    args = argument_parser.parse_args()
 
+    code = _read_input(args.input)
     result = deobfuscate(code, max_iterations=args.max_iterations)
 
     if args.output:
-        with open(args.output, 'w') as output_file:
-            output_file.write(result)
+        _write_output(args.output, result)
         return
 
     sys.stdout.write(result)