#TODO:
# - ask for permission to disable octoprint service

dwc2_install_routine(){
  if [ -d $KLIPPER_DIR ]; then
    # check for existing installation
      if [ -d $DWC2FK_DIR ] && [ -d $DWC2_DIR ]; then
        ERROR_MSG=" Looks like DWC2 is already installed!\n Skipping..."
        return
      fi
    stop_klipper
    #disable octoprint service if installed
      if systemctl is-enabled octoprint.service -q 2>/dev/null; then
        disable_octoprint_service
      fi
    install_tornado
    install_dwc2fk && dwc2fk_cfg
    install_dwc2
    dwc2_reverse_proxy_dialog
    create_custom_hostname
    start_klipper
  else
    ERROR_MSG=" Please install Klipper first!\n Skipping..."
  fi
}

install_tornado(){
  if [ "$(cd $KLIPPY_ENV_DIR/bin/ && $_/pip list 2>/dev/null | grep "tornado" | cut -d" " -f9)" = "5.1.1" ]; then
    ok_msg "Tornado 5.1.1 is already installed! Continue..."
  else
    status_msg "Installing Tornado 5.1.1 ..."
    cd ${HOME}
    PYTHONDIR="${HOME}/klippy-env"
    virtualenv ${PYTHONDIR}
    ${PYTHONDIR}/bin/pip install tornado==5.1.1
    ok_msg "Tornado 5.1.1 successfully installed!"
  fi
}

install_dwc2fk(){
  cd ${HOME}
  status_msg "Cloning DWC2-for-Klipper repository ..."
  git clone $DWC2FK_REPO && ok_msg "DWC2-for-Klipper successfully cloned!"
  #create a web_dwc2.py symlink if not already existing
  if [ -d $KLIPPER_DIR/klippy/extras ] && [ ! -e $KLIPPER_DIR/klippy/extras/web_dwc2.py ]; then
    status_msg "Creating web_dwc2.py Symlink ..."
    ln -s $DWC2FK_DIR/web_dwc2.py $KLIPPER_DIR/klippy/extras/web_dwc2.py && ok_msg "Symlink created!"
  fi
}

dwc2fk_cfg(){
  while true; do
    echo -e "${cyan}"
    read -p "###### Do you want to create the config now? (Y/n): " yn
    echo -e "${default}"
    case "$yn" in
    Y|y|Yes|yes|"") create_dwc2fk_cfg; break;;
    N|n|No|no) break;;
    esac
  done
}

create_dwc2fk_cfg(){
  echo -e "/=================================================\ "
  echo -e "|  1) [Default configuration]                     | "
  echo -e "|  2) [Custom configuration]                      | "
  echo -e "|  3) [Skip]                                      | "
  echo -e "\=================================================/ "
  while true; do
    read -p "Please select: " choice; echo
    case "$choice" in
    1) dwc2fk_default_cfg && ok_msg "Config written ..."; break;;
    2) create_dwc2fk_custom_cfg && ok_msg "Config written ..."; break;;
    3) echo "Skipping ..."; break;;
    esac
  done
}

dwc2fk_default_cfg(){
  cat <<DWC2 >> $PRINTER_CFG

##########################
### CREATED WITH KIAUH ###
##########################
[virtual_sdcard]
path: ~/sdcard

[web_dwc2]
printer_name: my_printer
listen_adress: 0.0.0.0
listen_port: 4750
web_path: dwc2/web
##########################
##########################
DWC2
}

create_dwc2fk_custom_cfg(){
  echo -e "${cyan}"
  read -e -p "Printer name: " -i "my_printer" PRINTER_NAME
  read -e -p "Listen adress: " -i "0.0.0.0" LISTEN_ADRESS
  read -e -p "Listen port: " -i "4750" LISTEN_PORT
  read -e -p "Web path: " -i "dwc2/web" WEB_PATH
  echo -e "${default}"
  DWC2_CFG=$(cat <<DWC2

##########################
### CREATED WITH KIAUH ###
##########################
[virtual_sdcard]
path: ~/sdcard

[web_dwc2]
printer_name: $PRINTER_NAME
listen_adress: $LISTEN_ADRESS
listen_port: $LISTEN_PORT
web_path: $WEB_PATH
##########################
##########################
DWC2
)
  echo "The following lines will be written:"
  echo -e "$DWC2_CFG"
  while true; do
    echo -e "${cyan}"
    read -p "###### Write now (Y) or start over (n)? (Y/n): " yn
    echo -e "${default}"
    case "$yn" in
      Y|y|Yes|yes|"") echo -e "$DWC2_CFG" >> $PRINTER_CFG; break;;
      N|n|No|no) create_dwc2fk_custom_cfg;;
    esac
  done
}

install_dwc2(){
  #check dependencies
  dep=(wget gzip tar curl)
  dependency_check
  #execute operation
  GET_DWC2_URL=`curl -s https://api.github.com/repositories/28820678/releases/latest | grep browser_download_url | cut -d'"' -f4`
  if [ ! -d $DWC2_DIR/web ]; then
    mkdir -p $DWC2_DIR/web
  fi
  cd $DWC2_DIR/web
  status_msg "Downloading DWC2 Web UI ..."
  wget -q $GET_DWC2_URL && ok_msg "Download complete!"
  status_msg "Unzipping archive ..."
  unzip -q -o *.zip && for f_ in $(find . | grep '.gz');do gunzip -f ${f_};done && ok_msg "Done!"
  status_msg "Writing version to file ..."
  echo $GET_DWC2_URL | cut -d/ -f8 > $DWC2_DIR/web/version && ok_msg "Done!"
  status_msg "Do a little cleanup ..."
  rm -rf DuetWebControl-SD.zip && ok_msg "Done!"
  ok_msg "DWC2 Web UI installed!"
}

dwc2_reverse_proxy_dialog(){
  top_border
  echo -e "|  If you want to have a nicer URL or simply need/want  | "
  echo -e "|  DWC2 to run on port 80 (http's default port) you     | "
  echo -e "|  can set up a reverse proxy to run DWC2 on port 80.   | "
  bottom_border
  while true; do
    echo -e "${cyan}"
    read -p "###### Do you want to set up a reverse proxy now? (Y/n): " yn
    echo -e "${default}"
    case "$yn" in
      Y|y|Yes|yes|"") create_reverse_proxy "dwc2"; break;;
      N|n|No|no) break;;
    esac
  done
}