#!/bin/bash

# Function to display command output in a dialog box
execute_command() {
  output=$(mktemp /tmp/command_output.XXXXXX)
  if "$@" >"$output" 2>&1; then
    dialog --title "Command Output" --textbox "$output" 20 60
  else
    dialog --title "Error Occurred" --textbox "$output" 20 60
  fi
  rm "$output"
}

# Function to install necessary packages for Debian
install_packages_debian() {
  echo "Starting Package installation on Debian. Please check the terminal output."
  sudo apt-get update && sudo apt-get install -y net-tools psmisc chromium tightvncserver xorg x11-xserver-utils unclutter openbox xdotool vnstat x11vnc
}

# Function to install necessary packages for Raspberry Pi OS Lite
install_packages_raspberry() {
  echo "Starting Package installation on Raspberry Pi OS Lite. Please check the terminal output."
  sudo apt-get update && sudo apt-get install -y net-tools psmisc chromium-browser tightvncserver xorg x11-xserver-utils unclutter openbox xdotool vnstat x11vnc
  sudo ln -s /usr/bin/chromium-browser /usr/bin/chromium
}

# Installation selector based on distro
install_packages() {
  DIALOG_RESULT=$(dialog --title "Select Distribution" --menu "Choose your Linux distribution:" 15 50 2 \
    1 "Debian" \
    2 "Raspberry Pi OS Lite" \
    3>&1 1>&2 2>&3 3>&-)

  case "$DIALOG_RESULT" in
    1) install_packages_debian ;;
    2) install_packages_raspberry ;;
  esac
}

# Function to configure .xinitrc
configure_xinitrc() {
  cat > /home/kiosk/.xinitrc << 'EOF'
#!/bin/sh

# Disable power management, screen blanking and make cursor disappear if not moved
xset s off
xset -dpms
xset s noblank
unclutter -idle 0.1 -root &

# Start Openbox session
openbox-session &
x11vnc -display :0 -rfbauth ~/.vnc/passwd -forever -rfbport 5901 &

# Start Kiosk Script
/bin/bash /home/kiosk/start.sh
EOF
  sudo chmod a+x /home/kiosk/.xinitrc
}

# Function to configure automatic sign-in
configure_autologin() {
  if [[ $1 != "--auto" ]]; then
    dialog --title "Automatic Login Configuration" --msgbox "Configuring automatic login..." 10 50
  fi
  sudo mkdir -p /etc/systemd/system/getty@tty1.service.d
  sudo bash -c "cat > /etc/systemd/system/getty@tty1.service.d/autologin.conf << 'EOF'
[Service]
ExecStart=
ExecStart=-/sbin/agetty --autologin $USER --noclear %I \$TERM
EOF"
}

# Function to configure start.sh with dynamic scaling factor for interactive or automated mode
configure_start_sh() {
  if [[ $# -eq 0 ]]; then  # If no arguments, it's 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>&-)
      VNC_PASSWORD=$(dialog --title "VNC Password" --inputbox "Please enter the VNC password:" 8 50 "" 3>&1 1>&2 2>&3 3>&-)
      set_start_script_single $SCALE_FACTOR $WEBSITE $VNC_PASSWORD
    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
      VNC_PASSWORD=$(dialog --title "VNC Password" --inputbox "Please enter the VNC password:" 8 50 "" 3>&1 1>&2 2>&3 3>&-)
      configure_tabs $SCALE_FACTOR $VNC_PASSWORD "${tab_urls[@]}"
    fi
  else  # Automated mode with URLs as arguments
    local scaling_factor=$1
    local vnc_password=$2
    shift 2  # Remove the first two arguments (scaling factor and VNC password) for URLs
    if [ $# -eq 1 ]; then  # Single website
      set_start_script_single $scaling_factor $1 $vnc_password
    else  # Multiple tabs
      configure_tabs $scaling_factor $vnc_password "$@"
    fi
  fi
}

# Helper function to configure start.sh for a single website
set_start_script_single() {
  local scaling_factor=$1
  local website=$2
  local vnc_password=$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

  # Write the script with properly formatted commands
  cat > /home/kiosk/start.sh <<EOF
#!/bin/bash
/usr/bin/chromium --force-device-scale-factor=$scaling_factor --temp-profile --disable-profiles --no-first-run --disable-translate --no-default-browser-check --disable-cache --ignore-certificate-errors --disable-web-security --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
  local vnc_password=$2
  shift 2  # Remove the first two arguments (scaling factor and VNC password) to get URLs
  local tab_urls=""
  for url in "$@"; do
    tab_urls+="'$url' "  # Ensure URLs are properly quoted
  done

  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

  # Write the script with properly formatted commands
  cat > /home/kiosk/start.sh <<EOF
#!/bin/bash
/usr/bin/chromium --force-device-scale-factor=$scaling_factor --temp-profile --disable-profiles --no-first-run --disable-translate --no-default-browser-check --disable-cache --ignore-certificate-errors --disable-web-security --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 edit .bashrc for automatic start of X session
edit_bashrc() {
  echo '[[ -z $DISPLAY && $XDG_VTNR -eq 1 ]] && exec startx' >> /home/$USER/.bashrc
}

# Check for help option
if [[ "$1" == "-h" || "$1" == "--help" ]]; then
    echo "Usage: $0 [--auto <distro> <scaling_factor> <vnc_password> <url1> [<url2> ... <urlN>]]"
    echo "Options:"
    echo "  --auto        Run the script in automatic mode with no user interaction."
    echo "                Requires specifying the distribution ('debian' or 'raspberry'),"
    echo "                scaling factor (e.g., '1' for 1080p, '2' for 4K), VNC password,"
    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 debian 1 myvncpassword \"https://www.example.com\""
    echo "  $0 --auto raspberry 2 myvncpassword \"https://www.example1.com\" \"https://www.example2.com\""
    exit 0
fi

# Automated or interactive mode check
if [[ $1 == "--auto" ]]; then
  distro=$2
  shift 2  # Remove the first two arguments (script name and distro) for URLs, scaling factor, and VNC password
  scaling_factor=$1
  vnc_password=$2
  shift 2  # Remove the scaling factor and VNC password to get URLs
  [[ $distro == "debian" ]] && install_packages_debian
  [[ $distro == "raspberry" ]] && install_packages_raspberry
  configure_xinitrc
  configure_start_sh $scaling_factor $vnc_password "$@"
  configure_autologin --auto
  edit_bashrc
else
dialog --title "WARNING !!!" --msgbox "Do not run this script under any user other than 'kiosk' and make sure the Kiosk user has sudo rights for the duration of the installation. In addition, check out the -h command on this script if you wish to do the installation more automated!" 10 50
  # Main menu for guided installation
  current_step=1
  while true; do
    case $current_step in
      1)
        install_packages
        ;;
      2)
        configure_xinitrc
        ;;
      3)
        configure_start_sh
        ;;
      4)
        configure_autologin
        ;;
      5)
        edit_bashrc
        ;;
      6)
        break
        ;;
      *)
        echo "Invalid option. Try another one." >&2
        ;;
    esac

    # Automatically proceed to the next step if the previous step was successful
    if [ $? -eq 0 ]; then
      current_step=$((current_step + 1))
      if [ $current_step -gt 5 ]; then
        current_step=6
      fi
    else
      dialog --title "Error" --msgbox "An error occurred. Please check the output and try again." 10 50
    fi

    # Exit the loop if the last step is completed
    if [ $current_step -eq 6 ]; then
      break
    fi
  done

  dialog --title "Exit" --msgbox "If you completed all steps and no errors were shown to you, reboot now to test if the installation was successful." 10 50
fi

clear