#!/bin/bash
set -e

# Ref: https://stackoverflow.com/questions/5947742/how-to-change-the-output-color-of-echo-in-linux
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BWHITE='\033[1;37m'
NC='\033[0m' # No Color
# --- helper functions for logs ---
info()
{
    echo -e "${GREEN}[INFO] " "$@" "$NC"
}
warn()
{
    echo -e "${YELLOW}[INFO] " "$@" "$NC"
}
fatal()
{
    echo -e "${RED}[INFO] " "$@" "$NC"
    exit 1
}

usr=$(whoami)

# Installing k3s
function install_k8s() {
    curl -sL https://get.k3s.io | sudo K3S_KUBECONFIG_MODE="644" INSTALL_K3S_VERSION='v1.25.6+k3s1' INSTALL_K3S_EXEC="--disable=traefik" sh -
    [[ -d ~/.kube ]] || mkdir ~/.kube
    sudo cp /etc/rancher/k3s/k3s.yaml ~/.kube/config
    sudo chmod 0644 ~/.kube/config
    sudo chown -R $usr ~/.kube/config
    sleep 10
}

# Checking whether the app exists or we do have to upgade.
function exists() {
  install_status=Upgrading
  [[ $UPGRADE_TOOLS -eq 1 ]] && {
    install_status=Upgrading
    return 100
  }
  which $1 &> /dev/null
  return $?
}

