CVE MCP Server

#!/bin/bash # Script to launch MCP Inspector for testing CVE MCP Server # Supports both local and Docker deployments set -e # Exit on error echo "=========================================" echo "MCP Inspector Testing Setup" echo "=========================================" echo "" # Check if npx is available if ! command -v npx &> /dev/null; then echo "Error: npx is not installed (part of Node.js)" echo "Please install Node.js from https://nodejs.org/" exit 1 fi # Detect deployment type echo "Which deployment do you want to test?" echo "" echo "1) Local (Python virtual environment)" echo "2) Docker container" echo "" read -p "Enter choice (1 or 2): " -r CHOICE echo "" case $CHOICE in 1) DEPLOYMENT="local" echo "Selected: Local deployment" echo "" # Check if virtual environment is activated if [ -z "$VIRTUAL_ENV" ]; then echo "Warning: Virtual environment not detected." echo "Please activate it first: source venv/bin/activate" exit 1 fi # Check if database exists if [ ! -f "data/cve.db" ]; then echo "Warning: Database not found at data/cve.db" echo "" echo "Load data first with one of these:" echo " ./run_load_limited.sh (quick test, ~100 CVEs)" echo " ./run_load_optimized.sh (full dataset, ~240K CVEs)" echo "" read -p "Continue anyway? (y/N) " -n 1 -r echo if [[ ! $REPLY =~ ^[Yy]$ ]]; then exit 1 fi fi COMMAND="python" ARGUMENTS="-m src.mcp_server" ENV_PYTHONPATH="$(pwd)" ;; 2) DEPLOYMENT="docker" echo "Selected: Docker deployment" echo "" # Check if Docker container is running if ! docker ps | grep -q cve-mcp-server; then echo "Error: Docker container 'cve-mcp-server' is not running" echo "" echo "Start it with: ./run_docker.sh" exit 1 fi # Check if database exists in container if ! docker exec cve-mcp-server test -f /app/data/cve.db 2>/dev/null; then echo "Warning: Database not found in container" echo "" echo "Load data with:" echo " docker exec cve-mcp-server python -m src.data_ingestion.loader --year 2024 --limit 100" echo "" read -p "Continue anyway? (y/N) " -n 1 -r echo if [[ ! $REPLY =~ ^[Yy]$ ]]; then exit 1 fi fi COMMAND="docker" ARGUMENTS="exec -i cve-mcp-server python -m src.mcp_server" ENV_PYTHONPATH="/app" ;; *) echo "Invalid choice. Exiting." exit 1 ;; esac echo "" echo "=========================================" echo "Launching MCP Inspector..." echo "=========================================" echo "" echo "Configuration for MCP Inspector:" echo "" echo " Command: $COMMAND" echo " Arguments: $ARGUMENTS" echo " Environment: PYTHONPATH=$ENV_PYTHONPATH" echo "" echo "Once the Inspector UI opens:" echo "1. Enter the configuration above in the connection settings" echo "2. Click 'Connect'" echo "3. Test the three available tools:" echo " - search_cves: Search for CVEs by keyword" echo " - get_cve_details: Get details for a specific CVE" echo " - get_stats: Get database statistics" echo "" echo "Press any key to launch MCP Inspector..." read -n 1 -s echo "" echo "Starting MCP Inspector (will open in browser)..." echo "" # Launch MCP Inspector npx @modelcontextprotocol/inspector echo "" echo "MCP Inspector closed." echo ""