#!/bin/bash
clear
set -e

### set some variables
ERROR_MSG=""
green="\e[92m"
yellow="\e[93m"
red="\e[91m"
cyan="\e[96m"
default="\e[39m"

### set some messages
warn_msg(){
  echo -e "${red}<!!!!> $1${default}"
}
status_msg(){
  echo; echo -e "${yellow}###### $1${default}"
}
ok_msg(){
  echo -e "${green}>>>>>> $1${default}"
}
title_msg(){
  echo -e "${cyan}$1${default}"
}
get_date(){
  current_date=$(date +"%Y-%m-%d_%H-%M")
}
print_unkown_cmd(){
  ERROR_MSG=" Sorry i don't know that command!"
}

### set important directories
#klipper
KLIPPER_DIR=${HOME}/klipper
KLIPPY_ENV_DIR=${HOME}/klippy-env
KLIPPER_SERVICE1=/etc/init.d/klipper
KLIPPER_SERVICE2=/etc/default/klipper
#dwc2
DWC2FK_DIR=${HOME}/dwc2-for-klipper
DWC2_DIR=${HOME}/sdcard/dwc2
WEB_DWC2=${HOME}/klipper/klippy/extras/web_dwc2.py
TORNADO_DIR1=${HOME}/klippy-env/lib/python2.7/site-packages/tornado
TORNADO_DIR2=${HOME}/klippy-env/lib/python2.7/site-packages/tornado-5.1.1.dist-info
#mainsail/moonraker
MAINSAIL_DIR=${HOME}/mainsail
MOONRAKER_DIR=${HOME}/moonraker
MOONRAKER_ENV_DIR=${HOME}/moonraker-env
MOONRAKER_SERVICE1=/etc/init.d/moonraker
MOONRAKER_SERVICE2=/etc/default/moonraker
#octoprint
OCTOPRINT_DIR=${HOME}/OctoPrint
OCTOPRINT_CFG_DIR=${HOME}/.octoprint
OCTOPRINT_SERVICE1=/etc/init.d/octoprint
OCTOPRINT_SERVICE2=/etc/default/octoprint
#misc
INI_FILE=${HOME}/kiauh/kiauh.ini
BACKUP_DIR=${HOME}/kiauh-backups
PRINTER_CFG=${HOME}/printer.cfg

### set github repos
KLIPPER_REPO=https://github.com/KevinOConnor/klipper.git
ARKSINE_REPO=https://github.com/Arksine/klipper.git
DMBUTYUGIN_REPO=https://github.com/dmbutyugin/klipper.git
DWC2FK_REPO=https://github.com/Stephan3/dwc2-for-klipper.git
MOONRAKER_REPO=https://github.com/Arksine/moonraker.git
#branches
BRANCH_MOONRAKER=Arksine/dev-moonraker-testing
BRANCH_SCURVE_SMOOTHING=dmbutyugin/scurve-smoothing
BRANCH_SCURVE_SHAPING=dmbutyugin/scurve-shaping

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

clear_msg(){
  CONFIRM_MSG="" && ERROR_MSG=""
}

main_menu(){
  print_header
  #print KIAUH update msg if update available
    if [ "$KIAUH_UPDATE_AVAIL" = "true" ]; then
      kiauh_update_msg
    fi
  #check install status
    klipper_status
    dwc2_status
    mainsail_status
    octoprint_status
    print_branch
  print_msg && clear_msg
  main_ui
  while true; do
    echo -e "${cyan}"
    read -p "Perform action: " action; echo
    echo -e "${default}"
    case "$action" in
      update)
        clear
        print_header
        update_kiauh
        print_msg && clear_msg
        main_ui;;
      0)
        clear
        print_header
        ERROR_MSG="Sorry this function is not implemented yet!"
        print_msg && clear_msg
        main_ui;;
      1)
        clear
        install_menu
        break;;
      2)
        clear
        update_menu
        break;;
      3)
        clear
        remove_menu
        break;;
      4)
        clear
        advanced_menu
        break;;
      5)
        clear
        backup_menu
        break;;
      Q|q)
        echo -e "${green}###### Happy printing! ######${default}"; echo
        exit -1;;
      *)
        clear
        print_header
        print_unkown_cmd
        print_msg && clear_msg
        main_ui;;
    esac
  done
  clear; main_menu
}

install_menu(){
  print_header
  install_ui
  while true; do
    echo -e "${cyan}"
    read -p "Perform action: " action; echo
    echo -e "${default}"
    case "$action" in
      1)
        clear
        print_header
        install_klipper
        print_msg && clear_msg
        install_ui;;
      2)
        clear
        print_header
        dwc2_install_routine
        print_msg && clear_msg
        install_ui;;
      3)
        clear
        print_header
        mainsail_install_routine
        print_msg && clear_msg
        install_ui;;
      4)
        clear
        print_header
        octoprint_install_routine
        print_msg && clear_msg
        install_ui;;
      Q|q)
        clear; main_menu; break;;
      *)
        clear
        print_header
        print_unkown_cmd
        print_msg && clear_msg
        install_ui;;
    esac
  done
  install_menu
}

