Added --ignore argument to ignore certain folders Made it only proccess .c and .h files - Added a recursively walk through folders to take a folder as an argument ** If it's a file, process directly ** If it's a directory, walk through recursively

Author: zouhairDe

c_formatter_42/__main__.py

@@ -12,16 +12,61 @@
 #                                                                              #
 # ############################################################################ #
 
+#!/usr/bin/env python3
 import argparse
 import sys
-
+import os
 from c_formatter_42.run import run_all
 
+def process_file(filepath, confirm=False):
+    """Process a single file with formatting."""
+    try:
+        with open(filepath, "r") as file:
+            content = file.read()
+        
+        if confirm:
+            result = input(f"Are you sure you want to overwrite {filepath}?[y/N] ")
+            if result != "y":
+                return True
+        
+        print(f"Formatting: {filepath}")
+        with open(filepath, "w") as file:
+            file.write(run_all(content))
+        return True
+    except OSError as e:
+        print(f"Error: {e.filename}: {e.strerror}", file=sys.stderr)
+        return False
 
-def main():
+def process_path(path, confirm=False, ignore_dirs=None):
+    """
+    Process a path (file or directory) recursively.
+    Formats .c and .h files in the directory.
+    """
+    ignore_dirs = ignore_dirs or []
+    
+    if os.path.isfile(path):
+        if path.endswith(('.c', '.h')):
+            return process_file(path, confirm)
+        return True
+    
+    if os.path.isdir(path):
+        success = True
+        for root, dirs, files in os.walk(path):
+            dirs[:] = [d for d in dirs if not any(ignore_dir in os.path.join(root, d) for ignore_dir in ignore_dirs)]
+            
+            for file in files:
+                if file.endswith(('.c', '.h')):
+                    filepath = os.path.join(root, file)
+                    success &= process_file(filepath, confirm)
+        return success
+    
+    print(f"Error: {path} is not a file or directory", file=sys.stderr)
+    return False
+
+def main() -> int:
     arg_parser = argparse.ArgumentParser(
         prog="c_formatter_42",
-        description="Format C source according to the norm",
+        description="Format C source files according to the norm",
         formatter_class=argparse.RawTextHelpFormatter,
     )
     arg_parser.add_argument(
@@ -31,33 +76,30 @@ def main():
         help="Ask confirmation before overwriting any file",
     )
     arg_parser.add_argument(
-        "filepaths",
-        metavar="FILE",
+        "-i",
+        "--ignore",
+        nargs='+',
+        default=[],
+        help="Ignore specified folders (e.g. .git/ .vscode/)",
+    )
+    arg_parser.add_argument(
+        "paths",
+        metavar="PATH",
         nargs="*",
-        help="File to format inplace, if no file is provided read STDIN",
+        help="Files or directories to format. If no path is provided, read STDIN",
     )
     args = arg_parser.parse_args()
-
-    if len(args.filepaths) == 0:
+    
+    if len(args.paths) == 0:
         content = sys.stdin.read()
         print(run_all(content), end="")
-    else:
-        for filepath in args.filepaths:
-            try:
-                with open(filepath, "r") as file:
-                    content = file.read()
-                if args.confirm:
-                    result = input(
-                        f"Are you sure you want to overwrite {filepath}?[y/N]"
-                    )
-                    if result != "y":
-                        continue
-                print(f"Writing to {filepath}")
-                with open(filepath, "w") as file:
-                    file.write(run_all(content))
-            except OSError as e:
-                print(f"Error: {e.filename}: {e.strerror}")
-
+        return 0
+    
+    success = True
+    for path in args.paths:
+        success &= process_path(path, args.confirm, args.ignore)
+    
+    return 0 if success else 1
 
 if __name__ == "__main__":
-    main()
+    sys.exit(main())