#!/usr/bin/env bash

# Copyright 2016 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 downloads and installs the Kubernetes client and server
# (and optionally test) binaries,
# It is intended to be called from an extracted Kubernetes release tarball.
#
# We automatically choose the correct client binaries to download.
#
# Options:
#  Set KUBERNETES_SERVER_ARCH to choose the server (Kubernetes cluster)
#  architecture to download:
#    * amd64 [default]
#    * arm
#    * arm64
#    * ppc64le
#    * s390x
#
#  Set KUBERNETES_SKIP_CONFIRM to skip the installation confirmation prompt.
#  Set KUBERNETES_RELEASE_URL to choose where to download binaries from.
#    (Defaults to https://storage.googleapis.com/kubernetes-release/release).
#  Set KUBERNETES_DOWNLOAD_TESTS to additionally download and extract the test
#    binaries tarball.

set -o errexit
set -o nounset
set -o pipefail

setvar KUBE_ROOT = $(cd $(dirname "${BASH_SOURCE}")/.. && pwd)

setvar KUBERNETES_RELEASE_URL = "${KUBERNETES_RELEASE_URL:-https://dl.k8s.io}"

proc detect_kube_release {
  if [[ -n "${KUBE_VERSION:-}" ]] {
    return 0  # Allow caller to explicitly set version
  }

  if [[ ! -e "${KUBE_ROOT}/version" ]] {
    echo "Can't determine Kubernetes release." >&2
    echo "${BASH_SOURCE} should only be run from a prebuilt Kubernetes release." >&2
    echo "Did you mean to use get-kube.sh instead?" >&2
    exit 1
  }

  setvar KUBE_VERSION = $(cat "${KUBE_ROOT}/version")
}

proc detect_client_info {
  local kernel=$(uname -s)
  case{
    Darwin {
      setvar CLIENT_PLATFORM = ""darwin""
      }
    Linux {
      setvar CLIENT_PLATFORM = ""linux""
      }
    * {
      echo "Unknown, unsupported platform: ${kernel}." >&2
      echo "Supported platforms: Linux, Darwin." >&2
      echo "Bailing out." >&2
      exit 2
  }
}

  # TODO: migrate the kube::util::host_platform function out of hack/lib and
  # use it here.
  local machine=$(uname -m)
  case{
    x86_64*|i?86_64*|amd64* {
      setvar CLIENT_ARCH = ""amd64""
      }
    aarch64*|arm64* {
      setvar CLIENT_ARCH = ""arm64""
      }
    arm* {
      setvar CLIENT_ARCH = ""arm""
      }
    i?86* {
      setvar CLIENT_ARCH = ""386""
      }
    s390x* {
      setvar CLIENT_ARCH = ""s390x""
      }	  
    * {
      echo "Unknown, unsupported architecture (${machine})." >&2
      echo "Supported architectures x86_64, i686, arm, arm64, s390x." >&2
      echo "Bailing out." >&2
      exit 3
      }
  }
}

proc md5sum_file {
  if which md5 >/dev/null 2>&1 {
    md5 -q $1
  } else {
    md5sum $1 | awk '{ print $1 }'
  }
}

proc sha1sum_file {
  if which sha1sum >/dev/null 2>&1 {
    sha1sum $1 | awk '{ print $1 }'
  } else {
    shasum -a1 $1 | awk '{ print $1 }'
  }
}

proc download_tarball {
  local -r download_path="$1"
  local -r file="$2"
  setvar url = ""${DOWNLOAD_URL_PREFIX}/${file}""
  mkdir -p ${download_path}
  if [[ $(which curl) ]] {
    curl -fL --retry 3 --keepalive-time 2 ${url} -o "${download_path}/${file}"
  } elif [[ $(which wget) ]] {
    wget ${url} -O "${download_path}/${file}"
  } else {
    echo "Couldn't find curl or wget.  Bailing out." >&2
    exit 4
  }
  echo
  local md5sum=$(md5sum_file "${download_path}/${file}")
  echo "md5sum(${file})=${md5sum}"
  local sha1sum=$(sha1sum_file "${download_path}/${file}")
  echo "sha1sum(${file})=${sha1sum}"
  echo
  # TODO: add actual verification
}

