#!/bin/bash
set -e

if [ -f /etc/os-release ]; then
    # freedesktop.org and systemd
    . /etc/os-release
    OS=$NAME
    VER=$VERSION_ID
else
    OS="none"
    VER="0"
fi

if [ $OS != "Ubuntu" ]; then
    printf -- '\033[33m Ubuntu not detected\n\033[37m'
    exit
fi

if [ "$(dpkg -l | awk '/ubuntu-desktop/ {print }'|wc -l)" -ge 1 ]; then
    DESKTOP="yes" 
else
    DESKTOP="no" 
fi

if [ "$(systemd-detect-virt)" == "wsl" ]; then
    WSL="yes"
else
    WSL="no"
fi

# Ubuntu 20.04 workaround
printf -- '\033[33m Remove neovim if already installed\n\033[37m'
if [ "$(dpkg -l | awk '/nala/ {print }'|wc -l)" -ge 1 ]; then
    sudo nala purge -y --autoremove neovim
else
    sudo apt -y purge --auto-remove neovim
fi

# Install system packages
printf -- '\033[33m Installing system packages\n\033[37m'
if [ "$(dpkg -l | awk '/nala/ {print }'|wc -l)" -ge 1 ]; then
    sudo nala update
else
    sudo apt update
    sudo apt install -y nala
    sudo nala update 
fi
sudo nala upgrade -y
sudo nala install -y gpg ninja-build gettext cmake unzip curl python3 python3-pip python3-venv \
    build-essential libssl-dev libffi-dev python3-dev file tmux libudev-dev pkg-config locales pipx btop ncdu ranger 

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

# 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 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
chmod +x ./vhclientx86_64
sudo mv ./vhclientx86_64 /usr/sbin
echo "$USER ALL=(ALL:ALL) NOPASSWD: /usr/sbin/vhclientx86_64" | sudo tee /etc/sudoers.d/$USER+vhclientx86_64
# 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 service udev restart
    sudo udevadm control --reload-rules
    sudo udevadm trigger
fi

# Install neovim
printf -- '\033[33m Installing neovim\n\033[37m'
cd ~
if [ -d ~/neovim ]; then rm -rf ~/neovim; fi
git clone https://github.com/neovim/neovim
cd neovim && git checkout stable
make CMAKE_BUILD_TYPE=RelWithDebInfo
cd build && cpack -G DEB && sudo dpkg -i nvim-linux64.deb
if [ -d ~/neovim ]; then rm -rf ~/neovim; fi

# Install rust
printf -- '\033[33m Installing Rust\n\033[37m'
cd ~
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
source "$HOME/.cargo/env"
rustup self update
rustup update 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

# 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 fd
printf -- '\033[33m Installing fd\n\033[37m'
sudo nala install -y fd-find

# 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

# Install gdu
printf -- '\033[33m Installing gdu\n\033[37m'
if [ $VER = "22.04" ]; then
    sudo add-apt-repository -y ppa:daniel-milde/gdu
fi
sudo nala update
sudo nala install -y gdu

# Install ruby
printf -- '\033[33m Installing ruby\n\033[37m'
sudo nala install -y ruby ruby-dev
sudo gem install neovim

# Install bottom
printf -- '\033[33m Installing bottom\n\033[37m'
cd ~
curl -LO https://github.com/ClementTsang/bottom/releases/download/0.9.1/bottom_0.9.1_amd64.deb
sudo dpkg -i bottom_0.9.1_amd64.deb
rm ~/bottom_0.9.1_amd64.deb

# Install go
printf -- '\033[33m Installing go\n\033[37m'
sudo nala install -y golang

# Install php
printf -- '\033[33m Installing php\n\033[37m'
sudo nala install -y php php-curl php-xml php-mbstring

# Install luarocks
printf -- '\033[33m Installing luarocks\n\033[37m'
sudo nala install -y luarocks

# Install composer
printf -- '\033[33m Installing composer\n\033[37m'
sudo nala install -y composer

# Install java
printf -- '\033[33m Installing java\n\033[37m'
sudo nala install -y default-jdk

# Install julia
printf -- '\033[33m Installing julia\n\033[37m'
wget https://julialang-s3.julialang.org/bin/linux/x64/1.9/julia-1.9.0-linux-x86_64.tar.gz
tar -xvzf julia-1.9.0-linux-x86_64.tar.gz
sudo cp -r julia-1.9.0 /opt/
sudo ln -sf /opt/julia-1.9.0/bin/julia /usr/local/bin/julia
rm julia-1.9.0-linux-x86_64.tar.gz*
rm -rf julia-1.9.0

