#!/bin/bash

set -euo pipefail

SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
source "$SCRIPT_DIR/common.sh"

ubuntu::print_banner() {
    local version="$1"
    local desktop="$2"
    local wsl="$3"
    printf -- ' \e[H\e[2J\n \e[0;31m                .-.\e[0;37m\n \e[0;31m          .-"``(|||)\e[0;37m\n \e[0;31m       ,`\\ \\    `-`.\e[0;37m               88                         88\n \e[0;31m      /   \\ "``-.   `\e[0;37m              88                         88\n \e[0;31m    .-.  ,       `___:\e[0;37m    88   88  88,888,  88   88  ,88888, 88888  88   88\n \e[0;31m   (:::) :        ___ \e[0;37m    88   88  88   88  88   88  88   88  88    88   88\n \e[0;31m    `-`  `       ,   :\e[0;37m    88   88  88   88  88   88  88   88  88    88   88\n \e[0;31m      \\   / ,..-`   ,\e[0;37m     88   88  88   88  88   88  88   88  88    88   88\n \e[0;31m       `./ /    .-.`\e[0;37m      "88888"  "88888"  "88888"  88   88  "8888 "88888"\n \e[0;31m         `-..-(   )\e[0;37m\n \e[0;31m               `-`\e[0;37m\n\n\n    \e[1;32mCMtec Ubuntu install/Update script\e[0;37m\n    \e[0;35mUbuntu version = '%s'\e[0;37m\n    \e[0;35mDesktop = '%s'\e[0;37m\n    \e[0;35mWSL = '%s'\e[0;37m\n' "$version" "$desktop" "$wsl"
}

