install.sh

#!/usr/bin/env bash
# Symlinks this repo's dotfiles into $HOME.
# Existing files/symlinks at the destination are backed up to ~/.dotfiles_backup/<timestamp>/.

set -euo pipefail

REPO_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
BACKUP_DIR="$HOME/.dotfiles_backup/$(date +%Y%m%d%H%M%S)"

echo "Checking prerequisites..."

if ! command -v cargo >/dev/null 2>&1; then
    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
    # shellcheck disable=SC1091
    . "$HOME/.cargo/env"
fi

if ! command -v jj >/dev/null 2>&1; then
    cargo install --locked jj-cli
fi

# map: repo file -> destination path in $HOME
declare -a LINKS=(
    "tmux.conf:.tmux.conf"
    "config_jj_config.toml:.config/jj/config.toml"
    "p10k.zsh:.p10k.zsh"
)

for entry in "${LINKS[@]}"; do
    src="${entry%%:*}"
    dest_rel="${entry##*:}"
    src_path="$REPO_DIR/$src"
    dest_path="$HOME/$dest_rel"

    mkdir -p "$(dirname "$dest_path")"

    if [ -e "$dest_path" ] || [ -L "$dest_path" ]; then
        if [ -L "$dest_path" ] && [ "$(readlink "$dest_path")" = "$src_path" ]; then
            echo "skip   $dest_rel (already linked)"
            continue
        fi
        mkdir -p "$BACKUP_DIR"
        mv "$dest_path" "$BACKUP_DIR/$(basename "$dest_rel")"
        echo "backed up existing $dest_rel -> $BACKUP_DIR/"
    fi

    ln -s "$src_path" "$dest_path"
    echo "linked $dest_rel -> $src_path"
done

if [ -d "$BACKUP_DIR" ]; then
    echo
    echo "Backups saved to $BACKUP_DIR"
fi

if grep -qE '^(name|email) = "\.\.\."$' "$REPO_DIR/config_jj_config.toml"; then
    echo
    echo "Reminder: set your name and email in ~/.config/jj/config.toml (jj [user] section) before using jj."
fi

echo
echo "Setting up oh-my-zsh plugins/theme..."

ZSH_CUSTOM="${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}"

clone_if_missing() {
    local repo_url="$1"
    local dest="$2"
    if [ -d "$dest" ]; then
        echo "skip   $dest (already cloned)"
    else
        git clone --depth 1 "$repo_url" "$dest"
    fi
}

clone_if_missing "https://github.com/romkatv/powerlevel10k.git" "$ZSH_CUSTOM/themes/powerlevel10k"
clone_if_missing "https://github.com/zsh-users/zsh-autosuggestions" "$ZSH_CUSTOM/plugins/zsh-autosuggestions"
clone_if_missing "https://github.com/zsh-users/zsh-syntax-highlighting.git" "$ZSH_CUSTOM/plugins/zsh-syntax-highlighting"

mkdir -p "$HOME/Downloads"
if [ -f "$HOME/Downloads/coolnight.itermcolors" ]; then
    echo "skip   coolnight.itermcolors (already downloaded)"
else
    curl -fsSL "https://raw.githubusercontent.com/josean-dev/dev-environment-files/main/coolnight.itermcolors" \
        --output "$HOME/Downloads/coolnight.itermcolors"
    echo "downloaded coolnight.itermcolors -> ~/Downloads/"
fi

if [ -f "$HOME/.zshrc" ]; then
    if grep -qE '^plugins=\(' "$HOME/.zshrc"; then
        sed -i.bak -E 's/^plugins=\(.*\)$/plugins=(git zsh-autosuggestions zsh-syntax-highlighting web-search)/' "$HOME/.zshrc"
        rm -f "$HOME/.zshrc.bak"
        echo "updated plugins= line in ~/.zshrc"
    else
        echo "warning: no plugins= line found in ~/.zshrc, skipping"
    fi

    if grep -qE '^ZSH_THEME=' "$HOME/.zshrc"; then
        sed -i.bak -E 's/^ZSH_THEME=.*$/ZSH_THEME="powerlevel10k\/powerlevel10k"/' "$HOME/.zshrc"
        rm -f "$HOME/.zshrc.bak"
        echo "updated ZSH_THEME= line in ~/.zshrc"
    else
        echo "warning: no ZSH_THEME= line found in ~/.zshrc, skipping"
    fi
fi

# Merge blocks from the repo's zshrc into ~/.zshrc, skipping any block whose
# key line is already present anywhere in ~/.zshrc (no duplication).
apply_zshrc_block() {
    local dest="$1" block="$2" key="$3"

    [ -n "$key" ] || return

    if grep -qF -- "$key" "$dest"; then
        return
    fi

    if [[ "$block" == *"p10k-instant-prompt"* ]]; then
        { printf '%s\n\n' "$block"; cat "$dest"; } > "$dest.tmp"
        mv "$dest.tmp" "$dest"
        echo "added p10k instant-prompt block to top of ~/.zshrc"
    else
        printf '\n%s\n' "$block" >> "$dest"
        echo "appended to ~/.zshrc: ${key}"
    fi
}

merge_zshrc() {
    local repo_zshrc="$REPO_DIR/zshrc"
    local dest="$HOME/.zshrc"

    [ -f "$dest" ] || touch "$dest"

    local block="" key=""
    while IFS= read -r line || [ -n "$line" ]; do
        if [ -z "$line" ]; then
            [ -n "$block" ] && apply_zshrc_block "$dest" "$block" "$key"
            block=""
            key=""
            continue
        fi
        if [ -n "$block" ]; then
            block="$block"$'\n'"$line"
        else
            block="$line"
        fi
        if [ -z "$key" ] && [[ "$line" != \#* ]]; then
            key="$line"
        fi
    done < "$repo_zshrc"
    [ -n "$block" ] && apply_zshrc_block "$dest" "$block" "$key"
}

merge_zshrc