#!/bin/bash

# Copyright 2015 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 will download latest version of kubectl command line tool and will
# bring up a local Kubernetes cluster with a single node.
set -o errexit
set -o nounset
set -o pipefail

setvar KUBE_HOST = ${KUBE_HOST:-localhost}
setvar KUBELET_KUBECONFIG = ${KUBELET_KUBECONFIG:-"/var/run/kubernetes/kubelet.kubeconfig"}

declare -r RED="\033[0;31m"
declare -r GREEN="\033[0;32m"
declare -r YELLOW="\033[0;33m"

proc echo_green {
  echo -e "${GREEN}$1"; tput sgr0
}

proc echo_red {
  echo -e "${RED}$1"; tput sgr0
}

proc echo_yellow {
  echo -e "${YELLOW}$1"; tput sgr0
}

proc run {
  # For a moment we need to change bash options to capture message if a command fails.
  set +o errexit
  setvar output = $($1 2>&1)
  setvar exit_code = ""$?
  set -o errexit
  if test $exit_code -eq 0 {
    echo_green "SUCCESS"
  } else {
    echo_red "FAILED"
    echo $output >&2
    exit 1
  }
}

# Creates a kubeconfig file for the kubelet.
# Args: destination file path
proc create-kubelet-kubeconfig {
  local destination="${2}"
  if [[ -z "${destination}" ]] {
    echo "Must provide destination path to create Kubelet kubeconfig file!"
    exit 1
  }
  echo "Creating Kubelet kubeconfig file"
  local dest_dir="$(dirname "${destination}")"
  mkdir -p ${dest_dir} &>/dev/null || sudo mkdir -p ${dest_dir}
  setvar sudo = $(test -w "${dest_dir}" || echo "sudo -E")
  cat <<< """ | ${sudo} tee "${destination}" > /dev/null
apiVersion: v1
kind: Config
clusters:
  - cluster:
      server: http://localhost:8080
    name: local
contexts:
  - context:
      cluster: local
    name: local
current-context: local
"""
${sudo} tee ${destination} > /dev/null
apiVersion: v1
kind: Config
clusters:
  - cluster:
      server: http://localhost:8080
    name: local
contexts:
  - context:
      cluster: local
    name: local
current-context: local
EOF
}


proc create_cluster {
  echo "Creating a local cluster:"
  echo -e -n "\tStarting kubelet..."
  create-kubelet-kubeconfig ${KUBELET_KUBECONFIG}
  run "docker run \
  --volume=/:/rootfs:ro \
  --volume=/sys:/sys:ro \
  --volume=/var/lib/docker/:/var/lib/docker:rw \
  --volume=/var/lib/kubelet/:/var/lib/kubelet:rw \
  --volume=/var/run:/var/run:rw \
  --volume=/run/xtables.lock:/run/xtables.lock:rw \
  --net=host \
  --pid=host \
  --privileged=true \
  -d \
  gcr.io/google_containers/hyperkube-${arch}:${release} \
    /hyperkube kubelet \
      --containerized \
      --hostname-override="127.0.0.1" \
      --address="0.0.0.0" \
      --kubeconfig=${KUBELET_KUBECONFIG}/kubelet.kubeconfig \
      --pod-manifest-path=/etc/kubernetes/manifests \
      --allow-privileged=true \
      --cluster-dns=10.0.0.10 \
      --cluster-domain=cluster.local \
      --v=2"

  echo -e -n "\tWaiting for master components to start..."
  while true {
    local running_count=$(kubectl -s=http://${KUBE_HOST}:8080 get pods --no-headers --namespace=kube-system 2>/dev/null | grep "Running" | wc -l)
    # We expect to have 3 running pods - etcd, master and kube-proxy.
    if test $running_count -ge 3 {
      break
    }
    echo -n "."
    sleep 1
  }
  echo_green "SUCCESS"
  echo_green "Cluster created!"
  echo ""
  kubectl -s http://${KUBE_HOST}:8080 clusterinfo
}

proc get_latest_version_number {
  local -r latest_url="https://storage.googleapis.com/kubernetes-release/release/stable.txt"
  if [[ $(which wget) ]] {
    wget -qO- ${latest_url}
  } elif [[ $(which curl) ]] {
    curl -Ss ${latest_url}
  } else {
    echo_red "Couldn't find curl or wget.  Bailing out."
    exit 4
  }
}

setvar latest_release = $(get_latest_version_number)
setvar release = ${KUBE_VERSION:-${latest_release}}

setvar uname = $(uname)
if [[ "${uname}" == "Darwin" ]] {
  setvar platform = ""darwin""
} elif [[ "${uname}" == "Linux" ]] {
  setvar platform = ""linux""
} else {
  echo_red "Unknown, unsupported platform: (${uname})."
  echo_red "Supported platforms: Linux, Darwin."
  echo_red "Bailing out."
  exit 2
}

setvar machine = $(uname -m)
if [[ "${machine}" == "x86_64" ]] {
  setvar arch = ""amd64""
} elif [[ "${machine}" == "i686" ]] {
  setvar arch = ""386""
} elif [[ "${machine}" == "arm*" ]] {
  setvar arch = ""arm""
} elif [[ "${machine}" == "s390x*" ]] {
  setvar arch = ""s390x""
} else {
  echo_red "Unknown, unsupported architecture (${machine})."
  echo_red "Supported architectures x86_64, i686, arm, s390x."
  echo_red "Bailing out."
  exit 3
}

setvar kubectl_url = ""https://storage.googleapis.com/kubernetes-release/release/${release}/bin/${platform}/${arch}/kubectl""

if [[ $(ls . | grep ^kubectl$ | wc -l) -lt 1 ]] {
  echo -n "Downloading kubectl binary..."
  if [[ $(which wget) ]] {
    run "wget ${kubectl_url}"
  } elif [[ $(which curl) ]] {
    run "curl -OL ${kubectl_url}"
  } else {
    echo_red "Couldn't find curl or wget.  Bailing out."
    exit 1
  }
  chmod a+x kubectl
  echo ""
} else {
  # TODO: We should detect version of kubectl binary if it too old
  # download newer version.
  echo "Detected existing kubectl binary. Skipping download."
}

create_cluster

echo ""
echo ""
echo "To list the nodes in your cluster run"
echo_yellow "\tkubectl -s=http://${KUBE_HOST}:8080 get nodes"
echo ""
echo "To run your first pod run"
echo_yellow "\tkubectl -s http://${KUBE_HOST}:8080 run nginx --image=nginx --port=80"