mirror of
https://github.com/tiramisulabs/seyfert.git
synced 2025-07-02 04:56:07 +00:00
feat: shard#ping method
This commit is contained in:
parent
c409628a43
commit
5338248ff8
@ -1,38 +1,54 @@
|
||||
import NodeWebSocket from 'ws';
|
||||
|
||||
export class BaseSocket {
|
||||
private internal: NodeWebSocket | WebSocket;
|
||||
|
||||
constructor(kind: 'ws' | 'bun', url: string) {
|
||||
this.internal = kind === 'ws' ? new NodeWebSocket(url) : new WebSocket(url);
|
||||
}
|
||||
|
||||
set onopen(callback: NodeWebSocket['onopen']) {
|
||||
this.internal.onopen = callback;
|
||||
}
|
||||
|
||||
set onmessage(callback: NodeWebSocket['onmessage']) {
|
||||
this.internal.onmessage = callback;
|
||||
}
|
||||
|
||||
set onclose(callback: NodeWebSocket['onclose']) {
|
||||
this.internal.onclose = callback;
|
||||
}
|
||||
|
||||
set onerror(callback: NodeWebSocket['onerror']) {
|
||||
this.internal.onerror = callback;
|
||||
}
|
||||
|
||||
send(data: string) {
|
||||
return this.internal.send(data);
|
||||
}
|
||||
|
||||
close(...args: Parameters<NodeWebSocket['close']>) {
|
||||
// @ts-expect-error
|
||||
return this.internal.close(...args);
|
||||
}
|
||||
|
||||
get readyState() {
|
||||
return this.internal.readyState;
|
||||
}
|
||||
}
|
||||
import { randomUUID } from 'node:crypto';
|
||||
import NodeWebSocket from 'ws';
|
||||
|
||||
export class BaseSocket {
|
||||
private internal: NodeWebSocket | WebSocket;
|
||||
|
||||
constructor(kind: 'ws' | 'bun', url: string) {
|
||||
this.internal = kind === 'ws' ? new NodeWebSocket(url) : new WebSocket(url);
|
||||
}
|
||||
|
||||
set onopen(callback: NodeWebSocket['onopen']) {
|
||||
this.internal.onopen = callback;
|
||||
}
|
||||
|
||||
set onmessage(callback: NodeWebSocket['onmessage']) {
|
||||
this.internal.onmessage = callback;
|
||||
}
|
||||
|
||||
set onclose(callback: NodeWebSocket['onclose']) {
|
||||
this.internal.onclose = callback;
|
||||
}
|
||||
|
||||
set onerror(callback: NodeWebSocket['onerror']) {
|
||||
this.internal.onerror = callback;
|
||||
}
|
||||
|
||||
send(data: string) {
|
||||
return this.internal.send(data);
|
||||
}
|
||||
|
||||
close(...args: Parameters<NodeWebSocket['close']>) {
|
||||
// @ts-expect-error
|
||||
return this.internal.close(...args);
|
||||
}
|
||||
|
||||
async ping() {
|
||||
if (!('ping' in this.internal)) throw new Error('Unexpected: Method ping not implemented');
|
||||
return new Promise<number>(res => {
|
||||
const nonce = randomUUID();
|
||||
const start = performance.now();
|
||||
const listener = (data: Buffer) => {
|
||||
if (data.toString() !== nonce) return;
|
||||
(this.internal as NodeWebSocket).removeListener('pong', listener);
|
||||
res(performance.now() - start);
|
||||
};
|
||||
(this.internal as NodeWebSocket).on('pong', listener);
|
||||
(this.internal as NodeWebSocket).ping(nonce);
|
||||
});
|
||||
}
|
||||
|
||||
get readyState() {
|
||||
return this.internal.readyState;
|
||||
}
|
||||
}
|
||||
|
||||
@ -1,327 +1,332 @@
|
||||
import type { GatewayReceivePayload, GatewaySendPayload } from 'discord-api-types/v10';
|
||||
import { GatewayCloseCodes, GatewayDispatchEvents, GatewayOpcodes } from 'discord-api-types/v10';
|
||||
import { inflateSync } from 'node:zlib';
|
||||
import type WS from 'ws';
|
||||
import { WebSocket, type CloseEvent, type ErrorEvent } from 'ws';
|
||||
import type { Logger } from '../../common';
|
||||
import { properties } from '../constants';
|
||||
import { DynamicBucket } from '../structures';
|
||||
import { ConnectTimeout } from '../structures/timeout';
|
||||
import { BaseSocket } from './basesocket';
|
||||
import type { ShardData, ShardOptions } from './shared';
|
||||
import { ShardSocketCloseCodes } from './shared';
|
||||
|
||||
export interface ShardHeart {
|
||||
interval: number;
|
||||
nodeInterval?: NodeJS.Timeout;
|
||||
lastAck?: number;
|
||||
lastBeat?: number;
|
||||
ack: boolean;
|
||||
}
|
||||
|
||||
export class Shard {
|
||||
debugger?: Logger;
|
||||
data: Partial<ShardData> | ShardData = {
|
||||
resumeSeq: null,
|
||||
};
|
||||
|
||||
websocket: BaseSocket | null = null;
|
||||
connectTimeout = new ConnectTimeout();
|
||||
heart: ShardHeart = {
|
||||
interval: 30e3,
|
||||
ack: true,
|
||||
};
|
||||
|
||||
bucket: DynamicBucket;
|
||||
offlineSendQueue: ((_?: unknown) => void)[] = [];
|
||||
|
||||
constructor(
|
||||
public id: number,
|
||||
public options: ShardOptions,
|
||||
) {
|
||||
this.options.ratelimitOptions ??= {
|
||||
rateLimitResetInterval: 60_000,
|
||||
maxRequestsPerRateLimitTick: 120,
|
||||
};
|
||||
|
||||
if (options.debugger) this.debugger = options.debugger;
|
||||
|
||||
const safe = this.calculateSafeRequests();
|
||||
this.bucket = new DynamicBucket({ refillInterval: 6e4, limit: safe, debugger: options.debugger });
|
||||
}
|
||||
|
||||
get latency() {
|
||||
return this.heart.lastAck && this.heart.lastBeat
|
||||
? this.heart.lastAck - this.heart.lastBeat
|
||||
: Number.POSITIVE_INFINITY;
|
||||
}
|
||||
|
||||
get isOpen() {
|
||||
return this.websocket?.readyState === WebSocket.OPEN;
|
||||
}
|
||||
|
||||
get gatewayURL() {
|
||||
return this.options.info.url;
|
||||
}
|
||||
|
||||
get resumeGatewayURL() {
|
||||
return this.data.resume_gateway_url;
|
||||
}
|
||||
|
||||
get currentGatewayURL() {
|
||||
const url = new URL(this.resumeGatewayURL ?? this.options.info.url);
|
||||
url.searchParams.set('v', '10');
|
||||
return url.href;
|
||||
}
|
||||
|
||||
async connect() {
|
||||
await this.connectTimeout.wait();
|
||||
if (this.isOpen) {
|
||||
this.debugger?.debug(`[Shard #${this.id}] attempted to connect while open`);
|
||||
return;
|
||||
}
|
||||
|
||||
this.debugger?.debug(`[Shard #${this.id}] Connecting to ${this.currentGatewayURL}`);
|
||||
|
||||
// @ts-expect-error @types/bun cause erros in compile
|
||||
// biome-ignore lint/correctness/noUndeclaredVariables: /\ bun lol
|
||||
this.websocket = new BaseSocket(typeof Bun === 'undefined' ? 'ws' : 'bun', this.currentGatewayURL);
|
||||
|
||||
this.websocket!.onmessage = (event: WS.MessageEvent) => this.handleMessage(event);
|
||||
|
||||
this.websocket!.onclose = (event: WS.CloseEvent) => this.handleClosed(event);
|
||||
|
||||
this.websocket!.onerror = (event: ErrorEvent) => this.debugger?.error(event);
|
||||
|
||||
this.websocket!.onopen = () => {
|
||||
this.heart.ack = true;
|
||||
};
|
||||
}
|
||||
|
||||
async send<T extends GatewaySendPayload = GatewaySendPayload>(force: boolean, message: T) {
|
||||
this.debugger?.info(
|
||||
`[Shard #${this.id}] Sending: ${GatewayOpcodes[message.op]} ${JSON.stringify(
|
||||
message.d,
|
||||
(_, value) => {
|
||||
if (typeof value === 'string')
|
||||
return value.replaceAll(this.options.token, v => {
|
||||
const split = v.split('.');
|
||||
return `${split[0]}.${'*'.repeat(split[1].length)}.${'*'.repeat(split[2].length)}`;
|
||||
});
|
||||
return value;
|
||||
},
|
||||
1,
|
||||
)}`,
|
||||
);
|
||||
await this.checkOffline(force);
|
||||
await this.bucket.acquire(force);
|
||||
await this.checkOffline(force);
|
||||
this.websocket?.send(JSON.stringify(message));
|
||||
}
|
||||
|
||||
async identify() {
|
||||
await this.send(true, {
|
||||
op: GatewayOpcodes.Identify,
|
||||
d: {
|
||||
token: `Bot ${this.options.token}`,
|
||||
compress: this.options.compress,
|
||||
properties: this.options.properties ?? properties,
|
||||
shard: [this.id, this.options.info.shards],
|
||||
intents: this.options.intents,
|
||||
presence: this.options.presence,
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
get resumable() {
|
||||
return !!(this.data.resume_gateway_url && this.data.session_id && this.data.resumeSeq !== null);
|
||||
}
|
||||
|
||||
async resume() {
|
||||
await this.send(true, {
|
||||
op: GatewayOpcodes.Resume,
|
||||
d: {
|
||||
seq: this.data.resumeSeq!,
|
||||
session_id: this.data.session_id!,
|
||||
token: `Bot ${this.options.token}`,
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
async heartbeat(requested: boolean) {
|
||||
this.debugger?.debug(
|
||||
`[Shard #${this.id}] Sending ${requested ? '' : 'un'}requested heartbeat (Ack=${this.heart.ack})`,
|
||||
);
|
||||
if (!requested) {
|
||||
if (!this.heart.ack) {
|
||||
await this.close(ShardSocketCloseCodes.ZombiedConnection, 'Zombied connection');
|
||||
return;
|
||||
}
|
||||
this.heart.ack = false;
|
||||
}
|
||||
|
||||
this.heart.lastBeat = Date.now();
|
||||
|
||||
this.websocket!.send(
|
||||
JSON.stringify({
|
||||
op: GatewayOpcodes.Heartbeat,
|
||||
d: this.data.resumeSeq ?? null,
|
||||
}),
|
||||
);
|
||||
}
|
||||
|
||||
async disconnect() {
|
||||
this.debugger?.info(`[Shard #${this.id}] Disconnecting`);
|
||||
await this.close(ShardSocketCloseCodes.Shutdown, 'Shard down request');
|
||||
}
|
||||
|
||||
async reconnect() {
|
||||
this.debugger?.info(`[Shard #${this.id}] Reconnecting`);
|
||||
await this.disconnect();
|
||||
await this.connect();
|
||||
}
|
||||
|
||||
async onpacket(packet: GatewayReceivePayload) {
|
||||
if (packet.s !== null) {
|
||||
this.data.resumeSeq = packet.s;
|
||||
}
|
||||
|
||||
this.debugger?.debug(`[Shard #${this.id}]`, packet.t ? packet.t : GatewayOpcodes[packet.op], this.data.resumeSeq);
|
||||
|
||||
switch (packet.op) {
|
||||
case GatewayOpcodes.Hello:
|
||||
{
|
||||
clearInterval(this.heart.nodeInterval);
|
||||
|
||||
this.heart.interval = packet.d.heartbeat_interval;
|
||||
|
||||
await this.heartbeat(false);
|
||||
this.heart.nodeInterval = setInterval(() => this.heartbeat(false), this.heart.interval);
|
||||
|
||||
if (this.resumable) {
|
||||
return this.resume();
|
||||
}
|
||||
await this.identify();
|
||||
}
|
||||
break;
|
||||
case GatewayOpcodes.HeartbeatAck:
|
||||
this.heart.ack = true;
|
||||
this.heart.lastAck = Date.now();
|
||||
break;
|
||||
case GatewayOpcodes.Heartbeat:
|
||||
this.heartbeat(true);
|
||||
break;
|
||||
case GatewayOpcodes.Reconnect:
|
||||
await this.reconnect();
|
||||
break;
|
||||
case GatewayOpcodes.InvalidSession:
|
||||
if (packet.d) {
|
||||
if (!this.resumable) {
|
||||
return this.debugger?.fatal(`[Shard #${this.id}] This is a completely unexpected error message.`);
|
||||
}
|
||||
await this.resume();
|
||||
} else {
|
||||
this.data.resumeSeq = 0;
|
||||
this.data.session_id = undefined;
|
||||
await this.identify();
|
||||
}
|
||||
break;
|
||||
case GatewayOpcodes.Dispatch:
|
||||
{
|
||||
switch (packet.t) {
|
||||
case GatewayDispatchEvents.Resumed:
|
||||
this.offlineSendQueue.map((resolve: () => any) => resolve());
|
||||
this.options.handlePayload(this.id, packet);
|
||||
break;
|
||||
case GatewayDispatchEvents.Ready: {
|
||||
this.data.resume_gateway_url = packet.d.resume_gateway_url;
|
||||
this.data.session_id = packet.d.session_id;
|
||||
this.offlineSendQueue.map((resolve: () => any) => resolve());
|
||||
this.options.handlePayload(this.id, packet);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
this.options.handlePayload(this.id, packet);
|
||||
break;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
protected async handleClosed(close: CloseEvent) {
|
||||
clearInterval(this.heart.nodeInterval);
|
||||
this.debugger?.warn(
|
||||
`[Shard #${this.id}] ${ShardSocketCloseCodes[close.code] ?? GatewayCloseCodes[close.code] ?? close.code} (${
|
||||
close.code
|
||||
})`,
|
||||
);
|
||||
|
||||
switch (close.code) {
|
||||
case ShardSocketCloseCodes.Shutdown:
|
||||
break;
|
||||
case 1000:
|
||||
case 1001:
|
||||
case 1006:
|
||||
case ShardSocketCloseCodes.ZombiedConnection:
|
||||
case GatewayCloseCodes.UnknownError:
|
||||
case GatewayCloseCodes.UnknownOpcode:
|
||||
case GatewayCloseCodes.DecodeError:
|
||||
case GatewayCloseCodes.NotAuthenticated:
|
||||
case GatewayCloseCodes.AlreadyAuthenticated:
|
||||
case GatewayCloseCodes.InvalidSeq:
|
||||
case GatewayCloseCodes.RateLimited:
|
||||
case GatewayCloseCodes.SessionTimedOut:
|
||||
this.debugger?.info(`[Shard #${this.id}] Trying to reconnect`);
|
||||
await this.reconnect();
|
||||
break;
|
||||
|
||||
case GatewayCloseCodes.AuthenticationFailed:
|
||||
case GatewayCloseCodes.DisallowedIntents:
|
||||
case GatewayCloseCodes.InvalidAPIVersion:
|
||||
case GatewayCloseCodes.InvalidIntents:
|
||||
case GatewayCloseCodes.InvalidShard:
|
||||
case GatewayCloseCodes.ShardingRequired:
|
||||
this.debugger?.fatal(`[Shard #${this.id}] cannot reconnect`);
|
||||
break;
|
||||
|
||||
default:
|
||||
this.debugger?.warn(`[Shard #${this.id}] Unknown close code, trying to reconnect anyways`);
|
||||
await this.reconnect();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
async close(code: number, reason: string) {
|
||||
if (this.websocket?.readyState !== WebSocket.OPEN) {
|
||||
return this.debugger?.warn(`[Shard #${this.id}] Is not open`);
|
||||
}
|
||||
this.debugger?.warn(`[Shard #${this.id}] Called close`);
|
||||
this.websocket?.close(code, reason);
|
||||
}
|
||||
|
||||
protected handleMessage({ data }: WS.MessageEvent) {
|
||||
if (data instanceof Buffer) {
|
||||
data = inflateSync(data);
|
||||
}
|
||||
return this.onpacket(JSON.parse(data as string));
|
||||
}
|
||||
|
||||
checkOffline(force: boolean) {
|
||||
if (!this.isOpen) {
|
||||
return new Promise(resolve => this.offlineSendQueue[force ? 'unshift' : 'push'](resolve));
|
||||
}
|
||||
return Promise.resolve();
|
||||
}
|
||||
|
||||
calculateSafeRequests(): number {
|
||||
const safeRequests =
|
||||
this.options.ratelimitOptions!.maxRequestsPerRateLimitTick -
|
||||
Math.ceil(this.options.ratelimitOptions!.rateLimitResetInterval / this.heart.interval) * 2;
|
||||
|
||||
if (safeRequests < 0) {
|
||||
return 0;
|
||||
}
|
||||
return safeRequests;
|
||||
}
|
||||
}
|
||||
import type { GatewayReceivePayload, GatewaySendPayload } from 'discord-api-types/v10';
|
||||
import { GatewayCloseCodes, GatewayDispatchEvents, GatewayOpcodes } from 'discord-api-types/v10';
|
||||
import { inflateSync } from 'node:zlib';
|
||||
import type WS from 'ws';
|
||||
import { WebSocket, type CloseEvent, type ErrorEvent } from 'ws';
|
||||
import type { Logger } from '../../common';
|
||||
import { properties } from '../constants';
|
||||
import { DynamicBucket } from '../structures';
|
||||
import { ConnectTimeout } from '../structures/timeout';
|
||||
import { BaseSocket } from './basesocket';
|
||||
import type { ShardData, ShardOptions } from './shared';
|
||||
import { ShardSocketCloseCodes } from './shared';
|
||||
|
||||
export interface ShardHeart {
|
||||
interval: number;
|
||||
nodeInterval?: NodeJS.Timeout;
|
||||
lastAck?: number;
|
||||
lastBeat?: number;
|
||||
ack: boolean;
|
||||
}
|
||||
|
||||
export class Shard {
|
||||
debugger?: Logger;
|
||||
data: Partial<ShardData> | ShardData = {
|
||||
resumeSeq: null,
|
||||
};
|
||||
|
||||
websocket: BaseSocket | null = null;
|
||||
connectTimeout = new ConnectTimeout();
|
||||
heart: ShardHeart = {
|
||||
interval: 30e3,
|
||||
ack: true,
|
||||
};
|
||||
|
||||
bucket: DynamicBucket;
|
||||
offlineSendQueue: ((_?: unknown) => void)[] = [];
|
||||
|
||||
constructor(
|
||||
public id: number,
|
||||
public options: ShardOptions,
|
||||
) {
|
||||
this.options.ratelimitOptions ??= {
|
||||
rateLimitResetInterval: 60_000,
|
||||
maxRequestsPerRateLimitTick: 120,
|
||||
};
|
||||
|
||||
if (options.debugger) this.debugger = options.debugger;
|
||||
|
||||
const safe = this.calculateSafeRequests();
|
||||
this.bucket = new DynamicBucket({ refillInterval: 6e4, limit: safe, debugger: options.debugger });
|
||||
}
|
||||
|
||||
get latency() {
|
||||
return this.heart.lastAck && this.heart.lastBeat
|
||||
? this.heart.lastAck - this.heart.lastBeat
|
||||
: Number.POSITIVE_INFINITY;
|
||||
}
|
||||
|
||||
get isOpen() {
|
||||
return this.websocket?.readyState === WebSocket.OPEN;
|
||||
}
|
||||
|
||||
get gatewayURL() {
|
||||
return this.options.info.url;
|
||||
}
|
||||
|
||||
get resumeGatewayURL() {
|
||||
return this.data.resume_gateway_url;
|
||||
}
|
||||
|
||||
get currentGatewayURL() {
|
||||
const url = new URL(this.resumeGatewayURL ?? this.options.info.url);
|
||||
url.searchParams.set('v', '10');
|
||||
return url.href;
|
||||
}
|
||||
|
||||
ping() {
|
||||
if (!this.websocket) return Promise.resolve(Number.POSITIVE_INFINITY);
|
||||
return this.websocket.ping();
|
||||
}
|
||||
|
||||
async connect() {
|
||||
await this.connectTimeout.wait();
|
||||
if (this.isOpen) {
|
||||
this.debugger?.debug(`[Shard #${this.id}] attempted to connect while open`);
|
||||
return;
|
||||
}
|
||||
|
||||
this.debugger?.debug(`[Shard #${this.id}] Connecting to ${this.currentGatewayURL}`);
|
||||
|
||||
// @ts-expect-error @types/bun cause erros in compile
|
||||
// biome-ignore lint/correctness/noUndeclaredVariables: /\ bun lol
|
||||
this.websocket = new BaseSocket(typeof Bun === 'undefined' ? 'ws' : 'bun', this.currentGatewayURL);
|
||||
|
||||
this.websocket!.onmessage = (event: WS.MessageEvent) => this.handleMessage(event);
|
||||
|
||||
this.websocket!.onclose = (event: WS.CloseEvent) => this.handleClosed(event);
|
||||
|
||||
this.websocket!.onerror = (event: ErrorEvent) => this.debugger?.error(event);
|
||||
|
||||
this.websocket!.onopen = () => {
|
||||
this.heart.ack = true;
|
||||
};
|
||||
}
|
||||
|
||||
async send<T extends GatewaySendPayload = GatewaySendPayload>(force: boolean, message: T) {
|
||||
this.debugger?.info(
|
||||
`[Shard #${this.id}] Sending: ${GatewayOpcodes[message.op]} ${JSON.stringify(
|
||||
message.d,
|
||||
(_, value) => {
|
||||
if (typeof value === 'string')
|
||||
return value.replaceAll(this.options.token, v => {
|
||||
const split = v.split('.');
|
||||
return `${split[0]}.${'*'.repeat(split[1].length)}.${'*'.repeat(split[2].length)}`;
|
||||
});
|
||||
return value;
|
||||
},
|
||||
1,
|
||||
)}`,
|
||||
);
|
||||
await this.checkOffline(force);
|
||||
await this.bucket.acquire(force);
|
||||
await this.checkOffline(force);
|
||||
this.websocket?.send(JSON.stringify(message));
|
||||
}
|
||||
|
||||
async identify() {
|
||||
await this.send(true, {
|
||||
op: GatewayOpcodes.Identify,
|
||||
d: {
|
||||
token: `Bot ${this.options.token}`,
|
||||
compress: this.options.compress,
|
||||
properties: this.options.properties ?? properties,
|
||||
shard: [this.id, this.options.info.shards],
|
||||
intents: this.options.intents,
|
||||
presence: this.options.presence,
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
get resumable() {
|
||||
return !!(this.data.resume_gateway_url && this.data.session_id && this.data.resumeSeq !== null);
|
||||
}
|
||||
|
||||
async resume() {
|
||||
await this.send(true, {
|
||||
op: GatewayOpcodes.Resume,
|
||||
d: {
|
||||
seq: this.data.resumeSeq!,
|
||||
session_id: this.data.session_id!,
|
||||
token: `Bot ${this.options.token}`,
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
async heartbeat(requested: boolean) {
|
||||
this.debugger?.debug(
|
||||
`[Shard #${this.id}] Sending ${requested ? '' : 'un'}requested heartbeat (Ack=${this.heart.ack})`,
|
||||
);
|
||||
if (!requested) {
|
||||
if (!this.heart.ack) {
|
||||
await this.close(ShardSocketCloseCodes.ZombiedConnection, 'Zombied connection');
|
||||
return;
|
||||
}
|
||||
this.heart.ack = false;
|
||||
}
|
||||
|
||||
this.heart.lastBeat = Date.now();
|
||||
|
||||
this.websocket!.send(
|
||||
JSON.stringify({
|
||||
op: GatewayOpcodes.Heartbeat,
|
||||
d: this.data.resumeSeq ?? null,
|
||||
}),
|
||||
);
|
||||
}
|
||||
|
||||
async disconnect() {
|
||||
this.debugger?.info(`[Shard #${this.id}] Disconnecting`);
|
||||
await this.close(ShardSocketCloseCodes.Shutdown, 'Shard down request');
|
||||
}
|
||||
|
||||
async reconnect() {
|
||||
this.debugger?.info(`[Shard #${this.id}] Reconnecting`);
|
||||
await this.disconnect();
|
||||
await this.connect();
|
||||
}
|
||||
|
||||
async onpacket(packet: GatewayReceivePayload) {
|
||||
if (packet.s !== null) {
|
||||
this.data.resumeSeq = packet.s;
|
||||
}
|
||||
|
||||
this.debugger?.debug(`[Shard #${this.id}]`, packet.t ? packet.t : GatewayOpcodes[packet.op], this.data.resumeSeq);
|
||||
|
||||
switch (packet.op) {
|
||||
case GatewayOpcodes.Hello:
|
||||
{
|
||||
clearInterval(this.heart.nodeInterval);
|
||||
|
||||
this.heart.interval = packet.d.heartbeat_interval;
|
||||
|
||||
await this.heartbeat(false);
|
||||
this.heart.nodeInterval = setInterval(() => this.heartbeat(false), this.heart.interval);
|
||||
|
||||
if (this.resumable) {
|
||||
return this.resume();
|
||||
}
|
||||
await this.identify();
|
||||
}
|
||||
break;
|
||||
case GatewayOpcodes.HeartbeatAck:
|
||||
this.heart.ack = true;
|
||||
this.heart.lastAck = Date.now();
|
||||
break;
|
||||
case GatewayOpcodes.Heartbeat:
|
||||
this.heartbeat(true);
|
||||
break;
|
||||
case GatewayOpcodes.Reconnect:
|
||||
await this.reconnect();
|
||||
break;
|
||||
case GatewayOpcodes.InvalidSession:
|
||||
if (packet.d) {
|
||||
if (!this.resumable) {
|
||||
return this.debugger?.fatal(`[Shard #${this.id}] This is a completely unexpected error message.`);
|
||||
}
|
||||
await this.resume();
|
||||
} else {
|
||||
this.data.resumeSeq = 0;
|
||||
this.data.session_id = undefined;
|
||||
await this.identify();
|
||||
}
|
||||
break;
|
||||
case GatewayOpcodes.Dispatch:
|
||||
{
|
||||
switch (packet.t) {
|
||||
case GatewayDispatchEvents.Resumed:
|
||||
this.offlineSendQueue.map((resolve: () => any) => resolve());
|
||||
this.options.handlePayload(this.id, packet);
|
||||
break;
|
||||
case GatewayDispatchEvents.Ready: {
|
||||
this.data.resume_gateway_url = packet.d.resume_gateway_url;
|
||||
this.data.session_id = packet.d.session_id;
|
||||
this.offlineSendQueue.map((resolve: () => any) => resolve());
|
||||
this.options.handlePayload(this.id, packet);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
this.options.handlePayload(this.id, packet);
|
||||
break;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
protected async handleClosed(close: CloseEvent) {
|
||||
clearInterval(this.heart.nodeInterval);
|
||||
this.debugger?.warn(
|
||||
`[Shard #${this.id}] ${ShardSocketCloseCodes[close.code] ?? GatewayCloseCodes[close.code] ?? close.code} (${
|
||||
close.code
|
||||
})`,
|
||||
);
|
||||
|
||||
switch (close.code) {
|
||||
case ShardSocketCloseCodes.Shutdown:
|
||||
break;
|
||||
case 1000:
|
||||
case 1001:
|
||||
case 1006:
|
||||
case ShardSocketCloseCodes.ZombiedConnection:
|
||||
case GatewayCloseCodes.UnknownError:
|
||||
case GatewayCloseCodes.UnknownOpcode:
|
||||
case GatewayCloseCodes.DecodeError:
|
||||
case GatewayCloseCodes.NotAuthenticated:
|
||||
case GatewayCloseCodes.AlreadyAuthenticated:
|
||||
case GatewayCloseCodes.InvalidSeq:
|
||||
case GatewayCloseCodes.RateLimited:
|
||||
case GatewayCloseCodes.SessionTimedOut:
|
||||
this.debugger?.info(`[Shard #${this.id}] Trying to reconnect`);
|
||||
await this.reconnect();
|
||||
break;
|
||||
|
||||
case GatewayCloseCodes.AuthenticationFailed:
|
||||
case GatewayCloseCodes.DisallowedIntents:
|
||||
case GatewayCloseCodes.InvalidAPIVersion:
|
||||
case GatewayCloseCodes.InvalidIntents:
|
||||
case GatewayCloseCodes.InvalidShard:
|
||||
case GatewayCloseCodes.ShardingRequired:
|
||||
this.debugger?.fatal(`[Shard #${this.id}] cannot reconnect`);
|
||||
break;
|
||||
|
||||
default:
|
||||
this.debugger?.warn(`[Shard #${this.id}] Unknown close code, trying to reconnect anyways`);
|
||||
await this.reconnect();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
async close(code: number, reason: string) {
|
||||
if (this.websocket?.readyState !== WebSocket.OPEN) {
|
||||
return this.debugger?.warn(`[Shard #${this.id}] Is not open`);
|
||||
}
|
||||
this.debugger?.warn(`[Shard #${this.id}] Called close`);
|
||||
this.websocket?.close(code, reason);
|
||||
}
|
||||
|
||||
protected handleMessage({ data }: WS.MessageEvent) {
|
||||
if (data instanceof Buffer) {
|
||||
data = inflateSync(data);
|
||||
}
|
||||
return this.onpacket(JSON.parse(data as string));
|
||||
}
|
||||
|
||||
checkOffline(force: boolean) {
|
||||
if (!this.isOpen) {
|
||||
return new Promise(resolve => this.offlineSendQueue[force ? 'unshift' : 'push'](resolve));
|
||||
}
|
||||
return Promise.resolve();
|
||||
}
|
||||
|
||||
calculateSafeRequests(): number {
|
||||
const safeRequests =
|
||||
this.options.ratelimitOptions!.maxRequestsPerRateLimitTick -
|
||||
Math.ceil(this.options.ratelimitOptions!.rateLimitResetInterval / this.heart.interval) * 2;
|
||||
|
||||
if (safeRequests < 0) {
|
||||
return 0;
|
||||
}
|
||||
return safeRequests;
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user