@discordjs/ws 0.1.0

package/dist/strategies/sharding/WorkerShardingStrategy.mjs

@@ -0,0 +1,165 @@
+import { once } from 'node:events';
+import { join } from 'node:path';
+import { Worker } from 'node:worker_threads';
+import { Collection } from '@discordjs/collection';
+import { IdentifyThrottler } from '../../utils/IdentifyThrottler.mjs';
+import { managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy.mjs';
+
+
+
+// -- Unbuild CommonJS Shims --
+import __cjs_url__ from 'url';
+import __cjs_path__ from 'path';
+import __cjs_mod__ from 'module';
+const __filename = __cjs_url__.fileURLToPath(import.meta.url);
+const __dirname = __cjs_path__.dirname(__filename);
+const require = __cjs_mod__.createRequire(import.meta.url);
+var __accessCheck = (obj, member, msg) => {
+  if (!member.has(obj))
+    throw TypeError("Cannot " + msg);
+};
+var __privateGet = (obj, member, getter) => {
+  __accessCheck(obj, member, "read from private field");
+  return getter ? getter.call(obj) : member.get(obj);
+};
+var __privateAdd = (obj, member, value) => {
+  if (member.has(obj))
+    throw TypeError("Cannot add the same private member more than once");
+  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
+};
+var __privateSet = (obj, member, value, setter) => {
+  __accessCheck(obj, member, "write to private field");
+  setter ? setter.call(obj, value) : member.set(obj, value);
+  return value;
+};
+var _workers, _workerByShardId;
+var WorkerSendPayloadOp = /* @__PURE__ */ ((WorkerSendPayloadOp2) => {
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["Connect"] = 0] = "Connect";
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["Destroy"] = 1] = "Destroy";
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["Send"] = 2] = "Send";
+  WorkerSendPayloadOp2[WorkerSendPayloadOp2["SessionInfoResponse"] = 3] = "SessionInfoResponse";
+  return WorkerSendPayloadOp2;
+})(WorkerSendPayloadOp || {});
+var WorkerRecievePayloadOp = /* @__PURE__ */ ((WorkerRecievePayloadOp2) => {
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["Connected"] = 0] = "Connected";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["Destroyed"] = 1] = "Destroyed";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["Event"] = 2] = "Event";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["RetrieveSessionInfo"] = 3] = "RetrieveSessionInfo";
+  WorkerRecievePayloadOp2[WorkerRecievePayloadOp2["UpdateSessionInfo"] = 4] = "UpdateSessionInfo";
+  return WorkerRecievePayloadOp2;
+})(WorkerRecievePayloadOp || {});
+class WorkerShardingStrategy {
+  constructor(manager, options) {
+    __privateAdd(this, _workers, []);
+    __privateAdd(this, _workerByShardId, new Collection());
+    this.connectPromises = new Collection();
+    this.destroyPromises = new Collection();
+    this.manager = manager;
+    this.throttler = new IdentifyThrottler(manager);
+    this.options = options;
+  }
+  async spawn(shardIds) {
+    const shardsPerWorker = this.options.shardsPerWorker === "all" ? shardIds.length : this.options.shardsPerWorker;
+    const strategyOptions = await managerToFetchingStrategyOptions(this.manager);
+    let shards = 0;
+    while (shards !== shardIds.length) {
+      const slice = shardIds.slice(shards, shardsPerWorker + shards);
+      const workerData = {
+        ...strategyOptions,
+        shardIds: slice
+      };
+      const worker = new Worker(join(__dirname, "worker.cjs"), { workerData });
+      await once(worker, "online");
+      worker.on("error", (err) => {
+        throw err;
+      }).on("messageerror", (err) => {
+        throw err;
+      }).on("message", (payload) => this.onMessage(worker, payload));
+      __privateGet(this, _workers).push(worker);
+      for (const shardId of slice) {
+        __privateGet(this, _workerByShardId).set(shardId, worker);
+      }
+      shards += slice.length;
+    }
+  }
+  async connect() {
+    const promises = [];
+    for (const [shardId, worker] of __privateGet(this, _workerByShardId).entries()) {
+      await this.throttler.waitForIdentify();
+      const payload = {
+        op: 0 /* Connect */,
+        shardId
+      };
+      const promise = new Promise((resolve) => this.connectPromises.set(shardId, resolve));
+      worker.postMessage(payload);
+      promises.push(promise);
+    }
+    await Promise.all(promises);
+  }
+  async destroy(options = {}) {
+    const promises = [];
+    for (const [shardId, worker] of __privateGet(this, _workerByShardId).entries()) {
+      const payload = {
+        op: 1 /* Destroy */,
+        shardId,
+        options
+      };
+      promises.push(new Promise((resolve) => this.destroyPromises.set(shardId, resolve)).then(() => worker.terminate()));
+      worker.postMessage(payload);
+    }
+    __privateSet(this, _workers, []);
+    __privateGet(this, _workerByShardId).clear();
+    await Promise.all(promises);
+  }
+  send(shardId, data) {
+    const worker = __privateGet(this, _workerByShardId).get(shardId);
+    if (!worker) {
+      throw new Error(`No worker found for shard ${shardId}`);
+    }
+    const payload = {
+      op: 2 /* Send */,
+      shardId,
+      payload: data
+    };
+    worker.postMessage(payload);
+  }
+  async onMessage(worker, payload) {
+    switch (payload.op) {
+      case 0 /* Connected */: {
+        const resolve = this.connectPromises.get(payload.shardId);
+        resolve();
+        this.connectPromises.delete(payload.shardId);
+        break;
+      }
+      case 1 /* Destroyed */: {
+        const resolve = this.destroyPromises.get(payload.shardId);
+        resolve();
+        this.destroyPromises.delete(payload.shardId);
+        break;
+      }
+      case 2 /* Event */: {
+        this.manager.emit(payload.event, { ...payload.data, shardId: payload.shardId });
+        break;
+      }
+      case 3 /* RetrieveSessionInfo */: {
+        const session = await this.manager.options.retrieveSessionInfo(payload.shardId);
+        const response = {
+          op: 3 /* SessionInfoResponse */,
+          nonce: payload.nonce,
+          session
+        };
+        worker.postMessage(response);
+        break;
+      }
+      case 4 /* UpdateSessionInfo */: {
+        await this.manager.options.updateSessionInfo(payload.shardId, payload.session);
+        break;
+      }
+    }
+  }
+}
+_workers = new WeakMap();
+_workerByShardId = new WeakMap();
+
+export { WorkerRecievePayloadOp, WorkerSendPayloadOp, WorkerShardingStrategy };
+//# sourceMappingURL=WorkerShardingStrategy.mjs.map

