#!/bin/bash
set -euo pipefail

echo "Claude Code Setup for VPS"
echo "========================="

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
NC='\033[0m'

check() { command -v "$1" &>/dev/null && echo -e "${GREEN}[ok]${NC} $1" || { echo -e "${RED}[missing]${NC} $1"; return 1; }; }

echo ""
echo "Checking dependencies..."
check node || { echo "Install Node.js first"; exit 1; }
check npm || { echo "Install npm first"; exit 1; }
check claude || { echo "Install Claude Code first: npm install -g @anthropic-ai/claude-code"; exit 1; }

CLAUDE_DIR="$HOME/.claude"
mkdir -p "$CLAUDE_DIR/commands" "$CLAUDE_DIR/skills" "$CLAUDE_DIR/agents"

# ============================================
# PLUGINS
# ============================================
echo ""
echo "Installing plugins..."

claude plugin install compound-engineering --marketplace every-marketplace 2>/dev/null || echo "compound-engineering: manual install needed"
claude plugin install ralph-wiggum 2>/dev/null || echo "ralph-wiggum: manual install needed"
claude plugin install code-simplifier 2>/dev/null || echo "code-simplifier: manual install needed"

# ============================================
# NPM TOOLS
# ============================================
echo ""
echo "Installing npm tools..."

npm install -g agent-browser 2>/dev/null || echo "agent-browser: npm install failed"

# ============================================
# VERCEL AGENT SKILLS
# ============================================
echo ""
echo "Installing Vercel agent-skills..."

npx add-skill vercel-labs/agent-skills 2>/dev/null || echo "vercel agent-skills: manual install needed"

# ============================================
# RAMS (design reviews)
# ============================================
echo ""
echo "Installing rams..."

npx add-skill elirousso/rams 2>/dev/null || {
    RAMS_TMP=$(mktemp -d)
    git clone --quiet --depth 1 https://github.com/elirousso/rams.git "$RAMS_TMP" 2>/dev/null && {
        cp -r "$RAMS_TMP/skills/"* "$CLAUDE_DIR/skills/" 2>/dev/null || true
        cp -r "$RAMS_TMP/commands/"* "$CLAUDE_DIR/commands/" 2>/dev/null || true
        rm -rf "$RAMS_TMP"
        echo "rams: installed from git"
    } || echo "rams: manual install needed"
}

# ============================================
# EVAL-SKILL
# ============================================
echo ""
echo "Installing eval-skill..."

EVAL_TMP=$(mktemp -d)
git clone --quiet --depth 1 https://github.com/harivansh-afk/eval-skill.git "$EVAL_TMP" 2>/dev/null && {
    mkdir -p "$CLAUDE_DIR/skills/eval" "$CLAUDE_DIR/agents" "$CLAUDE_DIR/evals"
    cp "$EVAL_TMP/skills/eval/SKILL.md" "$CLAUDE_DIR/skills/eval/" 2>/dev/null || true
    cp "$EVAL_TMP/agents/"*.md "$CLAUDE_DIR/agents/" 2>/dev/null || true
    cp "$EVAL_TMP/commands/"*.md "$CLAUDE_DIR/commands/" 2>/dev/null || true
    rm -rf "$EVAL_TMP"
    echo "eval-skill: installed"
} || echo "eval-skill: manual install needed"

# ============================================
# BROWSER SKILL (from official plugins)
# ============================================
echo ""
echo "Installing browser skill..."

mkdir -p "$CLAUDE_DIR/skills/browser"
cat > "$CLAUDE_DIR/skills/browser/SKILL.md" << 'BROWSER_SKILL'
---
name: browser
description: Automate browser interactions via CLI using agent-browser
triggers:
  - navigate websites
  - interact with web pages
  - fill forms
  - take screenshots
  - test web applications
  - extract information from web pages
---

# Browser Automation Skill

Use agent-browser CLI for headless browser automation.

## Commands

```bash
# Navigate to URL
agent-browser navigate "https://example.com"

# Take screenshot
agent-browser screenshot --output screenshot.png

# Click element
agent-browser click "button#submit"

# Fill form field
agent-browser type "input[name=email]" "user@example.com"

# Get page content
agent-browser content
```

