feat: @discordjs/ws (#8260)

Co-authored-by: Parbez <imranbarbhuiya.fsd@gmail.com>

packages/ws/src/strategies/sharding/SimpleShardingStrategy.ts

@@ -0,0 +1,64 @@
+import { Collection } from '@discordjs/collection';
+import type { GatewaySendPayload } from 'discord-api-types/v10';
+import type { IShardingStrategy } from './IShardingStrategy';
+import { IdentifyThrottler } from '../../utils/IdentifyThrottler';
+import type { WebSocketManager } from '../../ws/WebSocketManager';
+import { WebSocketShard, WebSocketShardDestroyOptions, WebSocketShardEvents } from '../../ws/WebSocketShard';
+import { managerToFetchingStrategyOptions } from '../context/IContextFetchingStrategy';
+import { SimpleContextFetchingStrategy } from '../context/SimpleContextFetchingStrategy';
+
+/**
+ * Simple strategy that just spawns shards in the current process
+ */
+export class SimpleShardingStrategy implements IShardingStrategy {
+	private readonly manager: WebSocketManager;
+	private readonly shards = new Collection<number, WebSocketShard>();
+
+	private readonly throttler: IdentifyThrottler;
+
+	public constructor(manager: WebSocketManager) {
+		this.manager = manager;
+		this.throttler = new IdentifyThrottler(manager);
+	}
+
+	public async spawn(shardIds: number[]) {
+		const strategyOptions = await managerToFetchingStrategyOptions(this.manager);
+		for (const shardId of shardIds) {
+			const strategy = new SimpleContextFetchingStrategy(this.manager, strategyOptions);
+			const shard = new WebSocketShard(strategy, shardId);
+			for (const event of Object.values(WebSocketShardEvents)) {
+				// @ts-expect-error
+				shard.on(event, (payload) => this.manager.emit(event, { ...payload, shardId }));
+			}
+			this.shards.set(shardId, shard);
+		}
+	}
+
+	public async connect() {
+		const promises = [];
+
+		for (const shard of this.shards.values()) {
+			await this.throttler.waitForIdentify();
+			promises.push(shard.connect());
+		}
+
+		await Promise.all(promises);
+	}
+
+	public async destroy(options?: Omit<WebSocketShardDestroyOptions, 'recover'>) {
+		const promises = [];
+
+		for (const shard of this.shards.values()) {
+			promises.push(shard.destroy(options));
+		}
+
+		await Promise.all(promises);
+		this.shards.clear();
+	}
+
+	public send(shardId: number, payload: GatewaySendPayload) {
+		const shard = this.shards.get(shardId);
+		if (!shard) throw new Error(`Shard ${shardId} not found`);
+		return shard.send(payload);
+	}
+}