#!/usr/bin/env bash

# Copyright (c) 2026 Joerg Heinemann (heinemannj)
# Author: Joerg Heinemann (heinemannj)
# License: MIT | https://github.com/heinemannj/ProxmoxVE-Admin/raw/main/LICENSE
# Source: https://raw.githubusercontent.com/heinemannj/ProxmoxVE-Admin/main/misc/whiptail.func

# ==============================================================================
# WHIPTAIL HELPER FUNCTIONS
# ==============================================================================
#
# This file contains several whiptail functions for displaying dialog boxes in the terminal.
#
# Whenever possible the dialog box
#   - width is automatically calculated based on the length of the title, text, tag, item and status.
#   - height and listheight are automatically calculated based on the number of options in the list or the number of lines in the text file.
#
# WHIPTAIL Box options: 
#   --msgbox <text> <height auto calculated> <width auto calculated>
#   --yesno <text> <height> <width> (N/A)
#   --infobox <text> <height> <width> (N/A)
#   --inputbox <text> <height static> <width auto calculated> [init] 
#   --passwordbox <text> <height> <width> [init] (N/A)
#   --textbox <file> <height auto calculated> <width static>
#   --menu <text> <height auto calculated> <width auto calculated> <listheight auto calculated> [tag item]
#   --checklist <text> <height auto calculated> <width auto calculated> <listheight auto calculated> [tag item status]
#   --radiolist <text> <height auto calculated> <width auto calculated> <listheight auto calculated> [tag item status]
#   --gauge <text> <height> <width> <percent> (N/A)
#