# Instal the toolings needed for installation/maintaining k8s
function install_tools() {
    ## installing kubectl
    exists kubectl || {
        info "$install_status kubectl"
        sudo curl -SsL https://dl.k8s.io/release/v1.20.0/bin/linux/amd64/kubectl -o /usr/local/bin/kubectl ; sudo chmod +x /usr/local/bin/kubectl
    }

    ## $install_status GH package manager
    exists eget || {
        info "$install_status eget"
        download_url=$(curl https://api.github.com/repos/zyedidia/eget/releases/latest -s | grep linux_amd64 | grep browser_download_url | cut -d '"' -f4)
        curl -SsL ${download_url} -o /tmp/eget.tar.gz
        tar -xf /tmp/eget.tar.gz --strip-components=1 -C /tmp/
        sudo mv /tmp/eget /usr/local/bin/eget
        sudo chmod +x /usr/local/bin/eget
    }

    ## installing stern, log viewer for K8s
    exists stern || {
        info "$install_status Stern"
        sudo /usr/local/bin/eget -q --to /usr/local/bin stern/stern
    }

    ## installing k9s, TUI K8s
    exists k9s || {
        info "$install_status K9s"
        sudo /usr/local/bin/eget -q --to /usr/local/bin derailed/k9s
    }

    ## installing helm, package manager for K8s
    exists helm || {
        info "$install_status Helm"
        sudo /usr/local/bin/eget -q --to /usr/local/bin https://get.helm.sh/helm-v3.10.2-linux-amd64.tar.gz -f helm
    }
}

# ## Installing openssl
# sudo apt update &> /dev/null
# sudo apt install openssl -y &> /dev/null

randomPass() {
    ## Installing openssl
    exists openssl || {
      sudo apt update &> /dev/null
      sudo apt install openssl -y &> /dev/null
    }
    openssl rand -hex 10
}

## Prepping the infra

# Mac os doesn't have gnu sed, which will cause compatibility issues.
# This wrapper will help to check the sed, and use the correct version="v1.10.0"
# Ref: https://stackoverflow.com/questions/37639496/how-can-i-check-the-version="v1.10.0"
function is_gnu_sed(){
  sed --version >/dev/null 2>&1
}

function sed_i_wrapper(){
  if is_gnu_sed; then
    $(which sed) "$@"
  else
    a=()
    for b in "$@"; do
      [[ $b == '-i' ]] && a=("${a[@]}" "$b" "") || a=("${a[@]}" "$b")
    done
    $(which sed) "${a[@]}"
  fi
}

function create_passwords() {
  # Error out only if the domain name is empty in vars.yaml
  existing_domain_name=$(awk '/domainName/ {print $2}' vars.yaml | xargs)
  [[ -z $existing_domain_name ]] && {
    [[ -z $DOMAIN_NAME ]] && {
      fatal 'DOMAIN_NAME variable is empty. Rerun the script `DOMAIN_NAME=openreplay.mycomp.org bash init.sh `'
    }
  }

  info "Creating dynamic passwords"
  sed_i_wrapper -i "s/postgresqlPassword: \"changeMePassword\"/postgresqlPassword: \"$(randomPass)\"/g" vars.yaml
  sed_i_wrapper -i "s/accessKey: \"changeMeMinioAccessKey\"/accessKey: \"$(randomPass)\"/g" vars.yaml
  sed_i_wrapper -i "s/secretKey: \"changeMeMinioPassword\"/secretKey: \"$(randomPass)\"/g" vars.yaml
  sed_i_wrapper -i "s/jwt_secret: \"SetARandomStringHere\"/jwt_secret: \"$(randomPass)\"/g" vars.yaml
  sed_i_wrapper -i "s/assistKey: \"SetARandomStringHere\"/assistKey: \"$(randomPass)\"/g" vars.yaml
  sed_i_wrapper -i "s/assistJWTSecret: \"SetARandomStringHere\"/assistJWTSecret: \"$(randomPass)\"/g" vars.yaml
  sed_i_wrapper -i "s/domainName: \"\"/domainName: \"${DOMAIN_NAME}\"/g" vars.yaml
}


function set_permissions() {
  info "Setting proper permission for shared folder"
  sudo mkdir -p /openreplay/storage/nfs
  sudo chown -R 1001:1001 /openreplay/storage/nfs
}

## Installing OpenReplay
function install_openreplay() {
  info "installing toolings"
  helm upgrade --install toolings ./toolings -n app --create-namespace --wait -f ./vars.yaml --atomic --debug
  info "installing databases"
  helm upgrade --install databases ./databases -n db --create-namespace --wait -f ./vars.yaml --atomic --debug
  info "installing application"
  helm upgrade --install openreplay ./openreplay -n app --create-namespace --wait -f ./vars.yaml --atomic --debug
}

function main() {
  [[ x$SKIP_K8S_INSTALL == "x1" ]] && {
      info "Skipping Kuberntes installation"
  } || {
    install_k8s
  }
  [[ x$SKIP_K8S_TOOLS == "x1" ]] && {
      info "Skipping Kuberntes tools installation"
  } || {
    install_tools
  }
  [[ x$SKIP_ROTATE_SECRETS == "x1" ]] && {
      info "Skipping random password generation"
  } || {
    create_passwords
  }
  [[ x$SKIP_OR_INSTALL == "x1" ]] && {
      info "Skipping OpenReplay installation"
  } || {
    set_permissions
    install_openreplay
    sudo mkdir -p /var/lib/openreplay
    sudo cp -f openreplay-cli /bin/openreplay
    [[ ! -d /var/lib/openreplay/openreplay ]] || {
      cd /var/lib/openreplay/openreplay
      date +%m-%d-%Y-%H%M%S | sudo tee -a /var/lib/openreplay/or_versions.txt
      sudo git log -1 2>&1 | sudo tee -a /var/lib/openreplay/or_versions.txt
      sudo rm -rf /var/lib/openreplay/openreplay
      cd -
    }
    sudo cp -rf $(cd ../.. && pwd) /var/lib/openreplay/openreplay
    sudo cp -rf ./vars.yaml /var/lib/openreplay/
  }
}

main

info "Configuration file is saved in /var/lib/openreplay/vars.yaml"
info "You can delete the directory $(echo $(cd ../.. && pwd)). Backup stored in /var/lib/openreplay"
info "Run ${BWHITE}openreplay -h${GREEN} to see the cli information to manage OpenReplay."