daenamnu/discordeno
1
0
Fork 0
mirror of https://github.com/discordeno/discordeno.git synced 2026-06-02 17:00:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove excess

Browse Source
This commit is contained in:
Skillz
2020-11-03 15:20:10 -05:00
parent 46af87f898
commit 70ba80b56e
1 changed files with 0 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
src/module/hugebot.ts
View File

@@ -1,105 +0,0 @@
import { delay } from "../../deps.ts";
import { DiscordBotGatewayData, RequestManager } from "../../mod.ts";
import { endpoints } from "../constants/discord.ts";
import { ClientOptions, EventHandlers } from "../types/options.ts";
import { botGatewayData } from "./client.ts";
const botOptions = {
workers: new Map<number, Worker>(),
eventHandlers: {} as EventHandlers,
botGatewayData: {} as DiscordBotGatewayData,
customShards: [] as number[],
shardsPerWorker: 25,
identifyPayload: {
token: "",
compress: true,
properties: {
$os: "linux",
$browser: "Discordeno",
$device: "Discordeno",
},
intents: 0,
shard: [0, 0],
},
};
/**
* This function should be used only by bot developers whose bots are in over 25,000 servers.
* Please be aware if you are a beginner developer using this, things will not work as per the guides. This is for advanced developers only!
*
* Advanced Devs: This function will allow you to have an insane amount of customization potential as when you get to large bots you need to be able to optimize every tiny detail to make you bot work the way you need.
*/
export async function startBigBrainBot(data: BigBrainBotOptions) {
botOptions.identifyPayload.token = `Bot ${data.token}`;
if (data.eventHandlers) botOptions.eventHandlers = data.eventHandlers;
if (data.shards) botOptions.customShards = data.shards;
if (data.compress) botOptions.identifyPayload.compress = data.compress;
if (data.shardsPerWorker) botOptions.shardsPerWorker = data.shardsPerWorker;
// Initial API connection to get info about bots connection
botOptions.botGatewayData = await RequestManager.get(
endpoints.GATEWAY_BOT,
) as DiscordBotGatewayData;
botOptions.identifyPayload.intents = data.intents.reduce(
(bits, next) => (bits |= next),
0,
);
botOptions.identifyPayload.shard = [0, botGatewayData.shards];
spawnBigBrainBotShards();
}
async function spawnBigBrainBotShards(shardID = 0, skipChecks = 0) {
if (shardID >= botOptions.botGatewayData.shards) return;
// 25 shards but shards start at 0 so we use 24
const workerID = shardID % 24;
const worker = botOptions.workers.get(workerID)
// High max concurrency allows starting shards faster
if (skipChecks) {
// If the worker exists we just need to add
if (worker) {
addShardToWorker(workerID, shardID);
} else {
createShardWorker(workerID, shardID);
}
spawnBigBrainBotShards(shardID + 1, skipChecks - 1);
}
// Make sure we can create a shard or we are waiting for shards to connect still.
if (createNextShard) {
// !(shardid % botOptions.botGatewayData.session_start_limit.max_concurrency)
createNextShard = false;
if (botOptions.botGatewayData.shards >= 25) createShardWorker();
// Start the next few shards based on max concurrency
spawnBigBrainBotShards(shardID + 1, botOptions.botGatewayData.session_start_limit.max_concurrency);
return;
}
await delay(1000);
spawnBigBrainBotShards(shardID);
}
export function createShardWorker(workerID: number, shardID: number) {
const path = new URL("./shard.ts", import.meta.url).toString();
const shard = new Worker(path, { type: "module", deno: true });
// Add to worker map
botOptions.workers.set(workerID, shard)
}
export interface BigBrainBotOptions extends ClientOptions {
/** This can be used to distribute your bot across different servers. For example, if you wanted 1 million shards per server you could control it using this. */
shards?: [number, number];
/** This can be used to forward the ws handling to a proxy. */
wsURL?: string;
/** This can be used to forward the REST handling to a proxy. */
restURL?: string;
/** This allows you to control how many shards per worker. For the times where you can optimize with more shards per worker as your bot has less tasks per shard.
* @default 25
*/
shardsPerWorker?: number;
}