#!/usr/bin/env bash

#=======================================================================#
# Copyright (C) 2020 - 2022 Dominik Willner <th33xitus@gmail.com>       #
#                                                                       #
# This file is part of KIAUH - Klipper Installation And Update Helper   #
# https://github.com/th33xitus/kiauh                                    #
#                                                                       #
# This file may be distributed under the terms of the GNU GPLv3 license #
#=======================================================================#

set -e
clear

### set color variables
green=$(echo -en "\e[92m")
yellow=$(echo -en "\e[93m")
red=$(echo -en "\e[91m")
cyan=$(echo -en "\e[96m")
default=$(echo -en "\e[39m")
white=$(echo -en "\e[39m")

### sourcing all additional scripts
SRCDIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )"/.. && pwd )"
for script in "${SRCDIR}/kiauh/scripts/"*.sh; do . "${script}"; done
for script in "${SRCDIR}/kiauh/scripts/ui/"*.sh; do . "${script}"; done

#nginx
NGINX_SA=/etc/nginx/sites-available
NGINX_SE=/etc/nginx/sites-enabled
NGINX_CONFD=/etc/nginx/conf.d
#misc
INI_FILE=${HOME}/.kiauh.ini

### set github repos
DMBUTYUGIN_REPO=https://github.com/dmbutyugin/klipper.git
#branches
BRANCH_SCURVE_SMOOTHING=dmbutyugin/scurve-smoothing
BRANCH_SCURVE_SHAPING=dmbutyugin/scurve-shaping

#===================================================#
#=================== UPDATE KIAUH ==================#
#===================================================#

function update_kiauh(){
  status_msg "Updating KIAUH ..."
  cd "${SRCDIR}/kiauh"
  git reset --hard && git pull
  ok_msg "Update complete! Please restart KIAUH."
  exit 0
}

#===================================================#
#=================== KIAUH STATUS ==================#
#===================================================#

function kiauh_update_avail(){
  [ ! -d "${SRCDIR}/kiauh/.git" ] && return
  local origin head
  cd "${SRCDIR}/kiauh"
  ### abort if not on master branch
  ! git branch -a | grep -q "\* master" && return
  ### compare commit hash
  git fetch -q
  origin=$(git rev-parse --short=8 origin/master)
  head=$(git rev-parse --short=8 HEAD)
  if [ "${origin}" != "${head}" ]; then
    echo "true"
  fi
}

### format some default message types
select_msg() {
  echo -e "${white}>>>>>> $1"
}
warn_msg(){
  echo -e "${red}>>>>>> $1${white}"
}
status_msg(){
  echo; echo -e "${yellow}###### $1${white}"
}
ok_msg(){
  echo -e "${green}>>>>>> $1${white}"
}
error_msg(){
  echo -e "${red}>>>>>> $1${white}"
}
abort_msg(){
  echo -e "${red}<<<<<< $1${white}"
}
title_msg(){
  echo -e "${cyan}$1${white}"
}
get_date(){
  current_date=$(date +"%y%m%d-%H%M")
  export current_date
}
print_unkown_cmd(){
  ERROR_MSG="Invalid command!"
}
invalid_option(){
  ERROR_MSG="Invalid command!"
}

print_msg(){
  if [ -n "${ERROR_MSG}" ]; then
    echo -e "${red}"
    echo -e "#########################################################"
    echo -e " ${ERROR_MSG} "
    echo -e "#########################################################"
    echo -e "${white}"
  fi
  if [ -n "${CONFIRM_MSG}" ]; then
    echo -e "${green}"
    echo -e "#########################################################"
    echo -e " ${CONFIRM_MSG} "
    echo -e "#########################################################"
    echo -e "${white}"
  fi
}

print_error(){
  [ -z "${1}" ] && return
  echo -e "${red}"
  echo -e "#########################################################"
  echo -e " ${1} "
  echo -e "#########################################################"
  echo -e "${white}"
}

print_confirm(){
  [ -z "${1}" ] && return
  echo -e "${green}"
  echo -e "#########################################################"
  echo -e " ${1} "
  echo -e "#########################################################"
  echo -e "${white}"
}

clear_msg(){
  unset CONFIRM_MSG
  unset ERROR_MSG
}

check_euid
init_ini
kiauh_update_avail
main_menu