#!/bin/bash

# Test script to verify eBPF integration with new system prompt format

echo "🧪 Testing eBPF Integration with TensorZero System Prompt Format"
echo "=============================================================="
echo ""

# Test 1: Check if agent can parse eBPF-enhanced responses
echo "Test 1: eBPF-Enhanced Response Parsing"
echo "--------------------------------------"

cat > /tmp/test_ebpf_response.json << 'EOF'
{
  "response_type": "diagnostic",
  "reasoning": "Network timeout issues require monitoring TCP connections and system calls to identify bottlenecks at the kernel level.",
  "commands": [
    {"id": "net_status", "command": "ss -tulpn | head -10", "description": "Current network connections"},
    {"id": "net_config", "command": "ip route show", "description": "Network routing configuration"}
  ],
  "ebpf_programs": [
    {
      "name": "tcp_connect_monitor",
      "type": "kprobe", 
      "target": "tcp_connect",
      "duration": 15,
      "description": "Monitor TCP connection attempts"
    },
    {
      "name": "connect_syscalls",
      "type": "tracepoint",
      "target": "syscalls/sys_enter_connect",
      "duration": 15,
      "filters": {"comm": "curl"},
      "description": "Monitor connect() system calls from applications"
    }
  ]
}
EOF

echo "✓ Created test eBPF-enhanced response format"
echo ""

# Test 2: Check agent capabilities
echo "Test 2: Agent eBPF Capabilities"
echo "-------------------------------"
./nannyagent-ebpf test-ebpf 2>/dev/null | grep -E "(eBPF|Capabilities|Programs)" || echo "No eBPF output found"
echo ""

# Test 3: Validate JSON format
echo "Test 3: JSON Format Validation"
echo "------------------------------"
if python3 -m json.tool /tmp/test_ebpf_response.json > /dev/null 2>&1; then
    echo "✓ JSON format is valid"
else
    echo "❌ JSON format is invalid"
fi
echo ""

# Test 4: Show eBPF program categories from system prompt
echo "Test 4: eBPF Program Categories (from system prompt)"
echo "---------------------------------------------------"
echo "📡 NETWORK issues:"
echo "   - tracepoint:syscalls/sys_enter_connect"
echo "   - kprobe:tcp_connect"
echo "   - kprobe:tcp_sendmsg"
echo ""
echo "🔄 PROCESS issues:"
echo "   - tracepoint:syscalls/sys_enter_execve" 
echo "   - tracepoint:sched/sched_process_exit"
echo "   - kprobe:do_fork"
echo ""
echo "📁 FILE I/O issues:"
echo "   - tracepoint:syscalls/sys_enter_openat"
echo "   - kprobe:vfs_read"
echo "   - kprobe:vfs_write"
echo ""
echo "⚡ PERFORMANCE issues:"
echo "   - tracepoint:syscalls/sys_enter_*"
echo "   - kprobe:schedule"
echo "   - tracepoint:irq/irq_handler_entry"
echo ""

# Test 5: Resolution response format
echo "Test 5: Resolution Response Format"
echo "---------------------------------"
cat > /tmp/test_resolution_response.json << 'EOF'
{
  "response_type": "resolution",
  "root_cause": "TCP connection timeouts are caused by iptables dropping packets on port 443 due to misconfigured firewall rules.",
  "resolution_plan": "1. Check iptables rules with 'sudo iptables -L -n'\n2. Remove blocking rule: 'sudo iptables -D INPUT -p tcp --dport 443 -j DROP'\n3. Verify connectivity: 'curl -I https://example.com'\n4. Persist rules: 'sudo iptables-save > /etc/iptables/rules.v4'",
  "confidence": "High",
  "ebpf_evidence": "eBPF tcp_connect traces show 127 connection attempts with immediate failures. System call monitoring revealed iptables netfilter hooks rejecting packets before reaching the application layer."
}
EOF

if python3 -m json.tool /tmp/test_resolution_response.json > /dev/null 2>&1; then
    echo "✓ Resolution response format is valid"
else
    echo "❌ Resolution response format is invalid"
fi
echo ""

echo "🎯 Integration Test Summary"
echo "=========================="
echo "✅ eBPF-enhanced diagnostic response format ready"
echo "✅ Resolution response format with eBPF evidence ready"  
echo "✅ System prompt includes comprehensive eBPF instructions"
echo "✅ Agent supports both traditional and eBPF-enhanced diagnostics"
echo ""
echo "📋 Next Steps:"
echo "1. Deploy the updated system prompt to TensorZero"
echo "2. Test with real network/process/file issues"
echo "3. Verify AI model understands eBPF program requests"
echo "4. Monitor eBPF trace data quality and completeness"
echo ""
echo "🔧 TensorZero Configuration:"
echo "   - Copy content from TENSORZERO_SYSTEM_PROMPT.md"
echo "   - Ensure model supports structured JSON responses"
echo "   - Test with sample diagnostic scenarios"

# Cleanup
rm -f /tmp/test_ebpf_response.json /tmp/test_resolution_response.json