#! /bin/bash

if ! command -v dialog; then
  sudo apt-get install dialog -y
fi

# Set dialog colors
cp -f dotfiles/.dialogrc  $HOME/.dialogrc

BACKTITLE="Trude's Linux Toolkit"
dialog --erase-on-exit \
       --backtitle "$BACKTITLE" \
       --checklist "Use the arrow keys and SPACE to select, then press ENTER." 30 90 5 \
        "1" "Update OS" "on"\
        "2" "Copy Dotfiles" "on"\
        "3" "Install DWM Desktop" "off" \
        "4" "Install GNOME Desktop" "on" \
        "5" "Install GitHub CLI" "off"\
        "6" "Install Ollama" "off"\
        "7" "Install MultiMC" "off" 2> choice.tmp
main_menu=$( cat choice.tmp )
rm choice.tmp

compile() {
  cd programs/$1
  sudo make clean install
  cd ../..
}

for selection in $main_menu; do
  if [ "$selection" = "1" ]; then
    # --- UPDATE OS ---

    dialogUpdate() {
      dialog --backtitle "$BACKTITLE" --title "Update Debian and Packages" \
	       --mixedgauge "Updating..." \
	       0 0 $1 \
         "Update repos" "$2" \
         "Update distro" "$3" \
         "Update packages" "$4" \
         "Clean up " "$5"
    }

    dialogUpdate 0 7 4 4 4

    sudo apt-get update &> /dev/null
    dialogUpdate 15 5 7 4 4

    sudo apt-get dist-upgrade -y &> /dev/null
    dialogUpdate 35 5 5 7 4DWM

    sudo apt-get upgrade -y &> /dev/null
    dialogUpdate 80 5 5 5 7

    sudo apt-get clean &> /dev/null
    sudo apt-get autoremove -y &> /dev/null
    dialogUpdate 100 5 5 5 5
  fi

  if [ "$selection" = "2" ]; then
    clear
    echo "---------------------"
    echo "--- Copy Dotfiles ---"
    echo "---------------------"
    echo
    echo

    # Neovim
    dialog --erase-on-exit \
           --backtitle "$BACKTITLE" \
           --title "Install/Update Neovim?" \
           --yesno "Nvim will be compiled from source. This may take a long time, depending on your device." 10 40

    if [ "$?" -eq 0 ]; then
      # NVIM has to be compiled from source to support arm64 and i386 devices, for example.
      sudo apt install -y ninja-build gettext cmake unzip curl build-essential
      git clone https://github.com/neovim/neovim --depth 1
      cd neovim
      git checkout stable
      make CMAKE_BUILD_TYPE=RelWithDebInfo
        sudo make install
      cd ..
      rm -rf neovim
    fi

    # Zed
    dialog --erase-on-exit \
           --backtitle "$BACKTITLE" \
           --title "Install Zed?" \
           --yesno "Zed (code editor) will be installed using the official install script." 10 40

    if [ "$?" -eq 0 ]; then
      curl https://zed.dev/install.sh | sh
    fi

    echo "Installing utilities..."
    sudo apt install -y htop fzf git wget curl bash-completion

    echo "Copying dotfiles..."
    cp -vrf dotfiles/.* $HOME
    cp -vrf dotfiles/* $HOME

    echo "Loading fonts..."
    fc-cache -f
  fi

  if [ "$selection" = "3" ]; then
    clear
    echo "---------------------------"
    echo "--- Install DWM Desktop ---"
    echo "---------------------------"
    echo
    echo

    # Install Dependencies
    sudo apt install -y xorg picom libx11-dev libxft-dev libxinerama-dev build-essential libxrandr-dev policykit-1-gnome dbus-x11 pipewire-audio wireplumber pipewire-pulse pipewire-alsa network-manager firefox-esr feh scrot dunst
    systemctl --user --now enable wireplumber.service
    sudo systemctl enable NetworkManager

    # Compile
    for program in "dwm" "dmenu" "slock" "st" "tabbed" "dwmblocks"; do
      compile $program
    done
  fi

  if [ "$selection" = "4" ]; then
    dialog --erase-on-exit \
           --backtitle "$BACKTITLE" \
           --title "Add Flatpak support?" \
           --yesno "Install Flatpak; Add the GNOME Software add-on and enable Flathub." 10 40
    flatpak_choice="$?"

    clear
    echo "-----------------------------"
    echo "--- Install GNOME Desktop ---"
    echo "-----------------------------"
    echo
    echo

    sudo apt install -y gnome-core power-profiles-daemon
    dconf load -f / < ./settings.dconf
  
    if [ $flatpak_choice -eq 0 ]; then
      echo "Enabling Flatpak..."
      sudo apt install -y flatpak
      sudo apt install -y gnome-software-plugin-flatpak
      sudo flatpak remote-add --if-not-exists flathub https://dl.flathub.org/repo/flathub.flatpakrepo
    fi


  fi

  if [ "$selection" = "5" ]; then
    # --- INSTALL GH CLI ---

    clear
    echo "----------------------"
    echo "--- Install GH CLI ---"
    echo "----------------------"
    echo
    echo

    sudo apt install wget -y
    sudo mkdir -p -m 755 /etc/apt/keyrings
    sudo rm -f /etc/apt/sources.list.d/github-cli.list
    wget -qO- https://cli.github.com/packages/githubcli-archive-keyring.gpg | sudo tee /etc/apt/keyrings/githubcli-archive-keyring.gpg
    sudo chmod go+r /etc/apt/keyrings/githubcli-archive-keyring.gpg
    echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/githubcli-archive-keyring.gpg] https://cli.github.com/packages stable main" | sudo tee /etc/apt/sources.list.d/github-cli.list
    sudo apt update
    sudo apt install gh -y
  fi

  if [ "$selection" = "6" ]; then
    # --- Install AI Tools ---

    clear
    echo "----------------------"
    echo "--- Install Ollama ---"
    echo "----------------------"
    echo
    echo

    # Ollama - LLM Server
    curl -fsSL https://ollama.com/install.sh | sh
  fi

  if [ "$selection" = "7" ]; then
    # --- Install MultiMC ---

    clear
    echo "-----------------------"
    echo "--- Install MultiMC ---"
    echo "-----------------------"
    echo
    echo

    sudo apt-get update
    sudo apt-get install -y libqt5core5a libqt5network5 libqt5gui5
    wget https://files.multimc.org/downloads/multimc_1.6-1.deb
    sudo apt-get install -y ./multimc_1.6-1.deb
    rm multimc_1.6-1.deb

    # Install java
    sudo mkdir -p /etc/apt/keyrings
    wget -qO- https://packages.adoptium.net/artifactory/api/gpg/key/public | sudo tee /etc/apt/keyrings/adoptium.asc
    echo "deb [signed-by=/etc/apt/keyrings/adoptium.asc] https://packages.adoptium.net/artifactory/deb $(awk -F= '/^VERSION_CODENAME/{print$2}' /etc/os-release) main" | sudo tee /etc/apt/sources.list.d/adoptium.list
    sudo apt update
    sudo apt install -y temurin-8-jdk temurin-21-jdk temurin-17-jdk
  fi
done

clear