Major refactor: Multi-user Chrome MCP extension with remote server architecture
This commit is contained in:
132
app/remote-server/test-streamable-http-client.js
Normal file
132
app/remote-server/test-streamable-http-client.js
Normal file
@@ -0,0 +1,132 @@
|
||||
/**
|
||||
* Test client for Streamable HTTP connection
|
||||
*/
|
||||
|
||||
import fetch from 'node-fetch';
|
||||
import { EventSource } from 'eventsource';
|
||||
|
||||
const SERVER_URL = 'http://localhost:3001';
|
||||
const MCP_URL = `${SERVER_URL}/mcp`;
|
||||
|
||||
console.log('🔌 Testing Streamable HTTP connection...');
|
||||
|
||||
let sessionId = null;
|
||||
|
||||
async function testStreamableHttp() {
|
||||
try {
|
||||
// Step 1: Send initialization request
|
||||
console.log('🚀 Sending initialization request...');
|
||||
|
||||
const initMessage = {
|
||||
jsonrpc: '2.0',
|
||||
id: 1,
|
||||
method: 'initialize',
|
||||
params: {
|
||||
protocolVersion: '2024-11-05',
|
||||
capabilities: {
|
||||
tools: {},
|
||||
},
|
||||
clientInfo: {
|
||||
name: 'test-streamable-http-client',
|
||||
version: '1.0.0',
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
const initResponse = await fetch(MCP_URL, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
Accept: 'application/json, text/event-stream',
|
||||
},
|
||||
body: JSON.stringify(initMessage),
|
||||
});
|
||||
|
||||
if (!initResponse.ok) {
|
||||
throw new Error(`Initialization failed: ${initResponse.status} ${initResponse.statusText}`);
|
||||
}
|
||||
|
||||
// Extract session ID from response headers
|
||||
sessionId = initResponse.headers.get('mcp-session-id');
|
||||
console.log(`✅ Initialization successful! Session ID: ${sessionId}`);
|
||||
|
||||
const initResult = await initResponse.text();
|
||||
console.log('📨 Initialization response (SSE):', initResult);
|
||||
|
||||
// Step 2: Establish SSE stream for this session
|
||||
console.log('🔌 Establishing SSE stream...');
|
||||
|
||||
const eventSource = new EventSource(MCP_URL, {
|
||||
headers: {
|
||||
'MCP-Session-ID': sessionId,
|
||||
},
|
||||
});
|
||||
|
||||
eventSource.onopen = () => {
|
||||
console.log('✅ SSE stream established!');
|
||||
// Test listing tools after stream is ready
|
||||
setTimeout(() => testListTools(), 1000);
|
||||
};
|
||||
|
||||
eventSource.onmessage = (event) => {
|
||||
try {
|
||||
const data = JSON.parse(event.data);
|
||||
console.log('📨 Received streaming message:', JSON.stringify(data, null, 2));
|
||||
} catch (error) {
|
||||
console.log('📨 Received streaming data:', event.data);
|
||||
}
|
||||
};
|
||||
|
||||
eventSource.onerror = (error) => {
|
||||
console.error('❌ SSE stream error:', error);
|
||||
};
|
||||
|
||||
// Close after 10 seconds
|
||||
setTimeout(() => {
|
||||
console.log('⏰ Closing connections...');
|
||||
eventSource.close();
|
||||
process.exit(0);
|
||||
}, 10000);
|
||||
} catch (error) {
|
||||
console.error('❌ Error in streamable HTTP test:', error);
|
||||
process.exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
async function testListTools() {
|
||||
if (!sessionId) {
|
||||
console.error('❌ No session ID available');
|
||||
return;
|
||||
}
|
||||
|
||||
console.log('📋 Testing tools/list via Streamable HTTP...');
|
||||
|
||||
const message = {
|
||||
jsonrpc: '2.0',
|
||||
id: 2,
|
||||
method: 'tools/list',
|
||||
params: {},
|
||||
};
|
||||
|
||||
try {
|
||||
const response = await fetch(MCP_URL, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'MCP-Session-ID': sessionId,
|
||||
},
|
||||
body: JSON.stringify(message),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
console.error('❌ Failed to send tools/list:', response.status, response.statusText);
|
||||
} else {
|
||||
console.log('✅ tools/list message sent successfully');
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('❌ Error sending tools/list:', error);
|
||||
}
|
||||
}
|
||||
|
||||
// Start the test
|
||||
testStreamableHttp();
|
Reference in New Issue
Block a user