update generate.sh

Author: Not-Dhananjay-Mishra

script/generate.sh

@@ -8,10 +8,28 @@ set -e
 
 CDPATH="" cd -- "$(dirname -- "$0")/.."
 
+CHECK_MODE=0
 if [ "$1" = "--check" ]; then
   export CHECK=1
+  CHECK_MODE=1
 fi
 
 go generate ./...
 
+MOD_DIRS="$(git ls-files '*go.mod' | xargs dirname | sort)"
+
+for dir in $MOD_DIRS; do
+  (
+    cd "$dir"
+    if [ "$CHECK_MODE" = "1" ]; then
+      if ! go mod tidy -diff; then
+        echo "go.mod/go.sum are out of date in $dir"
+        exit 1
+      fi
+    else
+      go mod tidy
+    fi
+  )
+done
+
 script/run-check-structfield-settings.sh -fix