#!/usr/bin/env bash
# ref: https://ignite.readthedocs.io/en/stable/installation/

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

TMP_DIR=/tmp/kubefire
ARCH=`uname -m`

KUBEFIRE_VERSION=${KUBEFIRE_VERSION:-}
CONTAINERD_VERSION=${CONTAINERD_VERSION:-""}
IGNITE_VERION=${IGNITE_VERION:-""}
CNI_VERSION=${CNI_VERSION:-""}
RUNC_VERSION=${RUNC_VERSION:-""}

if [ -z "$KUBEFIRE_VERSION" ] || [ -z "$CONTAINERD_VERSION" ] || [ -z "$IGNITE_VERION" ] || [ -z "$CNI_VERSION" ] || [ -z "$RUNC_VERSION" ]; then
  echo "incorrect versions provided!" >/dev/stderr
  exit 1
fi

STABLE_KUBEFIRE_VERSION=$(sed -E "s/(v[0-9]+\.[0-9]+\.[0-9]+)[a-zA-Z0-9\-]*/\1/g"< <(echo "$KUBEFIRE_VERSION"))

if [ ${ARCH} = aarch64 ] || [ ${ARCH} = arm64 ]; then
  ARCH_SUFFIX=arm64
elif [ ${ARCH} = x86_64 ]; then
  ARCH_SUFFIX=amd64
else
  echo "${ARCH} is not supported!" >/dev/stderr
  exit 1
fi

rm -rf $TMP_DIR && mkdir -p $TMP_DIR
pushd $TMP_DIR

function cleanup() {
  rm -rf $TMP_DIR || true
}

trap cleanup EXIT ERR INT TERM

function _check_version() {
  set +o pipefail

  local exec_name=$1
  local exec_version_cmd=$2
  local version=$3

  command -v "${exec_name}" && [[ "$(eval "$exec_name $exec_version_cmd 2>&1")" =~ $version ]]
  return $?
}

function check_virtualization() {
  if [ ${ARCH_SUFFIX} = arm64 ]; then
    return
  fi

  lscpu | grep "Virtuali[s|z]ation"
  lsmod | grep kvm
}

function install_containerd() {
  if _check_version /usr/local/bin/containerd --version $CONTAINERD_VERSION; then
    echo "containerd (${CONTAINERD_VERSION}) installed already!"
    return
  fi

  local version="${CONTAINERD_VERSION:1}"
  local dir=containerd-$version


  curl -sfSLO "https://github.com/containerd/containerd/releases/download/${CONTAINERD_VERSION}/containerd-${version}-linux-${ARCH_SUFFIX}.tar.gz"
  mkdir -p $dir
  tar -zxvf $dir*.tar.gz -C $dir
  chmod +x $dir/bin/*
  sudo mv $dir/bin/* /usr/local/bin/

  curl -sfSLO "https://raw.githubusercontent.com/containerd/containerd/${CONTAINERD_VERSION}/containerd.service"
  sudo groupadd containerd || true
  sudo mv containerd.service /etc/systemd/system/containerd.service

  chgrp_path=$(command -v chgrp | tr -d '\n')
  sudo sed -i -E "s#(ExecStart=/usr/local/bin/containerd)#\1\nExecStartPost=${chgrp_path} containerd /run/containerd/containerd.sock#g" /etc/systemd/system/containerd.service

  sudo mkdir -p /etc/containerd
  containerd config default | sudo tee /etc/containerd/config.toml >/dev/null
  sudo systemctl enable --now containerd
}

function install_runc() {
  if _check_version /usr/local/bin/runc -version $RUNC_VERSION; then
    echo "runc (${RUNC_VERSION}) installed already!"
    return
  fi


  curl -sfSL "https://github.com/opencontainers/runc/releases/download/${RUNC_VERSION}/runc.${ARCH_SUFFIX}" -o runc
  chmod +x runc
  sudo mv runc /usr/local/bin/
}

function install_cni() {
  if _check_version /opt/cni/bin/bridge --version $CNI_VERSION; then
    echo "CNI plugins (${CNI_VERSION}) installed already!"
    return
  fi

  mkdir -p /opt/cni/bin
  local f="https://github.com/containernetworking/plugins/releases/download/${CNI_VERSION}/cni-plugins-linux-${ARCH_SUFFIX}-${CNI_VERSION}.tgz"
  curl -sfSL "$f" | tar -C /opt/cni/bin -xz
}

function install_cni_patches() {
    curl -o host-local-rev -sfSL "https://github.com/innobead/kubefire/releases/download/${STABLE_KUBEFIRE_VERSION}/host-local-rev-linux-${ARCH_SUFFIX}"
    chmod +x host-local-rev
    sudo mv host-local-rev /opt/cni/bin/
}

function install_ignite() {
  if _check_version /usr/local/bin/ignite version $IGNITE_VERION; then
    echo "ignite (${IGNITE_VERION}) installed already!"
    return
  fi

  for binary in ignite ignited; do
    echo "Installing $binary..."

    local f="https://github.com/weaveworks/ignite/releases/download/${IGNITE_VERION}/${binary}-${ARCH_SUFFIX}"
    curl -sfSLo $binary "$f"
    chmod +x $binary
    sudo mv $binary /usr/local/bin
  done
}

function check_ignite() {
  ignite version
}

function create_cni_default_config() {
  mkdir -p /etc/cni/net.d/ || true
  sudo cat <<'EOF' > /etc/cni/net.d/00-kubefire.conflist
{
	"cniVersion": "0.4.0",
	"name": "kubefire-cni-bridge",
	"plugins": [
		{
			"type": "bridge",
			"bridge": "kubefire0",
			"isGateway": true,
			"isDefaultGateway": true,
			"promiscMode": true,
			"ipMasq": true,
			"ipam": {
				"type": "host-local-rev",
				"subnet": "10.62.0.0/16"
			}
		},
		{
			"type": "portmap",
			"capabilities": {
				"portMappings": true
			}
		},
		{
			"type": "firewall"
		}
	]
}
EOF
}

check_virtualization

install_runc
install_containerd
install_cni
install_cni_patches
install_ignite
check_ignite
create_cni_default_config

popd