ubuntu_setup.sh

#!/usr/bin/env bash
ORNG=$(echo "\033[38;5;202m")
CL=$(echo "\033[m")
function header_info {
    clear
    echo -e "$ORNG"
cat <<'EOF' 
                                          GGGGGGGGGGG                                          
                                   GGGGGGGGGGGGGGGGGGGGGGGGG                                   
                             GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                             
                          GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                          
                       GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                       
                     GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                     
                  GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                  
                 GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGí    íÞGGGGGGGGGGGGG                
               GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGí        {GGGGGGGGGGGGG               
             GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGÏ          zGGGGGGGGGGGGGG             
            GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGÞÇ6ÏÏzÏÏ6ÇÞGGGí          íGGGGGGGGGGGGGGG            
           GGGGGGGGGGGGGGGGGGGGGGGGGGüí›              6GÇ          6GGGGGGGGGGGGGGGG           
          GGGGGGGGGGGGGGGGGGGGGGGGGGG›                 ÇGG{      —GGGGGGGGGGGGGGGGGGG          
         GGGGGGGGGGGGGGGGGGGGGG6— íGGGí                 íGGGÞÇÇÇGGGGGGGGGGGGGGGGGGGGGG         
        GGGGGGGGGGGGGGGGGGGGGÞ›    íGGG{                   {6ÇÇ6zÇGGGGGGGGGGGGGGGGGGGGG        
       GGGGGGGGGGGGGGGGGGGGG{       íGGGzÏÇGGGGGGGGGÇÏ{           {GGGGGGGGGGGGGGGGGGGGG       
      GGGGGGGGGGGGGGGGGGGGÞ›         —GGGGGGGGGGGGGGGGGGÇ›          ÇGGGGGGGGGGGGGGGGGGGG      
      GGGGGGGGGGGGGGGGGGGÇ         íÞGGGGGGGGGGGGGGGGGGGGGÞí         6GGGGGGGGGGGGGGGGGGG      
      GGGGGGGGGGGGGGGGGGÇ         ÇGGGGGGGGGGGGGGGGGGGGGGGGGÞ         6GGGGGGGGGGGGGGGGGG      
     GGGGGGGGGGGGGGGGGGG{       ›ÞGGGGGGGGGGGGGGGGGGGGGGGGGGGÞ—       —GGGGGGGGGGGGGGGGGGG     
     GGGGGGGGGGGGGGGGGGÇ        ÇGGGGGGGGGGGGGGGGGGGGGGGGGGGGGÞ        üGGGGGGGGGGGGGGGGGG     
    GGGGGGGGGGÞí     zÞGÇ›     üGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGü       ›GGGGGGGGGGGGGGGGGGG    
    GGGGGGGGGz         ÏGG     GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG        ÞGGGGGGGGGGGGGGGGGG    
    GGGGGGGGÇ           ÇGz   {GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG    
    GGGGGGGGÇ           ÇGÏ   {GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG    
    GGGGGGGGGz         ÏGG     GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG›       ÞGGGGGGGGGGGGGGGGGG    
    GGGGGGGGGGÇí     íÞGÞ—     üGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGü       ›GGGGGGGGGGGGGGGGGGG    
     GGGGGGGGGGGGGGGGGG6        ÞGGGGGGGGGGGGGGGGGGGGGGGGGGGGGÞ        üGGGGGGGGGGGGGGGGGG     
     GGGGGGGGGGGGGGGGGGG{       ›ÞGGGGGGGGGGGGGGGGGGGGGGGGGGGÞ—       —GGGGGGGGGGGGGGGGGGG     
      GGGGGGGGGGGGGGGGGG6         ÞGGGGGGGGGGGGGGGGGGGGGGGGGÞ         6GGGGGGGGGGGGGGGGGG      
      GGGGGGGGGGGGGGGGGGGÇ         íÞGGGGGGGGGGGGGGGGGGGGGÞí         6GGGGGGGGGGGGGGGGGGG      
      GGGGGGGGGGGGGGGGGGGGÞ›         {GGGGGGGGGGGGGGGGGGÇ›          ÇGGGGGGGGGGGGGGGGGGGG      
       GGGGGGGGGGGGGGGGGGGGG{       {GGGÏüÇGGGGGGGGGÇü{           —ÞGGGGGGGGGGGGGGGGGGGG       
        GGGGGGGGGGGGGGGGGGGGGÇ›    íGGGí                   {üÇÇüzÇGGGGGGGGGGGGGGGGGGGGG        
         GGGGGGGGGGGGGGGGGGGGGG6› íGGGí                 íGGGÞÇÇÞGGGGGGGGGGGGGGGGGGGGGG         
          GGGGGGGGGGGGGGGGGGGGGGGGGGG—                 6GGí      {GGGGGGGGGGGGGGGGGGGG         
           GGGGGGGGGGGGGGGGGGGGGGGGGGü{›              6GÇ          6GGGGGGGGGGGGGGGG           
            GGGGGGGGGGGGGGGGGGGGGGGGGGGGGÞÇ6üÏzzzÏü6ÇÞGGz          íGGGGGGGGGGGGGGG            
             GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGz          zGGGGGGGGGGGGGG             
               GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGí        —GGGGGGGGGGGGG               
                GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGÞí    {ÞGGGGGGGGGGGGG                
                  GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                  
                     GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                     
                       GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                       
                          GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                          
                             GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG                             
                                  GGGGGGGGGGGGGGGGGGGGGGGGGGG                                  
                                          GGGGGGGGGGG                                          
EOF
echo -e "$CL"
echo "version 2.0"
}

YW=$(echo "\033[33m")
BL=$(echo "\033[36m")
RD=$(echo "\033[01;31m")
GN=$(echo "\033[1;92m")
BGN=$(echo "\033[4;92m")
DGN=$(echo "\033[32m")
BFR="\\r\\033[K"
HOLD="-"
BOLD=$(echo "\033[1m")
BFR="\\r\\033[K"
TAB="  "
CM="${GN}✓${CL}"
VERIFYPW="${TAB}${TAB}${CL}"
silent() { "$@" >/dev/null 2>&1; }
set -e
header_info
echo "Loading..."
function msg_info() {
  local msg="$1"
  echo -ne " ${HOLD} ${YW}${msg}..."
}

function msg_ok() {
  local msg="$1"
  echo -e "${BFR} ${CM} ${GN}${msg}${CL}"
}

set -eu -o pipefail # fail on error and report it, debug all lines
print_red (){
echo -e "\033[0;31m$1\033[0m"
}   
if [[ $EUID -ne 0 ]]; then
    print_red "This script must be run as root"
    exit 1
fi
userid=$SUDO_USER

install() {
    STD="silent"
    header_info
    rebootMenu
    vmwareMenu
    secondUser
    msg_info "Launching no touch in - 5"
    sleep 1
    msg_info "4"
    sleep 1
    msg_info "3"
    sleep 1
    msg_info "2"
    sleep 1
    msg_info "1"
    sleep 1
    header_info
    msg_ok "Launched"

    signalRepo
    fastFetchRepo
    letsUpdate
    apps
    flatHub
#    flathubTheme
    chromeInstall
    lxcInstall
    nvidiaInstall
    printInstall
    tmuxStuff
    case $vmwareq in
            [yY]) vmwareSux;;
            *) msg_ok "Skipping VMware Workstation install. Linux KVM FTW!";;
    esac
    allDone
}

signalRepo() {
      msg_info "Setting up Signal repository"
  #wget -q https://repo.netdata.cloud/repos/repoconfig/debian/bookworm/netdata-repo_2-2+debian12_all.deb
  #$STD dpkg -i netdata-repo_2-2+debian12_all.deb
  #rm -rf netdata-repo_2-2+debian12_all.deb
    wget -q -O- https://updates.signal.org/desktop/apt/keys.asc | gpg --dearmor > signal-desktop-keyring.gpg 
    cat signal-desktop-keyring.gpg | sudo tee /usr/share/keyrings/signal-desktop-keyring.gpg > /dev/null
    echo 'deb [arch=amd64 signed-by=/usr/share/keyrings/signal-desktop-keyring.gpg] https://updates.signal.org/desktop/apt xenial main' | $STD sudo tee /etc/apt/sources.list.d/signal-xenial.list
  msg_ok "Set up Signal repository"
}

fastFetchRepo() {
   msg_info "Setting up FastFetch repository"
  #wget -q https://repo.netdata.cloud/repos/repoconfig/debian/bookworm/netdata-repo_2-2+debian12_all.deb
  #$STD dpkg -i netdata-repo_2-2+debian12_all.deb
  #rm -rf netdata-repo_2-2+debian12_all.deb
  $STD sudo add-apt-repository -y ppa:zhangsongcui3371/fastfetch
  msg_ok "Set up FastFetch repository"
}

letsUpdate() {
    msg_info "Updating before install"
    $STD sudo apt-get update
    msg_ok "apt update done"
    msg_info "doing upgrade"
    $STD sudo DEBIAN_FRONTEND=noninteractive apt-get upgrade -y
    msg_ok "apt upgrade done"
    msg_info "doing dist-upgrade"
    $STD sudo DEBIAN_FRONTEND=noninteractive apt-get dist-upgrade -y
    msg_ok "dist-upgrade done"
    msg_info  "doing apt autoremove"
    $STD sudo DEBIAN_FRONTEND=noninteractive apt-get autoremove -y
    msg_ok "apt autoremove done"
    msg_info  "doing autoclean"
    $STD sudo DEBIAN_FRONTEND=noninteractive apt-get autoclean -y
    msg_ok "apt autoclean done"
    msg_info "updating snap"
    $STD sudo snap refresh
    msg_ok "snap updated"
    msg_ok "Updates done"
}

apps() {
    while read -r p ; do $STD sudo DEBIAN_FRONTEND=noninteractive apt-get install -y $p && msg_ok "$p installed" ; done < <(cat << "EOF"
    bpytop
    python3
    fastfetch
    curl
    gcc
    tree
    clamav
    tmux
    ncdu
    git
    fzf
    hashcat
    feroxbuster
    gufw
    signal-desktop
    make
    qemu-kvm
    libvirt-daemon-system
    libvirt-clients
    bridge-utils
    cifs-utils
    virtinst
    virt-manager
    net-tools
    gnome-shell-extension-manager
    sysvinit-utils
    wireshark
    ostree
    appstream-util
    rename
    parallel
    drawing
    flameshot
EOF
)
    while read -r p ; do $STD sudo snap install $p && msg_ok "$p installed" ; done < <(cat << "EOF"
    pycharm-community --classic
    obsidian --classic
    code --classic
    seclists
    tree
EOF
)
}

flatHub() {
    msg_info "Installing flathub"
    $STD sudo DEBIAN_FRONTEND=noninteractive apt install flatpak -y
    $STD sudo DEBIAN_FRONTEND=noninteractive apt install gnome-software-plugin-flatpak -y
    sudo flatpak remote-add --if-not-exists flathub https://flathub.org/repo/flathub.flatpakrepo
    msg_ok "Flathub installed"
    while read -r p ; do $STD sudo flatpak install --or-update flathub $p -y --noninteractive && msg_ok "flathub $p installed" ; done < <(cat << "EOF"
        org.onlyoffice.desktopeditors
        com.github.tchx84.Flatseal
        org.keepassxc.KeePassXC
        io.github.peazip.PeaZip
        io.missioncenter.MissionCenter
EOF
)
}

lxcInstall() {
#lxc configs
#sudo systemctl is-active libvirtd
msg_info "enabeling pcie passthrough"
sudo sed -i 's|GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"|GRUB_CMDLINE_LINUX_DEFAULT="quiet splash intel_iommu=on"|g' /etc/default/grub
$STD sudo update-grub
sudo usermod -aG kvm $userid
sudo usermod -aG libvirt $userid
msg_ok "pcie passthrough enabled"
}

gnomeExtFail() {
  # This Function does not work which makes me sad and mad, but mostly sad
    msg_info "adding gnome extension app-icons-taskbar"
    urls=(
      'https://extensions.gnome.org/extension/4944/app-icons-taskbar/'
    )

    # Loop through each URL
    for url in "${urls[@]}"; do
      echo "url = ${url}"
      # get package metadata
      id=$(echo "${url}" | cut --delimiter=/ --fields=5)
      url_pkg_metadata="https://extensions.gnome.org/extension-info/?pk=${id}"
      # Extract data for each extension
      uuid=$(curl -s "$url_pkg_metadata" | jq -r '.uuid' | tr -d '@')
      latest_extension_version=$(curl -s "$url_pkg_metadata" | jq -r '.shell_version_map | to_entries | max_by(.value.version) | .value.version')
      latest_shell_version="46"

      # get  package
      filename="${uuid}.v${latest_extension_version}.shell-extension.zip"
      url_pkg="https://extensions.gnome.org/extension-data/${filename}"
      wget -q -P /tmp "${url_pkg}"
      # install package
      gnome-extensions install "/tmp/${filename}" --force

      # Print the results
      echo "For URL: $url"
      echo "UUID: $uuid"
      echo "Latest extension version: $latest_extension_version"
      echo "Latest shell version: $latest_shell_version"
      echo "--------------------------------------"
    done
    msg_ok "gnome extension app-icons-taskbar installed"
}

flathubTheme() {
  theme=$(gsettings get org.gnome.desktop.interface gtk-theme | tr -d "'")
    msg_info "Updating flathub to Theme $theme"
  die() {
    echo "$@" >&2
    exit 1
  }
  cache_home="${XDG_CACHE_HOME:-$HOME/.cache}"
  data_home="${XDG_DATA_HOME:-$HOME/.local/share}"
  stylepak_cache="$cache_home/stylepak"
  repo_dir="$stylepak_cache/repo"
  install_target=user
  GTK_THEME_VER=3.22

  old_cache="$cache_home/pakitheme"
  if [[ -e "$old_cache" ]]; then
    # If the cache under the old name still exists, try to preserve it under the new name.
    # But, if the new cache also already exists, just trash the old one instead.
    [[ -e "$stylepak_cache" ]] && rm -rf "$old_cache" || mv "$old_cache" "$stylepak_cache"
  fi


  app_id=org.gtk.Gtk3theme.$theme
  for location in "$data_home/themes" "$HOME/.themes" /usr/share/themes; do
    if [[ -d "$location/$theme" ]]; then
      theme_path="$location/$theme"
      break
    fi
  done

  [[ -n "$theme_path" ]] || die 'Could not locate theme.'

  root_dir="$stylepak_cache/$theme"
  repo_dir="$root_dir/repo"

  rm -rf "$root_dir" "$repo_dir"
  mkdir -p "$repo_dir"
  ostree --repo="$repo_dir" init --mode=archive
  ostree --repo="$repo_dir" config set core.min-free-space-percent 0

  build_dir="$root_dir/build"
  rm -rf "$build_dir"
  mkdir -p "$build_dir/files"

  theme_gtk_version=$(ls -1d "$theme_path"/* 2>/dev/null \
    | grep -Po 'gtk-3\.\K\d+$' \
    | sort -nr \
    | head -1)

  [[ -n "$theme_gtk_version" ]] || \
    die "Theme directory did not contain any recognized GTK themes."

  cp -a "$theme_path/gtk-3.$theme_gtk_version/"* "$build_dir/files"
  mkdir -p "$build_dir/files/share/appdata"
cat >"$build_dir/files/share/appdata/$app_id.appdata.xml" <<EOF
<?xml version="1.0" encoding="UTF-8"?>
<component type="runtime">
  <id>$app_id</id>
  <metadata_license>CC0-1.0</metadata_license>
  <name>$theme GTK Theme</name>
  <summary>$theme (generated via stylepak)</summary>
</component>
EOF

  $STD appstream-compose \
    --prefix="$build_dir/files" \
    --basename="$app_id" \
    --origin=flatpak "$app_id"

  $STD ostree --repo="$repo_dir" commit -b base --tree=dir="$build_dir"

  bundles=()

  while read -r arch; do
    bundle="$root_dir/$app_id-$arch.flatpak"

    rm -rf "$build_dir"
    $STD ostree --repo="$repo_dir" checkout -U base "$build_dir"

  read -rd '' metadata <<EOF ||:
[Runtime]
name=$app_id
runtime=$app_id/$arch/$GTK_THEME_VER
sdk=$app_id/$arch/$GTK_THEME_VER
EOF
    # Make sure there is no trailing newline, so xa.metadata doesn't get confused later
    echo -n "$metadata" > "$build_dir/metadata"

     $STD ostree --repo="$repo_dir" commit \
      -b "runtime/$app_id/$arch/$GTK_THEME_VER" \
      --add-metadata-string "xa.metadata=$(cat $build_dir/metadata)" \
      --link-checkout-speedup \
      "$build_dir"
     $STD flatpak build-bundle --runtime --arch="$arch" \
      "$repo_dir" "$bundle" "$app_id" "$GTK_THEME_VER"

    trap 'rm "$bundle"' EXIT

    bundles+=("$bundle")
    # Note: a pipe can't be used because it will mess with subshells and cause the append
    # to bundles to fail.
  done < <(flatpak list --runtime --columns=arch:f | sort -u)

  for bundle in "${bundles[@]}"; do
    if [ "$install_target" == "system" ]; then
       $STD sudo flatpak install -y --$install_target "$bundle"
    else
       $STD flatpak install -y --$install_target "$bundle"
    fi
  done
msg_ok "Flathub Themes updated to $theme"
}

nvidiaInstall() {
    msg_info "Assessing Nvidia HW status"
    if ! lspci | grep -q "VGA compatible controller: NVIDIA"; then
        msg_ok "No Nvidia card found, skipping"
    else
        $STD sudo DEBIAN_FRONTEND=noninteractive apt install -y nvidia-cuda-toolkit
        msg_ok "Nvidia cuda toolkit installed"
    fi
}

printInstall() {
    msg_info "Assessing Fingerprint HW status"
    if ! lsusb | grep -q "Fingerprint"; then
        msg_ok "No fingerprint found, skipping"
    else
        $STD sudo pam-auth-update --enable fprintd
        msg_ok "Fingerprint auth enabled"
    fi
}

chromeInstall() {
  msg_info "installing Google Chrome"
  cd /home/$userid/Downloads
  chromedefault='google-chrome-stable_current_amd64.deb'
  #read -p "Enter desired version of Google Chrome [$chromedefault]:" chromev;
  chromefile=${chromev:-$chromedefault}
  #echo downloading $chromefile...
  test -f /home/$userid/Downloads/$chromefile&&msg_ok 'Chrome already downloaded'||wget -q https://dl.google.com/linux/direct/$chromefile -P /home/$userid/Downloads
  gc="google-chrome"
  which "$gc" | grep -o "$gc" > /dev/null &&  msg_ok $gc' already installed' || $STD sudo dpkg -i /home/$userid/Downloads/$chromefile && msg_ok "Google Chrome installed"

}

tmuxStuff() {
  msg_info "tmux fun"
  cd /home/$userid/
  test -d /home/$userid/.tmux.bk&&sudo rm -rf /home/$userid/.tmux.bk
  test -d /home/$userid/.tmux.conf.local.bk&&sudo rm -rf /home/$userid/.tmux.conf.local.bk
  test -d /home/$userid/.tmux&&sudo mv -f /home/$userid/.tmux /home/$userid/.tmux.bk
  test -d /home/$userid/.tmux.conf.local&&sudo mv -f /home/$userid/.tmux.conf.local /home/$userid/.tmux.conf.local.bk
  git clone -q https://github.com/gpakosz/.tmux.git
  ln -s -f .tmux/.tmux.conf
  cp .tmux/.tmux.conf.local /home/$userid/
  sudo chown $userid:$userid /home/$userid/.tmux
  sudo chown $userid:$userid /home/$userid/.tmux.conf.local
  msg_ok "tmux fun installed"

  msg_info "adding update alias and terminal candy"
  term=".$(echo $SHELL | grep -Eo 'bash|zsh')rc"
  sudo sed -z -i 's|HISTSIZE=1000\nHISTFILESIZE=2000\n|HISTSIZE=1000000\nHISTFILESIZE=2000000\nexport HISTFILE=~/.bash_history\n|g' /home/$userid/$term
  sudo sed -z -i "s|alias ll='ls -alF'\n|alias ll='ls -alFh'\n|g" /home/$userid/$term
  sudo sed -z -i "s|alias ll='ls -l'\n|alias ll='ls -alFh'\n|g" /home/$userid/$term
  sudo sed -z -i 's|HISTSIZE=1000\nSAVEHIST=2000\n|HISTSIZE=1000000\nSAVEHIST=2000000\n|g' /home/$userid/$term
  alreadythere="$(tail -n 1 /home/$userid/$term)"
  testv="#alreadydoneflag"
  if [ "$alreadythere" = "$testv" ]; then
    msg_ok "bashrc already updated"
  else
    cat << EOF >> /home/$userid/$term
fastfetch
alias update="sudo apt update && sudo apt upgrade -y && sudo snap refresh && sudo flatpak update"
alias cpu='watch -n 1 "cat /proc/cpuinfo | grep \"cpu MHz\" | awk '\''{printf \"%.2f GHz\\n\", \$NF/1000}'\'' " '
#alreadydoneflag
EOF
    sudo chown $userid:$userid /home/$userid/$term
    msg_ok "added update alias and terminal candy"
  fi
}

secondUser() {
  OPTIONS=( N "No, just for me"\
         Y "Create another user")

  seconduser=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Another User Option" --menu "Do you want to create another user?" 10 58 2 \
          "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
  case $seconduser in
    [yY]) another=$(whiptail --backtitle "Proxmox VE Helper Scripts" --inputbox "Set UserID" 8 58 --title "Please enter their ID: " 3>&1 1>&2 2>&3)
        passwordHandler
        msg_info "Creating $another user and shared vms folder in root along with vms group"
        grp="vms"
        ls / | grep -o "$grp" > /dev/null &&  msg_ok "folder $grp already created" || $STD sudo mkdir /$grp
        groups $userid | grep -o "$grp" > /dev/null &&  msg_ok "group $grp already created" || $STD sudo groupadd $grp && $STD sudo adduser $userid $grp
        cat /etc/passwd | grep -o "$another" > /dev/null && msg_ok "$another already created..." && return 0 || $STD sudo useradd -m -s /usr/bin/bash -G $grp $another
        enterhashhere=$(openssl passwd -1 "$PW1")
        echo "$another:$enterhashhere" | chpasswd -e
        PW1="just a really fantastic password"
        PW2="just a really fantastic password"
        sudo chown  root:vms /vms
        sudo chmod 2771 /vms
        msg_ok "$another created";;
    *) msg_ok "Skipping another user creation";;
esac
}

passwordHandler() {
    while true; do
    if PW1=$(whiptail --backtitle "Ubuntu Post-Install Script" --passwordbox "\nSet Password for $another" 9 58 --title "PASSWORD (leave blank for automatic login)" 3>&1 1>&2 2>&3); then
      if [[ ! -z "$PW1" ]]; then
        if [[ "$PW1" == *" "* ]]; then
          whiptail --msgbox "Password cannot contain spaces. Please try again." 8 58
        elif [ ${#PW1} -lt 5 ]; then
          whiptail --msgbox "Password must be at least 5 characters long. Please try again." 8 58
        else
          if PW2=$(whiptail --backtitle "Ubuntu Post-Install Script" --passwordbox "\nVerify $another's Password" 9 58 --title "PASSWORD VERIFICATION" 3>&1 1>&2 2>&3); then
            if [[ "$PW1" == "$PW2" ]]; then
              PW="-password $PW1"
              echo $PW1
              echo -e "${VERIFYPW}${BOLD}${DGN}$another Password: ${BGN}********${CL}"
              break
            else
              whiptail --msgbox "Passwords do not match. Please try again." 8 58
            fi
          else
            exit_script
          fi
        fi
      else
        PW1="Automatic Login"
        PW=""
        echo -e "${VERIFYPW}${BOLD}${DGN}Root Password: ${BGN}$PW1${CL}"
        break
      fi
    else
      exit_script
    fi
  done
}

vmwareSux() {
    vmwaredefault="https://softwareupdate.vmware.com/cds/vmw-desktop/ws/17.6.2/24409262/linux/core/VMware-Workstation-17.6.2-24409262.x86_64.bundle.tar"
    vmwareopt=$(echo $vmwaredefault | grep -oE "VMware-Workstation.*$")
    msg_info "Check here for latest version: https://softwareupdate.vmware.com/cds/vmw-desktop/ws/" && echo -e "\n" 
    read -p "Enter the FULL download url for the version you'd like ($vmwareopt) " vmwareDL 
    echo -e "\n" 
    vmwareFiledefault="VMware-Workstation-Full-17.6.3-24583834.x86_64.bundle"
    read -p  "Enter File Name here ($vmwareFiledefault): " vmwareFilea
    vmwareLink=${vmwareDL:-$vmwareFiledefault}
    vmwareBundle=${vmwareFilea:-$vmwareFiledefault}
    msg_info "downloading $vmwareBundle" && echo -e "\n" 
    vmwareV=$(echo $vmwareBundle | sed 's|VMware-Workstation-|VMware Workstation |g' | sed 's|.x86_64.bundle||g' | sed 's|-| build-|g')
    test -f /home/$userid/Downloads/$vmwareBundle&&msg_info "VMware already downloaded" && echo -e "\n" || wget -q $vmwareLink -P /home/$userid/Downloads 
    #msg_info "unzipping $vmwareFile" && echo -e "\n" && $STD sudo tar -xvf /home/$userid/Downloads/$vmwareFile -C /home/$userid/Downloads/  --not supporting tar due to VMware constantly changing shit
    $STD sudo chmod +x $vmwareBundle && $STD sudo chown $userid:$userid /home/$userid/Downloads/*
      #echo getting fixes
      #test -f /home/$userid/Downloads/workstation-$vmwareFIXversion.tar.gz&&echo 'VMware fix already downloaded' || wget https://github.com/mkubecek/vmware-host-modules/archive/workstation-$vmwareFIXversion.tar.gz -P /home/$userid/Downloads
      #cd /home/$userid/Downloads
      #tar -xzf workstation-$vmwareFIXversion.tar.gz;
      #cd vmware-host-modules-workstation-$vmwareFIXversion;
      #tar -cf vmmon.tar vmmon-only;
      #tar -cf vmnet.tar vmnet-only;
    vc="vmware"
    if which "$vc" > /dev/null 2>&1; then
        msg_ok "$vc already installed"
        echo -e "\n" 
    else
        msg_info "Installing $vmwareBundle"
        echo -e "\n" 
        if sudo bash "/home/$userid/Downloads/$vmwareBundle" --eulas-agreed --console --required; then
            msg_ok "$vmwareV installed"
            echo -e "\n" 
        else
            msg_error "Installation of $vmwareBundle failed"
            echo -e "\n" 
        fi
    fi
      #cp -v vmmon.tar vmnet.tar /usr/lib/vmware/modules/source/;
      #sudo vmware-modconfig --console --install-all;
    msg_info "if there are VMware service failures (vmmon vmnet) or anyother VMware issues, check if SecureBoot is enabled and visit; https://www.centennialsoftwaresolutions.com/post/ubuntu-20-04-3-lts-and-vmware-issues" && echo -e "\n"
}

rebootMenu() {
  OPTIONS=( N "No auto reboot"\
         Y "Auto reboot at end of script")

  rebootq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Reboot Option" --menu "Would you like to reboot at the end of this script?" 10 58 2 \
          "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

vmwareMenu() {
  OPTIONS=( N "No, KVM FTW"\
         Y "I'm a big dumb corpo who loves VMware")

  vmwareq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "VMware Install Option" --menu "Do you VMware like a dumb corpo?" 10 58 2 \
          "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

verboseMenu() {
  OPTIONS=( N "No"\
         Y "Yes! I want all the lines!!!")

  prompt=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Verbose Option" --menu "Would you like to run in verbose mode?" 10 58 2 \
          "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

chromeMenu() {
  OPTIONS=( N "No, skip Chrome" \
            Y "Yes, install Chrome")

  chromeq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Google Chrome Install" --menu "Do you want to install Google Chrome?" 10 58 2 \
            "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

kvmMenu() {
  OPTIONS=( N "No, skip PCIe passthrough" \
            Y "Yes, enable PCIe passthrough")

  kvmq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "PCIe Passthrough" --menu "Do you want to enable PCIe passthrough?" 10 58 2 \
           "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

nvidiaMenu() {
  OPTIONS=( N "No, skip Nvidia/CUDA" \
            Y "Yes, install Nvidia/CUDA for Hashcat")

  nvidiaq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Nvidia/CUDA Install" --menu "Do you want to install Nvidia/CUDA for Hashcat?" 10 58 2 \
            "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

flatMenu() {
  OPTIONS=( N "No, skip Flathub goodies" \
            Y "Yes, install Flathub goodies")

  flatq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Flathub Goodies Install" --menu "Do you want to install Flathub goodies?" 10 58 2 \
           "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

fingerprintMenu() {
  OPTIONS=( N "No, skip fingerprint in terminal" \
            Y "Yes, enable fingerprint in terminal")

  printq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Fingerprint in Terminal" --menu "Do you want to enable fingerprint in terminal?" 10 58 2 \
            "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}

flathubThemeMenu() {
  OPTIONS=( N "No, skip. Not sure this is even necessary" \
            Y "Yes, Try and match theme across flathub apps")

  flathubThemeq=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Match Flathub apps Theme" --menu "Do you want to match user theme across flathub apps?" 10 58 2 \
            "${OPTIONS[@]}" 3>&1 1>&2 2>&3)
}
  
allDone() {
  msg_ok "Ok, I think we're done!\n"
  case $rebootq in 
    [yY]) echo "install app icons taskbar by visiting the following link: https://extensions.gnome.org/extension/4944/app-icons-taskbar/"
        sleep 3
        sudo reboot
        return 1
        ;;
    *) msg_info "you should restart soonest"
        echo " install app icons taskbar by visiting the following link: https://extensions.gnome.org/extension/4944/app-icons-taskbar/"
        exit
        ;;
  esac
}

custom() {
  header_info
  verboseMenu
  if [[ ${prompt,,} =~ ^(y|yes)$ ]]; then
  STD=""
  else
  STD="silent"
  fi
  header_info

    rebootMenu
    vmwareMenu
    secondUser
    chromeMenu
    kvmMenu
    nvidiaMenu
    flatMenu
#    case $flatq in
#            [nN]) ;;
#            *) flathubThemeq;;
#    esac
    fingerprintMenu
    signalRepo
    fastFetchRepo
    letsUpdate
    apps
        case $chromeq in
                [nN]) msg_ok "Skipping Google Chrome install";;
                *) chromeInstall;;
        esac
        case $kvmq in
                [nN]) msg_ok "Skipping pcie passthrough";;
                *) lxcInstall;;
        esac
        case $vmwareq in
                [nN]) msg_ok "Skipping VMware Workstation install. Linux KVM FTW!";;
                *) vmwareSux;;
        esac
        case $nvidiaq in
                [nN]) msg_ok "Skipping Nvidia install";;
                *) nvidiaInstall;;
        esac
        case $flatq in
                [nN]) msg_ok "Skipping Flathub goodies install";;
                *) flatHub;;
        esac
#        case $flathubThemeq in
#                [nN]) msg_ok "Skipping Flathub goodies install";;
#                *) flatHub;;
#        esac
        case $printq in
                [nN]) msg_ok "Skipping fingerprint in terminal";;
                *) printInstall;;
        esac
    tmuxStuff
    allDone
}


if ! hostnamectl | grep -Eq "Operating System: Ubuntu 24\.[\.0-9]{2,}"; then
  echo -e "This version of Linux is not supported"
  echo -e "Requires Ubuntu 24.04 or higher"
  echo -e "Exiting..."
  sleep 2
  exit
fi

OPTIONS=(Full "Install (Google Chrome, Signal, nvidia, etc)" \
         Custom "Install")

CHOICE=$(whiptail --backtitle "Ubuntu Post-Install Script" --title "Install Packages" --menu "Select an option:" 10 58 2 \
          "${OPTIONS[@]}" 3>&1 1>&2 2>&3)

case $CHOICE in
  "Full")
    install
    ;;
  "Custom")
    custom
    ;;
  *)
    echo "Exiting..."
    exit 0
    ;;
esac