package/dist/strategies/sharding/WorkerShardingStrategy.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerShardingStrategy.mjs","sources":["../../../src/strategies/sharding/WorkerShardingStrategy.ts"],"sourcesContent":["import { once } from 'node:events';\nimport { join } from 'node:path';\nimport { Worker } from 'node:worker_threads';\nimport { Collection } from '@discordjs/collection';\nimport type { GatewaySendPayload } from 'discord-api-types/v10';\nimport type { IShardingStrategy } from './IShardingStrategy';\nimport { IdentifyThrottler } from '../../utils/IdentifyThrottler';\nimport type { SessionInfo, WebSocketManager } from '../../ws/WebSocketManager';\nimport type { WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';\nimport { FetchingStrategyOptions, managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy';\n\nexport interface WorkerData extends FetchingStrategyOptions {\n\tshardIds: number[];\n}\n\nexport enum WorkerSendPayloadOp {\n\tConnect,\n\tDestroy,\n\tSend,\n\tSessionInfoResponse,\n}\n\nexport type WorkerSendPayload =\n\t| { op: WorkerSendPayloadOp.Connect; shardId: number }\n\t| { op: WorkerSendPayloadOp.Destroy; shardId: number; options?: WebSocketShardDestroyOptions }\n\t| { op: WorkerSendPayloadOp.Send; shardId: number; payload: GatewaySendPayload }\n\t| { op: WorkerSendPayloadOp.SessionInfoResponse; nonce: number; session: SessionInfo | null };\n\nexport enum WorkerRecievePayloadOp {\n\tConnected,\n\tDestroyed,\n\tEvent,\n\tRetrieveSessionInfo,\n\tUpdateSessionInfo,\n}\n\nexport type WorkerRecievePayload =\n\t| { op: WorkerRecievePayloadOp.Connected; shardId: number }\n\t| { op: WorkerRecievePayloadOp.Destroyed; shardId: number }\n\t// Can't seem to get a type-safe union based off of the event, so I'm sadly leaving data as any for now\n\t| { op: WorkerRecievePayloadOp.Event; shardId: number; event: WebSocketShardEvents; data: any }\n\t| { op: WorkerRecievePayloadOp.RetrieveSessionInfo; shardId: number; nonce: number }\n\t| { op: WorkerRecievePayloadOp.UpdateSessionInfo; shardId: number; session: SessionInfo | null };\n\n/**\n * Options for a {@link WorkerShardingStrategy}\n */\nexport interface WorkerShardingStrategyOptions {\n\t/**\n\t * Dictates how many shards should be spawned per worker thread.\n\t */\n\tshardsPerWorker: number | 'all';\n}\n\n/**\n * Strategy used to spawn threads in worker_threads\n */\nexport class WorkerShardingStrategy implements IShardingStrategy {\n\tprivate readonly manager: WebSocketManager;\n\tprivate readonly options: WorkerShardingStrategyOptions;\n\n\t#workers: Worker[] = [];\n\treadonly #workerByShardId = new Collection<number, Worker>();\n\n\tprivate readonly connectPromises = new Collection<number, () => void>();\n\tprivate readonly destroyPromises = new Collection<number, () => void>();\n\n\tprivate readonly throttler: IdentifyThrottler;\n\n\tpublic constructor(manager: WebSocketManager, options: WorkerShardingStrategyOptions) {\n\t\tthis.manager = manager;\n\t\tthis.throttler = new IdentifyThrottler(manager);\n\t\tthis.options = options;\n\t}\n\n\tpublic async spawn(shardIds: number[]) {\n\t\tconst shardsPerWorker = this.options.shardsPerWorker === 'all' ? shardIds.length : this.options.shardsPerWorker;\n\t\tconst strategyOptions = await managerToFetchingStrategyOptions(this.manager);\n\n\t\tlet shards = 0;\n\t\twhile (shards !== shardIds.length) {\n\t\t\tconst slice = shardIds.slice(shards, shardsPerWorker + shards);\n\t\t\tconst workerData: WorkerData = {\n\t\t\t\t...strategyOptions,\n\t\t\t\tshardIds: slice,\n\t\t\t};\n\n\t\t\tconst worker = new Worker(join(__dirname, 'worker.cjs'), { workerData });\n\t\t\tawait once(worker, 'online');\n\t\t\tworker\n\t\t\t\t.on('error', (err) => {\n\t\t\t\t\tthrow err;\n\t\t\t\t})\n\t\t\t\t.on('messageerror', (err) => {\n\t\t\t\t\tthrow err;\n\t\t\t\t})\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-misused-promises\n\t\t\t\t.on('message', (payload: WorkerRecievePayload) => this.onMessage(worker, payload));\n\n\t\t\tthis.#workers.push(worker);\n\t\t\tfor (const shardId of slice) {\n\t\t\t\tthis.#workerByShardId.set(shardId, worker);\n\t\t\t}\n\n\t\t\tshards += slice.length;\n\t\t}\n\t}\n\n\tpublic async connect() {\n\t\tconst promises = [];\n\n\t\tfor (const [shardId, worker] of this.#workerByShardId.entries()) {\n\t\t\tawait this.throttler.waitForIdentify();\n\n\t\t\tconst payload: WorkerSendPayload = {\n\t\t\t\top: WorkerSendPayloadOp.Connect,\n\t\t\t\tshardId,\n\t\t\t};\n\n\t\t\tconst promise = new Promise<void>((resolve) => this.connectPromises.set(shardId, resolve));\n\t\t\tworker.postMessage(payload);\n\t\t\tpromises.push(promise);\n\t\t}\n\n\t\tawait Promise.all(promises);\n\t}\n\n\tpublic async destroy(options: Omit<WebSocketShardDestroyOptions, 'recover'> = {}) {\n\t\tconst promises = [];\n\n\t\tfor (const [shardId, worker] of this.#workerByShardId.entries()) {\n\t\t\tconst payload: WorkerSendPayload = {\n\t\t\t\top: WorkerSendPayloadOp.Destroy,\n\t\t\t\tshardId,\n\t\t\t\toptions,\n\t\t\t};\n\n\t\t\tpromises.push(\n\t\t\t\tnew Promise<void>((resolve) => this.destroyPromises.set(shardId, resolve)).then(() => worker.terminate()),\n\t\t\t);\n\t\t\tworker.postMessage(payload);\n\t\t}\n\n\t\tthis.#workers = [];\n\t\tthis.#workerByShardId.clear();\n\n\t\tawait Promise.all(promises);\n\t}\n\n\tpublic send(shardId: number, data: GatewaySendPayload) {\n\t\tconst worker = this.#workerByShardId.get(shardId);\n\t\tif (!worker) {\n\t\t\tthrow new Error(`No worker found for shard ${shardId}`);\n\t\t}\n\n\t\tconst payload: WorkerSendPayload = {\n\t\t\top: WorkerSendPayloadOp.Send,\n\t\t\tshardId,\n\t\t\tpayload: data,\n\t\t};\n\t\tworker.postMessage(payload);\n\t}\n\n\tprivate async onMessage(worker: Worker, payload: WorkerRecievePayload) {\n\t\tswitch (payload.op) {\n\t\t\tcase WorkerRecievePayloadOp.Connected: {\n\t\t\t\tconst resolve = this.connectPromises.get(payload.shardId)!;\n\t\t\t\tresolve();\n\t\t\t\tthis.connectPromises.delete(payload.shardId);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.Destroyed: {\n\t\t\t\tconst resolve = this.destroyPromises.get(payload.shardId)!;\n\t\t\t\tresolve();\n\t\t\t\tthis.destroyPromises.delete(payload.shardId);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.Event: {\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-argument\n\t\t\t\tthis.manager.emit(payload.event, { ...payload.data, shardId: payload.shardId });\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.RetrieveSessionInfo: {\n\t\t\t\tconst session = await this.manager.options.retrieveSessionInfo(payload.shardId);\n\t\t\t\tconst response: WorkerSendPayload = {\n\t\t\t\t\top: WorkerSendPayloadOp.SessionInfoResponse,\n\t\t\t\t\tnonce: payload.nonce,\n\t\t\t\t\tsession,\n\t\t\t\t};\n\t\t\t\tworker.postMessage(response);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerRecievePayloadOp.UpdateSessionInfo: {\n\t\t\t\tawait this.manager.options.updateSessionInfo(payload.shardId, payload.session);\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t}\n}\n"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AAOU;AACV;AACA;AACA;AACA;AACA;AACA;AACU;AACV;AACA;AACA;AACA;AACA;AACA;AACA;AACO;AACP;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;"}

package/dist/strategies/sharding/worker.cjs

@@ -0,0 +1,78 @@
+'use strict';
+
+const node_worker_threads = require('node:worker_threads');
+const collection = require('@discordjs/collection');
+const WorkerShardingStrategy = require('./WorkerShardingStrategy.cjs');
+const WebSocketShard = require('../../ws/WebSocketShard.cjs');
+const WorkerContextFetchingStrategy = require('../context/WorkerContextFetchingStrategy.cjs');
+
+if (node_worker_threads.isMainThread) {
+  throw new Error("Expected worker script to not be ran within the main thread");
+}
+const data = node_worker_threads.workerData;
+const shards = new collection.Collection();
+async function connect(shardId) {
+  const shard = shards.get(shardId);
+  if (!shard) {
+    throw new Error(`Shard ${shardId} does not exist`);
+  }
+  await shard.connect();
+}
+async function destroy(shardId, options) {
+  const shard = shards.get(shardId);
+  if (!shard) {
+    throw new Error(`Shard ${shardId} does not exist`);
+  }
+  await shard.destroy(options);
+}
+for (const shardId of data.shardIds) {
+  const shard = new WebSocketShard.WebSocketShard(new WorkerContextFetchingStrategy.WorkerContextFetchingStrategy(data), shardId);
+  for (const event of Object.values(WebSocketShard.WebSocketShardEvents)) {
+    shard.on(event, (data2) => {
+      const payload = {
+        op: WorkerShardingStrategy.WorkerRecievePayloadOp.Event,
+        event,
+        data: data2,
+        shardId
+      };
+      node_worker_threads.parentPort.postMessage(payload);
+    });
+  }
+  shards.set(shardId, shard);
+}
+node_worker_threads.parentPort.on("messageerror", (err) => {
+  throw err;
+}).on("message", async (payload) => {
+  switch (payload.op) {
+    case WorkerShardingStrategy.WorkerSendPayloadOp.Connect: {
+      await connect(payload.shardId);
+      const response = {
+        op: WorkerShardingStrategy.WorkerRecievePayloadOp.Connected,
+        shardId: payload.shardId
+      };
+      node_worker_threads.parentPort.postMessage(response);
+      break;
+    }
+    case WorkerShardingStrategy.WorkerSendPayloadOp.Destroy: {
+      await destroy(payload.shardId, payload.options);
+      const response = {
+        op: WorkerShardingStrategy.WorkerRecievePayloadOp.Destroyed,
+        shardId: payload.shardId
+      };
+      node_worker_threads.parentPort.postMessage(response);
+      break;
+    }
+    case WorkerShardingStrategy.WorkerSendPayloadOp.Send: {
+      const shard = shards.get(payload.shardId);
+      if (!shard) {
+        throw new Error(`Shard ${payload.shardId} does not exist`);
+      }
+      await shard.send(payload.payload);
+      break;
+    }
+    case WorkerShardingStrategy.WorkerSendPayloadOp.SessionInfoResponse: {
+      break;
+    }
+  }
+});
+//# sourceMappingURL=worker.cjs.map

package/dist/strategies/sharding/worker.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.cjs","sources":["../../../src/strategies/sharding/worker.ts"],"sourcesContent":["import { isMainThread, workerData, parentPort } from 'node:worker_threads';\nimport { Collection } from '@discordjs/collection';\nimport {\n\tWorkerData,\n\tWorkerRecievePayload,\n\tWorkerRecievePayloadOp,\n\tWorkerSendPayload,\n\tWorkerSendPayloadOp,\n} from './WorkerShardingStrategy';\nimport { WebSocketShard, WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';\nimport { WorkerContextFetchingStrategy } from '../context/WorkerContextFetchingStrategy';\n\nif (isMainThread) {\n\tthrow new Error('Expected worker script to not be ran within the main thread');\n}\n\nconst data = workerData as WorkerData;\nconst shards = new Collection<number, WebSocketShard>();\n\nasync function connect(shardId: number) {\n\tconst shard = shards.get(shardId);\n\tif (!shard) {\n\t\tthrow new Error(`Shard ${shardId} does not exist`);\n\t}\n\tawait shard.connect();\n}\n\nasync function destroy(shardId: number, options?: WebSocketShardDestroyOptions) {\n\tconst shard = shards.get(shardId);\n\tif (!shard) {\n\t\tthrow new Error(`Shard ${shardId} does not exist`);\n\t}\n\tawait shard.destroy(options);\n}\n\nfor (const shardId of data.shardIds) {\n\tconst shard = new WebSocketShard(new WorkerContextFetchingStrategy(data), shardId);\n\tfor (const event of Object.values(WebSocketShardEvents)) {\n\t\t// @ts-expect-error\n\t\tshard.on(event, (data) => {\n\t\t\tconst payload: WorkerRecievePayload = {\n\t\t\t\top: WorkerRecievePayloadOp.Event,\n\t\t\t\tevent,\n\t\t\t\tdata,\n\t\t\t\tshardId,\n\t\t\t};\n\t\t\tparentPort!.postMessage(payload);\n\t\t});\n\t}\n\tshards.set(shardId, shard);\n}\n\nparentPort!\n\t.on('messageerror', (err) => {\n\t\tthrow err;\n\t})\n\t// eslint-disable-next-line @typescript-eslint/no-misused-promises\n\t.on('message', async (payload: WorkerSendPayload) => {\n\t\tswitch (payload.op) {\n\t\t\tcase WorkerSendPayloadOp.Connect: {\n\t\t\t\tawait connect(payload.shardId);\n\t\t\t\tconst response: WorkerRecievePayload = {\n\t\t\t\t\top: WorkerRecievePayloadOp.Connected,\n\t\t\t\t\tshardId: payload.shardId,\n\t\t\t\t};\n\t\t\t\tparentPort!.postMessage(response);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerSendPayloadOp.Destroy: {\n\t\t\t\tawait destroy(payload.shardId, payload.options);\n\t\t\t\tconst response: WorkerRecievePayload = {\n\t\t\t\t\top: WorkerRecievePayloadOp.Destroyed,\n\t\t\t\t\tshardId: payload.shardId,\n\t\t\t\t};\n\t\t\t\tparentPort!.postMessage(response);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerSendPayloadOp.Send: {\n\t\t\t\tconst shard = shards.get(payload.shardId);\n\t\t\t\tif (!shard) {\n\t\t\t\t\tthrow new Error(`Shard ${payload.shardId} does not exist`);\n\t\t\t\t}\n\t\t\t\tawait shard.send(payload.payload);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerSendPayloadOp.SessionInfoResponse: {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t});\n"],"names":["isMainThread","workerData","Collection","WebSocketShard","WorkerContextFetchingStrategy","WebSocketShardEvents","WorkerRecievePayloadOp","parentPort","WorkerSendPayloadOp"],"mappings":";;;;;;;;AAQA,IAAIA,gCAAY,EAAE;AAClB,EAAE,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;AACjF,CAAC;AACD,MAAM,IAAI,GAAGC,8BAAU,CAAC;AACxB,MAAM,MAAM,GAAG,IAAIC,qBAAU,EAAE,CAAC;AAChC,eAAe,OAAO,CAAC,OAAO,EAAE;AAChC,EAAE,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACpC,EAAE,IAAI,CAAC,KAAK,EAAE;AACd,IAAI,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;AACvD,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;AACxB,CAAC;AACD,eAAe,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;AACzC,EAAE,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACpC,EAAE,IAAI,CAAC,KAAK,EAAE;AACd,IAAI,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;AACvD,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAC/B,CAAC;AACD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE;AACrC,EAAE,MAAM,KAAK,GAAG,IAAIC,6BAAc,CAAC,IAAIC,2DAA6B,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;AACrF,EAAE,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAACC,mCAAoB,CAAC,EAAE;AAC3D,IAAI,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,KAAK;AAC/B,MAAM,MAAM,OAAO,GAAG;AACtB,QAAQ,EAAE,EAAEC,6CAAsB,CAAC,KAAK;AACxC,QAAQ,KAAK;AACb,QAAQ,IAAI,EAAE,KAAK;AACnB,QAAQ,OAAO;AACf,OAAO,CAAC;AACR,MAAMC,8BAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AACtC,KAAK,CAAC,CAAC;AACP,GAAG;AACH,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AAC7B,CAAC;AACDA,8BAAU,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,GAAG,KAAK;AACvC,EAAE,MAAM,GAAG,CAAC;AACZ,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,OAAO,OAAO,KAAK;AACpC,EAAE,QAAQ,OAAO,CAAC,EAAE;AACpB,IAAI,KAAKC,0CAAmB,CAAC,OAAO,EAAE;AACtC,MAAM,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACrC,MAAM,MAAM,QAAQ,GAAG;AACvB,QAAQ,EAAE,EAAEF,6CAAsB,CAAC,SAAS;AAC5C,QAAQ,OAAO,EAAE,OAAO,CAAC,OAAO;AAChC,OAAO,CAAC;AACR,MAAMC,8BAAU,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;AACvC,MAAM,MAAM;AACZ,KAAK;AACL,IAAI,KAAKC,0CAAmB,CAAC,OAAO,EAAE;AACtC,MAAM,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;AACtD,MAAM,MAAM,QAAQ,GAAG;AACvB,QAAQ,EAAE,EAAEF,6CAAsB,CAAC,SAAS;AAC5C,QAAQ,OAAO,EAAE,OAAO,CAAC,OAAO;AAChC,OAAO,CAAC;AACR,MAAMC,8BAAU,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;AACvC,MAAM,MAAM;AACZ,KAAK;AACL,IAAI,KAAKC,0CAAmB,CAAC,IAAI,EAAE;AACnC,MAAM,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAChD,MAAM,IAAI,CAAC,KAAK,EAAE;AAClB,QAAQ,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;AACnE,OAAO;AACP,MAAM,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACxC,MAAM,MAAM;AACZ,KAAK;AACL,IAAI,KAAKA,0CAAmB,CAAC,mBAAmB,EAAE;AAClD,MAAM,MAAM;AACZ,KAAK;AACL,GAAG;AACH,CAAC,CAAC;;"}

package/dist/strategies/sharding/worker.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=worker.d.ts.map

package/dist/strategies/sharding/worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../../src/strategies/sharding/worker.ts"],"names":[],"mappings":""}

package/dist/strategies/sharding/worker.mjs

@@ -0,0 +1,76 @@
+import { isMainThread, parentPort, workerData } from 'node:worker_threads';
+import { Collection } from '@discordjs/collection';
+import { WorkerRecievePayloadOp, WorkerSendPayloadOp } from './WorkerShardingStrategy.mjs';
+import { WebSocketShard, WebSocketShardEvents } from '../../ws/WebSocketShard.mjs';
+import { WorkerContextFetchingStrategy } from '../context/WorkerContextFetchingStrategy.mjs';
+
+if (isMainThread) {
+  throw new Error("Expected worker script to not be ran within the main thread");
+}
+const data = workerData;
+const shards = new Collection();
+async function connect(shardId) {
+  const shard = shards.get(shardId);
+  if (!shard) {
+    throw new Error(`Shard ${shardId} does not exist`);
+  }
+  await shard.connect();
+}
+async function destroy(shardId, options) {
+  const shard = shards.get(shardId);
+  if (!shard) {
+    throw new Error(`Shard ${shardId} does not exist`);
+  }
+  await shard.destroy(options);
+}
+for (const shardId of data.shardIds) {
+  const shard = new WebSocketShard(new WorkerContextFetchingStrategy(data), shardId);
+  for (const event of Object.values(WebSocketShardEvents)) {
+    shard.on(event, (data2) => {
+      const payload = {
+        op: WorkerRecievePayloadOp.Event,
+        event,
+        data: data2,
+        shardId
+      };
+      parentPort.postMessage(payload);
+    });
+  }
+  shards.set(shardId, shard);
+}
+parentPort.on("messageerror", (err) => {
+  throw err;
+}).on("message", async (payload) => {
+  switch (payload.op) {
+    case WorkerSendPayloadOp.Connect: {
+      await connect(payload.shardId);
+      const response = {
+        op: WorkerRecievePayloadOp.Connected,
+        shardId: payload.shardId
+      };
+      parentPort.postMessage(response);
+      break;
+    }
+    case WorkerSendPayloadOp.Destroy: {
+      await destroy(payload.shardId, payload.options);
+      const response = {
+        op: WorkerRecievePayloadOp.Destroyed,
+        shardId: payload.shardId
+      };
+      parentPort.postMessage(response);
+      break;
+    }
+    case WorkerSendPayloadOp.Send: {
+      const shard = shards.get(payload.shardId);
+      if (!shard) {
+        throw new Error(`Shard ${payload.shardId} does not exist`);
+      }
+      await shard.send(payload.payload);
+      break;
+    }
+    case WorkerSendPayloadOp.SessionInfoResponse: {
+      break;
+    }
+  }
+});
+//# sourceMappingURL=worker.mjs.map

package/dist/strategies/sharding/worker.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.mjs","sources":["../../../src/strategies/sharding/worker.ts"],"sourcesContent":["import { isMainThread, workerData, parentPort } from 'node:worker_threads';\nimport { Collection } from '@discordjs/collection';\nimport {\n\tWorkerData,\n\tWorkerRecievePayload,\n\tWorkerRecievePayloadOp,\n\tWorkerSendPayload,\n\tWorkerSendPayloadOp,\n} from './WorkerShardingStrategy';\nimport { WebSocketShard, WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';\nimport { WorkerContextFetchingStrategy } from '../context/WorkerContextFetchingStrategy';\n\nif (isMainThread) {\n\tthrow new Error('Expected worker script to not be ran within the main thread');\n}\n\nconst data = workerData as WorkerData;\nconst shards = new Collection<number, WebSocketShard>();\n\nasync function connect(shardId: number) {\n\tconst shard = shards.get(shardId);\n\tif (!shard) {\n\t\tthrow new Error(`Shard ${shardId} does not exist`);\n\t}\n\tawait shard.connect();\n}\n\nasync function destroy(shardId: number, options?: WebSocketShardDestroyOptions) {\n\tconst shard = shards.get(shardId);\n\tif (!shard) {\n\t\tthrow new Error(`Shard ${shardId} does not exist`);\n\t}\n\tawait shard.destroy(options);\n}\n\nfor (const shardId of data.shardIds) {\n\tconst shard = new WebSocketShard(new WorkerContextFetchingStrategy(data), shardId);\n\tfor (const event of Object.values(WebSocketShardEvents)) {\n\t\t// @ts-expect-error\n\t\tshard.on(event, (data) => {\n\t\t\tconst payload: WorkerRecievePayload = {\n\t\t\t\top: WorkerRecievePayloadOp.Event,\n\t\t\t\tevent,\n\t\t\t\tdata,\n\t\t\t\tshardId,\n\t\t\t};\n\t\t\tparentPort!.postMessage(payload);\n\t\t});\n\t}\n\tshards.set(shardId, shard);\n}\n\nparentPort!\n\t.on('messageerror', (err) => {\n\t\tthrow err;\n\t})\n\t// eslint-disable-next-line @typescript-eslint/no-misused-promises\n\t.on('message', async (payload: WorkerSendPayload) => {\n\t\tswitch (payload.op) {\n\t\t\tcase WorkerSendPayloadOp.Connect: {\n\t\t\t\tawait connect(payload.shardId);\n\t\t\t\tconst response: WorkerRecievePayload = {\n\t\t\t\t\top: WorkerRecievePayloadOp.Connected,\n\t\t\t\t\tshardId: payload.shardId,\n\t\t\t\t};\n\t\t\t\tparentPort!.postMessage(response);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerSendPayloadOp.Destroy: {\n\t\t\t\tawait destroy(payload.shardId, payload.options);\n\t\t\t\tconst response: WorkerRecievePayload = {\n\t\t\t\t\top: WorkerRecievePayloadOp.Destroyed,\n\t\t\t\t\tshardId: payload.shardId,\n\t\t\t\t};\n\t\t\t\tparentPort!.postMessage(response);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerSendPayloadOp.Send: {\n\t\t\t\tconst shard = shards.get(payload.shardId);\n\t\t\t\tif (!shard) {\n\t\t\t\t\tthrow new Error(`Shard ${payload.shardId} does not exist`);\n\t\t\t\t}\n\t\t\t\tawait shard.send(payload.payload);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase WorkerSendPayloadOp.SessionInfoResponse: {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t});\n"],"names":[],"mappings":";;;;;;AAQA,IAAI,YAAY,EAAE;AAClB,EAAE,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;AACjF,CAAC;AACD,MAAM,IAAI,GAAG,UAAU,CAAC;AACxB,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;AAChC,eAAe,OAAO,CAAC,OAAO,EAAE;AAChC,EAAE,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACpC,EAAE,IAAI,CAAC,KAAK,EAAE;AACd,IAAI,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;AACvD,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;AACxB,CAAC;AACD,eAAe,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE;AACzC,EAAE,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACpC,EAAE,IAAI,CAAC,KAAK,EAAE;AACd,IAAI,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;AACvD,GAAG;AACH,EAAE,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAC/B,CAAC;AACD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE;AACrC,EAAE,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC,IAAI,6BAA6B,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;AACrF,EAAE,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE;AAC3D,IAAI,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,KAAK;AAC/B,MAAM,MAAM,OAAO,GAAG;AACtB,QAAQ,EAAE,EAAE,sBAAsB,CAAC,KAAK;AACxC,QAAQ,KAAK;AACb,QAAQ,IAAI,EAAE,KAAK;AACnB,QAAQ,OAAO;AACf,OAAO,CAAC;AACR,MAAM,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AACtC,KAAK,CAAC,CAAC;AACP,GAAG;AACH,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AAC7B,CAAC;AACD,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,GAAG,KAAK;AACvC,EAAE,MAAM,GAAG,CAAC;AACZ,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,OAAO,OAAO,KAAK;AACpC,EAAE,QAAQ,OAAO,CAAC,EAAE;AACpB,IAAI,KAAK,mBAAmB,CAAC,OAAO,EAAE;AACtC,MAAM,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACrC,MAAM,MAAM,QAAQ,GAAG;AACvB,QAAQ,EAAE,EAAE,sBAAsB,CAAC,SAAS;AAC5C,QAAQ,OAAO,EAAE,OAAO,CAAC,OAAO;AAChC,OAAO,CAAC;AACR,MAAM,UAAU,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;AACvC,MAAM,MAAM;AACZ,KAAK;AACL,IAAI,KAAK,mBAAmB,CAAC,OAAO,EAAE;AACtC,MAAM,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;AACtD,MAAM,MAAM,QAAQ,GAAG;AACvB,QAAQ,EAAE,EAAE,sBAAsB,CAAC,SAAS;AAC5C,QAAQ,OAAO,EAAE,OAAO,CAAC,OAAO;AAChC,OAAO,CAAC;AACR,MAAM,UAAU,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;AACvC,MAAM,MAAM;AACZ,KAAK;AACL,IAAI,KAAK,mBAAmB,CAAC,IAAI,EAAE;AACnC,MAAM,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAChD,MAAM,IAAI,CAAC,KAAK,EAAE;AAClB,QAAQ,MAAM,IAAI,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;AACnE,OAAO;AACP,MAAM,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AACxC,MAAM,MAAM;AACZ,KAAK;AACL,IAAI,KAAK,mBAAmB,CAAC,mBAAmB,EAAE;AAClD,MAAM,MAAM;AACZ,KAAK;AACL,GAAG;AACH,CAAC,CAAC"}

package/dist/utils/IdentifyThrottler.cjs

@@ -0,0 +1,33 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+const promises = require('node:timers/promises');
+
+class IdentifyThrottler {
+  constructor(manager) {
+    this.manager = manager;
+    this.identifyState = {
+      remaining: 0,
+      resetsAt: Infinity
+    };
+  }
+  async waitForIdentify() {
+    if (this.identifyState.remaining <= 0) {
+      const diff = this.identifyState.resetsAt - Date.now();
+      if (diff <= 5e3) {
+        const time = diff + Math.random() * 1500;
+        await promises.setTimeout(time);
+      }
+      const info = await this.manager.fetchGatewayInformation();
+      this.identifyState = {
+        remaining: info.session_start_limit.max_concurrency,
+        resetsAt: Date.now() + 5e3
+      };
+    }
+    this.identifyState.remaining--;
+  }
+}
+
+exports.IdentifyThrottler = IdentifyThrottler;
+//# sourceMappingURL=IdentifyThrottler.cjs.map

package/dist/utils/IdentifyThrottler.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"IdentifyThrottler.cjs","sources":["../../src/utils/IdentifyThrottler.ts"],"sourcesContent":["import { setTimeout as sleep } from 'node:timers/promises';\nimport type { WebSocketManager } from '../ws/WebSocketManager';\n\nexport class IdentifyThrottler {\n\tprivate identifyState = {\n\t\tremaining: 0,\n\t\tresetsAt: Infinity,\n\t};\n\n\tpublic constructor(private readonly manager: WebSocketManager) {}\n\n\tpublic async waitForIdentify(): Promise<void> {\n\t\tif (this.identifyState.remaining <= 0) {\n\t\t\tconst diff = this.identifyState.resetsAt - Date.now();\n\t\t\tif (diff <= 5_000) {\n\t\t\t\tconst time = diff + Math.random() * 1_500;\n\t\t\t\tawait sleep(time);\n\t\t\t}\n\n\t\t\tconst info = await this.manager.fetchGatewayInformation();\n\t\t\tthis.identifyState = {\n\t\t\t\tremaining: info.session_start_limit.max_concurrency,\n\t\t\t\tresetsAt: Date.now() + 5_000,\n\t\t\t};\n\t\t}\n\n\t\tthis.identifyState.remaining--;\n\t}\n}\n"],"names":["sleep"],"mappings":";;;;;;AACO,MAAM,iBAAiB,CAAC;AAC/B,EAAE,WAAW,CAAC,OAAO,EAAE;AACvB,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,IAAI,IAAI,CAAC,aAAa,GAAG;AACzB,MAAM,SAAS,EAAE,CAAC;AAClB,MAAM,QAAQ,EAAE,QAAQ;AACxB,KAAK,CAAC;AACN,GAAG;AACH,EAAE,MAAM,eAAe,GAAG;AAC1B,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,IAAI,CAAC,EAAE;AAC3C,MAAM,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;AAC5D,MAAM,IAAI,IAAI,IAAI,GAAG,EAAE;AACvB,QAAQ,MAAM,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;AACjD,QAAQ,MAAMA,mBAAK,CAAC,IAAI,CAAC,CAAC;AAC1B,OAAO;AACP,MAAM,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;AAChE,MAAM,IAAI,CAAC,aAAa,GAAG;AAC3B,QAAQ,SAAS,EAAE,IAAI,CAAC,mBAAmB,CAAC,eAAe;AAC3D,QAAQ,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG;AAClC,OAAO,CAAC;AACR,KAAK;AACL,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;AACnC,GAAG;AACH;;;;"}

package/dist/utils/IdentifyThrottler.d.ts

@@ -0,0 +1,8 @@
+import type { WebSocketManager } from '../ws/WebSocketManager';
+export declare class IdentifyThrottler {
+    private readonly manager;
+    private identifyState;
+    constructor(manager: WebSocketManager);
+    waitForIdentify(): Promise<void>;
+}
+//# sourceMappingURL=IdentifyThrottler.d.ts.map

package/dist/utils/IdentifyThrottler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"IdentifyThrottler.d.ts","sourceRoot":"","sources":["../../src/utils/IdentifyThrottler.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,qBAAa,iBAAiB;IAMV,OAAO,CAAC,QAAQ,CAAC,OAAO;IAL3C,OAAO,CAAC,aAAa,CAGnB;gBAEkC,OAAO,EAAE,gBAAgB;IAEhD,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;CAiB7C"}

package/dist/utils/IdentifyThrottler.mjs

@@ -0,0 +1,29 @@
+import { setTimeout } from 'node:timers/promises';
+
+class IdentifyThrottler {
+  constructor(manager) {
+    this.manager = manager;
+    this.identifyState = {
+      remaining: 0,
+      resetsAt: Infinity
+    };
+  }
+  async waitForIdentify() {
+    if (this.identifyState.remaining <= 0) {
+      const diff = this.identifyState.resetsAt - Date.now();
+      if (diff <= 5e3) {
+        const time = diff + Math.random() * 1500;
+        await setTimeout(time);
+      }
+      const info = await this.manager.fetchGatewayInformation();
+      this.identifyState = {
+        remaining: info.session_start_limit.max_concurrency,
+        resetsAt: Date.now() + 5e3
+      };
+    }
+    this.identifyState.remaining--;
+  }
+}
+
+export { IdentifyThrottler };
+//# sourceMappingURL=IdentifyThrottler.mjs.map

package/dist/utils/IdentifyThrottler.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"IdentifyThrottler.mjs","sources":["../../src/utils/IdentifyThrottler.ts"],"sourcesContent":["import { setTimeout as sleep } from 'node:timers/promises';\nimport type { WebSocketManager } from '../ws/WebSocketManager';\n\nexport class IdentifyThrottler {\n\tprivate identifyState = {\n\t\tremaining: 0,\n\t\tresetsAt: Infinity,\n\t};\n\n\tpublic constructor(private readonly manager: WebSocketManager) {}\n\n\tpublic async waitForIdentify(): Promise<void> {\n\t\tif (this.identifyState.remaining <= 0) {\n\t\t\tconst diff = this.identifyState.resetsAt - Date.now();\n\t\t\tif (diff <= 5_000) {\n\t\t\t\tconst time = diff + Math.random() * 1_500;\n\t\t\t\tawait sleep(time);\n\t\t\t}\n\n\t\t\tconst info = await this.manager.fetchGatewayInformation();\n\t\t\tthis.identifyState = {\n\t\t\t\tremaining: info.session_start_limit.max_concurrency,\n\t\t\t\tresetsAt: Date.now() + 5_000,\n\t\t\t};\n\t\t}\n\n\t\tthis.identifyState.remaining--;\n\t}\n}\n"],"names":["sleep"],"mappings":";;AACO,MAAM,iBAAiB,CAAC;AAC/B,EAAE,WAAW,CAAC,OAAO,EAAE;AACvB,IAAI,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;AAC3B,IAAI,IAAI,CAAC,aAAa,GAAG;AACzB,MAAM,SAAS,EAAE,CAAC;AAClB,MAAM,QAAQ,EAAE,QAAQ;AACxB,KAAK,CAAC;AACN,GAAG;AACH,EAAE,MAAM,eAAe,GAAG;AAC1B,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,IAAI,CAAC,EAAE;AAC3C,MAAM,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;AAC5D,MAAM,IAAI,IAAI,IAAI,GAAG,EAAE;AACvB,QAAQ,MAAM,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;AACjD,QAAQ,MAAMA,UAAK,CAAC,IAAI,CAAC,CAAC;AAC1B,OAAO;AACP,MAAM,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;AAChE,MAAM,IAAI,CAAC,aAAa,GAAG;AAC3B,QAAQ,SAAS,EAAE,IAAI,CAAC,mBAAmB,CAAC,eAAe;AAC3D,QAAQ,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG;AAClC,OAAO,CAAC;AACR,KAAK;AACL,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;AACnC,GAAG;AACH;;;;"}

package/dist/utils/constants.cjs

@@ -0,0 +1,63 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+const node_fs = require('node:fs');
+const node_path = require('node:path');
+const collection = require('@discordjs/collection');
+const v10 = require('discord-api-types/v10');
+const utils = require('./utils.cjs');
+
+var Encoding = /* @__PURE__ */ ((Encoding2) => {
+  Encoding2["JSON"] = "json";
+  return Encoding2;
+})(Encoding || {});
+var CompressionMethod = /* @__PURE__ */ ((CompressionMethod2) => {
+  CompressionMethod2["ZlibStream"] = "zlib-stream";
+  return CompressionMethod2;
+})(CompressionMethod || {});
+const packageJson = node_fs.readFileSync(node_path.join(__dirname, "..", "..", "package.json"), "utf8");
+const Package = JSON.parse(packageJson);
+const DefaultDeviceProperty = `@discordjs/ws ${Package.version}`;
+const getDefaultSessionStore = utils.lazy(() => new collection.Collection());
+const DefaultWebSocketManagerOptions = {
+  shardCount: null,
+  shardIds: null,
+  largeThreshold: null,
+  initialPresence: null,
+  identifyProperties: {
+    browser: DefaultDeviceProperty,
+    device: DefaultDeviceProperty,
+    os: process.platform
+  },
+  version: v10.APIVersion,
+  encoding: "json" /* JSON */,
+  compression: null,
+  retrieveSessionInfo(shardId) {
+    const store = getDefaultSessionStore();
+    return store.get(shardId) ?? null;
+  },
+  updateSessionInfo(shardId, info) {
+    const store = getDefaultSessionStore();
+    if (info) {
+      store.set(shardId, info);
+    } else {
+      store.delete(shardId);
+    }
+  },
+  handshakeTimeout: 3e4,
+  helloTimeout: 6e4,
+  readyTimeout: 15e3
+};
+const ImportantGatewayOpcodes = /* @__PURE__ */ new Set([
+  v10.GatewayOpcodes.Heartbeat,
+  v10.GatewayOpcodes.Identify,
+  v10.GatewayOpcodes.Resume
+]);
+
+exports.CompressionMethod = CompressionMethod;
+exports.DefaultDeviceProperty = DefaultDeviceProperty;
+exports.DefaultWebSocketManagerOptions = DefaultWebSocketManagerOptions;
+exports.Encoding = Encoding;
+exports.ImportantGatewayOpcodes = ImportantGatewayOpcodes;
+//# sourceMappingURL=constants.cjs.map

package/dist/utils/constants.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.cjs","sources":["../../src/utils/constants.ts"],"sourcesContent":["import { readFileSync } from 'node:fs';\nimport { join } from 'node:path';\nimport { Collection } from '@discordjs/collection';\nimport { APIVersion, GatewayOpcodes } from 'discord-api-types/v10';\nimport { lazy } from './utils';\nimport type { OptionalWebSocketManagerOptions, SessionInfo } from '../ws/WebSocketManager';\n\n/**\n * Valid encoding types\n */\nexport enum Encoding {\n\tJSON = 'json',\n}\n\n/**\n * Valid compression methods\n */\nexport enum CompressionMethod {\n\tZlibStream = 'zlib-stream',\n}\n\nconst packageJson = readFileSync(join(__dirname, '..', '..', 'package.json'), 'utf8');\n// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\nconst Package = JSON.parse(packageJson);\n\n// eslint-disable-next-line @typescript-eslint/restrict-template-expressions, @typescript-eslint/no-unsafe-member-access\nexport const DefaultDeviceProperty = `@discordjs/ws ${Package.version}`;\n\nconst getDefaultSessionStore = lazy(() => new Collection<number, SessionInfo | null>());\n\n/**\n * Default options used by the manager\n */\nexport const DefaultWebSocketManagerOptions: OptionalWebSocketManagerOptions = {\n\tshardCount: null,\n\tshardIds: null,\n\tlargeThreshold: null,\n\tinitialPresence: null,\n\tidentifyProperties: {\n\t\tbrowser: DefaultDeviceProperty,\n\t\tdevice: DefaultDeviceProperty,\n\t\tos: process.platform,\n\t},\n\tversion: APIVersion,\n\tencoding: Encoding.JSON,\n\tcompression: null,\n\tretrieveSessionInfo(shardId) {\n\t\tconst store = getDefaultSessionStore();\n\t\treturn store.get(shardId) ?? null;\n\t},\n\tupdateSessionInfo(shardId: number, info: SessionInfo | null) {\n\t\tconst store = getDefaultSessionStore();\n\t\tif (info) {\n\t\t\tstore.set(shardId, info);\n\t\t} else {\n\t\t\tstore.delete(shardId);\n\t\t}\n\t},\n\thandshakeTimeout: 30_000,\n\thelloTimeout: 60_000,\n\treadyTimeout: 15_000,\n};\n\nexport const ImportantGatewayOpcodes = new Set([\n\tGatewayOpcodes.Heartbeat,\n\tGatewayOpcodes.Identify,\n\tGatewayOpcodes.Resume,\n]);\n"],"names":["readFileSync","join","lazy","Collection","APIVersion","GatewayOpcodes"],"mappings":";;;;;;;;;;AAKU,IAAC,QAAQ,mBAAmB,CAAC,CAAC,SAAS,KAAK;AACtD,EAAE,SAAS,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;AAC7B,EAAE,OAAO,SAAS,CAAC;AACnB,CAAC,EAAE,QAAQ,IAAI,EAAE,EAAE;AACT,IAAC,iBAAiB,mBAAmB,CAAC,CAAC,kBAAkB,KAAK;AACxE,EAAE,kBAAkB,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;AACnD,EAAE,OAAO,kBAAkB,CAAC;AAC5B,CAAC,EAAE,iBAAiB,IAAI,EAAE,EAAE;AAC5B,MAAM,WAAW,GAAGA,oBAAY,CAACC,cAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,MAAM,CAAC,CAAC;AACtF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;AAC5B,MAAC,qBAAqB,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;AACxE,MAAM,sBAAsB,GAAGC,UAAI,CAAC,MAAM,IAAIC,qBAAU,EAAE,CAAC,CAAC;AAChD,MAAC,8BAA8B,GAAG;AAC9C,EAAE,UAAU,EAAE,IAAI;AAClB,EAAE,QAAQ,EAAE,IAAI;AAChB,EAAE,cAAc,EAAE,IAAI;AACtB,EAAE,eAAe,EAAE,IAAI;AACvB,EAAE,kBAAkB,EAAE;AACtB,IAAI,OAAO,EAAE,qBAAqB;AAClC,IAAI,MAAM,EAAE,qBAAqB;AACjC,IAAI,EAAE,EAAE,OAAO,CAAC,QAAQ;AACxB,GAAG;AACH,EAAE,OAAO,EAAEC,cAAU;AACrB,EAAE,QAAQ,EAAE,MAAM;AAClB,EAAE,WAAW,EAAE,IAAI;AACnB,EAAE,mBAAmB,CAAC,OAAO,EAAE;AAC/B,IAAI,MAAM,KAAK,GAAG,sBAAsB,EAAE,CAAC;AAC3C,IAAI,OAAO,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC;AACtC,GAAG;AACH,EAAE,iBAAiB,CAAC,OAAO,EAAE,IAAI,EAAE;AACnC,IAAI,MAAM,KAAK,GAAG,sBAAsB,EAAE,CAAC;AAC3C,IAAI,IAAI,IAAI,EAAE;AACd,MAAM,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;AAC/B,KAAK,MAAM;AACX,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAC5B,KAAK;AACL,GAAG;AACH,EAAE,gBAAgB,EAAE,GAAG;AACvB,EAAE,YAAY,EAAE,GAAG;AACnB,EAAE,YAAY,EAAE,IAAI;AACpB,EAAE;AACU,MAAC,uBAAuB,mBAAmB,IAAI,GAAG,CAAC;AAC/D,EAAEC,kBAAc,CAAC,SAAS;AAC1B,EAAEA,kBAAc,CAAC,QAAQ;AACzB,EAAEA,kBAAc,CAAC,MAAM;AACvB,CAAC;;;;;;;;"}

package/dist/utils/constants.d.ts

@@ -0,0 +1,21 @@
+import { GatewayOpcodes } from 'discord-api-types/v10';
+import type { OptionalWebSocketManagerOptions } from '../ws/WebSocketManager';
+/**
+ * Valid encoding types
+ */
+export declare enum Encoding {
+    JSON = "json"
+}
+/**
+ * Valid compression methods
+ */
+export declare enum CompressionMethod {
+    ZlibStream = "zlib-stream"
+}
+export declare const DefaultDeviceProperty: string;
+/**
+ * Default options used by the manager
+ */
+export declare const DefaultWebSocketManagerOptions: OptionalWebSocketManagerOptions;
+export declare const ImportantGatewayOpcodes: Set<GatewayOpcodes>;
+//# sourceMappingURL=constants.d.ts.map

package/dist/utils/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/utils/constants.ts"],"names":[],"mappings":"AAGA,OAAO,EAAc,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,KAAK,EAAE,+BAA+B,EAAe,MAAM,wBAAwB,CAAC;AAE3F;;GAEG;AACH,oBAAY,QAAQ;IACnB,IAAI,SAAS;CACb;AAED;;GAEG;AACH,oBAAY,iBAAiB;IAC5B,UAAU,gBAAgB;CAC1B;AAOD,eAAO,MAAM,qBAAqB,QAAqC,CAAC;AAIxE;;GAEG;AACH,eAAO,MAAM,8BAA8B,EAAE,+BA4B5C,CAAC;AAEF,eAAO,MAAM,uBAAuB,qBAIlC,CAAC"}

package/dist/utils/constants.mjs

@@ -0,0 +1,64 @@
+import { readFileSync } from 'node:fs';
+import { join } from 'node:path';
+import { Collection } from '@discordjs/collection';
+import { APIVersion, GatewayOpcodes } from 'discord-api-types/v10';
+import { lazy } from './utils.mjs';
+
+
+
+// -- Unbuild CommonJS Shims --
+import __cjs_url__ from 'url';
+import __cjs_path__ from 'path';
+import __cjs_mod__ from 'module';
+const __filename = __cjs_url__.fileURLToPath(import.meta.url);
+const __dirname = __cjs_path__.dirname(__filename);
+const require = __cjs_mod__.createRequire(import.meta.url);
+var Encoding = /* @__PURE__ */ ((Encoding2) => {
+  Encoding2["JSON"] = "json";
+  return Encoding2;
+})(Encoding || {});
+var CompressionMethod = /* @__PURE__ */ ((CompressionMethod2) => {
+  CompressionMethod2["ZlibStream"] = "zlib-stream";
+  return CompressionMethod2;
+})(CompressionMethod || {});
+const packageJson = readFileSync(join(__dirname, "..", "..", "package.json"), "utf8");
+const Package = JSON.parse(packageJson);
+const DefaultDeviceProperty = `@discordjs/ws ${Package.version}`;
+const getDefaultSessionStore = lazy(() => new Collection());
+const DefaultWebSocketManagerOptions = {
+  shardCount: null,
+  shardIds: null,
+  largeThreshold: null,
+  initialPresence: null,
+  identifyProperties: {
+    browser: DefaultDeviceProperty,
+    device: DefaultDeviceProperty,
+    os: process.platform
+  },
+  version: APIVersion,
+  encoding: "json" /* JSON */,
+  compression: null,
+  retrieveSessionInfo(shardId) {
+    const store = getDefaultSessionStore();
+    return store.get(shardId) ?? null;
+  },
+  updateSessionInfo(shardId, info) {
+    const store = getDefaultSessionStore();
+    if (info) {
+      store.set(shardId, info);
+    } else {
+      store.delete(shardId);
+    }
+  },
+  handshakeTimeout: 3e4,
+  helloTimeout: 6e4,
+  readyTimeout: 15e3
+};
+const ImportantGatewayOpcodes = /* @__PURE__ */ new Set([
+  GatewayOpcodes.Heartbeat,
+  GatewayOpcodes.Identify,
+  GatewayOpcodes.Resume
+]);
+
+export { CompressionMethod, DefaultDeviceProperty, DefaultWebSocketManagerOptions, Encoding, ImportantGatewayOpcodes };
+//# sourceMappingURL=constants.mjs.map

package/dist/utils/constants.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.mjs","sources":["../../src/utils/constants.ts"],"sourcesContent":["import { readFileSync } from 'node:fs';\nimport { join } from 'node:path';\nimport { Collection } from '@discordjs/collection';\nimport { APIVersion, GatewayOpcodes } from 'discord-api-types/v10';\nimport { lazy } from './utils';\nimport type { OptionalWebSocketManagerOptions, SessionInfo } from '../ws/WebSocketManager';\n\n/**\n * Valid encoding types\n */\nexport enum Encoding {\n\tJSON = 'json',\n}\n\n/**\n * Valid compression methods\n */\nexport enum CompressionMethod {\n\tZlibStream = 'zlib-stream',\n}\n\nconst packageJson = readFileSync(join(__dirname, '..', '..', 'package.json'), 'utf8');\n// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\nconst Package = JSON.parse(packageJson);\n\n// eslint-disable-next-line @typescript-eslint/restrict-template-expressions, @typescript-eslint/no-unsafe-member-access\nexport const DefaultDeviceProperty = `@discordjs/ws ${Package.version}`;\n\nconst getDefaultSessionStore = lazy(() => new Collection<number, SessionInfo | null>());\n\n/**\n * Default options used by the manager\n */\nexport const DefaultWebSocketManagerOptions: OptionalWebSocketManagerOptions = {\n\tshardCount: null,\n\tshardIds: null,\n\tlargeThreshold: null,\n\tinitialPresence: null,\n\tidentifyProperties: {\n\t\tbrowser: DefaultDeviceProperty,\n\t\tdevice: DefaultDeviceProperty,\n\t\tos: process.platform,\n\t},\n\tversion: APIVersion,\n\tencoding: Encoding.JSON,\n\tcompression: null,\n\tretrieveSessionInfo(shardId) {\n\t\tconst store = getDefaultSessionStore();\n\t\treturn store.get(shardId) ?? null;\n\t},\n\tupdateSessionInfo(shardId: number, info: SessionInfo | null) {\n\t\tconst store = getDefaultSessionStore();\n\t\tif (info) {\n\t\t\tstore.set(shardId, info);\n\t\t} else {\n\t\t\tstore.delete(shardId);\n\t\t}\n\t},\n\thandshakeTimeout: 30_000,\n\thelloTimeout: 60_000,\n\treadyTimeout: 15_000,\n};\n\nexport const ImportantGatewayOpcodes = new Set([\n\tGatewayOpcodes.Heartbeat,\n\tGatewayOpcodes.Identify,\n\tGatewayOpcodes.Resume,\n]);\n"],"names":[],"mappings":";;;;;;;;;;;;;;;AAKU;AACV;AACA;AACA;AACU;AACV;AACA;AACA;AACA;AACA;AACY;AACZ;AACY;AACZ;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACY;AACZ;AACA;AACA;AACA;;"}

package/dist/utils/utils.cjs

@@ -0,0 +1,15 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+function range({ start, end }) {
+  return Array.from({ length: end - start + 1 }, (_, i) => i + start);
+}
+function lazy(cb) {
+  let defaultValue;
+  return () => defaultValue ?? (defaultValue = cb());
+}
+
+exports.lazy = lazy;
+exports.range = range;
+//# sourceMappingURL=utils.cjs.map