diff --git a/src/gateway/opcodes/Identify.ts b/src/gateway/opcodes/Identify.ts index e9b91a3d5..77421765b 100644 --- a/src/gateway/opcodes/Identify.ts +++ b/src/gateway/opcodes/Identify.ts @@ -96,6 +96,8 @@ export async function onIdentify(this: WebSocket, data: Payload) { }), ); + this.accessToken = identify.token; + taskSw.reset(); // don't include checkToken time... const user = tokenData.user; @@ -634,7 +636,7 @@ export async function onIdentify(this: WebSocket, data: Payload) { }); if (this.capabilities.has(Capabilities.FLAGS.AUTH_TOKEN_REFRESH) && tokenData.tokenVersion != CurrentTokenFormatVersion) { - d.auth_token = await generateToken(this.user_id); + d.auth_token = this.accessToken = (await generateToken(this.user_id))!; } // const buildReadyEventDataTime = taskSw.getElapsedAndReset(); diff --git a/src/gateway/util/WebSocket.ts b/src/gateway/util/WebSocket.ts index af0854215..301b0515d 100644 --- a/src/gateway/util/WebSocket.ts +++ b/src/gateway/util/WebSocket.ts @@ -27,6 +27,7 @@ export interface WebSocket extends WS { version: number; user_id: string; session_id: string; + accessToken: string; encoding: "etf" | "json"; compress?: "zlib-stream" | "zstd-stream"; ipAddress?: string;