feat: indomitable strategy

package.json

@@ -51,8 +51,10 @@
 		"lint:fix": "eslint src --fix",
 		"prepack": "pnpm run build"
 	},
+	"dependencies": {
+		"@discordjs/ws": "^2.0.4"
+	},
 	"devDependencies": {
-		"@discordjs/ws": "^2.0.4",
 		"@types/node": "^25.9.1",
 		"@types/ws": "^8.18.1",
 		"discord.js": "^14.26.4",

src/client/ShardClientUtil.ts

@@ -1,11 +1,13 @@
 import EventEmitter from "node:events";
 import process from "node:process";
 import { clearTimeout } from "node:timers";
+import type { WebSocketManager } from "@discordjs/ws";
 import type { Client } from "discord.js";
 import type { Indomitable } from "../Indomitable.js";
 import type { AbortableData, InternalOpsData, Message, SessionObject, Transportable } from "../Util.js";
 import { EnvProcessData, MakeAbortableRequest, InternalOps } from "../Util.js";
 import { ClientWorker } from "../ipc/ClientWorker.js";
+import type { IndomitableShardingStrategyInterface } from "../strategies/IndomitableShardingStrategyInterface.js";
 
 export interface ShardClientUtilEvents {
 	message: [message: Message];
@@ -53,6 +55,15 @@ export class ShardClientUtil extends EventEmitter<ShardClientUtilEvents> {
 		return Number(BigInt(end) - start);
 	}
 
+	/**
+	 * Returns the current WebSocket sharding strategy.
+	 */
+	public get strategy(): IndomitableShardingStrategyInterface {
+		// @ts-expect-error internal field
+		// eslint-disable-next-line @typescript-eslint/dot-notation
+		return (this.client.ws["_ws"] as WebSocketManager).strategy;
+	}
+
 	/**
 	 * Evaluates a script or function on all clusters in the context of the client
 	 *

src/index.ts

@@ -1,6 +1,7 @@
 export * from "./client/ShardClient.js";
 export * from "./client/ShardClientUtil.js";
 export * from "./concurrency/ConcurrencyManager.js";
+export * from "./strategies/IndomitableWorkerShardingStrategy.js";
 export * from "./ipc/BaseIpc.js";
 export * from "./ipc/MainWorker.js";
 export * from "./ipc/BaseWorker.js";

src/strategies/IndomitableShardingStrategyInterface.ts

@@ -0,0 +1,14 @@
+/**
+ * This file is adapted from discord.js and includes additional modifications.
+ *
+ * Original Apache 2.0 license:
+ * https://github.com/discordjs/discord.js/blob/3d6121589f9c0d91f7cf4976307e8be07053a277/LICENSE
+ */
+import type { IShardingStrategy, WebSocketShardDestroyOptions } from "@discordjs/ws";
+import type { Awaitable } from "discord.js";
+import type { WebsocketShardState } from "./IndomitableWorkerShardingStrategy";
+
+export interface IndomitableShardingStrategyInterface extends Omit<IShardingStrategy, "fetchStatus"> {
+	destroyShards(shardIds: number[], options?: WebSocketShardDestroyOptions): Awaitable<void>;
+	fetchStatus(): Awaitable<Map<number, WebsocketShardState>>;
+}