#! /bin/bash

CSI_PROW_SIDECAR_E2E_IMPORT_PATH="github.com/kubernetes-csi/external-provisioner/v5/test/e2e"
CSI_PROW_SIDECAR_E2E_PATH="github.com/kubernetes-csi/external-provisioner/test/e2e"

. release-tools/prow.sh

volume_mode_conversion () {
   [ "${VOLUME_MODE_CONVERSION_TESTS}" == "true" ]
}

if volume_mode_conversion; then
  install_snapshot_controller() {
      CONTROLLER_DIR="https://raw.githubusercontent.com/kubernetes-csi/external-snapshotter/${CSI_SNAPSHOTTER_VERSION}"
      if [[ ${REPO_DIR} == *"external-snapshotter"* ]]; then
          CONTROLLER_DIR="${REPO_DIR}"
      fi
      SNAPSHOT_RBAC_YAML="${CONTROLLER_DIR}/deploy/kubernetes/snapshot-controller/rbac-snapshot-controller.yaml"
      echo "kubectl apply -f ${SNAPSHOT_RBAC_YAML}"
      # Ignore: Double quote to prevent globbing and word splitting.
      # shellcheck disable=SC2086
      kubectl apply -f ${SNAPSHOT_RBAC_YAML}

      cnt=0
      until kubectl get clusterrolebinding snapshot-controller-role; do
         if [ $cnt -gt 30 ]; then
            echo "Cluster role bindings:"
            kubectl describe clusterrolebinding
            echo >&2 "ERROR: snapshot controller RBAC not ready after over 5 min"
            exit 1
        fi
        echo "$(date +%H:%M:%S)" "waiting for snapshot RBAC setup complete, attempt #$cnt"
    	cnt=$((cnt + 1))
        sleep 10
      done

    SNAPSHOT_CONTROLLER_YAML="${CSI_PROW_WORK}/snapshot-controller.yaml"
    run curl "${CONTROLLER_DIR}/deploy/kubernetes/snapshot-controller/setup-snapshot-controller.yaml" --output "${SNAPSHOT_CONTROLLER_YAML}" --silent --location

    echo "Enabling prevent-volume-mode-conversion in snapshot-controller"
    sed -i -e 's/# end snapshot controller args/- \"--prevent-volume-mode-conversion=true\"\n            # end snapshot controller args/' "${SNAPSHOT_CONTROLLER_YAML}"

    echo "kubectl apply -f $SNAPSHOT_CONTROLLER_YAML"
    kubectl apply -f "$SNAPSHOT_CONTROLLER_YAML"

    cnt=0
    expected_running_pods=$(kubectl apply --dry-run=client -o "jsonpath={.spec.replicas}" -f "$SNAPSHOT_CONTROLLER_YAML")
    expected_namespace=$(kubectl apply --dry-run=client -o "jsonpath={.metadata.namespace}" -f "$SNAPSHOT_CONTROLLER_YAML")
    while [ "$(kubectl get pods -n "$expected_namespace" -l app=snapshot-controller | grep 'Running' -c)" -lt "$expected_running_pods" ]; do
      if [ $cnt -gt 30 ]; then
        echo "snapshot-controller pod status:"
        kubectl describe pods -n "$expected_namespace" -l app=snapshot-controller
        echo >&2 "ERROR: snapshot controller not ready after over 5 min"
        exit 1
      fi
      echo "$(date +%H:%M:%S)" "waiting for snapshot controller deployment to complete, attempt #$cnt"
	  cnt=$((cnt + 1))
      sleep 10
    done
  }
fi

main