fix: Wait for connection to be established before returning
- Add waitForConnection to ensure store is updated - Poll isConnected state with timeout - Prevents tool registration before connection is ready
This commit is contained in:
@@ -404,9 +404,39 @@ export async function connectWithToken(token: string): Promise<boolean> {
|
|||||||
// Connect passing the token directly
|
// Connect passing the token directly
|
||||||
try {
|
try {
|
||||||
await webmcpInstance.connect(finalToken)
|
await webmcpInstance.connect(finalToken)
|
||||||
return true
|
|
||||||
|
// Wait for connection to be fully established (event handler updates store)
|
||||||
|
const connected = await waitForConnection(3000)
|
||||||
|
if (!connected) {
|
||||||
|
console.warn('[WebMCP] Connection timeout - store not updated')
|
||||||
|
}
|
||||||
|
return connected
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('[WebMCP] Failed to connect:', e)
|
console.error('[WebMCP] Failed to connect:', e)
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Wait for isConnected to become true
|
||||||
|
*/
|
||||||
|
async function waitForConnection(timeoutMs: number = 3000): Promise<boolean> {
|
||||||
|
const canvasStore = useCanvasStore()
|
||||||
|
|
||||||
|
// Already connected
|
||||||
|
if (canvasStore.isConnected) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for connection with polling
|
||||||
|
const startTime = Date.now()
|
||||||
|
while (Date.now() - startTime < timeoutMs) {
|
||||||
|
await new Promise(resolve => setTimeout(resolve, 100))
|
||||||
|
if (canvasStore.isConnected) {
|
||||||
|
console.log('[WebMCP] Connection confirmed')
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user