#!/usr/bin/env bash
# Copyright 2019 The Kubernetes Authors.
# SPDX-License-Identifier: Apache-2.0

set -x
set -e
set -o pipefail
set -o nounset

if [[ -z "${1-}" ]] ; then
  echo "Usage: $0 <cmd>"
  echo "Example: $0 lint"
  exit 1
fi

cmd=$1
skip_pattern="${2-}"
expected_module_count=${3:-48}

seen=()
# Hack scripts must be run from the root of the repository.
KUSTOMIZE_ROOT=$(pwd)
export KUSTOMIZE_ROOT

# Build find command with multiple -not -path options
find_cmd="find . -name go.mod -not -path \"./site/*\""
if [[ -n "$skip_pattern" ]]; then
  # Split skip_pattern by | and add -not -path for each
  IFS='|' read -ra PATTERNS <<< "$skip_pattern"
  for pattern in "${PATTERNS[@]}"; do
    find_cmd+=" -not -path \"$pattern\""
  done
fi

# verify all modules pass validation
for i in $(eval "$find_cmd"); do
  pushd .
  cd $(dirname "$i");

  set +x
  dir=$(pwd)
  module="${dir#"$KUSTOMIZE_ROOT"}"
  echo -e "\n----------------------------------------------------------"
  echo "Running command in $module"
  echo -e "----------------------------------------------------------"
  set -x

  bash -c "$cmd"
  seen+=("$module")
  popd
done

set +x
echo -e "\n\n----------------------------------------------------------"
echo -e "SUCCESS: Ran '$cmd' on the following modules:"
printf "  - %s\n" "${seen[@]}"

if [[ "${#seen[@]}" -ne $expected_module_count ]]; then
  echo
  echo "SANITY CHECK FAILURE: Expected to see $expected_module_count modules, but saw ${#seen[@]}"
  exit 1
fi