PHP-102055: Fix access token retrieval URL (#264)

PHP-102055: Fix access token retrieval URL
Remove staging refs

Author: june-hua

startupscript/butane/004-git-clone-devcontainer.sh

@@ -17,6 +17,19 @@ if [[ $# -lt 1 ]]; then
     usage
 fi
 
+
+# Map the server to appropriate service path
+function get_service_url() {
+  case "$1" in
+    "dev-stable") echo "https://workbench-dev.verily.com/api/$2" ;;
+    "dev-unstable") echo "https://workbench-dev-unstable.verily.com/api/$2" ;;
+    "test") echo "https://workbench-test.verily.com/api/$2" ;;
+    "prod") echo "https://workbench.verily.com/api/$2" ;;
+    *) return 1 ;;
+  esac
+}
+readonly -f get_service_url
+
 source /home/core/metadata-utils.sh
 
 # To accommodate the use of SSH URLs for public Git repositories, set the following Git configuration:
@@ -44,11 +57,21 @@ api_url="${api_url%.git}"
 # Check if repo is private
 private_status=$(curl --retry 5 -s "${api_url}" | jq -r ".status")
 if [[ "${PRIVATE_DEVCONTAINER_ENABLED}" == "TRUE" && "${private_status}" == 404 ]]; then
+  # Get ECM service URL
+  SERVER="$(get_metadata_value "terra-cli-server" "")"
+  readonly SERVER
+  if [[ -z "${SERVER}" ]]; then
+    SERVER="prod"
+  fi
+  if ! ECM_SERVICE_URL="$(get_service_url "${SERVER}" "ecm")"; then
+    exit 1
+  fi
+
   # disable logs to not expose access token
   set +o xtrace
 
   # Retrieve GitHub access token
-  response=$(curl https://workbench-dev.verily.com/api/ecm/api/oauth/v1/github/access-token \
+  response=$(curl "${ECM_SERVICE_URL}/api/oauth/v1/github/access-token" \
   -w "\n%{http_code}" \
   -H "Authorization: Bearer $(/home/core/wb.sh auth print-access-token)")
   http_code=$(echo "${response}" | tail -n1)

startupscript/dataproc/startup.sh

@@ -202,7 +202,6 @@ function get_service_url() {
   "dev-stable") echo "https://workbench-dev.verily.com/api/$2" ;;
   "dev-unstable") echo "https://workbench-dev-unstable.verily.com/api/$2" ;;
   "test") echo "https://workbench-test.verily.com/api/$2" ;;
-  "staging") echo "https://workbench-staging.verily.com/api/$2" ;;
   "prod") echo "https://workbench.verily.com/api/$2" ;;
   *) return 1 ;;
   esac
@@ -231,7 +230,7 @@ function get_ui_uri() {
     echo "https://workbench.verily.com"
   elif [[ "${env}" == "dev-stable" ]]; then
     echo "https://workbench-dev.verily.com"
-  elif [[ "${env}" == "dev-unstable" || "${env}" == "test" || "${env}" == "staging" ]]; then
+  elif [[ "${env}" == "dev-unstable" || "${env}" == "test" ]]; then
     echo "https://workbench-${env}.verily.com"
   else
     echo "https://workbench.verily.com"

startupscript/install-cli.sh

@@ -29,7 +29,6 @@ function get_axon_version_url() {
     "dev-stable") echo "https://workbench-dev.verily.com/api/axon/version" ;;
     "dev-unstable") echo "https://workbench-dev-unstable.verily.com/api/axon/version" ;;
     "test") echo "https://workbench-test.verily.com/api/axon/version" ;;
-    "staging") echo "https://workbench-staging.verily.com/api/axon/version" ;;
     "prod") echo "https://workbench.verily.com/api/axon/version" ;;
     *) return 1 ;;
   esac

startupscript/vertex-ai-user-managed-notebook/post-startup.sh

@@ -238,7 +238,6 @@ function get_service_url() {
     "dev-stable") echo "https://workbench-dev.verily.com/api/$2" ;;
     "dev-unstable") echo "https://workbench-dev-unstable.verily.com/api/$2" ;;
     "test") echo "https://workbench-test.verily.com/api/$2" ;;
-    "staging") echo "https://workbench-staging.verily.com/api/$2" ;;
     "prod") echo "https://workbench.verily.com/api/$2" ;;
     *) return 1 ;;
   esac