setup.sh

#!/bin/bash
set -e

echo "🐉 Kali Linux MCP Server for Docker Desktop"
echo "==========================================="
echo ""
echo "⚠️  DISK SPACE REQUIREMENT: ~20GB"
echo "   This installation uses kali-linux-everything (ALL tools)"
echo ""

# Check if Docker is running
if ! docker info > /dev/null 2>&1; then
    echo "❌ Error: Docker is not running. Please start Docker Desktop."
    exit 1
fi

echo "✅ Docker is running"

# Check if Docker Compose is available
if ! docker-compose --version > /dev/null 2>&1; then
    echo "❌ Error: docker-compose not found. Please install Docker Compose."
    exit 1
fi

echo "✅ Docker Compose is available"

# Check if docker mcp command is available
if ! docker mcp version > /dev/null 2>&1; then
    echo "⚠️  Warning: 'docker mcp' command not found."
    echo "   This is required for Docker Desktop MCP Toolkit integration."
    echo "   Make sure you have Docker Desktop with MCP Toolkit support."
    echo ""
    read -p "Continue anyway? (y/N) " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        exit 1
    fi
fi

echo "✅ Docker MCP CLI is available"

# Build the MCP server
echo ""
echo "📦 Building MCP server container..."
docker-compose build

# Start the services
echo ""
echo "🚀 Starting services..."
docker-compose up -d

# Wait for containers to start
echo ""
echo "⏳ Waiting for containers to start..."
sleep 5

# Check if containers are running
if docker ps | grep -q kali-mcp-server && docker ps | grep -q kali-tools-container; then
    echo ""
    echo "✅ Services started successfully!"
    echo ""
    echo "📊 Container Status:"
    docker-compose ps
    echo ""
    
    # Add to Docker Desktop MCP Toolkit
    echo "🔧 Adding to Docker Desktop MCP Toolkit..."
    echo ""
    
    # Check if catalog already exists
    if docker mcp catalog ls 2>/dev/null | grep -q "my-security-tools"; then
        echo "⚠️  Catalog 'my-security-tools' already exists."
        read -p "Recreate it? This will overwrite existing configuration. (y/N) " -n 1 -r
        echo
        if [[ $REPLY =~ ^[Yy]$ ]]; then
            echo "🗑️  Removing existing catalog..."
            docker mcp server disable kali-linux-tools 2>/dev/null || true
            docker mcp catalog rm my-security-tools 2>/dev/null || true
        else
            echo "ℹ️  Skipping catalog creation. Using existing catalog."
        fi
    fi
    
    # Create custom catalog if it doesn't exist
    if ! docker mcp catalog ls 2>/dev/null | grep -q "my-security-tools"; then
        echo "📚 Creating custom catalog..."
        docker mcp catalog create my-security-tools
        echo "✅ Custom catalog created"
    fi
    
    # Add Kali server to catalog
    echo "➕ Adding Kali server to catalog..."
    if docker mcp catalog add my-security-tools kali-linux-tools ./kali-catalog.yaml --force; then
        echo "✅ Kali server added to catalog"
    else
        echo "❌ Failed to add Kali server to catalog"
        echo "You can add it manually with:"
        echo "   docker mcp catalog add my-security-tools kali-linux-tools ./kali-catalog.yaml"
    fi
    
    # Enable the server
    echo "🔌 Enabling Kali server..."
    if docker mcp server enable kali-linux-tools; then
        echo "✅ Kali server enabled"
    else
        echo "⚠️  Server may already be enabled or failed to enable"
        echo "You can enable it manually with:"
        echo "   docker mcp server enable kali-linux-tools"
    fi
    
    # Verify installation
    echo ""
    echo "🔍 Verifying installation..."
    if docker mcp server ls 2>/dev/null | grep -q "kali-linux-tools"; then
        echo "✅ Kali Linux Security Tools is now available in Docker Desktop!"
    else
        echo "⚠️  Server may not be visible yet. Try restarting Docker Desktop."
    fi
    
    echo ""
    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
    echo "🎉 Installation Complete!"
    echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
    echo ""
    echo "📝 Next Steps:"
    echo "   1. Open Docker Desktop"
    echo "   2. Go to MCP Toolkit in the sidebar"
    echo "   3. Look for 'Kali Linux Security Tools' in your servers"
    echo "   4. Connect to Ask Gordon or your preferred AI client"
    echo ""
    echo "🧪 Test It:"
    echo "   Ask Gordon: 'List available Kali tools using kali_list_tools'"
    echo "   Claude Desktop: 'Use kali_help to show me how to use nmap'"
    echo "   VS Code: Run 'docker mcp client connect vscode' first"
    echo ""
    echo "⏱️  First Run: ALL Kali tools will be installed (1-2 hours, ~15GB)"
    echo "   Monitor progress: docker-compose logs -f kali-mcp-server"
    echo ""
    echo "🔧 Useful Commands:"
    echo "   List servers:  docker mcp server ls"
    echo "   View catalog:  docker mcp catalog show my-security-tools"
    echo "   View logs:     docker-compose logs -f"
    echo "   Restart:       docker-compose restart"
    echo "   Stop:          docker-compose down"
    echo "   Disable:       docker mcp server disable kali-linux-tools"
    echo ""
    echo "⚠️  Security Warning:"
    echo "   Only use these tools on systems you own or have permission to test!"
    echo ""
else
    echo ""
    echo "❌ Error: Containers failed to start"
    echo "Check logs with: docker-compose logs"
    exit 1
fi