# Install node
printf -- '\033[33m Installing node\n\033[37m'
cd ~
curl -sL https://deb.nodesource.com/setup_18.x | sudo -E bash -
sudo nala install -y nodejs

# Install platformio
printf -- '\033[33m Installing platformio\n\033[37m'
if [ $VER == "24.04" ]; then
    pipx install platformio
    pipx ensurepath
else
    pip3 install platformio
fi

# Install pynvim
printf -- '\033[33m Installing pynvim\n\033[37m'
if [ $VER == "24.04" ]; then
    sudo nala install -y python3-pynvim
else
    pip3 install pynvim
fi

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

if [ $DESKTOP == "yes" ]; then
    # Install flatpak 
    printf -- '\033[33m Installing desktop dep.\n\033[37m'
    sudo add-apt-repository -y ppa:flatpak/stable
    sudo nala update
    sudo nala install -y flatpak 
    flatpak remote-add --if-not-exists flathub https://flathub.org/repo/flathub.flatpakrepo

    printf -- '\033[33m Installing Visual Code\n\033[37m'
    sudo snap install --classic code

    printf -- '\033[33m Installing Discord\n\033[37m'
    flatpak install -y flathub com.discordapp.Discord

    printf -- '\033[33m Installing KiCad\n\033[37m'
    flatpak install -y flathub org.kicad.KiCad

    printf -- '\033[33m Installing FreeCad\n\033[37m'
    flatpak install -y flathub org.freecadweb.FreeCAD

    printf -- '\033[33m Installing Moonlight\n\033[37m'
    flatpak install -y flathub com.moonlight_stream.Moonlight

    printf -- '\033[33m Installing DrawIo\n\033[37m'
    flatpak install -y flathub com.jgraph.drawio.desktop

    printf -- '\033[33m Installing StreamController\n\033[37m'
    flatpak install -y flathub com.core447.StreamController
    sudo sh -c "echo -e 'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"0060\", TAG+=\"uaccess\"\\n'\
'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"0063\", TAG+=\"uaccess\"\\n'\
'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"006c\", TAG+=\"uaccess\"\\n'\
'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"006d\", TAG+=\"uaccess\"\\n'\
'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"0080\", TAG+=\"uaccess\"\\n'\
'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"0084\", TAG+=\"uaccess\"\\n'\
'SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"0fd9\", ATTRS{idProduct}==\"008f\", TAG+=\"uaccess\"\\n'\
'KERNEL==\"uinput\", SUBSYSTEM==\"misc\", OPTIONS+=\"static_node=uinput\", TAG+=\"uaccess\", GROUP=\"input\", MODE=\"0660\"' > /etc/udev/rules.d/60-streamdeck.rules"
    sudo udevadm trigger

    # Install gnome-shell-extension-manager
    printf -- '\033[33m Installing gnome-shell-extension-manager\n\033[37m'
    sudo nala install -y gnome-shell-extension-manager 
    
    # Install remmina 
    printf -- '\033[33m Installing remmina\n\033[37m'
    sudo nala install -y remmina 
    
    # 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 Wezterm 
    printf -- '\033[33m Installing wezterm\n\033[37m'
    curl -fsSL https://apt.fury.io/wez/gpg.key | sudo gpg --yes --dearmor -o /usr/share/keyrings/wezterm-fury.gpg
    echo 'deb [signed-by=/usr/share/keyrings/wezterm-fury.gpg] https://apt.fury.io/wez/ * *' | sudo tee /etc/apt/sources.list.d/wezterm.list
    sudo nala update
    sudo nala install -y wezterm

    # Symlink wezterm config
    printf -- '\033[33m Symlinking wezterm config\n\033[37m'
    ln -sf ~/code_server/wezterm.lua ~/.wezterm.lua

    # Install Virtualbox 
    if [ "$(dpkg -l | awk '/virtualbox/ {print }'|wc -l)" -eq 0 ]; then
        printf -- '\033[33m Installing Virtualbox\n\033[37m'
        curl https://www.virtualbox.org/download/oracle_vbox_2016.asc | gpg --dearmor > oracle_vbox_2016.gpg
        curl https://www.virtualbox.org/download/oracle_vbox.asc | gpg --dearmor > oracle_vbox.gpg
        sudo install -o root -g root -m 644 oracle_vbox_2016.gpg /etc/apt/trusted.gpg.d/
        sudo install -o root -g root -m 644 oracle_vbox.gpg /etc/apt/trusted.gpg.d/
        echo "deb [arch=amd64] http://download.virtualbox.org/virtualbox/debian $(lsb_release -sc) contrib" | sudo tee /etc/apt/sources.list.d/virtualbox.list
        sudo nala update
        sudo nala install -y linux-headers-$(uname -r) dkms
        sudo nala install virtualbox-7.0 -y
        wget https://download.virtualbox.org/virtualbox/7.0.0/Oracle_VM_VirtualBox_Extension_Pack-7.0.0.vbox-extpack
        sudo VBoxManage extpack install --replace Oracle_VM_VirtualBox_Extension_Pack-7.0.0.vbox-extpack --accept-license=33d7284dc4a0ece381196fda3cfe2ed0e1e8e7ed7f27b9a9ebc4ee22e24bd23c
        sudo usermod -a -G vboxusers $USER
    fi
