#!/bin/bash

# Final Sprocket Integration Test
# This script tests the complete sprocket integration with the relay

set -e

echo "🧪 Final Sprocket Integration Test"
echo "================================="

# Configuration
RELAY_PORT="3334"
TEST_CONFIG_DIR="$HOME/.config/ORLY_TEST"

# Clean up any existing test processes
echo "🧹 Cleaning up existing processes..."
pkill -f "ORLY_TEST" || true
sleep 2

# Create test configuration directory
echo "📁 Setting up test environment..."
mkdir -p "$TEST_CONFIG_DIR"

# Copy the Python sprocket script
cp test-sprocket.py "$TEST_CONFIG_DIR/sprocket.py"

# Create bash wrapper for the Python script
cat > "$TEST_CONFIG_DIR/sprocket.sh" << 'EOF'
#!/bin/bash
python3 "$(dirname "$0")/sprocket.py"
EOF

chmod +x "$TEST_CONFIG_DIR/sprocket.sh"

echo "✅ Sprocket script created at: $TEST_CONFIG_DIR/sprocket.sh"

# Set environment variables for the relay
export ORLY_APP_NAME="ORLY_TEST"
export ORLY_DATA_DIR="/tmp/orly_test_data"
export ORLY_LISTEN="127.0.0.1"
export ORLY_PORT="$RELAY_PORT"
export ORLY_LOG_LEVEL="info"
export ORLY_SPROCKET_ENABLED="true"
export ORLY_ADMINS=""
export ORLY_OWNERS=""

# Clean up test data directory
rm -rf "$ORLY_DATA_DIR"
mkdir -p "$ORLY_DATA_DIR"

# Function to cleanup
cleanup() {
    echo "🧹 Cleaning up..."
    pkill -f "ORLY_TEST" || true
    sleep 2
    rm -rf "$ORLY_DATA_DIR"
    echo "✅ Cleanup complete"
}

# Set trap for cleanup
trap cleanup EXIT

# Start the relay
echo "🚀 Starting relay with sprocket enabled..."
go run . test > /tmp/orly_test.log 2>&1 &
RELAY_PID=$!

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

# Check if relay is running
if ! kill -0 $RELAY_PID 2>/dev/null; then
    echo "❌ Relay failed to start"
    echo "Log output:"
    cat /tmp/orly_test.log
    exit 1
fi

echo "✅ Relay started successfully (PID: $RELAY_PID)"

# Check if websocat is available
if ! command -v websocat &> /dev/null; then
    echo "❌ websocat is required for testing"
    echo "Install it with: cargo install websocat"
    echo "Or use: go install github.com/gorilla/websocket/examples/echo@latest"
    exit 1
fi

# Test sprocket functionality
echo "🧪 Testing sprocket functionality..."

# Test 1: Normal event (should be accepted)
echo "📤 Test 1: Normal event (should be accepted)"
current_time=$(date +%s)
normal_event="{
    \"id\": \"test_normal_123\",
    \"pubkey\": \"1234567890abcdef1234567890abcdef12345678\",
    \"created_at\": $current_time,
    \"kind\": 1,
    \"content\": \"Hello, world! This is a normal message.\",
    \"sig\": \"test_sig_normal\"
}"

normal_message="[\"EVENT\",$normal_event]"
normal_response=$(echo "$normal_message" | websocat "ws://127.0.0.1:$RELAY_PORT" --text)
echo "Response: $normal_response"

if echo "$normal_response" | grep -q '"OK","test_normal_123",true'; then
    echo "✅ Test 1 PASSED: Normal event accepted"
else
    echo "❌ Test 1 FAILED: Normal event not accepted"
fi

# Test 2: Spam content (should be rejected)
echo "📤 Test 2: Spam content (should be rejected)"
spam_event="{
    \"id\": \"test_spam_456\",
    \"pubkey\": \"1234567890abcdef1234567890abcdef12345678\",
    \"created_at\": $current_time,
    \"kind\": 1,
    \"content\": \"This message contains spam content\",
    \"sig\": \"test_sig_spam\"
}"

spam_message="[\"EVENT\",$spam_event]"
spam_response=$(echo "$spam_message" | websocat "ws://127.0.0.1:$RELAY_PORT" --text)
echo "Response: $spam_response"

if echo "$spam_response" | grep -q '"OK","test_spam_456",false'; then
    echo "✅ Test 2 PASSED: Spam content rejected"
else
    echo "❌ Test 2 FAILED: Spam content not rejected"
fi

# Test 3: Test kind 9999 (should be shadow rejected)
echo "📤 Test 3: Test kind 9999 (should be shadow rejected)"
kind_event="{
    \"id\": \"test_kind_789\",
    \"pubkey\": \"1234567890abcdef1234567890abcdef12345678\",
    \"created_at\": $current_time,
    \"kind\": 9999,
    \"content\": \"Test message with special kind\",
    \"sig\": \"test_sig_kind\"
}"

kind_message="[\"EVENT\",$kind_event]"
kind_response=$(echo "$kind_message" | websocat "ws://127.0.0.1:$RELAY_PORT" --text)
echo "Response: $kind_response"

if echo "$kind_response" | grep -q '"OK","test_kind_789",true'; then
    echo "✅ Test 3 PASSED: Test kind shadow rejected (OK=true but not processed)"
else
    echo "❌ Test 3 FAILED: Test kind not shadow rejected"
fi

# Test 4: Blocked hashtag (should be rejected)
echo "📤 Test 4: Blocked hashtag (should be rejected)"
hashtag_event="{
    \"id\": \"test_hashtag_101\",
    \"pubkey\": \"1234567890abcdef1234567890abcdef12345678\",
    \"created_at\": $current_time,
    \"kind\": 1,
    \"content\": \"Message with blocked hashtag\",
    \"tags\": [[\"t\", \"blocked\"]],
    \"sig\": \"test_sig_hashtag\"
}"

hashtag_message="[\"EVENT\",$hashtag_event]"
hashtag_response=$(echo "$hashtag_message" | websocat "ws://127.0.0.1:$RELAY_PORT" --text)
echo "Response: $hashtag_response"

if echo "$hashtag_response" | grep -q '"OK","test_hashtag_101",false'; then
    echo "✅ Test 4 PASSED: Blocked hashtag rejected"
else
    echo "❌ Test 4 FAILED: Blocked hashtag not rejected"
fi

echo ""
echo "🎉 Sprocket integration test completed!"
echo "📊 Check the results above to verify sprocket functionality"
echo ""
echo "📝 Relay logs are available at: /tmp/orly_test.log"
echo "💡 To view logs: cat /tmp/orly_test.log"