update_menu(){
  print_header
  #compare versions
    ui_print_versions
  print_msg && clear_msg
  read_bb4u_stat
  update_ui
  while true; do
    echo -e "${cyan}"
    read -p "Perform action: " action; echo
    echo -e "${default}"
    case "$action" in
      0)
        clear
        print_header
        toggle_backups
        print_msg && clear_msg
        update_ui;;
      1)
        clear
        print_header
        update_klipper && ui_print_versions
        print_msg && clear_msg
        update_ui;;
      2)
        clear
        print_header
        update_dwc2fk && ui_print_versions
        print_msg && clear_msg
        update_ui;;
      3)
        clear
        print_header
        update_dwc2 && ui_print_versions
        print_msg && clear_msg
        update_ui;;
      4)
        clear
        print_header
        update_moonraker && ui_print_versions
        print_msg && clear_msg
        update_ui;;
      5)
        clear
        print_header
        update_mainsail && ui_print_versions
        print_msg && clear_msg
        update_ui;;
      Q|q)
        clear; main_menu; break;;
      *)
        clear
        print_header
        print_unkown_cmd
        print_msg && clear_msg
        ui_print_versions
        update_ui;;
    esac
  done
  update_menu
}

remove_menu(){
  print_header
  remove_ui
  while true; do
    echo -e "${cyan}"
    read -p "Perform action: " action; echo
    echo -e "${default}"
    case "$action" in
      1)
        clear
        print_header
        remove_klipper
        print_msg && clear_msg
        remove_ui;;
      2)
        clear
        print_header
        remove_dwc2
        print_msg && clear_msg
        remove_ui;;
      3)
        clear
        print_header
        remove_mainsail
        print_msg && clear_msg
        remove_ui;;
      4)
        clear
        print_header
        remove_octoprint
        print_msg && clear_msg
        remove_ui;;
      5)
        clear
        print_header
        remove_nginx
        print_msg && clear_msg
        remove_ui;;
      Q|q)
        clear; main_menu; break;;
      *)
        clear
        print_header
        print_unkown_cmd
        print_msg && clear_msg
        remove_ui;;
    esac
  done
  remove_menu
}

advanced_menu(){
  print_header
  print_msg && clear_msg
  read_octoprint_service_status
  advanced_ui
  while true; do
    echo -e "${cyan}"
    read -p "Perform action: " action; echo
    echo -e "${default}"
    case "$action" in
      0)
        clear
        print_header
        toggle_octoprint_service
        read_octoprint_service_status
        print_msg && clear_msg
        advanced_ui;;
      1)
        clear
        print_header
        switch_menu
        print_msg && clear_msg
        advanced_ui;;
      2)
        clear
        print_header
        load_klipper_state
        print_msg && clear_msg
        advanced_ui;;
      3)
        clear
        print_header
        build_fw
        print_msg && clear_msg
        advanced_ui;;
      4)
        clear
        print_header
        flash_routine
        print_msg && clear_msg
        advanced_ui;;
      5)
        clear
        print_header
        get_printer_usb
        print_msg && clear_msg
        advanced_ui;;
      6)
        clear
        print_header
        get_printer_usb && write_printer_usb
        print_msg && clear_msg
        advanced_ui;;
      7)
        clear
        print_header
        create_dwc2fk_cfg
        print_msg && clear_msg
        advanced_ui;;
      8)
        clear
        print_header
        create_custom_hostname
        print_msg && clear_msg
        advanced_ui;;
      9)
        clear
        print_header
        remove_branding
        print_msg && clear_msg
        advanced_ui;;
      Q|q)
        clear; main_menu; break;;
      *)
        clear
        print_header
        print_unkown_cmd
        print_msg && clear_msg
        advanced_ui;;
    esac
  done
  advanced_menu
}

switch_menu(){
  if [ -d $KLIPPER_DIR ]; then
    read_branch
    print_msg && clear_msg
    switch_ui
    while true; do
      echo -e "${cyan}"
      read -p "Perform action: " action; echo
      echo -e "${default}"
      case "$action" in
        1)
          clear
          print_header
          switch_to_origin
          read_branch
          print_msg && clear_msg
          switch_ui;;
        2)
          clear
          print_header
          switch_to_scurve_shaping
          read_branch
          print_msg && clear_msg
          switch_ui;;
        3)
          clear
          print_header
          switch_to_scurve_smoothing
          read_branch
          print_msg && clear_msg
          switch_ui;;
        4)
          clear
          print_header
          switch_to_moonraker
          read_branch
          print_msg && clear_msg
          switch_ui;;
        Q|q)
          clear; advanced_menu; break;;
        *)
          clear
          print_header
          print_unkown_cmd
          print_msg && clear_msg
          switch_ui;;
      esac
    done
  else
    ERROR_MSG="No Klipper directory found! Download Klipper first!"
  fi
}

#rollback_menu(){
#  load_klipper_state
#  print_msg && clear_msg
#  advanced_menu
#}

backup_menu(){
  print_header
  print_msg && clear_msg
  backup_ui
  while true; do
    echo -e "${cyan}"
    read -p "Perform action: " action; echo
    echo -e "${default}"
    case "$action" in
      1)
        clear
        print_header
        backup_klipper
        print_msg && clear_msg
        backup_ui;;
      2)
        clear
        print_header
        backup_dwc2
        print_msg && clear_msg
        backup_ui;;
      3)
        clear
        print_header
        backup_mainsail
        print_msg && clear_msg
        backup_ui;;
      4)
        clear
        print_header
        backup_moonraker
        print_msg && clear_msg
        backup_ui;;
      5)
        clear
        print_header
        backup_octoprint
        print_msg && clear_msg
        backup_ui;;
      Q|q)
        clear; main_menu; break;;
      *)
        clear
        print_header
        print_unkown_cmd
        print_msg && clear_msg
        backup_ui;;
    esac
  done
  backup_menu
}

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

check_euid
kiauh_status
main_menu