custom-checks: handle multiple dirs

This also fixes a [shellcheck issue](https://github.com/koalaman/shellcheck/wiki/SC2013).

1 files changed, 11 insertions, 9 deletions

diff --git a/custom-checks b/custom-checks
index 440dafe5..37d2840c 100755
--- a/custom-checks
+++ b/custom-checks
@@ -44,24 +44,26 @@ fi
 
 shopt -s globstar
 
-directory="$1"
+directories=("$@")
 
 check_errors() {
     regex="$1"
     message="$2"
 
-    for match in $(
-        grep -n "$regex" "$directory"/**/*.vim \
-        | grep -o '^[^:]\+:[0-9]\+' \
-        | sed 's:^\./::'
-    ); do
-        RETURN_CODE=1
-        echo "$match $message"
+    for directory in "${directories[@]}"; do
+        while IFS= read -r match; do
+            RETURN_CODE=1
+            echo "$match $message"
+        done < <(grep -n "$regex" "$directory"/**/*.vim \
+            | grep -o '^[^:]\+:[0-9]\+' \
+            | sed 's:^\./::')
     done
 }
 
 if (( FIX_ERRORS )); then
-    sed -i "s/^\(function.*)\) *$/\1 abort/" "$directory"/**/*.vim
+    for directory in "${directories[@]}"; do
+        sed -i "s/^\(function.*)\) *$/\1 abort/" "$directory"/**/*.vim
+    done
 fi
 
 check_errors \