## Usage Pattern

1. Navigate to target URL
2. Wait for page load
3. Interact with elements using CSS selectors
4. Capture results or screenshots

Requires: `npm install -g agent-browser`
BROWSER_SKILL
echo "browser skill: installed"

# ============================================
# MCP SERVERS CONFIG
# ============================================
echo ""
echo "Configuring MCP servers..."

SETTINGS_FILE="$CLAUDE_DIR/settings.json"
if [[ ! -f "$SETTINGS_FILE" ]]; then
    echo '{}' > "$SETTINGS_FILE"
fi

cat > "$CLAUDE_DIR/mcp-servers.json" << 'EOF'
{
  "context7": {
    "type": "http",
    "url": "https://mcp.context7.com/mcp",
    "headers": {
      "CONTEXT7_API_KEY": "YOUR_CONTEXT7_API_KEY"
    }
  },
  "axiom": {
    "type": "stdio",
    "command": "npx",
    "args": ["-y", "@axiomhq/mcp"],
    "env": {
      "AXIOM_TOKEN": "YOUR_AXIOM_TOKEN",
      "AXIOM_ORG_ID": "YOUR_AXIOM_ORG_ID"
    }
  },
  "github": {
    "type": "stdio",
    "command": "npx",
    "args": ["-y", "@modelcontextprotocol/server-github"],
    "env": {
      "GITHUB_PERSONAL_ACCESS_TOKEN": "YOUR_GITHUB_PAT"
    }
  },
  "firecrawl": {
    "type": "stdio",
    "command": "npx",
    "args": ["-y", "firecrawl-mcp"],
    "env": {
      "FIRECRAWL_API_KEY": "YOUR_FIRECRAWL_API_KEY"
    }
  },
  "playwright": {
    "type": "stdio",
    "command": "npx",
    "args": ["-y", "@anthropic-ai/playwright-mcp"]
  }
}
EOF

echo "MCP config written to $CLAUDE_DIR/mcp-servers.json"

# ============================================
# CLAUDE.MD
# ============================================
echo ""
echo "Installing CLAUDE.md..."

cat > "$CLAUDE_DIR/CLAUDE.md" << 'CLAUDEMD'
<?xml version="1.0" encoding="UTF-8"?>
<claude-instructions>

<project-defaults>
    <python>
        <rule>Always use 'uv' as the default package manager and virtual environment tool</rule>
        <rule>Prefer 'uv run' for executing Python scripts</rule>
        <rule>Use 'uv pip' instead of bare 'pip'</rule>
        <rule>Use 'uv venv' for creating virtual environments</rule>
    </python>
</project-defaults>

<universal-constraints>
    <style>
        <rule>Never use emojis in any output or code comments</rule>
        <rule>Never use em dashes - use hyphens or colons instead</rule>
    </style>

    <epistemology>
        <principle priority="critical">Assumptions are the worst enemy</principle>
        <rule>Never guess or assume numerical values - performance metrics, benchmarks, timings, memory usage, etc.</rule>
        <rule>When uncertain about any quantifiable result, implement the code and measure/visualize the actual results</rule>
        <rule>Do not cite expected performance improvements or statistics without empirical data</rule>
        <rule>If a claim requires a number, either cite a source, run a test, or explicitly state "this needs to be measured"</rule>
        <rule>Prefer "let's benchmark this" over "this should be about X% faster"</rule>
    </epistemology>

    <interaction-model>
        <rule>If a user request is unclear, ask clarifying questions until the execution steps are perfectly clear</rule>
        <rule>Once clarified, proceed autonomously without asking for human intervention</rule>
        <ask-for-help-only-when>
            <condition>A script runs longer than 2 minutes - use timeout, then ask user to run manually</condition>
            <condition>Elevated privileges are required (sudo)</condition>
            <condition>Other critical blockers that cannot be resolved programmatically</condition>
        </ask-for-help-only-when>
    </interaction-model>

    <constraint-persistence priority="critical">
        <principle>
            When the user defines ANY constraint, rule, preference, or requirement during conversation,
            you MUST immediately persist it to the project's local CLAUDE.md. This is NOT optional.
        </principle>

        <triggers>
            <pattern>never do X</pattern>
            <pattern>always do X</pattern>
            <pattern>from now on</pattern>
            <pattern>going forward</pattern>
            <pattern>I want you to</pattern>
            <pattern>make sure to</pattern>
            <pattern>do not ever</pattern>
            <pattern>remember to</pattern>
            <pattern>the rule is</pattern>
            <pattern>use X instead of Y</pattern>
            <pattern>prefer X over Y</pattern>
            <pattern>avoid X</pattern>
            <pattern>stop doing X</pattern>
        </triggers>

        <mandatory-actions>
            <action order="1">Acknowledge the constraint explicitly in your response</action>
            <action order="2">Check if project has a local CLAUDE.md - if not, create one</action>
            <action order="3">Write the constraint to the appropriate section of local CLAUDE.md</action>
            <action order="4">Confirm the constraint has been persisted</action>
            <action order="5">Apply the constraint immediately and in all future actions</action>
        </mandatory-actions>
    </constraint-persistence>