ubuntu::detect_environment() {
    if [[ -f /etc/os-release ]]; then
        . /etc/os-release
        VER=$VERSION_ID
    else
        VER="0"
    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::update_system() {
    common::log_section "*** System applications ***"

    if [[ "$WSL" == "yes" ]]; then
        common::log_step "Ubuntu 20.04 workaround"
        sudo apt -y purge --auto-remove neovim
    fi

    common::log_step "Updating system"
    sudo apt update
    sudo apt upgrade -y
    if [[ "$WSL" == "no" ]]; then
        if dpkg -s flatpak &>/dev/null; then
            flatpak update -y
        else
            sudo add-apt-repository -y ppa:flatpak/stable
            sudo apt update
            sudo apt install -y flatpak
            flatpak remote-add --if-not-exists flathub https://flathub.org/repo/flathub.flatpakrepo
            flatpak install -y flathub com.github.tchx84.Flatseal
        fi
    fi
    if dpkg -s snap &>/dev/null; then
        sudo snap refresh
    else
        sudo apt install -y snap
    fi
}

ubuntu::install_system_packages() {
    common::log_step "Installing system packages"
    common::ensure_local_bin
    sudo apt install -y gawk imagemagick gpg ninja-build gettext cmake unzip curl build-essential libssl-dev libffi-dev file libudev-dev pkg-config locales btop ncdu ranger timeshift
    common::link_shell_profiles ubuntu
    if [[ "$WSL" == "no" ]]; then
        mkdir -p "$HOME/.config/alacritty"
        common::link_alacritty_config
    fi
    common::link_update_shortcut
    common::link_start_scripts
}

ubuntu::install_homebrew() {
    common::log_step "Installing Homebrew"
    sudo apt-get install -y -qq build-essential curl file git

    export NONINTERACTIVE=1
    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
    BREW_PREFIX="/home/linuxbrew/.linuxbrew"
    if [[ -d "$BREW_PREFIX" ]]; then
        eval "$("$BREW_PREFIX"/bin/brew shellenv)"
    else
        echo "[!] Homebrew install failed or brew not found at expected location."
        exit 1
    fi
    brew --version >/dev/null && echo "[✓] Homebrew installed successfully."
}

ubuntu::setup_locales() {
    common::log_step "Setup locals"
    sudo locale-gen "en_US.UTF-8"
    sudo update-locale LANG=en_US.UTF-8
}

ubuntu::install_rust() {
    common::log_step "Installing Rust"
    (
        cd "$HOME"
        curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- --no-modify-path -y
    )
    if [[ "$WSL" == "yes" ]]; then
        # shellcheck disable=SC1091
        source "$HOME/.cargo/env"
    fi
    rustup self update
    rustup update stable
    rustup default stable
}

ubuntu::install_python() {
    common::log_step "Installing python"
    sudo apt install -y python3 python3-pip python3-venv python3-dev pipx
    if [[ "$VER" == "22.04" ]]; then
        pip3 install pynvim
    else
        sudo apt install -y python3-pynvim
    fi
}

ubuntu::install_node() {
    common::log_step "Installing node"
    (
        cd "$HOME"
        curl -sL https://deb.nodesource.com/setup_18.x | sudo -E bash -
    )
    sudo apt install -y nodejs
    npm config set prefix "${HOME}/.npm"
}

ubuntu::install_tmux() {
    common::log_step "Installing tmux"
    sudo apt install -y libevent-dev ncurses-dev build-essential bison pkg-config
    (
        cd "$HOME"
        wget https://github.com/tmux/tmux/releases/download/3.3a/tmux-3.3a.tar.gz
        tar xzvf tmux-3.3a.tar.gz
        cd tmux-3.3a
        ./configure
        make && sudo make install
    )
    common::link_tmux_conf
}

ubuntu::install_virtualhere() {
    common::install_virtualhere_client skip
}

ubuntu::install_teensy_rules() {
    if [[ ! -f /.dockerenv ]]; then
        common::log_step "Installing teensy udev rules"
        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
}

ubuntu::install_neovim() {
    common::log_step "Installing neovim"
    sudo apt install -y bat fd-find
    (
        cd "$HOME"
        if [[ -d "$HOME/neovim" ]]; then
            rm -rf "$HOME/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*.deb
        rm -rf "$HOME/neovim"
    )
    common::link_nvim_config
}

ubuntu::install_caffeine() {
    if [[ "$WSL" == "no" ]]; then
        common::log_step "Installing caffeine"
        sudo apt install -y caffeine
    fi
}

ubuntu::install_lazygit() {
    common::log_step "Installing lazygit"
    (
        cd "$HOME"
        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 "$HOME/lazygit" && rm "$HOME/lazygit.tar.gz"
    )
    common::link_lazygit_config
}

ubuntu::install_gdu() {
    common::log_step "Installing gdu"
    if [[ "$VER" == "22.04" ]]; then
        sudo add-apt-repository -y ppa:daniel-milde/gdu
    fi
    sudo apt update
    sudo apt install -y gdu
}

ubuntu::install_ruby() {
    common::log_step "Installing ruby"
    sudo apt install -y ruby ruby-dev
    sudo gem install neovim
}

ubuntu::install_bottom() {
    common::log_step "Installing bottom"
    (
        cd "$HOME"
        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 "$HOME/bottom_0.9.1_amd64.deb"
    )
}

ubuntu::install_go() {
    common::log_step "Installing go"
    sudo apt install -y golang
}

ubuntu::install_php() {
    common::log_step "Installing php"
    sudo apt install -y php php-curl php-xml php-mbstring
}

ubuntu::install_luarocks() {
    common::log_step "Installing luarocks"
    sudo apt install -y luarocks
}

ubuntu::install_composer() {
    common::log_step "Installing composer"
    sudo apt install -y composer
}

ubuntu::install_java() {
    common::log_step "Installing java"
    if [[ "$WSL" == "yes" ]]; then
        sudo add-apt-repository -y ppa:linuxuprising/java
        sudo apt update
        sudo apt install -y oracle-java17-installer
    else
        sudo apt install -y default-jdk
    fi
}

ubuntu::install_julia() {
    common::log_step "Installing julia"
    (
        cd "$HOME"
        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
    )
}

ubuntu::install_ble_sh() {
    common::log_step "Installing ble.sh"
    (
        cd "$HOME"
        if [[ -d "$HOME/ble.sh" ]]; then
            rm -rf "$HOME/ble.sh"
        fi
        git clone --recursive --depth 1 --shallow-submodules https://github.com/akinomyoga/ble.sh.git
        make -C ble.sh install PREFIX=~/.local
        rm -rf "$HOME/ble.sh"
    )
    common::symlink "$PROJECT_ROOT/blerc" "$HOME/.blerc"
}

ubuntu::install_platformio() {
    common::log_step "Installing platformio"
    if [[ "$VER" == "24.04" ]]; then
        pipx install platformio
        pipx ensurepath
    else
        pip3 install platformio
    fi
}

ubuntu::install_node_neovim() {
    common::log_step "Installing neovim for node"
    sudo npm install -g neovim
}

ubuntu::install_rust_embedded() {
    common::log_step "Installing Rust embedded stuff"
    (
        cd "$HOME"
        rustup target add thumbv6m-none-eabi
        cargo install elf2uf2-rs --locked
        cargo install probe-run
        mkdir -p "$HOME/.local/bin"
        common::symlink "$PROJECT_ROOT/pico-load.sh" "$HOME/.local/bin/pico-load"
    )
}

ubuntu::install_cli_utilities() {
    common::log_step "Installing tree sitter cli"
    cargo install tree-sitter-cli

    common::log_step "Installing ripgrep"
    cargo install ripgrep

    common::log_step "Installing eza"
    cargo install eza

    common::log_step "Installing zoxide"
    cargo install zoxide --locked

    common::log_step "Installing starship"
    cargo install starship --locked
    common::link_starship_config
}

ubuntu::install_lsp_servers() {
    common::log_step "Installing LSP servers"
    rustup component add rust-analyzer clippy rustfmt
    sudo npm i -g bash-language-server pyright
    sudo npm install --save-dev prettier
    sudo apt install -y shfmt clangd clang-format
    pipx install black
    pipx install isort
    pipx install mdformat
    brew install lua-language-server marksman
    cargo install stylua
    common::link_lua_check
}

ubuntu::install_desktop_apps() {
    if [[ "$WSL" == "no" ]]; then
        common::log_section "*** Desktop applications ***"

        if [[ "$DESKTOP" == "yes" ]]; then
            common::log_step "Installing Krita"
            flatpak install -y flathub org.kde.krita

            common::log_step "Installing KiCad"
            flatpak install -y flathub org.kicad.KiCad

            common::log_step "Installing FreeCad"
            flatpak install -y flathub org.freecadweb.FreeCAD

            common::log_step "Installing Moonlight"
            flatpak install -y flathub com.moonlight_stream.Moonlight

            common::log_step "Installing DrawIo"
            flatpak install -y flathub com.jgraph.drawio.desktop

            common::log_step "Installing VSCode"
            sudo snap install --classic code

            common::log_step "Installing gnome-shell-extension-manager"
            sudo apt install -y gnome-shell-extension-manager

            common::log_step "Installing gnome tweaks"
            sudo apt install -y gnome-tweaks

            common::log_step "Installing remmina"
            sudo apt install -y remmina

            common::log_step "Installing feh"
            sudo apt install -y feh

            common::log_step "Installing NerdFonts"
            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
            )

            common::log_step "Installing Alacritty"
            sudo add-apt-repository ppa:aslatter/ppa
            sudo apt update
            sudo apt install -y alacritty

            common::log_step "Installing Brave Browser"
            sudo curl -fsSLo /usr/share/keyrings/brave-browser-archive-keyring.gpg https://brave-browser-apt-release.s3.brave.com/brave-browser-archive-keyring.gpg
            echo "deb [signed-by=/usr/share/keyrings/brave-browser-archive-keyring.gpg] https://brave-browser-apt-release.s3.brave.com/ stable main" | sudo tee /etc/apt/sources.list.d/brave-browser-release.list
            sudo apt update
            sudo apt install -y brave-browser

            if [[ "$VER" == "22.04" ]]; then
                common::log_step "Installing Audio control"
                sudo apt install -y pavucontrol
            fi
            if [[ "$VER" == "24.04" ]]; then
                common::log_step "Installing Audio control"
                sudo apt install -y pavucontrol helvum
            fi
        fi
    fi
}

