setup.sh

#!/usr/bin/env bash
set -euo pipefail

# AgentSource Plugin — Setup Script
# Installs the CLI, saves API key, and registers the plugin with Claude Code.

PLUGIN_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
INSTALL_DIR="$HOME/.agentsource"
BIN_DIR="$INSTALL_DIR/bin"
SESSIONS_DIR="$INSTALL_DIR/sessions"
CONFIG_FILE="$INSTALL_DIR/config.json"
CLAUDE_PLUGINS_DIR="$HOME/.claude/plugins"

echo "=== AgentSource Plugin Setup ==="
echo "Plugin directory : $PLUGIN_DIR"
echo "Install directory: $INSTALL_DIR"
echo ""

# ---------------------------------------------------------------------------
# 1. Verify Python 3
# ---------------------------------------------------------------------------
if ! command -v python3 &>/dev/null; then
    echo "[ERROR] Python 3 not found. Please install Python 3.8+ and retry."
    exit 1
fi

PYTHON_VERSION=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
PYTHON_MAJOR=$(python3 -c 'import sys; print(sys.version_info.major)')
PYTHON_MINOR=$(python3 -c 'import sys; print(sys.version_info.minor)')

if [ "$PYTHON_MAJOR" -lt 3 ] || { [ "$PYTHON_MAJOR" -eq 3 ] && [ "$PYTHON_MINOR" -lt 8 ]; }; then
    echo "[ERROR] Python 3.8+ required. Found: $PYTHON_VERSION"
    exit 1
fi

echo "[OK] Python $PYTHON_VERSION"

# ---------------------------------------------------------------------------
# 2. Create directory structure
# ---------------------------------------------------------------------------
mkdir -p "$BIN_DIR" "$SESSIONS_DIR"
echo "[OK] Directories created"

# ---------------------------------------------------------------------------
# 3. Install CLI
# ---------------------------------------------------------------------------
cp "$PLUGIN_DIR/bin/agentsource.py" "$BIN_DIR/agentsource.py"
chmod +x "$BIN_DIR/agentsource.py"
echo "[OK] CLI installed to $BIN_DIR/agentsource.py"

# ---------------------------------------------------------------------------
# 4. API key configuration
# ---------------------------------------------------------------------------
echo ""
if [ -n "${EXPLORIUM_API_KEY:-}" ]; then
    echo "[OK] EXPLORIUM_API_KEY is set in environment"
    read -r -p "Save it to ~/.agentsource/config.json for persistence? [y/N] " save_key
    if [[ "${save_key:-}" =~ ^[Yy]$ ]]; then
        printf '{\n  "api_key": "%s"\n}\n' \
            "$EXPLORIUM_API_KEY" > "$CONFIG_FILE"
        chmod 600 "$CONFIG_FILE"
        echo "[OK] API key saved to $CONFIG_FILE (mode 600)"
    fi
else
    echo "EXPLORIUM_API_KEY is not set."
    read -r -p "Enter your Explorium API key (or press Enter to skip): " api_key
    if [ -n "${api_key:-}" ]; then
        printf '{\n  "api_key": "%s"\n}\n' \
            "$api_key" > "$CONFIG_FILE"
        chmod 600 "$CONFIG_FILE"
        echo "[OK] API key saved to $CONFIG_FILE (mode 600)"
        echo ""
        echo "  Add this to your shell profile (~/.zshrc or ~/.bashrc) for future sessions:"
        echo "    export EXPLORIUM_API_KEY='$api_key'"
    else
        echo "[WARN] No API key configured. Set EXPLORIUM_API_KEY before using the plugin."
    fi
fi

# ---------------------------------------------------------------------------
# 5. Smoke-test the CLI
# ---------------------------------------------------------------------------
echo ""
echo "Running CLI smoke test..."
if python3 "$BIN_DIR/agentsource.py" --help &>/dev/null; then
    echo "[OK] CLI smoke test passed"
else
    echo "[ERROR] CLI smoke test failed. Check your Python installation."
    exit 1
fi

# ---------------------------------------------------------------------------
# 6. Register plugin with Claude Code (copy files)
# ---------------------------------------------------------------------------
echo ""

# Standard plugin location (local Claude Code)
PLUGIN_DEST="$CLAUDE_PLUGINS_DIR/agentsource-plugin"
mkdir -p "$PLUGIN_DEST"
cp -r "$PLUGIN_DIR/.claude-plugin" "$PLUGIN_DEST/"
cp -r "$PLUGIN_DIR/skills"         "$PLUGIN_DEST/"
cp -r "$PLUGIN_DIR/bin"            "$PLUGIN_DEST/"
echo "[OK] Plugin installed to $PLUGIN_DEST"

# Also copy to Cowork plugin cache (Claude desktop app uploads from here)
COWORK_BASE="$HOME/Library/Application Support/Claude/local-agent-mode-sessions"
if [ -d "$COWORK_BASE" ]; then
    # Find all cowork plugin directories for this plugin
    while IFS= read -r -d '' cowork_dest; do
        cp -r "$PLUGIN_DIR/.claude-plugin" "$cowork_dest/"
        cp -r "$PLUGIN_DIR/skills"         "$cowork_dest/"
        cp -r "$PLUGIN_DIR/bin"            "$cowork_dest/"
        echo "[OK] Cowork plugin updated: $cowork_dest"
    done < <(find "$COWORK_BASE" -type d -name "agentsource-plugin" -path "*/marketplaces/*" -print0 2>/dev/null)
fi

# ---------------------------------------------------------------------------
# 7. Done
# ---------------------------------------------------------------------------
echo ""
echo "=== Setup Complete ==="
echo ""
echo "Optional: Add the CLI to your PATH:"
echo "  echo 'export PATH=\"\$HOME/.agentsource/bin:\$PATH\"' >> ~/.zshrc"
echo ""
echo "Usage in Claude Code:"
echo "  Open any project, then describe what you're looking for, e.g.:"
echo "  'Find CTOs at Series B SaaS companies in California'"
echo "  'Show me fintech companies using Stripe with 50-200 employees'"
echo "  'Get emails for VP Sales at companies that recently raised Series A'"
echo ""
echo "Direct CLI usage:"
echo "  python3 ~/.agentsource/bin/agentsource.py --help"
echo "  python3 ~/.agentsource/bin/agentsource.py autocomplete --entity-type businesses --field linkedin_category --query 'saas' --semantic"