#!/usr/bin/env bash

# Copyright 2021 The Kubernetes Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

###
# This script is to be used when updating Kubernetes and its staging
# repositories to *tagged* releases. This is the ideal case, but another
# script, submodule-k8s.sh, is available as a break-glass solution if we must
# switch to an unreleased commit.
###

set -o errexit
set -o pipefail

KUBE_ROOT="$(dirname "${BASH_SOURCE[0]}")/../.."
cd "${KUBE_ROOT}"

VERSION=${1#"v"}
APIS_VERSION=${2#"v"}
FORK=${3:-git@github.com:kubernetes/kubernetes.git}
SED=${4:-sed}

# $1: The k8s version to download.
cluster_autoscaler:list_mods:init() {
  k8s_version="${1:-${VERSION}}"

  workdir="$(mktemp -d)"
  repo="${workdir}/kubernetes"
  git clone --depth 1 "${FORK}" "${repo}"
  pushd "${repo}"
  git fetch --depth 1 origin "v${k8s_version}"
  git checkout FETCH_HEAD
}

cluster_autoscaler:list_mods:cleanup() {
  popd
  rm -rf "${workdir}"
}

# $1: The k8s version to download.
cluster_autoscaler:list_mods() {
  k8s_version="${1:-${VERSION}}"

  if [ -z "${k8s_version}" ]; then
    echo "Usage: hack/update-deps.sh <k8s version> <k8s version for apis> <k8s fork:-git@github.com:kubernetes/kubernetes.git>"
    exit 1
  fi
  cluster_autoscaler:list_mods:init "${k8s_version}" > /dev/null
  mods=($(
        cat go.mod | sed -n 's|.*k8s.io/\(.*\) => ./staging/src/k8s.io/.*|k8s.io/\1|p'
  ))
  cluster_autoscaler:list_mods:cleanup > /dev/null
  echo "${mods[@]}"
}

# $1: The package path.
# $2: The k8s version.
# $3: and later: The module names.
cluster_autoscaler:update_deps() {
  args=("${@}")
  pkg="${args[0]}"
  k8s_version="${args[1]}"
  mods=("${args[@]:2}")

  pushd "${pkg}"

  for mod in "${mods[@]}"; do
    echo "${pkg}: ${mod}"
    gomod="${mod}@kubernetes-${k8s_version}"
    gomod_json="$(go mod download -json "${gomod}")"
    retval="${?}"
    if [ "${retval}" -ne 0 ]; then
      echo "Error downloading module ${gomod}."
      exit 1
    fi
    mod_version=$(echo "${gomod_json}" | "${SED}" -n 's|.*"Version": "\(.*\)".*|\1|p')
    if [ "${pkg}" = "./cluster-autoscaler" ]; then
      go mod edit "-replace=${mod}=${mod}@${mod_version}"
    else
      go get "${mod}@${mod_version}"
    fi
  done

  go mod tidy

  if [ "${pkg}" = "./cluster-autoscaler" ]; then
    go get "k8s.io/kubernetes@v${k8s_version}"
    go mod tidy
    "${SED}" -i "s|\(const ClusterAutoscalerVersion = \)\".*\"|\1\"${k8s_version}\"|" "version/version.go"
  fi

  git rm -r --force --ignore-unmatch kubernetes
  popd
}

# k8s.io/autoscaler/cluster-autoscaler/go.mod
mods=($(cluster_autoscaler:list_mods "${VERSION}"))
cluster_autoscaler:update_deps "./cluster-autoscaler" "${VERSION}" "${mods[@]}"

# k8s.io/autoscaler/cluster-autoscaler/apis/go.mod
apis_mods=($(cluster_autoscaler:list_mods "${APIS_VERSION}"))
cluster_autoscaler:update_deps "./cluster-autoscaler/apis" "${APIS_VERSION}" "${apis_mods[@]}"