From 1c01c586ede3ff8fa2d879afd4ef567e0346f041 Mon Sep 17 00:00:00 2001
From: Yuki Iwai <yuki.iwai.tz@gmail.com>
Date: Mon, 22 May 2023 23:47:39 +0900
Subject: [PATCH] Avoid to depend on local env when installing the
 code-generators (#1810)

Signed-off-by: Yuki Iwai <yuki.iwai.tz@gmail.com>
---
 hack/update-codegen.sh | 27 +++++++++++++--------------
 1 file changed, 13 insertions(+), 14 deletions(-)

diff --git a/hack/update-codegen.sh b/hack/update-codegen.sh
index 9600cf159b..4b79001030 100755
--- a/hack/update-codegen.sh
+++ b/hack/update-codegen.sh
@@ -29,26 +29,25 @@ set -o pipefail
 SCRIPT_ROOT=$(dirname "${BASH_SOURCE[0]}")/..
 ROOT_PKG=github.com/kubeflow/training-operator
 
-# Grab code-generator version from go.sum
-CODEGEN_VERSION=$(grep 'k8s.io/code-generator' go.mod | awk '{print $2}')
-CODEGEN_PKG=$(echo $(go env GOPATH)"/pkg/mod/k8s.io/code-generator@${CODEGEN_VERSION}")
+GET_PKG_LOCATION() {
+  pkg_name="${1:-}"
+
+  pkg_location="$(go list -m -f '{{.Dir}}' "${pkg_name}" 2>/dev/null)"
+  if [ "${pkg_location}" = "" ]; then
+    echo "${pkg_name} is missing. Running 'go mod download'."
 
-if [[ ! -d ${CODEGEN_PKG} ]]; then
-    echo "${CODEGEN_PKG} is missing. Running 'go mod download'."
     go mod download
-fi
+    pkg_location=$(go list -m -f '{{.Dir}}' "${pkg_name}")
+  fi
+  echo "${pkg_location}"
+}
 
+# Grab code-generator version from go.sum
+CODEGEN_PKG="$(GET_PKG_LOCATION "k8s.io/code-generator")"
 echo ">> Using ${CODEGEN_PKG}"
 
 # Grab openapi-gen version from go.mod
-OPENAPI_VERSION=$(grep 'k8s.io/kube-openapi' go.mod | awk '{print $2}')
-OPENAPI_PKG=$(echo $(go env GOPATH)"/pkg/mod/k8s.io/kube-openapi@${OPENAPI_VERSION}")
-
-if [[ ! -d ${OPENAPI_PKG} ]]; then
-    echo "${OPENAPI_PKG} is missing. Running 'go mod download'."
-    go mod download
-fi
-
+OPENAPI_PKG="$(GET_PKG_LOCATION 'k8s.io/kube-openapi')"
 echo ">> Using ${OPENAPI_PKG}"
 
 # code-generator does work with go.mod but makes assumptions about