#!/bin/bash

# Function to display the menu
show_menu() {
    DIALOG_RESULT=$(dialog --title "Configuration Menu" --menu "Choose an option:" 15 50 3 \
      1 "Change Scaling, Kiosk Type, and Websites" \
      2 "Change VNC Password" \
      3 "Exit" \
      3>&1 1>&2 2>&3 3>&-)
    echo $DIALOG_RESULT
}

# Function to configure start.sh
configure_start_sh() {
    if [[ $# -eq 0 ]]; then  # Interactive mode
        SCALE_FACTOR=$(dialog --title "Display Scaling Factor" --inputbox "Enter the display scaling factor (e.g., 1 for 1080p, 2 for 4K):" 8 50 "1" 3>&1 1>&2 2>&3 3>&-)

        DIALOG_RESULT=$(dialog --title "start.sh Configuration" --menu "Choose the script example:" 15 50 2 \
          1 "Simple website script" \
          2 "Two or more tabs alternating every 30 seconds" \
          3>&1 1>&2 2>&3 3>&-)

        if [ "$DIALOG_RESULT" = "1" ]; then
            WEBSITE=$(dialog --title "Enter Website URL" --inputbox "Please enter the URL for the simple website:" 8 50 "https://www.example.com" 3>&1 1>&2 2>&3 3>&-)
            set_start_script_single $SCALE_FACTOR $WEBSITE
        elif [ "$DIALOG_RESULT" = "2" ]; then
            TAB_COUNT=$(dialog --title "Number of Tabs" --inputbox "Enter number of tabs:" 8 50 "3" 3>&1 1>&2 2>&3 3>&-)
            local tab_urls=()
            for (( i=1; i<=TAB_COUNT; i++ )); do
                URL=$(dialog --title "Enter URL for Tab $i" --inputbox "Enter URL:" 8 50 "http://example$i.com" 3>&1 1>&2 2>&3 3>&-)
                tab_urls+=("$URL")
            done
            configure_tabs $SCALE_FACTOR "${tab_urls[@]}"
        fi
    else  # Automated mode
        local scaling_factor=$1
        shift  # Remove the first argument (scaling factor) to get URLs
        if [ $# -eq 1 ]; then  # Single website
            set_start_script_single $scaling_factor $1
        else  # Multiple tabs
            configure_tabs $scaling_factor "$@"
        fi
    fi
}

# Helper function to configure start.sh for a single website
set_start_script_single() {
    local scaling_factor=$1
    local website=$2

    # Write the script with properly formatted commands
    cat > /home/kiosk/start.sh <<EOF
#!/bin/bash
/usr/bin/chromium --no-first-run --disable-translate --no-default-browser-check --disable-cache --kiosk $website
EOF
    chmod a+x /home/kiosk/start.sh
}

# Helper function to configure start.sh for multiple tabs
configure_tabs() {
    local scaling_factor=$1
    shift  # Remove the first argument (scaling factor) to loop over URLs
    local tab_urls=""
    for url in "$@"; do
        tab_urls+="'$url' "  # Ensure URLs are properly quoted
    done

    # Write the script with properly formatted commands
    cat > /home/kiosk/start.sh <<EOF
#!/bin/bash
/usr/bin/chromium --incognito --force-device-scale-factor=$scaling_factor --temp-profile --disable-profiles --no-first-run --disable-translate --no-default-browser-check --disable-cache --kiosk $tab_urls &
sleep 10  # Allow time for Chromium to launch

# Function definitions should be placed before they are called
switch_tabs() {
    WINDOW_ID=\$(xdotool search --onlyvisible --class chromium | head -n 1)
    xdotool windowactivate --sync \$WINDOW_ID
    xdotool key --window \$WINDOW_ID ctrl+Tab
}

is_vnc_active() {
    netstat -an | grep ':5901' | grep -q 'ESTABLISHED'
    return \$?
}

# Main loop to handle tab switching
while true; do
    if is_vnc_active; then
        echo "VNC session active, pausing tab switch..."
        sleep 5
    else
        echo "VNC session inactive, switching tabs..."
        switch_tabs
        sleep 30
    fi
done
EOF
    chmod a+x /home/kiosk/start.sh
}

# Function to change VNC password
change_vnc_password() {
    VNC_PASSWORD=$(dialog --title "VNC Password" --inputbox "Please enter the new VNC password:" 8 50 "" 3>&1 1>&2 2>&3 3>&-)
    mkdir -p /home/kiosk/.vnc
    rm -f /home/kiosk/.vnc/passwd  # Remove existing password file if it exists
    echo "$VNC_PASSWORD" | vncpasswd -f > /home/kiosk/.vnc/passwd
    chmod 600 /home/kiosk/.vnc/passwd
    echo "VNC password changed"
}

# Check for help option
if [[ "$1" == "-h" || "$1" == "--help" ]]; then
    echo "Usage: $0 [--auto <scaling_factor> <url1> [<url2> ... <urlN>]]"
    echo "Options:"
    echo "  --auto        Run the script in automatic mode with no user interaction."
    echo "                Requires specifying the scaling factor (e.g., '1' for 1080p, '2' for 4K),"
    echo "                and at least one URL. Additional URLs are optional and will open in new tabs."
    echo "  -h, --help    Display this help and exit."
    echo ""
    echo "Example:"
    echo "  $0 --auto 1 \"https://www.example.com\""
    echo "  $0 --auto 2 \"https://www.example1.com\" \"https://www.example2.com\""
    exit 0
fi

# Automated or interactive mode check
if [[ $1 == "--auto" ]]; then
    shift  # Remove the first argument (script name)
    scaling_factor=$1
    shift  # Remove the scaling factor to get URLs
    configure_start_sh $scaling_factor "$@"
else
    # Main loop
    while true; do
        choice=$(show_menu)
        case $choice in
            1) configure_start_sh ;;
            2) change_vnc_password ;;
            3) exit 0 ;;
            *) echo "Invalid option, please try again" ;;
        esac
    done
fi