fix script

format.bat

@@ -1,30 +1,74 @@
 @echo off
-REM Variable that will hold the name of the clang-format command
+SETLOCAL ENABLEDELAYEDEXPANSION
+
+REM =========================
+REM Variables
+REM =========================
 SET FMT=clang-format
+SET MODE=
+SET EXIT_CODE=0
 
-REM Function to format files
-:format
-for /r %%f in (*.h *.hpp *.m *.mm *.c *.cpp) do (
-    echo %%~nxf | findstr /i "_generated$" >nul
-    if errorlevel 1 (
-        echo format %%f
-        %FMT% -i "%%f"
+REM =========================
+REM Parse arguments
+REM =========================
+:parse_args
+IF "%~1"=="" GOTO end_parse_args
+IF "%~1"=="--check" (
+    SET MODE=check
+) ELSE IF "%~1"=="--fix" (
+    SET MODE=fix
+) ELSE (
+    SET DIRS=!DIRS! "%~1"
+)
+SHIFT
+GOTO parse_args
+:end_parse_args
+
+IF "%MODE%"=="" (
+    ECHO Usage: %~nx0 --check|--fix <directory> [<directory> ...]
+    EXIT /B 1
+)
+
+IF "%DIRS%"=="" (
+    ECHO Please provide at least one directory.
+    EXIT /B 1
+)
+
+REM =========================
+REM Iterate over directories
+REM =========================
+FOR %%D IN (%DIRS%) DO (
+    IF NOT EXIST "%%~D" (
+        ECHO %%~D is not a valid directory.
+        SET EXIT_CODE=1
+        GOTO :continue_dirs
     )
-)
-echo ~~~ %1 Done ~~~
-exit /b
 
-REM Check if argument is provided
-if "%1"=="" (
-    echo Please provide a directory as an argument.
-    exit /b
+    REM Recursively find source files
+    FOR /R "%%~D" %%F IN (*.h *.hpp *.c *.cpp *.m *.mm) DO (
+        SET FILE=%%F
+        REM Skip *_generated* files
+        ECHO !FILE! | FINDSTR /I "_generated" >nul
+        IF ERRORLEVEL 1 (
+            IF "%MODE%"=="fix" (
+                ECHO Formatting !FILE!
+                %FMT% -i "!FILE!"
+            ) ELSE (
+                REM Check mode: run clang-format --dry-run --Werror
+                %FMT% --dry-run --Werror "!FILE!" 2>nul
+                IF ERRORLEVEL 1 (
+                    REM clang-format will already print diagnostic
+                    ECHO --- Diff for !FILE! ---
+                    REM Generate diff using fc
+                    %FMT% "!FILE!" > "%TEMP%\fmt.tmp"
+                    fc "!FILE!" "%TEMP%\fmt.tmp"
+                    SET EXIT_CODE=1
+                )
+            )
+        )
+    )
+
+    :continue_dirs
 )
 
-REM Check if directory exists
-if not exist "%1" (
-    echo %1 is not a valid directory.
-    exit /b
-)
-
-cd %1
-call :format
+EXIT /B %EXIT_CODE%