#!/bin/bash
set -e

# Install system packages
printf -- '\033[33m Installing system packages\n\033[37m' 
sudo apt update && sudo apt -y upgrade 
sudo apt -y install ninja-build gettext cmake unzip curl python3 python3-pip python3-venv \
    build-essential libssl-dev libffi-dev python3-dev file tmux libudev-dev

# 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 
# 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
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 

# 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' 
sudo apt install -y ripgrep 

# Install fd
printf -- '\033[33m Installing fd\n\033[37m' 
sudo apt 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' 
sudo add-apt-repository -y ppa:daniel-milde/gdu 
sudo apt update 
sudo apt install -y gdu 

# Install ruby
printf -- '\033[33m Installing ruby\n\033[37m' 
sudo apt 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 apt install -y golang 

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

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

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

# Install java
printf -- '\033[33m Installing java\n\033[37m' 
sudo apt 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 apt install -y nodejs 

# Install platformio
printf -- '\033[33m Installing platformio\n\033[37m' 
pip3 install platformio 

# Install pynvim
printf -- '\033[33m Installing pynvim\n\033[37m' 
pip3 install pynvim 

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

# Symlink nvim config 
printf -- '\033[33m Symlinking nvim config\n\033[37m' 
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' 
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 update script
printf -- '\033[33m Symlinking update script\n\033[37m' 
ln -sf ~/code_server/update_wrapper.sh ~/.local/bin/update 

# 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 

# 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)' >> ~/.bashrc 
echo "fi" >> ~/.bashrc 
echo "fi" >> ~/.bashrc 

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