discordeno/module/Client.ts

import { endpoints } from "../constants/discord.ts";
import RequestManager from "../managers/RequestManager.ts";
import { DiscordBotGateway, DiscordPayload, DiscordHeartbeatPayload } from "../types/discord.ts";
import ShardingManager from "../managers/ShardingManager.ts";
import {
  connectWebSocket,
  isWebSocketCloseEvent,
  isWebSocketPingEvent,
  isWebSocketPongEvent,
  WebSocket
} from "https://deno.land/std/ws/mod.ts";
// import { encode } from "https://deno.land/std/strings/mod.ts"
// import { BufReader } from "https://deno.land/std/io/bufio.ts"
// import { TextProtoReader } from "https://deno.land/std/textproto/mod.ts"
import { blue, green, red, yellow } from "https://deno.land/std/fmt/colors.ts"
import { keepDiscordWebsocketAlive } from "./websocket.ts";

class Client {
  /** The bot's token. This should never be used by end users. It is meant to be used internally to make requests to the Discord API. */
  token: string;
  /** The Rate limit manager to handle all outgoing requests to discord. Not meant to be used by users. */
  RequestManager: RequestManager;
  /** Creates and handles all the shards necessary for the bot. */
  ShardingManager: ShardingManager;

  constructor(token: string) {
    this.token = `Bot ${token}`;
    this.RequestManager = new RequestManager(this, this.token);
    this.ShardingManager = new ShardingManager();
  }

  /** Begins initial handshake, creates the websocket with Discord and spawns all necessary shards. */
  async connect() {
    const data = (await this.RequestManager.get(
      endpoints.GATEWAY_BOT
    )) as DiscordBotGateway;
    // Open a WS with the url from discord.
    const sock = await connectWebSocket(data.url);
    console.log(sock)
    console.log(green("ws connected! (type 'close' to quit)"));

    for await (const msg of sock.receive()) {
      if (typeof msg === "string") {
        try {
          const json = JSON.parse(msg)
          this.handleDiscordPayload(json, sock)
        } catch {
          console.log(red(`Invalid JSON String send by discord: ${msg}`))
        }
        console.log(yellow("< " + msg));
      } else if (isWebSocketPingEvent(msg)) {
        console.log(blue("< ping"));
      } else if (isWebSocketPongEvent(msg)) {
        console.log(blue("< pong"));
      } else if (isWebSocketCloseEvent(msg)) {
        console.log(red(`closed: code=${msg.code}, reason=${msg.reason}`));
      }
    }

    // Begin spawning all necessary shards
    this.spawnShards(data.shards);
  }

  handleDiscordPayload(data: DiscordPayload, socket: WebSocket) {
    switch (data.op) {
      case 10: // Initial Heartbeat
        keepDiscordWebsocketAlive(socket, (data.d as DiscordHeartbeatPayload).heartbeat_interval, data.s)
    }
  }

  spawnShards(total: number, id = 1) {
    // this.ShardingManager.spawnShard(id);
    if (id < total) this.spawnShards(total, id + 1);
  }
}

export default Client;