function whiptail_msgbox() {
  local TITLE=$1
  local TEXT=$2
  local LEN
  LEN=$(echo "$TEXT" | wc -l)
  local HIGHT=$(( LEN + 7 ))
  (( HIGHT > 30 )) && HIGHT=30
  local WIDTH=$(( ${#TITLE} + 16 ))
  local WIDTH_ARRAY=( "$WIDTH" $(( ${#TEXT} + 4 )) )
  for i in "${WIDTH_ARRAY[@]}"; do
    (( i > WIDTH )) && WIDTH=$i
  done
  (( WIDTH > 150 )) && WIDTH=150

  whiptail --backtitle "$APP_BACKTITLE" --title "$TITLE" --scrolltext --msgbox "$TEXT" "$HIGHT" "$WIDTH" 3>&1 1>&2 2>&3
}

function whiptail_inputbox() {
  local TITLE=$1
  local TEXT=$2
  local VALUE_INIT=$3
  local VALUE_INPUT
  local HIGHT=10
  local WIDTH=$(( ${#TITLE} + 16 ))
  local WIDTH_ARRAY=( "$WIDTH" $(( ${#TEXT} + 4 )) $(( ${#VALUE_INIT} + 8 )) )
  for i in "${WIDTH_ARRAY[@]}"; do
    (( i > WIDTH )) && WIDTH=$i
  done

  if VALUE_INPUT=$(whiptail --backtitle "$APP_BACKTITLE" --title "$TITLE" --inputbox "\n$TEXT" \
    "$HIGHT" "$WIDTH" "$VALUE_INIT" 3>&1 1>&2 2>&3); then
    echo "$VALUE_INPUT"
  else
    echo "$VALUE_INIT"
  fi
}

function whiptail_textbox() {
  local TITLE=$1
  local FILE=$2
  local LEN
  LEN=$(wc -l < "$FILE")
  local HIGHT=$(( LEN + 7 ))
  (( HIGHT > 30 )) && HIGHT=30
  local WIDTH=150

  whiptail --backtitle "$APP_BACKTITLE" --title "$TITLE" --scrolltext --textbox "$FILE" "$HIGHT" "$WIDTH" 3>&1 1>&2 2>&3
}

function whiptail_menu() {
  local TITLE=$1
  local TEXT="\nSelect an option:"
  local LEN=$(( ${#OPTIONS[@]} / 2 ))
  local HIGHT=$(( LEN + 9 ))
  local WIDTH=$(( ${#TITLE} + 16 ))
  local WIDTH_OFFSET=5
  local CHOICE
  local MAX_LEFT=0
  local MAX_RIGHT=0
  for ((i=0; i<${#OPTIONS[@]}; i+=2)); do
    (( ${#OPTIONS[i]} > MAX_LEFT )) && MAX_LEFT=${#OPTIONS[i]}
    (( ${#OPTIONS[$(( i+1 ))]} > MAX_RIGHT )) && MAX_RIGHT=${#OPTIONS[$(( i+1 ))]}
  done
  (( MAX_LEFT + MAX_RIGHT + WIDTH_OFFSET > WIDTH )) && WIDTH=$(( MAX_LEFT + MAX_RIGHT + WIDTH_OFFSET ))

  if CHOICE=$(whiptail --backtitle "$APP_BACKTITLE" --title "$TITLE" --menu "$TEXT" \
    "$HIGHT" "$WIDTH" "$LEN" "${OPTIONS[@]}" 3>&1 1>&2 2>&3 || true); then
    echo "$CHOICE"
  fi
}

function whiptail_checklist() {
  local TITLE=$1
  local TEXT=$2
  local -n LIST=$3
  local CHOICE
  local OPTIONS=()
  local WIDTH=$(( ${#TITLE} + 16 ))
  local WIDTH_OFFSET=15
  local MAX_LEFT=0
  local MAX_RIGHT=0

  for ((i=0; i<${#LIST[@]}; i+=2)); do
    local j=$(( i+1 ))
    (( ${#LIST[i]} > MAX_LEFT )) && MAX_LEFT=${#LIST[i]}
    (( ${#LIST[j]} > MAX_RIGHT )) && MAX_RIGHT=${#LIST[j]}
    OPTIONS+=("${LIST[i]}" "${LIST[j]}" "OFF")  
  done
  (( MAX_LEFT + MAX_RIGHT + WIDTH_OFFSET > WIDTH )) && WIDTH=$(( MAX_LEFT + MAX_RIGHT + WIDTH_OFFSET ))
  local LEN=$(( ${#OPTIONS[@]} / 2 ))
  (( LEN > 21 )) && LEN=21
  local HIGHT=$(( LEN + 9 ))

  if CHOICE=$(whiptail --backtitle "$APP_BACKTITLE" --title "$TITLE" --scrolltext --checklist "$TEXT" \
    "$HIGHT" "$WIDTH" "$LEN" "${OPTIONS[@]}" 3>&1 1>&2 2>&3 | tr -d '"'); then
    echo "$CHOICE"
  fi
}

function whiptail_radiolist() {
  local TITLE=$1
  local TEXT=$2
  local -n LIST=$3
  local CHOICE
  local OPTIONS=()
  local WIDTH=$(( ${#TITLE} + 16 ))
  local WIDTH_OFFSET=15
  local MAX_LEFT=0
  local MAX_RIGHT=0

  for ((i=0; i<${#LIST[@]}; i+=2)); do
    local j=$(( i+1 ))
    (( ${#LIST[i]} > MAX_LEFT )) && MAX_LEFT=${#LIST[i]}
    (( ${#LIST[j]} > MAX_RIGHT )) && MAX_RIGHT=${#LIST[j]}
    OPTIONS+=("${LIST[i]}" "${LIST[j]}" "OFF")  
  done
  (( MAX_LEFT + MAX_RIGHT + WIDTH_OFFSET > WIDTH )) && WIDTH=$(( MAX_LEFT + MAX_RIGHT + WIDTH_OFFSET ))
  local LEN=$(( ${#OPTIONS[@]} / 2 ))
  (( LEN > 21 )) && LEN=21
  local HIGHT=$(( LEN + 9 ))

  if CHOICE=$(whiptail --backtitle "$APP_BACKTITLE" --title "$TITLE" --scrolltext --radiolist "$TEXT" \
    "$HIGHT" "$WIDTH" "$LEN" "${OPTIONS[@]}" 3>&1 1>&2 2>&3 | tr -d '"'); then
    echo "$CHOICE"
  fi
}