install.sh

#!/bin/bash
# Hotkey Whisper - Installer
set -eu

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Source shared libraries
source "$SCRIPT_DIR/lib/colors.sh"
source "$SCRIPT_DIR/lib/constants.sh"
source "$SCRIPT_DIR/lib/banner.sh"

print_banner

# Helper function to check if a command exists
command_exists() {
    command -v "$1" &> /dev/null
}

# Helper function to check if apt package is installed
pkg_installed() {
    dpkg -s "$1" &> /dev/null 2>&1
}

# ─────────────────────────────────────────────────────────────
# Step 1: System Dependencies
# ─────────────────────────────────────────────────────────────
echo -e "${WHITE}${BOLD}[1/4] System Dependencies${NC}"

SYSTEM_PKGS=(python3-venv portaudio19-dev xdotool netcat-openbsd gir1.2-appindicator3-0.1 python3-gi)
MISSING_PKGS=()

for pkg in "${SYSTEM_PKGS[@]}"; do
    if pkg_installed "$pkg"; then
        echo -e "  $CHECK ${GRAY}$pkg${NC}"
    else
        echo -e "  $ARROW $pkg ${GRAY}(will install)${NC}"
        MISSING_PKGS+=("$pkg")
    fi
done

if [ ${#MISSING_PKGS[@]} -gt 0 ]; then
    echo ""
    echo -e "  ${GRAY}Installing missing packages...${NC}"
    sudo apt-get update -qq
    sudo apt-get install -y -qq "${MISSING_PKGS[@]}" > /dev/null
    echo -e "  $CHECK ${GRAY}Installed ${#MISSING_PKGS[@]} package(s)${NC}"
fi
echo ""

# ─────────────────────────────────────────────────────────────
# Step 2: Python Environment
# ─────────────────────────────────────────────────────────────
echo -e "${WHITE}${BOLD}[2/4] Python Environment${NC}"

mkdir -p "$DATA_DIR" "$BIN_DIR" "$CONFIG_DIR"

if [ -d "$VENV_DIR" ] && [ -f "$VENV_DIR/bin/activate" ]; then
    echo -e "  $CHECK ${GRAY}Virtual environment exists${NC}"
else
    echo -e "  $ARROW Creating virtual environment..."
    python3 -m venv "$VENV_DIR"
    echo -e "  $CHECK ${GRAY}Virtual environment created${NC}"
fi

source "$VENV_DIR/bin/activate"
echo ""

# ─────────────────────────────────────────────────────────────
# Step 3: Python Packages
# ─────────────────────────────────────────────────────────────
echo -e "${WHITE}${BOLD}[3/4] Python Packages${NC}"

# Check if packages are already installed
check_pip_pkg() {
    pip show "$1" &> /dev/null 2>&1
}

NEED_TORCH=false
NEED_STT=false

if check_pip_pkg "torch" && check_pip_pkg "torchaudio"; then
    echo -e "  $CHECK ${GRAY}torch & torchaudio${NC}"
else
    echo -e "  $ARROW torch & torchaudio ${GRAY}(will install)${NC}"
    NEED_TORCH=true
fi

if check_pip_pkg "RealtimeSTT"; then
    echo -e "  $CHECK ${GRAY}RealtimeSTT${NC}"
else
    echo -e "  $ARROW RealtimeSTT ${GRAY}(will install)${NC}"
    NEED_STT=true
fi

if $NEED_TORCH || $NEED_STT; then
    echo ""
    pip install -q --upgrade pip
    
    if $NEED_TORCH; then
        echo -e "  ${GRAY}Installing PyTorch (this may take a few minutes)...${NC}"
        pip install -q torch torchaudio --index-url https://download.pytorch.org/whl/cu121
        echo -e "  $CHECK ${GRAY}Installed torch & torchaudio${NC}"
    fi
    
    if $NEED_STT; then
        echo -e "  ${GRAY}Installing RealtimeSTT...${NC}"
        pip install -q RealtimeSTT
        echo -e "  $CHECK ${GRAY}Installed RealtimeSTT${NC}"
    fi
fi
echo ""

# ─────────────────────────────────────────────────────────────
# Step 4: Setup Commands
# ─────────────────────────────────────────────────────────────
echo -e "${WHITE}${BOLD}[4/4] Setup Commands${NC}"

chmod +x "$SCRIPT_DIR/main.sh" "$SCRIPT_DIR/src/server.py" "$SCRIPT_DIR/src/tray.py"
ln -sf "$SCRIPT_DIR/main.sh" "$BIN_DIR/hotkey-whisper"
echo -e "  $CHECK ${GRAY}Linked hotkey-whisper to PATH${NC}"

# Create tray launcher wrapper (avoids snap environment conflicts)
cat > "$BIN_DIR/hotkey-whisper-tray" << 'TRAY_EOF'
#!/bin/bash
# Launch tray with clean environment to avoid snap library conflicts
exec env -i \
    HOME="$HOME" \
    PATH="/usr/bin:/bin" \
    DISPLAY="${DISPLAY:-:0}" \
    XAUTHORITY="${XAUTHORITY:-$HOME/.Xauthority}" \
    DBUS_SESSION_BUS_ADDRESS="$DBUS_SESSION_BUS_ADDRESS" \
    XDG_RUNTIME_DIR="$XDG_RUNTIME_DIR" \
    python3 "${0%-tray}/../../repos/hotkey-whisper/src/tray.py" "$@"
TRAY_EOF
# Fix path in wrapper
sed -i "s|\${0%-tray}/../../repos/hotkey-whisper|$SCRIPT_DIR|" "$BIN_DIR/hotkey-whisper-tray"
chmod +x "$BIN_DIR/hotkey-whisper-tray"
echo -e "  $CHECK ${GRAY}Created tray launcher${NC}"

# Copy default config to user config dir if not exists
if [ ! -f "$CONFIG_DIR/config.json" ]; then
    cp "$SCRIPT_DIR/config.json" "$CONFIG_DIR/config.json"
    echo -e "  $CHECK ${GRAY}Created user config${NC}"
fi
echo ""

# ─────────────────────────────────────────────────────────────
# Done!
# ─────────────────────────────────────────────────────────────
echo -e "${PURPLE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
echo -e "${GREEN}${BOLD}  ✓ Installation Complete!${NC}"
echo -e "${PURPLE}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${NC}"
echo ""
echo -e "  ${WHITE}${BOLD}Quick Start${NC}"
echo ""
echo -e "    ${CYAN}hotkey-whisper${NC}"
echo -e "    ${GRAY}(First run downloads the model ~1.5GB)${NC}"
echo ""
echo -e "  ${WHITE}${BOLD}Set Up Keyboard Shortcut${NC}"
echo ""
echo -e "    ${GRAY}Settings → Keyboard → Custom Shortcuts${NC}"
echo -e "    ${GRAY}Command:${NC} ${CYAN}$BIN_DIR/hotkey-whisper${NC}"
echo ""