#!/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

function get_date() {
  local current_date
  current_date=$(date +"%y%m%d-%H%M")
  echo "${current_date}"
}

function check_for_backup_dir() {
  [[ -d ${BACKUP_DIR} ]] && return

  status_msg "Create KIAUH backup directory ..."
  mkdir -p "${BACKUP_DIR}" && ok_msg "Directory created!"
}

function backup_before_update() {
  read_kiauh_ini "${FUNCNAME[0]}"
  local state="${backup_before_update}"
  [[ ${state} = "false" ]] && return
  backup_"${1}"
}

function backup_klipper_config_dir() {
  check_for_backup_dir
  local current_date config_folder_name

  if [[ -d "${KLIPPER_CONFIG}" ]]; then
    current_date=$(get_date)
    config_folder_name="$(echo "${KLIPPER_CONFIG}" | rev | cut -d"/" -f1 | rev)"

    status_msg "Timestamp: ${current_date}"
    status_msg "Create backup of the Klipper config directory ..."

    mkdir -p "${BACKUP_DIR}/${config_folder_name}/${current_date}"
    cp -r "${KLIPPER_CONFIG}" "${_}"

    print_confirm "Configuration directory backup complete!"
  else
    ok_msg "No config directory found! Skipping backup ..."
  fi
}

function backup_moonraker_database() {
  check_for_backup_dir
  local current_date databases target_dir regex=".moonraker_database(_[0-9a-zA-Z]+)?"
  databases=$(find "${HOME}" -maxdepth 1 -type d -regextype posix-extended -regex "${HOME}/${regex}" | sort)

  if [[ -n ${databases} ]]; then
    current_date=$(get_date)
    target_dir="${BACKUP_DIR}/moonraker_database_backup/${current_date}"

    status_msg "Timestamp: ${current_date}"
    mkdir -p "${target_dir}"

    for database in ${databases}; do
      status_msg "Create backup of ${database} ..."
      cp -r "${database}" "${target_dir}"
      ok_msg "Done!"
    done

    print_confirm "Moonraker database backup complete!"
  else
    print_error "No Moonraker database found! Skipping backup ..."
  fi
  return
}

function backup_klipper() {
  local current_date

  if [[ -d ${KLIPPER_DIR} && -d ${KLIPPY_ENV} ]]; then
    status_msg "Creating Klipper backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/klipper-backups/${current_date}"
    cp -r "${KLIPPER_DIR}" "${_}" && cp -r "${KLIPPY_ENV}" "${_}"
    print_confirm "Klipper backup complete!"
  else
    print_error "Can't back up 'klipper' and/or 'klipper-env' directory! Not found!"
  fi
}

function backup_mainsail() {
  local current_date

  if [[ -d ${MAINSAIL_DIR} ]]; then
    status_msg "Creating Mainsail backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/mainsail-backups/${current_date}"
    cp -r "${MAINSAIL_DIR}" "${_}"
    print_confirm "Mainsail backup complete!"
  else
    print_error "Can't back up 'mainsail' directory! Not found!"
  fi
}

function backup_fluidd() {
  local current_date

  if [[ -d ${FLUIDD_DIR} ]]; then
    status_msg "Creating Fluidd backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/fluidd-backups/${current_date}"
    cp -r "${FLUIDD_DIR}" "${_}"
    print_confirm "Fluidd backup complete!"
  else
    print_error "Can't back up 'fluidd' directory! Not found!"
  fi
}

function backup_moonraker() {
  local current_date

  if [[ -d ${MOONRAKER_DIR} && -d ${MOONRAKER_ENV} ]]; then
    status_msg "Creating Moonraker backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/moonraker-backups/${current_date}"
    cp -r "${MOONRAKER_DIR}" "${_}" && cp -r "${MOONRAKER_ENV}" "${_}"
    print_confirm "Moonraker backup complete!"
  else
    print_error "Can't back up moonraker and/or moonraker-env directory! Not found!"
  fi
}

function backup_octoprint() {
  local current_date

  if [[ -d ${OCTOPRINT_DIR} && -d ${OCTOPRINT_CFG_DIR} ]]; then
    status_msg "Creating OctoPrint backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/octoprint-backups/${current_date}"
    cp -r "${OCTOPRINT_DIR}" "${_}" && cp -r "${OCTOPRINT_CFG_DIR}" "${_}"
    print_confirm " OctoPrint backup complete!"
  else
    print_error "Can't back up OctoPrint and/or .octoprint directory!\n Not found!"
  fi
}

function backup_klipperscreen() {
  local current_date
  if [[ -d ${KLIPPERSCREEN_DIR} ]] ; then
    status_msg "Creating KlipperScreen backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/klipperscreen-backups/${current_date}"
    cp -r "${KLIPPERSCREEN_DIR}" "${_}"
    print_confirm "KlipperScreen backup complete!"
  else
    print_error "Can't back up KlipperScreen directory!\n Not found!"
  fi
}

function backup_telegram_bot() {
  local current_date

  if [[ -d ${TELEGRAM_BOT_DIR} ]] ; then
    status_msg "Creating MoonrakerTelegramBot backup ..."
    check_for_backup_dir
    current_date=$(get_date)
    status_msg "Timestamp: ${current_date}"
    mkdir -p "${BACKUP_DIR}/MoonrakerTelegramBot-backups/${current_date}"
    cp -r "${TELEGRAM_BOT_DIR}" "${_}"
    print_confirm "MoonrakerTelegramBot backup complete!"
  else
    print_error "Can't back up MoonrakerTelegramBot directory!\n Not found!"
  fi
}