#!/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
    while true
    do
        case $selection in
            0)
                exit
                ;;
            1)
                help="true"
                exit
                ;;
            2)
                dns="true"
                exit
                ;;
            3)
                mount="true"
                exit
                ;;
            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"
                exit
                ;;
            6)
                deleteBackup="true"
                exit
                ;;
            7)
                self_update="true"
                exit
                ;;
            8)
                script=$(readlink -f "$0")
                script_path=$(dirname "$script")
                script_name="heavy_script.sh"
                cd "$script_path" || exit
                while true 
                do
                    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
                        if [[ $up_async == 0 ]]; then
                            echo "Error: \"$up_async\" is less than 1"
                            echo "NOT adding it to the list"
                            sleep 5
                            continue
                        elif ! [[ $up_async =~ ^[0-9]+$  ]]; then
                            echo "Error: \"$up_async\" is invalid, it needs to be an integer"
                            echo "NOT adding it to the list"
                            sleep 5
                            continue
                        else
                            update_selection+=("-U" "$up_async")
                            break
                        fi
                    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
                        if [[ $up_async == 0 ]]; then
                            echo "Error: \"$up_async\" is less than 1"
                            echo "NOT adding it to the list"
                            sleep 5
                            continue
                        elif ! [[ $up_async =~ ^[0-9]+$  ]]; then
                            echo "Error: \"$up_async\" is invalid, it needs to be an integer"
                            echo "NOT adding it to the list"
                            sleep 5
                            continue
                        else
                            update_selection+=("-u" "$up_async")
                            break
                        fi
                    elif [[ $current_selection == 0 ]]; then
                        echo "Exiting.." 
                        exit
                    else
                        echo "$current_selection was not an option, try again"
                        exit
                    fi
                done
                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 "---------------"
                    echo "bash heavy_script.sh ${update_selection[*]}"
                    echo
                    read -rt 600 -p "Please type the number associated with the flag above: " current_selection

                    if [[ $current_selection == 0 ]]; then
                        clear -x
                        echo "Running \"bash heavy_script.sh ${update_selection[*]}\""
                        echo
                        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_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_selection+=("-i" "$up_ignore")                
                    elif [[ $current_selection == 3 ]]; then

                        update_selection+=("-r")
                    elif [[ $current_selection == 4 ]]; then

                        update_selection+=("-S")
                    elif [[ $current_selection == 5 ]]; then

                        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_selection+=("-t" "$up_timeout")
                    elif [[ $current_selection == 7 ]]; then

                        update_selection+=("-s") 
                    elif [[ $current_selection == 8 ]]; then

                        update_selection+=("-p")  
                    else
                        echo "$current_selection was not an option, try again" && sleep 5 && continue                                                                  
                    fi
                done
                exit
                ;;
            *)
                echo "That was not an option, please try again" && sleep 5
                ;;
        esac
        echo
    done
}
export -f menu