</universal-constraints>

<mcp-guidance>
    <principle>
        When uncertain about syntax, APIs, or current best practices - ALWAYS use an MCP
        server first. Do not guess or rely on potentially outdated knowledge.
    </principle>

    <server name="context7">
        <purpose>Up-to-date library documentation</purpose>
        <use-when>Need current documentation for any library or framework</use-when>
        <tools>resolve-library-id, get-library-docs</tools>
    </server>

    <server name="axiom">
        <purpose>Query observability data</purpose>
        <use-when>Need to analyze logs, traces, or metrics</use-when>
        <tools>queryApl, getDatasets</tools>
    </server>

    <server name="github">
        <purpose>GitHub API operations</purpose>
        <use-when>Need to interact with repos, issues, PRs</use-when>
    </server>

    <server name="firecrawl">
        <purpose>Web scraping and extraction</purpose>
        <use-when>Need to extract data from websites</use-when>
        <tools>firecrawl_scrape, firecrawl_crawl, firecrawl_extract</tools>
    </server>

    <server name="playwright">
        <purpose>Browser automation</purpose>
        <use-when>Need to test web apps or automate browser tasks</use-when>
    </server>
</mcp-guidance>

</claude-instructions>
CLAUDEMD

echo "CLAUDE.md installed"

# ============================================
# ENABLE PLUGINS IN SETTINGS
# ============================================
echo ""
echo "Enabling plugins in settings..."

node -e "
const fs = require('fs');
const settingsPath = '$SETTINGS_FILE';
let settings = {};
try { settings = JSON.parse(fs.readFileSync(settingsPath, 'utf8')); } catch {}

settings.enabledPlugins = {
  ...settings.enabledPlugins,
  'ralph-wiggum@claude-plugins-official': true,
  'code-simplifier@claude-plugins-official': true,
  'compound-engineering@every-marketplace': true
};

const mcpPath = '$CLAUDE_DIR/mcp-servers.json';
try {
  const mcp = JSON.parse(fs.readFileSync(mcpPath, 'utf8'));
  settings.mcpServers = { ...settings.mcpServers, ...mcp };
} catch {}

fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
console.log('Settings updated');
" 2>/dev/null || echo "Settings: manual update needed"

# ============================================
# DONE
# ============================================
echo ""
echo "========================="
echo -e "${GREEN}Setup complete!${NC}"
echo ""
echo "Next steps:"
echo "1. Edit ~/.claude/settings.json and add your API keys:"
echo "   - CONTEXT7_API_KEY"
echo "   - AXIOM_TOKEN + AXIOM_ORG_ID"
echo "   - GITHUB_PERSONAL_ACCESS_TOKEN"
echo "   - FIRECRAWL_API_KEY (optional)"
echo ""
echo "2. Verify plugins:"
echo "   claude plugin list"
echo ""
echo "3. Start Claude:"
echo "   claude"
echo ""