From 72cc7b581b7e74156b9f6e6e73a054223d12bc67 Mon Sep 17 00:00:00 2001
From: Kevin Veen-Birkenbach <kevin@veen.world>
Date: Thu, 14 Sep 2023 08:54:21 +0200
Subject: [PATCH] added compress function

---
 scan.py | 31 ++++++++++++++++++++-----------
 1 file changed, 20 insertions(+), 11 deletions(-)

diff --git a/scan.py b/scan.py
index 64c2571..57b079e 100644
--- a/scan.py
+++ b/scan.py
@@ -1,6 +1,7 @@
 import os
 import argparse
 import re
+import zlib
 
 def remove_comments(content, file_type):
     """Remove comments from the content based on file type."""
@@ -17,6 +18,10 @@ def remove_comments(content, file_type):
     # Add more file types and their comment styles as needed
     return content.strip()
 
+def compress_code(content):
+    """Compress code using zlib."""
+    return zlib.compress(content.encode())
+
 def filter_directories(dirs, ignore_strings, ignore_hidden):
     """Filter out directories based on ignore criteria."""
     if ignore_hidden:
@@ -33,52 +38,56 @@ def should_print_file(file, file_filters, ignore_strings, ignore_hidden):
         return False
     return True
 
-def print_file_content(file_path, no_comments):
+def print_file_content(file_path, no_comments, compress):
     """Print the content of a file."""
     try:
         with open(file_path, 'r') as f:
             content = f.read()
         if no_comments:
             file_type = os.path.splitext(file_path)[1]
-            print(file_type)
             content = remove_comments(content, file_type)
         print(f"======== File Path: {file_path} ========")
-        print(content)
+        if compress:
+            compressed_content = compress_code(content)
+            print(f"======== Compressed Code ========")
+            print(compressed_content)
+        else:
+            print(content)
         print("==================================\n")
     except UnicodeDecodeError:
         print(f"Warning: Could not read file due to encoding issues: {file_path}")
         exit(1)
 
-def handle_directory(directory, file_filters, ignore_strings, ignore_hidden, verbose, no_comments):
+def handle_directory(directory, file_filters, ignore_strings, ignore_hidden, verbose, no_comments, compress):
     """Handle scanning and printing for directories."""
     for root, dirs, files in os.walk(directory):
         filter_directories(dirs, ignore_strings, ignore_hidden)
         for file in files:
             if should_print_file(file, file_filters, ignore_strings, ignore_hidden):
-                print_file_content(os.path.join(root, file), no_comments)
+                print_file_content(os.path.join(root, file), no_comments, compress)
             elif verbose:
                 print(f"Skipped file: {file}")
 
-def handle_file(file_path, file_filters, ignore_strings, ignore_hidden):
+def handle_file(file_path, file_filters, ignore_strings, ignore_hidden, no_comments, compress):
     """Handle scanning and printing for individual files."""
-    if should_print_file(os.path.basename(file_path), file_filters, ignore_strings, ignore_hidden):
-        print_file_content(file_path, no_comments)
+    print_file_content(file_path, no_comments, compress)
 
 def main():
-    parser = argparse.ArgumentParser(description="Scan directories and print file contents.")
+    parser = argparse.ArgumentParser(description="Scan directories and print/compile file contents.")
     parser.add_argument("paths", nargs='+', help="List of files or directories to scan.")
     parser.add_argument("-f", "--filetype", nargs='+', default=[], help="Filter by file types (e.g., .txt .log).")
     parser.add_argument("-i", "--ignore", nargs='+', default=[], help="Ignore files and folders containing these strings.")
     parser.add_argument("--ignore-hidden", action='store_true', help="Ignore hidden directories and files.")
     parser.add_argument("-v", "--verbose", action='store_true', help="Enable verbose mode.")
     parser.add_argument("--no-comments", action='store_true', help="Remove comments from the displayed content based on file type.")
+    parser.add_argument("--compress", action='store_true', help="Compress code (for Python files).")
     args = parser.parse_args()
 
     for path in args.paths:
         if os.path.isdir(path):
-            handle_directory(path, args.filetype, args.ignore, args.ignore_hidden, args.verbose, args.no_comments)
+            handle_directory(path, args.filetype, args.ignore, args.ignore_hidden, args.verbose, args.no_comments, args.compress)
         elif os.path.isfile(path):
-            handle_file(path, args.filetype, args.ignore, args.ignore_hidden, args.no_comments)
+            handle_file(path, args.filetype, args.ignore, args.ignore_hidden, args.no_comments, args.compress)
         else:
             print(f"Error: {path} is neither a valid file nor a directory.")
             exit(1)