diff --git a/src/services/apis/chatgpt-web.mjs b/src/services/apis/chatgpt-web.mjs index 3b98452..1938be0 100644 --- a/src/services/apis/chatgpt-web.mjs +++ b/src/services/apis/chatgpt-web.mjs @@ -127,8 +127,13 @@ export async function registerWebsocket(accessToken) { const response = JSON.parse( (await request(accessToken, 'POST', '/register-websocket')).responseText, ) + let resolve if (response.wss_url) { websocket = new WebSocket(response.wss_url) + websocket.onopen = () => { + console.debug('global websocket opened') + resolve() + } websocket.onclose = () => { websocket = null expires_at = null @@ -139,6 +144,7 @@ export async function registerWebsocket(accessToken) { } expires_at = new Date(response.expires_at) } + return new Promise((r) => (resolve = r)) } /**