fi

# Symlink nvim config
printf -- '\033[33m Symlinking nvim config\n\033[37m'
mkdir -p ~/.config/nvim
ln -sf ~/code_server/config/nvim/init.lua ~/.config/nvim/init.lua

# Symlink git config
printf -- '\033[33m Symlinking git config\n\033[37m'
ln -sf ~/code_server/gitconfig ~/.gitconfig

# Symlink tmux config
printf -- '\033[33m Symlinking tmux config\n\033[37m'
ln -sf ~/code_server/tmux.conf ~/.tmux.conf

# Symlink clang-format config
printf -- '\033[33m Symlinking clang-format config\n\033[37m'
ln -sf ~/code_server/clang-format ~/.clang-format

# Symlink lazygit config
printf -- '\033[33m Symlinking lazygit config\n\033[37m'
mkdir -p ~/.config/lazygit
ln -sf ~/code_server/config/lazygit/config.yml ~/.config/lazygit/config.yml

# Symlink luacheck config
printf -- '\033[33m Symlinking luacheck config\n\033[37m'
ln -sf ~/code_server/luacheckrc ~/.luacheckrc

# Symlink rp2040 flash script
printf -- '\033[33m Symlinking rp2040 flash script\n\033[37m'
mkdir -p ~/.local/bin
ln -sf ~/code_server/pico-load.sh ~/.local/bin/pico-load

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

# Symlink start script
printf -- '\033[33m Symlinking start script\n\033[37m'
ln -sf ~/code_server/start.sh ~/.start

# Restore default .bashrc
printf -- '\033[33m Restoring default .bashrc\n\033[37m'
sudo /bin/cp /etc/skel/.bashrc ~/.bashrc

# Add path to bashrc
printf -- '\033[33m Adding path to bashrc\n\033[37m'
echo 'PATH="$HOME/.local/bin:$HOME/.cargo/bin:$PATH"' >> ~/.bashrc
echo 'alias vi="nvim"' >> ~/.bashrc
echo 'alias vim="nvim"' >> ~/.bashrc
echo 'alias ls="eza -1 -l --icons=auto -h -g -o --no-permissions -a"' >> ~/.bashrc
echo 'alias lsg="eza -1 -l --icons=auto --git-repos --git -h -g -o --no-permissions -a"' >> ~/.bashrc
echo 'alias lss="eza -1 -l --icons=auto -h -g --total-size -o --no-permissions -a"' >> ~/.bashrc
echo 'source ~/.fzf.bash' >> ~/.bashrc
echo 'eval "$(zoxide init bash)"' >> ~/.bashrc

if [ $DESKTOP == "no" ] && [ $WSL == "no" ]; then
    # Add tmux/nvim autostart to bashrc
    printf -- '\033[33m Adding tmux/nvim autostart to bashrc\n\033[37m'
    echo "# TMUX" >> ~/.bashrc
    echo "if [[ ! \$TERM =~ screen ]]; then" >> ~/.bashrc
    echo "if which tmux >/dev/null 2>&1; then" >> ~/.bashrc
    echo '   test -z "$TMUX" && (tmux attach || tmux new-session ~/.start)' >> ~/.bashrc
    echo "fi" >> ~/.bashrc
    echo "fi" >> ~/.bashrc
fi

if [ $DESKTOP == "yes" ] || [ $WSL == "yes" ]; then
    # Add tmux/nvim autostart to bashrc
    printf -- '\033[33m Adding tmux/nvim autostart to bashrc\n\033[37m'
    echo "# TMUX" >> ~/.bashrc
    echo "if [[ ! \$TERM =~ screen ]]; then" >> ~/.bashrc
    echo "if which tmux >/dev/null 2>&1; then" >> ~/.bashrc
    echo '   tmux new-session' >> ~/.bashrc
    echo "fi" >> ~/.bashrc
    echo "fi" >> ~/.bashrc
fi

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