#!/bin/bash

set -e 

echo "=================================================="
echo "MCP Registry Publish Endpoint Test Script"
echo "=================================================="
echo "This script expects the MCP Registry server to be running locally."
echo "Please ensure the server is started using one of the following methods:"
echo "  • Docker Compose: docker compose up"
echo "  • Direct execution: go run cmd/registry/main.go"
echo "  • Built binary: ./build/registry"
echo ""
echo "REQUIRED: Set the BEARER_TOKEN environment variable with a valid GitHub token"
echo "Example: export BEARER_TOKEN=your_github_token_here"
echo "=================================================="
echo ""

# Default values
HOST="http://localhost:8080"
VERBOSE=false

# Display usage information
function show_usage {
  echo "Usage: $0 [options]"
  echo "Options:"
  echo "  -h, --host      Base URL of the MCP Registry service (default: http://localhost:8080)"
  echo "  -v, --verbose   Show verbose output including full request payload"
  echo "  --help          Show this help message"
  echo ""
  echo "Environment Variables:"
  echo "  BEARER_TOKEN    Required: GitHub token for authentication"
  exit 1
}

# Check if bearer token is set
if [[ -z "$BEARER_TOKEN" ]]; then
  echo "Error: BEARER_TOKEN environment variable is not set."
  echo "Please set your GitHub token as an environment variable:"
  echo "  export BEARER_TOKEN=your_github_token_here"
  exit 1
fi

# Check if jq is installed
if ! command -v jq &> /dev/null; then
  echo "Error: jq is required but not installed."
  echo "Please install jq using your package manager, for example:"
  echo "  brew install jq (macOS)"
  echo "  apt-get install jq (Debian/Ubuntu)"
  echo "  yum install jq (CentOS/RHEL)"
  exit 1
fi

# Check if the API is running
echo "Checking if the MCP Registry API is running at $HOST..."
health_check=$(curl -s -o /dev/null -w "%{http_code}" "$HOST/v0/health" 2>/dev/null)
if [[ "$health_check" != "200" ]]; then
  echo "Error: MCP Registry API is not running at $HOST (health check returned $health_check)"
  echo "Please start the server using one of the methods mentioned above and try again."
  exit 1
else
  echo "✓ MCP Registry API is running at $HOST"
fi

# Parse command line arguments
while [[ "$#" -gt 0 ]]; do
  case $1 in
    -h|--host) HOST="$2"; shift ;;
    -v|--verbose) VERBOSE=true ;;
    --help) show_usage ;;
    *) echo "Unknown parameter: $1"; show_usage ;;
  esac
  shift
done

# Create a temporary file for our JSON payload
PAYLOAD_FILE=$(mktemp)

# Create sample server detail payload based on current model structure
cat > "$PAYLOAD_FILE" << EOF
{
  "name": "io.github.example/test-mcp-server",
  "description": "A test server for MCP Registry validation - published at $(date)",
  "repository": {
    "url": "https://github.com/example/test-mcp-server",
    "source": "github",
    "id": "example/test-mcp-server"
  },
  "version_detail": {
    "version": "1.0.$(date +%s)"
  },
  "packages": [
    {
      "registry_name": "npm",
      "name": "test-mcp-server",
      "version": "1.0.$(date +%s)",
      "runtime_hint": "node",
      "runtime_arguments": [
        {
          "type": "positional",
          "name": "config",
          "description": "Configuration file path",
          "format": "file_path",
          "is_required": false,
          "default": "./config.json"
        }
      ],
      "environment_variables": [
        {
          "name": "PORT",
          "description": "Port to run the server on",
          "format": "number",
          "is_required": false,
          "default": "3000"
        },
        {
          "name": "API_KEY",
          "description": "API key for external service",
          "format": "string",
          "is_required": true,
          "is_secret": true
        }
      ]
    }
  ]
}
EOF

# Show the payload if verbose mode is enabled
if $VERBOSE; then
  echo "Request Payload:"
  cat "$PAYLOAD_FILE" | jq '.'
  echo "-------------------------------------"
fi

# Test publish endpoint
echo "Testing publish endpoint: $HOST/v0/publish"
echo "Using Bearer Token: ${BEARER_TOKEN:0:10}..." # Show only first 10 chars for security

# Get response and status code in a single request
response_file=$(mktemp)
headers_file=$(mktemp)

# Execute curl with response body to file and headers+status to another file
curl -s -X POST \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer ${BEARER_TOKEN}" \
  -d "@$PAYLOAD_FILE" \
  -D "$headers_file" \
  -o "$response_file" \
  "$HOST/v0/publish"

# Read the response body
http_response=$(<"$response_file")

# Extract the status code from the headers file
status_code=$(head -n1 "$headers_file" | awk '{print $2}')

# Clean up temp files
rm "$response_file" "$headers_file"

echo "Status Code: $status_code"

# Check for status code in 2xx range (200, 201, 202, etc)
if [[ "${status_code:0:1}" == "2" ]]; then
  # Parse JSON response with jq
  echo "Response:"
  echo "$http_response" | jq '.' 2>/dev/null || echo "$http_response"
  
  # Check for server added message and extract UUID
  message=$(echo "$http_response" | jq -r '.message // empty' 2>/dev/null)
  server_id=$(echo "$http_response" | jq -r '.id // .server_id // empty' 2>/dev/null)
  
  # Validate the response contains success indicators
  success_indicators=0
  
  if [[ ! -z "$message" && "$message" != "null" ]]; then
    echo "✓ Success message received: $message"
    if [[ "$message" == *"server"* && ("$message" == *"added"* || "$message" == *"published"* || "$message" == *"created"*) ]]; then
      ((success_indicators++))
      echo "✓ Message indicates server was successfully added"
    fi
  fi
  
  if [[ ! -z "$server_id" && "$server_id" != "null" && "$server_id" != "empty" ]]; then
    echo "✓ Server UUID received: $server_id"
    # Validate UUID format (basic check for UUID pattern)
    if [[ "$server_id" =~ ^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}$ ]]; then
      ((success_indicators++))
      echo "✓ Server ID appears to be a valid UUID format"
    else
      echo "⚠ Server ID format may not be a standard UUID: $server_id"
      ((success_indicators++)) # Still count as success if we got an ID
    fi
  fi
  
  if [[ $success_indicators -ge 2 ]]; then
    echo ""
    echo "🎉 PUBLISH TEST PASSED!"
    echo "   ✓ Server successfully published with ID: $server_id"
    echo "   ✓ Success message: $message"
  else
    echo ""
    echo "❌ PUBLISH TEST FAILED!"
    echo "   Expected: Success message about server being added AND a server UUID"
    echo "   Received: message='$message', id='$server_id'"
    exit 1
  fi
  
else
  echo ""
  echo "❌ PUBLISH TEST FAILED!"
  echo "   Expected: 2xx status code"
  echo "   Received: $status_code"
  echo "   Response:"
  echo "$http_response" | jq '.' 2>/dev/null || echo "$http_response"
  exit 1
fi

echo "-------------------------------------"

# Clean up temp file
rm "$PAYLOAD_FILE"