Merge pull request #513 from blueyed/custom-checks-dirs

custom-checks: handle multiple dirs

custom-checks

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