#!/bin/bash
set -e

# vibe installation script
# This script installs the vibe CLI tool and sets up Claude Code integration

REPO_URL="https://github.com/Blurjp/vibe.git"
INSTALL_DIR="${VIBE_INSTALL_DIR:-$HOME/.vibe}"
CLAUDE_COMMANDS_DIR="$HOME/.claude/commands"

echo "[vibe] Installing vibe..."

# Detect if we're already in the repo
if [ -d "tools/vibe" ] && [ -f "tools/vibe/vibe" ]; then
    echo "[vibe] Already in vibe repo, skipping clone..."
    VIBE_DIR="$(pwd)"
else
    # Create install directory
    mkdir -p "$INSTALL_DIR"

    # Clone or update the repo
    if [ -d "$INSTALL_DIR/vibe" ]; then
        echo "[vibe] Updating existing installation..."
        cd "$INSTALL_DIR/vibe"
        git pull
    else
        echo "[vibe] Cloning repo to $INSTALL_DIR/vibe..."
        git clone "$REPO_URL" "$INSTALL_DIR/vibe"
        cd "$INSTALL_DIR/vibe"
    fi
    VIBE_DIR="$INSTALL_DIR/vibe"
fi

# NOTE: vibe init should be run in YOUR project folder, not here
# This only sets up the vibe tool itself
echo "[vibe] Note: Run 'vibe init' in your project folder to set up .vibedbg/"

# Set up Claude Code integration
echo "[vibe] Setting up Claude Code integration..."

# Install to standard Claude Code CLI using symlinks
# This way bug fixes in the repo automatically apply to installed commands
mkdir -p "$CLAUDE_COMMANDS_DIR"
# Remove old copies if they exist
rm -f "$CLAUDE_COMMANDS_DIR"/vibe-*.md
# Create symlinks
for cmd_file in tools/vibe/claude-commands/*.md; do
    cmd_name=$(basename "$cmd_file")
    ln -sf "$VIBE_DIR/$cmd_file" "$CLAUDE_COMMANDS_DIR/$cmd_name"
done
echo "[vibe] Symlinked commands to $CLAUDE_COMMANDS_DIR"

# Also install to ccg (claude-glm) if it exists
CCG_COMMANDS_DIR="$HOME/.claude-glm/commands"
if [ -d "$HOME/.claude-glm" ] || command -v ccg &>/dev/null || command -v claude-glm &>/dev/null; then
    mkdir -p "$CCG_COMMANDS_DIR"
    # Remove old copies if they exist
    rm -f "$CCG_COMMANDS_DIR"/vibe-*.md
    # Create symlinks
    for cmd_file in tools/vibe/claude-commands/*.md; do
        cmd_name=$(basename "$cmd_file")
        ln -sf "$VIBE_DIR/$cmd_file" "$CCG_COMMANDS_DIR/$cmd_name"
    done
    echo "[vibe] Also symlinked commands to $CCG_COMMANDS_DIR (for ccg)"
fi

# Make vibe executable
chmod +x tools/vibe/vibe

# Create symlink in ~/.local/bin for global access
LOCAL_BIN="$HOME/.local/bin"
mkdir -p "$LOCAL_BIN"
ln -sf "$VIBE_DIR/tools/vibe/vibe" "$LOCAL_BIN/vibe"
echo "[vibe] Created symlink: $LOCAL_BIN/vibe"

# Add ~/.local/bin to PATH if not already there
SHELL_CONFIG="$HOME/.zshrc"
if [ -n "$BASH_VERSION" ]; then
    SHELL_CONFIG="$HOME/.bashrc"
fi

if ! grep -q "\.local/bin" "$SHELL_CONFIG" 2>/dev/null; then
    echo "[vibe] Adding ~/.local/bin to PATH in $SHELL_CONFIG..."
    echo "" >> "$SHELL_CONFIG"
    echo "# vibe: add ~/.local/bin to PATH" >> "$SHELL_CONFIG"
    echo "export PATH=\"$LOCAL_BIN:\$PATH\"" >> "$SHELL_CONFIG"
    echo "[vibe] Run 'source $SHELL_CONFIG' or restart your shell to use 'vibe' command."
else
    echo "[vibe] ~/.local/bin already in PATH"
fi

echo ""
echo "[vibe] Installation complete!"
echo ""
echo "Installed to: $VIBE_DIR"
echo ""
echo "Usage in any project folder:"
echo "  cd /path/to/your/project"
echo "  vibe init                    # Initialize .vibedbg/ in current project"
echo "  vibe select [--note \"...\"]   # Capture screen region"
echo "  vibe ask \"...\"               # Ask Claude to analyze"
echo ""
echo "Or use within Claude Code CLI (from any project):"
echo "  /vibe-select                  # Capture screen region"
echo "  /vibe-ask                     # Analyze and recommend fixes"
echo ""
echo "Note: Commands are symlinked, so bug fixes auto-apply when you"
echo "      'cd ~/.vibe/vibe && git pull' in the install directory."
echo ""
echo "Next steps:"
echo "  1. Run: source $SHELL_CONFIG"
echo "  2. cd to your project folder"
echo "  3. Run: vibe init"