#!/bin/bash
set -e
if ! pacman -Qs inetutils >/dev/null; then
    sudo pacman -Syy --noconfirm inetutils
fi
if ! pacman -Qs awk >/dev/null; then
    sudo pacman -Syy --noconfirm awk
fi

HOSTNAME=$(hostname | awk '{ print toupper($0) }')

echo -e ' \e[H\e[2J
           \e[0;34m.
          \e[0;34m/ \
         \e[0;34m/   \      \e[1;37m               #     \e[1;34m| *
        \e[0;34m/^.   \     \e[1;37m a##e #%" a#"e 6##%  \e[1;34m| | |-^-. |   | \ /
       \e[0;34m/  .-.  \    \e[1;37m.oOo# #   #    #  #  \e[1;34m| | |   | |   |  X
      \e[0;34m/  (   ) _\   \e[1;37m%OoO# #   %#e" #  #  \e[1;34m| | |   | ^._.| / \ \e[0;37mTM
     \e[1;34m/ _.~   ~._^\
    \e[1;34m/.^         ^.\ \e[0;37mTM


    \e[1;32mCMtec '$HOSTNAME' install/Update script\e[0;37m
'

# Create backup/snapshot
if pacman -Qs timeshift >/dev/null; then
    CONFIG_FILE="/etc/timeshift/timeshift.json"
    if [ -f "$CONFIG_FILE" ]; then
        BTRFS_MODE=$(jq '.btrfs_mode' "$CONFIG_FILE")
        if [ "$BTRFS_MODE" == '"true"' ]; then
            printf -- '\033[33m Creating backup/snapshot\n\033[37m'
            sudo timeshift --create --comments "Update script"
        fi
    fi
fi

# Enable multilib (if applicable)
if [ "$HOSTNAME" == "STEAMBOX" ]; then
    if ! grep -q "^\[multilib\]" /etc/pacman.conf; then
        printf -- '\033[33m Enabling multilib\n\033[37m'
        sudo tee -a /etc/pacman.conf >/dev/null <<EOT

[multilib]
Include = /etc/pacman.d/mirrorlist
EOT
    fi
fi

# Symlink bashrc
printf -- '\033[33m Symlinking bashrc config\n\033[37m'
ln -sf ~/linuxbox/bashrc ~/.bashrc

# Update all pacman packages
printf -- '\033[33m Updating pacman packages\n\033[37m'
sudo pacman -Suyy --noconfirm

# Install drivers (if applicable)
if [ "$HOSTNAME" == "STEAMBOX" ]; then
    # Install nvidia drivers
    printf -- '\033[33m Installing NVIDIA drivers\n\033[37m'
    sudo pacman --noconfirm --needed -S base-devel linux-headers git
    sudo pacman --noconfirm --needed -S nvidia-open-dkms nvidia-utils lib32-nvidia-utils nvidia-settings opencl-nvidia

fi

# Update all AUR packages
printf -- '\033[33m Updating AUR packages\n\033[37m'
if pacman -Qs yay >/dev/null; then
    yay --noconfirm --aur
else
    sudo pacman -S --needed --noconfirm git base-devel
    if [ -d ~/yay-bin ]; then rm -rf ~/yay-bin; fi
    cd ~
    git clone https://aur.archlinux.org/yay-bin.git
    cd yay-bin
    makepkg --noconfirm -si
    cd ~
    if [ -d ~/yay-bin ]; then rm -rf ~/yay-bin; fi
fi
# Update all Flatpak packages
printf -- '\033[33m Updating Flatpak packages\n\033[37m'
if pacman -Qs flatpak >/dev/null; then
    flatpak update -y
else
    sudo pacman --noconfirm --needed -S flatpak
    flatpak remote-add --if-not-exists flathub https://flathub.org/repo/flathub.flatpakrepo
    flatpak install -y flathub com.github.tchx84.Flatseal
fi

# Install initial system packages
printf -- '\033[33m Installing initial system packages\n\033[37m'
sudo pacman --noconfirm --needed -S base-devel cmake gawk wget gettext unzip curl inetutils

# Setup locals
printf -- '\033[33m Setup locals\n\033[37m'
sudo localectl set-locale LANG=en_US.UTF-8

# Install git
printf -- '\033[33m Installing git\n\033[37m'
sudo pacman --noconfirm --needed -S git
ln -sf ~/linuxbox/gitconfig ~/.gitconfig

# Install hyprland
printf -- '\033[33m Installing hyprland\n\033[37m'
sudo pacman --noconfirm --needed -S usbutils plymouth dracut dunst hypridle hyprland hyprlock xdg-desktop-portal-hyprland polkit-gnome xorg-xhost gnome-keyring qt5ct qt6ct gnome-themes-extra qt5-wayland qt6-wayland lxappearance

yay --noconfirm -S --needed --aur adwaita-qt5-git
yay --noconfirm -S --needed --aur adwaita-qt6-git
yay --noconfirm -S --needed --aur wlogout
yay --noconfirm -S --needed --aur hyprshot

mkdir -p ~/.local/bin
mkdir -p ~/.config/hypr
mkdir -p ~/.config/gtk-3.0
mkdir -p ~/.config/qt5ct
mkdir -p ~/.config/qt6ct
mkdir -p ~/.config/dunst

if [ -f "~/linuxbox/config/hypr/hyprland_$HOSTNAME.conf" ]; then
    ln -sf ~/linuxbox/config/hypr/hyprland_$HOSTNAME.conf ~/.config/hypr/hyprland.conf
else
    ln -sf ~/linuxbox/config/hypr/hyprland_CMBOX.conf ~/.config/hypr/hyprland.conf
fi

ln -sf ~/linuxbox/wrappedhl ~/.local/bin/wrappedhl
ln -sf ~/linuxbox/config/hypr/hypridle.conf ~/.config/hypr/hypridle.conf
ln -sf ~/linuxbox/config/gtk-3.0/settings.ini ~/.config/gtk-3.0/settings.ini
ln -sf ~/linuxbox/config/qt5ct/qt5ct.conf ~/.config/qt5ct/qt5ct.conf
ln -sf ~/linuxbox/config/qt6ct/qt6ct.conf ~/.config/qt6ct/qt6ct.conf
ln -sf ~/linuxbox/config/dunst/dunstrc ~/.config/dunst/dunstrc

# Install waybar
printf -- '\033[33m Installing waybar\n\033[37m'
sudo pacman --noconfirm --needed -S waybar
mkdir -p ~/.config/waybar
ln -sf ~/linuxbox/config/waybar/config ~/.config/waybar/config
ln -sf ~/linuxbox/config/waybar/style.css ~/.config/waybar/style.css

# Install Audio
printf -- '\033[33m Installing Audio\n\033[37m'
sudo pacman --noconfirm --needed -S pipewire pipewire-pulse pipewire-alsa pavucontrol helvum
mkdir -p ~/.config/pipewire
ln -sf ~/linuxbox/config/pipewire/pipewire.conf ~/.config/pipewire/pipewire.conf

# Install networkmanager
printf -- '\033[33m Installing networkmanager\n\033[37m'
sudo pacman --noconfirm --needed -S network-manager-applet
yay --noconfirm -S --needed --aur networkmanager-dmenu-git
mkdir -p ~/.config/networkmanager-dmenu
ln -sf ~/linuxbox/config/networkmanager-dmenu/config.ini ~/.config/networkmanager-dmenu/config.ini

# Install autologin
printf -- '\033[33m Installing autologin\n\033[37m'
yay --noconfirm -S --needed --aur pam_autologin
sudo /bin/cp -rf ~/linuxbox/login /etc/pam.d/login
sudo /bin/cp -rf ~/linuxbox/getty@.service /usr/lib/systemd/system/getty@.service
sudo touch /etc/security/autologin.conf

# Install udisk
printf -- '\033[33m Installing udisk\n\033[37m'
sudo pacman --noconfirm --needed -S udisks2
echo 'ENV{ID_FS_USAGE}=="filesystem|other|crypto", ENV{UDISKS_FILESYSTEM_SHARED}="1"' | sudo tee /etc/udev/rules.d/99-udisks2.rules
echo 'D /media 0755 root root 0 -' | sudo tee /etc/tmpfiles.d/media.conf

# Install btop
printf -- '\033[33m Installing btop\n\033[37m'
sudo pacman --noconfirm --needed -S btop

# Install ranger
printf -- '\033[33m Installing ranger\n\033[37m'
sudo pacman --noconfirm --needed -S ranger

# Install kitty
printf -- '\033[33m Installing kitty\n\033[37m'
sudo pacman --noconfirm --needed -S kitty
mkdir -p ~/.config/kitty
ln -sf ~/linuxbox/config/kitty/kitty.conf ~/.config/kitty/kitty.conf

# Install tmux
printf -- '\033[33m Installing tmux\n\033[37m'
sudo pacman --noconfirm --needed -S tmux
ln -sf ~/linuxbox/tmux.conf ~/.tmux.conf

# Install python
printf -- '\033[33m Installing python\n\033[37m'
sudo pacman --noconfirm --needed -S python python-pip python-pipx python-pipenv python-pynvim

# Install fzf
printf -- '\033[33m Installing fzf\n\033[37m'
rm -rf ~/.fzf
git clone --depth 1 https://github.com/junegunn/fzf.git ~/.fzf
~/.fzf/install --all
source ~/.fzf.bash

# Install fzfmenu
printf -- '\033[33m Installing fzfmenu\n\033[37m'
mkdir -p ~/.local/bin
ln -sf ~/linuxbox/fzfmenu.sh ~/.local/bin/fzfmenu

# # Install VirtualHere server
# curl https://raw.githubusercontent.com/virtualhere/script/main/install_server | sudo sh

# Install VirtualHere client
printf -- '\033[33m Installing VirtualHere client\n\033[37m'
cd ~
wget https://www.virtualhere.com/sites/default/files/usbclient/scripts/virtualhereclient.service
wget https://www.virtualhere.com/sites/default/files/usbclient/vhclientx86_64
wget https://www.virtualhere.com/sites/default/files/usbclient/vhuit64
chmod +x ./vhclientx86_64
chmod +x ./vhuit64
sudo mv ./vhclientx86_64 /usr/sbin
sudo mv ./vhuit64 /usr/sbin
echo "$USER ALL=(ALL:ALL) NOPASSWD: /usr/sbin/vhclientx86_64" | sudo tee /etc/sudoers.d/$USER+vhclientx86_64
echo "$USER ALL=(ALL:ALL) NOPASSWD: /usr/sbin/vhuit64" | sudo tee /etc/sudoers.d/$USER+vhuit64
sudo mv virtualhereclient.service /etc/systemd/system/virtualhereclient.service
sudo systemctl daemon-reload
sudo systemctl enable virtualhereclient.service
sudo systemctl start virtualhereclient.service

# Install teensy udev rules (only for non docker system)
if [ ! -f /.dockerenv ]; then
    printf -- '\033[33m Installing teensy udev rules\n\033[37m'
    sudo rm -f /tmp/00-teensy.rules /etc/udev/rules.d/00-teensy.rules /lib/udev/rules.d/00-teensy.rules
    sudo wget -O /tmp/00-teensy.rules https://www.pjrc.com/teensy/00-teensy.rules
    sudo install -o root -g root -m 0664 /tmp/00-teensy.rules /lib/udev/rules.d/00-teensy.rules
    sudo udevadm control --reload-rules
    sudo udevadm trigger
fi

# Install neovim
printf -- '\033[33m Installing neovim\n\033[37m'
sudo pacman --noconfirm --needed -S neovim
mkdir -p ~/.config/nvim
ln -sf ~/linuxbox/config/nvim/init.lua ~/.config/nvim/init.lua
ln -sf ~/linuxbox/clang-format ~/.clang-format
ln -sf ~/linuxbox/luacheckrc ~/.luacheckrc
mkdir -p ~/.local/bin
ln -sf ~/linuxbox/pico-load.sh ~/.local/bin/pico-load

# Install fd
printf -- '\033[33m Installing fd\n\033[37m'
sudo pacman --noconfirm --needed -S fd

# Install lazygit
printf -- '\033[33m Installing lazygit\n\033[37m'
cd ~
LAZYGIT_VERSION=$(curl -s "https://api.github.com/repos/jesseduffield/lazygit/releases/latest" | grep -Po '"tag_name": "v\K[^"]*')
curl -Lo lazygit.tar.gz "https://github.com/jesseduffield/lazygit/releases/latest/download/lazygit_${LAZYGIT_VERSION}_Linux_x86_64.tar.gz"
tar xf lazygit.tar.gz lazygit
sudo install lazygit /usr/local/bin
rm ~/lazygit && rm ~/lazygit.tar.gz
mkdir -p ~/.config/lazygit
ln -sf ~/linuxbox/config/lazygit/config.yml ~/.config/lazygit/config.yml

# Install gdu
printf -- '\033[33m Installing gdu\n\033[37m'
sudo pacman --noconfirm --needed -S gdu

# Install ruby
printf -- '\033[33m Installing ruby\n\033[37m'
sudo pacman --noconfirm --needed -S ruby

# Install bottom
printf -- '\033[33m Installing bottom\n\033[37m'
sudo pacman --noconfirm --needed -S bottom

# Install go
printf -- '\033[33m Installing go\n\033[37m'
sudo pacman --noconfirm --needed -S go

# Install php
printf -- '\033[33m Installing php\n\033[37m'
sudo pacman --noconfirm --needed -S php

# Install luarocks
printf -- '\033[33m Installing luarocks\n\033[37m'
sudo pacman --noconfirm --needed -S luarocks

# Install composer
printf -- '\033[33m Installing composer\n\033[37m'
sudo pacman --noconfirm --needed -S composer

# Install java
printf -- '\033[33m Installing java\n\033[37m'
sudo pacman --noconfirm --needed -S jdk-openjdk

# Install julia
printf -- '\033[33m Installing julia\n\033[37m'
sudo pacman --noconfirm --needed -S julia

# Install node
printf -- '\033[33m Installing node\n\033[37m'
sudo pacman --noconfirm --needed -S nodejs npm

# Install ble.sh
printf -- '\033[33m Installing ble.sh\n\033[37m'
if [ -d ~/ble.sh ]; then rm -rf ~/ble.sh; fi
cd ~
git clone --recursive --depth 1 --shallow-submodules https://github.com/akinomyoga/ble.sh.git
make -C ble.sh install
if [ -d ~/ble.sh ]; then rm -rf ~/ble.sh; fi
ln -sf ~/linuxbox/blerc ~/.blerc

# Install platformio
printf -- '\033[33m Installing platformio\n\033[37m'
sudo pacman --noconfirm --needed -S platformio-core

# Install neovim for node
printf -- '\033[33m Installing neovim for node\n\033[37m'
sudo npm install -g neovim

# Install rust
printf -- '\033[33m Installing Rust\n\033[37m'
sudo pacman --noconfirm --needed -S rustup
rustup update stable
rustup default stable

# Install rust embedded stuff
printf -- '\033[33m Installing Rust embedded stuff\n\033[37m'
cd ~
rustup target add thumbv6m-none-eabi
cargo install elf2uf2-rs --locked
cargo install probe-run
cargo install flip-link

# Install tree sitter cli
printf -- '\033[33m Installing tree sitter cli\n\033[37m'
cd ~
cargo install tree-sitter-cli

# Install ripgrep
printf -- '\033[33m Installing ripgrep\n\033[37m'
cargo install ripgrep

# Install eza
printf -- '\033[33m Installing eza\n\033[37m'
cargo install eza

# Install zoxide
printf -- '\033[33m Installing zoxide\n\033[37m'
cargo install zoxide --locked

# Install starship
printf -- '\033[33m Installing starship\n\033[37m'
cargo install starship --locked
ln -sf ~/linuxbox/config/starship.toml ~/.config/starship.toml

# Install Discord (Equibop)
printf -- '\033[33m Installing Discord (Equibop)\n\033[37m'
flatpak install -y flathub io.github.equicord.equibop

# Install XAir
printf -- '\033[33m Installing XAir\n\033[37m'
flatpak install -y flathub com.behringer.XAirEdit

# Install Krita
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing Krita\n\033[37m'
    flatpak install -y flathub org.kde.krita
fi

# Install KiCad
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing KiCad\n\033[37m'
    sudo pacman --noconfirm --needed -S kicad
fi

# Install FreeCad
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing FreeCad\n\033[37m'
    sudo pacman --noconfirm --needed -S freecad
fi

# Install Prusaslicer
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing Prusaslicer\n\033[37m'
    flatpak install -y flathub com.prusa3d.PrusaSlicer
fi

# Install Moonlight
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing Moonlight\n\033[37m'
    flatpak install -y flathub com.moonlight_stream.Moonlight
fi

# Install DrawIo
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing DrawIo\n\033[37m'
    flatpak install -y flathub com.jgraph.drawio.desktop
fi

# Install Gimp
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing Gimp\n\033[37m'
    flatpak install -y flathub org.gimp.GIMP
fi

# Install VSCode
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing VSCode\n\033[37m'
    sudo pacman --noconfirm --needed -S code
fi

# Install remmina
if [ "$HOSTNAME" == "CMBOX" ]; then
    printf -- '\033[33m Installing remmina\n\033[37m'
    flatpak install -y flathub org.remmina.Remmina
fi

# Install feh
printf -- '\033[33m Installing feh\n\033[37m'
sudo pacman --noconfirm --needed -S feh

# Install NerdFonts
printf -- '\033[33m Installing NerdFonts\n\033[37m'
mkdir -p /home/$USER/.local/share/fonts
cd /home/$USER/.local/share/fonts
wget https://github.com/ryanoasis/nerd-fonts/releases/download/v3.2.1/CodeNewRoman.zip
unzip -o CodeNewRoman.zip
rm CodeNewRoman.zip
wget https://github.com/ryanoasis/nerd-fonts/releases/download/v3.2.1/NerdFontsSymbolsOnly.zip
unzip -o NerdFontsSymbolsOnly.zip
rm NerdFontsSymbolsOnly.zip
fc-cache -fv

# Install other fonts
printf -- '\033[33m Installing other fonts\n\033[37m'
sudo pacman --noconfirm --needed -S noto-fonts poppler-data adobe-source-code-pro-fonts

# Install Brave Browser
printf -- '\033[33m Installing Brave Browser\n\033[37m'
yay --noconfirm -S --needed --aur brave-bin

# Install timeshift
printf -- '\033[33m Installing Timeshift\n\033[37m'
sudo pacman --noconfirm --needed -S timeshift

# Install sunshine
printf -- '\033[33m Installing Sunshine\n\033[37m'
sudo pacman --noconfirm --needed -S intel-media-driver
yay --noconfirm --needed -S sunshine
echo 'KERNEL=="uinput", SUBSYSTEM=="misc", OPTIONS+="static_node=uinput", TAG+="uaccess"' | sudo tee /etc/udev/rules.d/60-sunshine.rules
mkdir -p ~/.config/systemd/user
cp -f ~/linuxbox/config/systemd/user/sunshine.service ~/.config/systemd/user/sunshine.service
sudo udevadm control --reload-rules
sudo udevadm trigger
sudo setcap cap_sys_admin+p $(readlink -f $(which sunshine))
sudo systemctl daemon-reload
systemctl --user start sunshine

# Install steam
if [ "$HOSTNAME" == "STEAMBOX" ]; then
    printf -- '\033[33m Installing steam\n\033[37m'
    sudo pacman -S --noconfirm --needed steam
    yay --noconfirm -S --needed --aur protonup-qt
fi

# Install starcitizen
if [ "$HOSTNAME" == "STEAMBOX" ]; then
    printf -- '\033[33m Installing starcitizen\n\033[37m'
    sudo pacman -S --noconfirm --needed wine-staging winetricks wine-mono wine-gecko
    yay --noconfirm -S --needed --aur lug-helper
fi

# # Install game services
# if [ "$HOSTNAME" == "STEAMBOX" ]; then
#     printf -- '\033[33m Installing Game services\n\033[37m'
#     # sudo systemctl enable fstrim.timer
#     sudo pacman -S --noconfirm --needed steam lutris wine-staging winetricks gamemode lib32-gamemode giflib lib32-giflib libpng lib32-libpng libldap lib32-libldap gnutls lib32-gnutls mpg123 lib32-mpg123 openal lib32-openal \
#         v4l-utils lib32-v4l-utils libgpg-error lib32-libgpg-error alsa-plugins lib32-alsa-plugins alsa-lib lib32-alsa-lib libjpeg-turbo lib32-libjpeg-turbo sqlite lib32-sqlite libxcomposite lib32-libxcomposite libxinerama \
#         lib32-libxinerama ncurses lib32-ncurses opencl-icd-loader lib32-opencl-icd-loader libxslt lib32-libxslt libva lib32-libva gtk3 lib32-gtk3 gst-plugins-base-libs lib32-gst-plugins-base-libs vulkan-icd-loader \
#         lib32-vulkan-icd-loader mangohud lib32-mangohud goverlay gamescope solaar lib32-libpulse
#     yay --noconfirm -S --needed --aur vkbasalt lib32-vkbasalt proton-ge-custom-bin xone-dkms-git dxvk-bin vkd3d-proton-bin
# fi

# Symlink update script
printf -- '\033[33m Symlinking update wrapper\n\033[37m'
mkdir -p ~/.local/bin
ln -sf ~/linuxbox/update_wrapper.sh ~/.local/bin/update

printf -- '\033[32m \n\n***** Update complete! Please reboot. *****\n\n\033[37m'