ubuntu::final_message() {
    printf -- '\033[32m \n\n***** Update complete! Please restart your terminal. *****\n\n\033[37m'
}

ubuntu::run() {
    sudo -v
    ubuntu::detect_environment

    ubuntu::print_banner "$VER" "$DESKTOP" "$WSL"

    ubuntu::update_system
    ubuntu::install_system_packages
    ubuntu::install_homebrew
    ubuntu::setup_locales
    ubuntu::install_rust
    ubuntu::install_python
    ubuntu::install_node
    common::install_fzf
    ubuntu::install_tmux
    ubuntu::install_virtualhere
    ubuntu::install_teensy_rules
    ubuntu::install_neovim
    ubuntu::install_caffeine
    ubuntu::install_lazygit
    ubuntu::install_gdu
    ubuntu::install_ruby
    ubuntu::install_bottom
    ubuntu::install_go
    ubuntu::install_php
    ubuntu::install_luarocks
    ubuntu::install_composer
    ubuntu::install_java
    ubuntu::install_julia
    ubuntu::install_ble_sh
    ubuntu::install_platformio
    ubuntu::install_node_neovim
    ubuntu::install_rust_embedded
    ubuntu::install_cli_utilities
    ubuntu::install_lsp_servers
    ubuntu::install_desktop_apps

    ubuntu::final_message
}