#!/bin/bash

menu(){
  clear -x
  title
  echo "1)  Help"
  echo "2)  List DNS Names"
  echo "3)  Mount and Unmount PVC storage"
  echo "4)  Create a Backup"
  echo "5)  Restore a Backup"
  echo "6)  Delete a Backup"
  echo "7)  Update HeavyScript"
  echo "8)  Update Applications"
  echo
  echo "0)  Exit"
  read -rt 600 -p "Please select an option by number: " selection

  case $selection in
    0)
        exit
        ;;
    1)
        help="true"
        ;;
    2)
        dns="true"
        ;;
    3)
        mount="true"
        ;;
    4)
        read -rt 600 -p "Please type the max number of backups to keep: " number_of_backups
        number_of_backups=$number_of_backups
        ! [[ $number_of_backups =~ ^[0-9]+$  ]] && echo -e "Error: -b needs to be assigned an interger\n\"""$number_of_backups""\" is not an interger" >&2 && exit
        [[ "$number_of_backups" -le 0 ]] && echo "Error: Number of backups is required to be at least 1" && exit
        echo "Generating backup, please be patient for output.."
        backup "$number_of_backups"
      ;;
    5)
        restore="true"
        ;;
    6)
        deleteBackup="true"
        ;;
    7)
        self_update="true"
        ;;
    8)
        script=$(readlink -f "$0")
        script_path=$(dirname "$script")
        script_name="heavy_script.sh"
        cd "$script_path" || exit
        clear -x
        title
        echo "Choose Your Update Type"
        echo "-----------------------"
        echo "1) -U | Update all applications, ignores versions"
        echo "2) -u | Update all applications, does not update Major releases"
        echo
        echo "0) Exit"
        echo
        read -rt 600 -p "Please type the number associated with the flag above: " current_selection
        if [[ $current_selection == 1 ]]; then
            echo -e "\nHow many applications do you want updating at the same time?"
            read -rt 600 -p "Please type an integer greater than 0: " up_async
        elif [[ $current_selection == 2 ]]; then
            echo -e "\nHow many applications do you want updating at the same time?"
            read -rt 600 -p "Please type an integer greater than 0: " up_async
        elif [[ $current_selection == 0 ]]; then
            echo "Exiting.." 
            exit
        else
            echo "$current_selection was not an option, try again"
            exit
        fi
        if [[ $up_async == 0 ]]; then
            echo "0 was not an option.. exiting"
            exit
        elif ! [[ $up_async =~ ^[0-9]+$  ]]; then
            echo "Error: \"$up_async\" is invalid, it needs to be an integer"
            exit
        else
            update_selection+=("-u" "$up_async")
        fi
        while true 
        do
            clear -x
            title
            echo "Choose your update options"
            echo "--------------------------"
            echo "1) -b | Back-up your ix-applications dataset, specify a number after -b"
            echo "2) -i | Add application to ignore list, one by one, see example below."
            echo "3) -r | Roll-back applications if they fail to update"
            echo "4) -S | Shutdown applications prior to updating"
            echo "5) -v | verbose output"
            echo "6) -t | Set a custom timeout in seconds when checking if either an App or Mountpoint correctly Started, Stopped or (un)Mounted. Defaults to 500 seconds"
            echo "7) -s | sync catalog"
            echo "8) -p | Prune unused/old docker images"
            echo
            echo "0) Done making selections, proceed with update"
            echo 
            echo "Current Choices"
            echo "---------------"
            for i in "${update_list[@]}"
            do
                echo "$i"
            done
            echo
            read -rt 600 -p "Please type the number associated with the flag above: " current_selection


            if [[ $current_selection == 0 ]]; then
                exec bash "$script_name" "${update_selection[@]}"
                exit
            elif [[ $current_selection == 1 ]]; then
                echo "Up to how many backups should we keep?"
                read -rt 600 -p "Please type an integer: " up_backups
                ! [[ $up_backups =~ ^[0-9]+$ ]] && echo -e "Error: \"$up_backups\" is invalid, it needs to be an integer\nNOT adding it to the list" && sleep 5 && continue
                [[ $up_backups == 0 ]] && echo -e "Error: Number of backups cannot be 0\nNOT adding it to the list" && sleep 5 && continue
                update_list+=("-b $up_backups")
                update_selection+=("-b" "$up_backups")
            elif [[ $current_selection == 2 ]]; then
                read -rt 600 -p "What is the name of the application we should ignore?: " up_ignore
                update_list+=("-b $up_ignore")
                update_selection+=("-i" "$up_ignore")                
            elif [[ $current_selection == 3 ]]; then
                update_list+=("-r")
                update_selection+=("-r")
            elif [[ $current_selection == 4 ]]; then
                update_list+=("-S")
                update_selection+=("-S")
            elif [[ $current_selection == 5 ]]; then
                update_list+=("-v")
                update_selection+=("-v")
            elif [[ $current_selection == 6 ]]; then
                echo "What do you want your timeout to be?"
                read -rt 600 -p "Please type an integer: " up_timeout
                ! [[ $up_timeout =~ ^[0-9]+$ ]] && echo -e "Error: \"$up_timeout\" is invalid, it needs to be an integer\nNOT adding it to the list" && sleep 5 && continue
                update_list+=("-t $up_timeout")
                update_selection+=("-t" "$up_timeout")
            elif [[ $current_selection == 7 ]]; then
                update_list+=("-s")
                update_selection+=("-s") 
            elif [[ $current_selection == 8 ]]; then
                update_list+=("-p")
                update_selection+=("-p")  
            else
                echo "$current_selection was not an option, try again" && sleep 5 && continue                                                                  
            fi
        done
        ;;
    *)
        echo "Unknown option" && exit 1
        ;;
  esac
  echo ""
}
export -f menu