setup.sh

#!/usr/bin/env bash
# Refocus Shell - Setup
set -euo pipefail

INSTALL_DIR="$HOME/.local/refocus"
BIN_DIR="$HOME/.local/bin"
SRC_DIR="$(dirname "$(realpath "${BASH_SOURCE[0]}")")"
_info()  { echo "  $1"; }
_ok()    { echo "$1"; }
_warn()  { echo "  $1"; }
_die()   { echo "$1" >&2; exit 1; }

install_deps() {
    local missing=()
    command -v sqlite3      &>/dev/null || missing+=(sqlite3)
    command -v notify-send  &>/dev/null || missing+=(libnotify-bin)
    command -v crontab      &>/dev/null || missing+=(cron)

    [[ ${#missing[@]} -eq 0 ]] && { _ok "Dependencies present."; return; }

    _info "Installing: ${missing[*]}"
    if command -v apt-get &>/dev/null; then
        sudo apt-get install -y "${missing[@]}"
    elif command -v pacman &>/dev/null; then
        local pkgs=()
        for p in "${missing[@]}"; do
            case $p in
                libnotify-bin) pkgs+=(libnotify);;
                cron)          pkgs+=(cronie);;
                *)             pkgs+=("$p");;
            esac
        done
        sudo pacman -S --noconfirm "${pkgs[@]}"
    elif command -v dnf &>/dev/null; then
        local pkgs=()
        for p in "${missing[@]}"; do
            case $p in
                libnotify-bin) pkgs+=(libnotify);;
                cron)          pkgs+=(cronie);;
                *)             pkgs+=("$p");;
            esac
        done
        sudo dnf install -y "${pkgs[@]}"
    else
        _warn "Unknown package manager. Install manually: ${missing[*]}"
    fi
}

install_files() {
    local db_tmp="" env_tmp=""

    if [[ -d "$INSTALL_DIR" ]]; then
        echo -n "Existing installation found. Code will be updated; data and config preserved. Continue? (yes/N): "
        read -r ans
        [[ "$ans" == "yes" ]] || { echo "Aborted."; exit 0; }

        # Stash data before the code wipe
        if [[ -f "$INSTALL_DIR/refocus.db" ]]; then
            db_tmp=$(mktemp)
            cp "$INSTALL_DIR/refocus.db" "$db_tmp"
        fi
        if [[ -f "$INSTALL_DIR/.env" ]]; then
            env_tmp=$(mktemp)
            cp "$INSTALL_DIR/.env" "$env_tmp"
        fi
    fi

    # Strip any stale nudge cron entry before wiping (fixed-string — dots in path are literal)
    local tmp; tmp=$(mktemp)
    crontab -l 2>/dev/null | grep -vF "$INSTALL_DIR/focus-nudge" > "$tmp" || true
    crontab "$tmp"
    rm -f "$tmp"

    # Wipe and rebuild
    rm -rf "$INSTALL_DIR"
    mkdir -p "$INSTALL_DIR/services" "$INSTALL_DIR/lib" "$INSTALL_DIR/core" "$BIN_DIR"

    cp "$SRC_DIR/env.sh"      "$INSTALL_DIR/"
    cp "$SRC_DIR/focus"          "$INSTALL_DIR/"
    cp "$SRC_DIR/focus-nudge"    "$INSTALL_DIR/"
    cp "$SRC_DIR/services/"*.sh  "$INSTALL_DIR/services/"
    cp "$SRC_DIR/lib/"*.sh       "$INSTALL_DIR/lib/"
    cp "$SRC_DIR/core/"*.sh      "$INSTALL_DIR/core/"
    [[ -d "$SRC_DIR/docs" ]]     && cp -r "$SRC_DIR/docs" "$INSTALL_DIR/docs"

    chmod +x "$INSTALL_DIR/focus"
    chmod +x "$INSTALL_DIR/focus-nudge"
    chmod +x "$INSTALL_DIR/lib/"*.sh
    chmod +x "$INSTALL_DIR/services/"*.sh

    # Restore stashed data
    [[ -n "$db_tmp"  ]] && { mv "$db_tmp"  "$INSTALL_DIR/refocus.db"; _ok "Database preserved."; }
    [[ -n "$env_tmp" ]] && { mv "$env_tmp" "$INSTALL_DIR/.env";       _ok "Config (.env) preserved."; }

    ln -sf "$INSTALL_DIR/focus" "$BIN_DIR/focus"
    _ok "Files installed to $INSTALL_DIR"
}

install_shell() {
    local rc="$HOME/.bashrc"
    local line="source $INSTALL_DIR/services/focus-function.sh"

    if grep -qF "$line" "$rc" 2>/dev/null; then
        _ok "Shell integration already in $rc"
    else
        {
            echo ""
            echo "# Refocus Shell"
            echo "$line"
        } >> "$rc"
        _ok "Shell integration added to $rc"
        _warn "Run: source ~/.bashrc"
    fi
}

install_desktop_entry() {
    # Registers Refocus with the desktop notification subsystem so nudges are
    # logged in notification history. Plasma (and other XDG-compliant daemons)
    # match the desktop-entry hint sent by focus-nudge against this file's
    # basename: 'refocus'. Without it, the bubble shows but is discarded.
    local app_dir="$HOME/.local/share/applications"
    local desktop="$app_dir/refocus.desktop"
    mkdir -p "$app_dir"
    cat > "$desktop" <<'DESKTOP'
[Desktop Entry]
Type=Application
Name=Refocus
Comment=Focus and time tracker
Icon=appointment-soon
NoDisplay=true
NotifyRcName=refocus
DESKTOP
    if command -v update-desktop-database &>/dev/null; then
        update-desktop-database "$app_dir" 2>/dev/null || true
    fi
    _ok "Desktop entry installed (notifications will appear in history)."
}

init_and_enable() {
    export REFOCUS_ROOT="$INSTALL_DIR"
    source "$INSTALL_DIR/env.sh"
    source "$INSTALL_DIR/services/database.sh"
    source "$INSTALL_DIR/services/cron.sh"
    db_init                 # INSERT OR IGNORE — no-op on a preserved DB
    set_focus_enabled
    cron_install || _warn "Could not arm cron nudge — run 'focus enable' manually."
    _ok "Database ready and nudging armed."
}

case "${1:-install}" in
    install)
        echo "Installing Refocus Shell..."
        install_deps
        install_files
        [[ -f "$INSTALL_DIR/refocus.db" ]] && _ok "Existing database preserved."
        init_and_enable
        install_shell
        install_desktop_entry
        echo ""
        echo "Done. Open a new terminal or run: source ~/.bashrc"
        ;;
    uninstall)
        echo -n "Remove $INSTALL_DIR and shell integration? (yes/N): "
        read -r ans
        [[ "$ans" == "yes" ]] || { echo "Cancelled."; exit 0; }
        if [[ -f "$INSTALL_DIR/services/cron.sh" ]]; then
            export REFOCUS_ROOT="$INSTALL_DIR"
            source "$INSTALL_DIR/services/cron.sh"
            cron_remove 2>/dev/null || true
        fi
        rm -rf "$INSTALL_DIR"
        rm -f  "$BIN_DIR/focus"
        rm -f  "$HOME/.local/share/applications/refocus.desktop"
        rc="$HOME/.bashrc"
        sed -i '/# Refocus Shell/d' "$rc"
        sed -i '/focus-function\.sh/d' "$rc"
        _ok "Uninstalled."
        ;;
    *)
        echo "Usage: ./setup.sh [install|uninstall]" >&2; exit 2
        ;;
esac