install.sh

#!/usr/bin/env bash
# Agent-Litmus Installer
# "The test your tests have to pass."
#
# Usage: curl -fsSL https://raw.githubusercontent.com/saisumantatgit/Agent-Litmus/main/install.sh | bash

set -euo pipefail

REPO="saisumantatgit/Agent-Litmus"
BRANCH="main"
RAW_BASE="https://raw.githubusercontent.com/${REPO}/${BRANCH}"

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

echo -e "${CYAN}"
echo '  ╔═══════════════════════════════════════════╗'
echo '  ║          Agent-Litmus Installer           ║'
echo '  ║  "The test your tests have to pass."      ║'
echo '  ╚═══════════════════════════════════════════╝'
echo -e "${NC}"

# ─── Detect CLI ─────────────────────────────────────────────────────

detect_cli() {
    if command -v claude &>/dev/null; then
        echo "claude-code"
    elif command -v cursor &>/dev/null || [ -d ".cursor" ]; then
        echo "cursor"
    elif command -v codex &>/dev/null; then
        echo "codex"
    elif command -v aider &>/dev/null; then
        echo "aider"
    else
        echo "generic"
    fi
}

CLI=$(detect_cli)
echo -e "${GREEN}Detected CLI:${NC} ${CLI}"

# ─── Create directories ─────────────────────────────────────────────

echo -e "${YELLOW}Creating directories...${NC}"

mkdir -p .claude/skills/test-audit
mkdir -p .claude/skills/edge-analysis
mkdir -p .claude/skills/strength-analysis
mkdir -p .claude/skills/test-fix
mkdir -p .claude/skills/test-report
mkdir -p .claude/commands
mkdir -p .claude/agents

# ─── Download core files ─────────────────────────────────────────────

download() {
    local src="$1"
    local dst="$2"
    echo -e "  ${CYAN}Downloading${NC} ${dst}"
    curl -fsSL "${RAW_BASE}/${src}" -o "${dst}" 2>/dev/null || {
        echo -e "  ${RED}Failed:${NC} ${src}"
        return 1
    }
}

echo -e "${YELLOW}Downloading skills...${NC}"
download ".claude/skills/test-audit/SKILL.md" ".claude/skills/test-audit/SKILL.md"
download ".claude/skills/edge-analysis/SKILL.md" ".claude/skills/edge-analysis/SKILL.md"
download ".claude/skills/strength-analysis/SKILL.md" ".claude/skills/strength-analysis/SKILL.md"
download ".claude/skills/test-fix/SKILL.md" ".claude/skills/test-fix/SKILL.md"
download ".claude/skills/test-report/SKILL.md" ".claude/skills/test-report/SKILL.md"

echo -e "${YELLOW}Downloading commands...${NC}"
download ".claude/commands/litmus-scan.md" ".claude/commands/litmus-scan.md"
download ".claude/commands/litmus-edge.md" ".claude/commands/litmus-edge.md"
download ".claude/commands/litmus-strength.md" ".claude/commands/litmus-strength.md"
download ".claude/commands/litmus-fix.md" ".claude/commands/litmus-fix.md"
download ".claude/commands/litmus-report.md" ".claude/commands/litmus-report.md"

echo -e "${YELLOW}Downloading agents...${NC}"
download ".claude/agents/assertion-analyzer.md" ".claude/agents/assertion-analyzer.md"
download ".claude/agents/edge-detector.md" ".claude/agents/edge-detector.md"
download ".claude/agents/mutation-reasoner.md" ".claude/agents/mutation-reasoner.md"
download ".claude/agents/test-improver.md" ".claude/agents/test-improver.md"
download ".claude/agents/quality-aggregator.md" ".claude/agents/quality-aggregator.md"

echo -e "${YELLOW}Downloading references...${NC}"
mkdir -p references
download "references/violation-types.md" "references/violation-types.md"
download "references/assertion-classification.md" "references/assertion-classification.md"
download "references/edge-case-taxonomy.md" "references/edge-case-taxonomy.md"

echo -e "${YELLOW}Downloading template...${NC}"
mkdir -p templates
download "templates/litmus-protocol.yaml" "templates/litmus-protocol.yaml"

# ─── CLI-specific setup ─────────────────────────────────────────────

case "$CLI" in
    claude-code)
        echo -e "${YELLOW}Setting up Claude Code plugin...${NC}"
        mkdir -p .claude-plugin
        download ".claude-plugin/plugin.json" ".claude-plugin/plugin.json"
        download ".claude-plugin/hooks.json" ".claude-plugin/hooks.json"
        ;;
    cursor)
        echo -e "${YELLOW}Setting up Cursor rules...${NC}"
        mkdir -p .cursor/rules
        download "adapters/cursor/.cursor/rules/litmus.md" ".cursor/rules/litmus.md"
        ;;
    codex)
        echo -e "${YELLOW}Setting up Codex agent...${NC}"
        download "adapters/codex/AGENTS.md" "AGENTS.md"
        ;;
    aider)
        echo -e "${YELLOW}Setting up Aider config...${NC}"
        if [ -f ".aider.conf.yml" ]; then
            echo -e "  ${YELLOW}Warning:${NC} .aider.conf.yml exists. Saving Litmus config as .aider.litmus.yml"
            download "adapters/aider/.aider.conf.yml" ".aider.litmus.yml"
        else
            download "adapters/aider/.aider.conf.yml" ".aider.conf.yml"
        fi
        ;;
    generic)
        echo -e "${YELLOW}Setting up generic adapter...${NC}"
        mkdir -p prompts
        download "prompts/litmus-scan.md" "prompts/litmus-scan.md"
        download "prompts/litmus-edge.md" "prompts/litmus-edge.md"
        download "prompts/litmus-strength.md" "prompts/litmus-strength.md"
        download "prompts/litmus-fix.md" "prompts/litmus-fix.md"
        download "prompts/litmus-report.md" "prompts/litmus-report.md"
        ;;
esac

# ─── Done ────────────────────────────────────────────────────────────

echo ""
echo -e "${GREEN}Agent-Litmus installed successfully.${NC}"
echo ""
echo -e "  Commands available:"
echo -e "    ${CYAN}/litmus-scan${NC}     Scan a test file for 12 violation types"
echo -e "    ${CYAN}/litmus-edge${NC}     Identify edge cases and check coverage"
echo -e "    ${CYAN}/litmus-strength${NC} Thought-experiment mutation testing"
echo -e "    ${CYAN}/litmus-fix${NC}      Generate improved test code"
echo -e "    ${CYAN}/litmus-report${NC}   Project-wide Test Quality Score"
echo ""
echo -e "  Quick start:"
echo -e "    ${YELLOW}/litmus-scan path/to/your.test.ts${NC}"
echo ""