proc extract_arch_tarball {
  local -r tarfile="$1"
  local -r platform="$2"
  local -r arch="$3"

  setvar platforms_dir = ""${KUBE_ROOT}/platforms/${platform}/${arch}""
  echo "Extracting ${tarfile} into ${platforms_dir}"
  mkdir -p ${platforms_dir}
  # Tarball looks like kubernetes/{client,server}/bin/BINARY"
  tar -xzf ${tarfile} --strip-components 3 -C ${platforms_dir}
  # Create convenience symlink
  ln -sf ${platforms_dir} "$(dirname ${tarfile})/bin"
  echo "Add '$(dirname ${tarfile})/bin' to your PATH to use newly-installed binaries."
}

detect_kube_release
setvar DOWNLOAD_URL_PREFIX = ""${KUBERNETES_RELEASE_URL}/${KUBE_VERSION}""

setvar SERVER_PLATFORM = ""linux""
setvar SERVER_ARCH = "${KUBERNETES_SERVER_ARCH:-amd64}"
setvar SERVER_TAR = ""kubernetes-server-${SERVER_PLATFORM}-${SERVER_ARCH}.tar.gz""

detect_client_info
setvar CLIENT_TAR = ""kubernetes-client-${CLIENT_PLATFORM}-${CLIENT_ARCH}.tar.gz""

echo "Kubernetes release: ${KUBE_VERSION}"
echo "Server: ${SERVER_PLATFORM}/${SERVER_ARCH}  (to override, set KUBERNETES_SERVER_ARCH)"
echo "Client: ${CLIENT_PLATFORM}/${CLIENT_ARCH}  (autodetected)"
echo

# TODO: remove this check and default to true when we stop shipping server
# tarballs in kubernetes.tar.gz
setvar DOWNLOAD_SERVER_TAR = 'false'
if [[ ! -e "${KUBE_ROOT}/server/${SERVER_TAR}" ]] {
  setvar DOWNLOAD_SERVER_TAR = 'true'
  echo "Will download ${SERVER_TAR} from ${DOWNLOAD_URL_PREFIX}"
}

# TODO: remove this check and default to true when we stop shipping kubectl
# in kubernetes.tar.gz
setvar DOWNLOAD_CLIENT_TAR = 'false'
if [[ ! -x "${KUBE_ROOT}/platforms/${CLIENT_PLATFORM}/${CLIENT_ARCH}/kubectl" ]] {
  setvar DOWNLOAD_CLIENT_TAR = 'true'
  echo "Will download and extract ${CLIENT_TAR} from ${DOWNLOAD_URL_PREFIX}"
}

setvar TESTS_TAR = ""kubernetes-test.tar.gz""
setvar DOWNLOAD_TESTS_TAR = 'false'
if [[ -n "${KUBERNETES_DOWNLOAD_TESTS-}" ]] {
  setvar DOWNLOAD_TESTS_TAR = 'true'
  echo "Will download and extract ${TESTS_TAR} from ${DOWNLOAD_URL_PREFIX}"
}

if [[ "${DOWNLOAD_CLIENT_TAR}" == false && \
      "${DOWNLOAD_SERVER_TAR}" == false && \
      "${DOWNLOAD_TESTS_TAR}" == false ]] {
  echo "Nothing additional to download."
  exit 0
}

if [[ -z "${KUBERNETES_SKIP_CONFIRM-}" ]] {
  echo "Is this ok? [Y]/n"
  read confirm
  if [[ "${confirm}" =~ ^[nN]$ ]] {
    echo "Aborting."
    exit 1
  }
}

if ${DOWNLOAD_SERVER_TAR} {
  download_tarball "${KUBE_ROOT}/server" ${SERVER_TAR}
}

if ${DOWNLOAD_CLIENT_TAR} {
  download_tarball "${KUBE_ROOT}/client" ${CLIENT_TAR}
  extract_arch_tarball "${KUBE_ROOT}/client/${CLIENT_TAR}" ${CLIENT_PLATFORM} ${CLIENT_ARCH}
}

if ${DOWNLOAD_TESTS_TAR} {
  download_tarball "${KUBE_ROOT}/test" ${TESTS_TAR}
  echo "Extracting ${TESTS_TAR} into ${KUBE_ROOT}"
  # Strip leading "kubernetes/"
  tar -xzf "${KUBE_ROOT}/test/${TESTS_TAR}" --strip-components 1 -C ${KUBE_ROOT}
}