lunibee 0.1.0 → 0.1.2

package/dist/ws/index.js

@@ -0,0 +1,582 @@
+// @bun
+// packages/types/src/index.ts
+var GatewayIntentBits = {
+  Guilds: 1 << 0,
+  GuildMembers: 1 << 1,
+  GuildModeration: 1 << 2,
+  GuildExpressions: 1 << 3,
+  GuildIntegrations: 1 << 4,
+  GuildWebhooks: 1 << 5,
+  GuildInvites: 1 << 6,
+  GuildVoiceStates: 1 << 7,
+  GuildPresences: 1 << 8,
+  GuildMessages: 1 << 9,
+  GuildMessageReactions: 1 << 10,
+  GuildMessageTyping: 1 << 11,
+  DirectMessages: 1 << 12,
+  DirectMessageReactions: 1 << 13,
+  DirectMessageTyping: 1 << 14,
+  MessageContent: 1 << 15,
+  GuildScheduledEvents: 1 << 16,
+  AutoModerationConfiguration: 1 << 20,
+  AutoModerationExecution: 1 << 21,
+  GuildMessagePolls: 1 << 24,
+  DirectMessagePolls: 1 << 25
+};
+var IntentBits = {
+  guild: 1 << 0,
+  guilds: 1 << 0,
+  guildMembers: 1 << 1,
+  guildModeration: 1 << 2,
+  guildBans: 1 << 2,
+  guildExpressions: 1 << 3,
+  guildEmojis: 1 << 3,
+  guildEmojisAndStickers: 1 << 3,
+  guildIntegrations: 1 << 4,
+  guildWebhooks: 1 << 5,
+  guildInvites: 1 << 6,
+  guildVoiceStates: 1 << 7,
+  guildPresences: 1 << 8,
+  guildMessages: 1 << 9,
+  guildMessage: 1 << 9,
+  guildMessageReactions: 1 << 10,
+  guildMessageTyping: 1 << 11,
+  directMessages: 1 << 12,
+  directMessageReactions: 1 << 13,
+  directMessageTyping: 1 << 14,
+  messageContent: 1 << 15,
+  guildScheduledEvents: 1 << 16,
+  autoModerationConfiguration: 1 << 20,
+  autoModerationExecution: 1 << 21,
+  guildMessagePolls: 1 << 24,
+  directMessagePolls: 1 << 25
+};
+function resolveGatewayIntents(intents) {
+  if (typeof intents === "number")
+    return intents;
+  if (Array.isArray(intents)) {
+    return intents.reduce((acc, intent) => {
+      return acc | resolveGatewayIntents(intent);
+    }, 0);
+  }
+  if (typeof intents === "string") {
+    if (intents in GatewayIntentBits)
+      return GatewayIntentBits[intents];
+    if (intents in IntentBits)
+      return IntentBits[intents];
+    const lower = intents.charAt(0).toLowerCase() + intents.slice(1);
+    if (lower in IntentBits)
+      return IntentBits[lower];
+    const num = Number(intents);
+    if (!Number.isNaN(num))
+      return num;
+  }
+  return 0;
+}
+
+// packages/ws/src/index.ts
+var GatewayOpcodes = {
+  Dispatch: 0,
+  Heartbeat: 1,
+  Identify: 2,
+  PresenceUpdate: 3,
+  VoiceStateUpdate: 4,
+  Resume: 6,
+  Reconnect: 7,
+  RequestGuildMembers: 8,
+  InvalidSession: 9,
+  Hello: 10,
+  HeartbeatAck: 11
+};
+var GatewayState;
+((GatewayState2) => {
+  GatewayState2["Connect"] = "CONNECT";
+  GatewayState2["Hello"] = "HELLO";
+  GatewayState2["Identify"] = "IDENTIFY";
+  GatewayState2["Resume"] = "RESUME";
+  GatewayState2["Ready"] = "READY";
+  GatewayState2["Dispatch"] = "DISPATCH";
+  GatewayState2["Heartbeat"] = "HEARTBEAT";
+  GatewayState2["Reconnect"] = "RECONNECT";
+  GatewayState2["Closed"] = "CLOSED";
+})(GatewayState ||= {});
+
+class GatewayError extends Error {
+  code;
+  constructor(message, code, options) {
+    super(message, options);
+    this.name = "GatewayError";
+    this.code = code;
+  }
+}
+
+class Gateway {
+  state = "CONNECT" /* Connect */;
+  #options;
+  #ws;
+  #sequence = null;
+  #sessionId;
+  #resumeURL;
+  #heartbeatTimer;
+  #initialHeartbeat;
+  #heartbeatAckTimer;
+  #heartbeatACK = true;
+  #heartbeatInterval = 0;
+  #heartbeatSentAt = 0;
+  #lastMessageAt = 0;
+  #zombieTimer;
+  #zombieReported = false;
+  #closed = false;
+  #attempt = 0;
+  #reconnectTimer;
+  #connectPromise;
+  #resolveConnect;
+  #rejectConnect;
+  #listeners = new Map;
+  #sendTimestamps = [];
+  constructor(options) {
+    if (!options.token?.trim())
+      throw new TypeError("A Gateway token is required.");
+    const resolvedIntents = resolveGatewayIntents(options.intents);
+    if (!Number.isInteger(resolvedIntents) || resolvedIntents < 0)
+      throw new TypeError("Gateway intents must be a non-negative integer.");
+    this.#options = {
+      shardId: 0,
+      shardCount: 1,
+      reconnect: true,
+      maxReconnectAttempts: Infinity,
+      reconnectBaseDelay: 1000,
+      reconnectMaxDelay: 30000,
+      heartbeatAckTimeout: 1e4,
+      zombieTimeout: 30000,
+      properties: {
+        os: "Android",
+        browser: "Discord Android",
+        device: "Discord Android"
+      },
+      presence: { status: "online", activities: [], afk: false, since: null },
+      ...options
+    };
+    if (this.#options.shardId < 0 || this.#options.shardId >= this.#options.shardCount)
+      throw new RangeError("Gateway shardId must be within the configured shard count.");
+    if (this.#options.zombieTimeout <= this.#options.heartbeatAckTimeout)
+      throw new RangeError("Gateway zombieTimeout must be greater than heartbeatAckTimeout");
+  }
+  connect(url = "wss://gateway.discord.gg/?v=10&encoding=json") {
+    if (this.state === "CLOSED" /* Closed */)
+      throw new GatewayError("Gateway has been permanently closed.");
+    if (this.#connectPromise)
+      return this.#connectPromise;
+    this.#closed = false;
+    this.#setState("CONNECT" /* Connect */);
+    this.#connectPromise = new Promise((resolve, reject) => {
+      this.#resolveConnect = resolve;
+      this.#rejectConnect = reject;
+      this.#open(url);
+    });
+    return this.#connectPromise;
+  }
+  close() {
+    this.#closed = true;
+    this.#clearTimers();
+    this.#settleConnect(new GatewayError("Gateway connection closed before socket open."));
+    const ws = this.#ws;
+    this.#ws = undefined;
+    try {
+      ws?.close(1000, "Client closed connection");
+    } catch (error) {
+      this.#emitError(error);
+    }
+    this.#setState("CLOSED" /* Closed */);
+  }
+  emit(event, data) {
+    this.#emit(event, data);
+  }
+  on(event, listener) {
+    if (!event || typeof listener !== "function")
+      throw new TypeError("Gateway event and listener are required.");
+    let listeners = this.#listeners.get(event);
+    if (!listeners)
+      this.#listeners.set(event, listeners = new Set);
+    listeners.add(listener);
+    return this;
+  }
+  off(event, listener) {
+    this.#listeners.get(event)?.delete(listener);
+    return this;
+  }
+  send(payload) {
+    if (!payload || !Number.isInteger(payload.op))
+      throw new TypeError("Gateway payload must contain an integer opcode.");
+    if (this.#ws?.readyState !== WebSocket.OPEN)
+      return false;
+    const now = Date.now();
+    while (this.#sendTimestamps.length && now - this.#sendTimestamps[0] >= 60000)
+      this.#sendTimestamps.shift();
+    if (this.#sendTimestamps.length >= 115) {
+      this.#emitError(new GatewayError("Gateway send rate budget exhausted."));
+      return false;
+    }
+    try {
+      this.#ws.send(JSON.stringify(payload));
+      this.#sendTimestamps.push(now);
+      return true;
+    } catch (error) {
+      this.#emitError(error);
+      return false;
+    }
+  }
+  setPresence(data) {
+    this.#options.presence = { ...this.#options.presence, ...data };
+    const payload = {
+      since: this.#options.presence.since ?? null,
+      activities: this.#options.presence.activities ?? [],
+      status: this.#options.presence.status ?? "online",
+      afk: Boolean(this.#options.presence.afk)
+    };
+    return this.send({
+      op: GatewayOpcodes.PresenceUpdate,
+      d: payload,
+      s: null,
+      t: null
+    });
+  }
+  setVoiceState(data) {
+    return this.send({
+      op: GatewayOpcodes.VoiceStateUpdate,
+      d: data,
+      s: null,
+      t: null
+    });
+  }
+  requestGuildMembers(data) {
+    return this.send({
+      op: GatewayOpcodes.RequestGuildMembers,
+      d: data,
+      s: null,
+      t: null
+    });
+  }
+  #open(url) {
+    if (this.#closed)
+      return;
+    let ws;
+    try {
+      ws = new WebSocket(url);
+    } catch (error) {
+      const failure = this.#normalizeError(error);
+      this.#emitError(failure);
+      if (this.#options.reconnect) {
+        this.#setState("RECONNECT" /* Reconnect */);
+        this.#scheduleReconnect();
+      } else
+        this.#settleConnect(failure);
+      return;
+    }
+    this.#ws = ws;
+    ws.addEventListener("open", () => {
+      this.#lastMessageAt = Date.now();
+      this.#zombieReported = false;
+      this.#startZombieDetection();
+      this.#emit("open", undefined);
+      this.#settleConnect();
+    });
+    ws.addEventListener("message", (event) => {
+      this.#lastMessageAt = Date.now();
+      this.#zombieReported = false;
+      this.#message(ws, String(event.data));
+    });
+    ws.addEventListener("close", (event) => this.#close(ws, event.code));
+    ws.addEventListener("error", () => this.#emitError(new GatewayError("Gateway WebSocket error")));
+  }
+  #startZombieDetection() {
+    if (this.#zombieTimer)
+      clearInterval(this.#zombieTimer);
+    const interval = Math.max(1, Math.min(250, this.#options.heartbeatAckTimeout, this.#options.zombieTimeout / 2));
+    this.#zombieTimer = setInterval(() => {
+      if (this.#closed || this.#ws?.readyState !== WebSocket.OPEN || this.#lastMessageAt === 0)
+        return;
+      const silentFor = Date.now() - this.#lastMessageAt;
+      const deadline = Math.max(this.#options.zombieTimeout, this.#heartbeatInterval + this.#options.heartbeatAckTimeout);
+      if (silentFor < deadline || this.#zombieReported)
+        return;
+      this.#zombieReported = true;
+      const error = new GatewayError(`Gateway connection appears stale after ${silentFor}ms without traffic.`);
+      this.#emit("zombie", { silentFor, timeout: deadline });
+      this.#emitError(error);
+      try {
+        this.#ws.close(1001, "Zombie Gateway connection");
+      } catch (closeError) {
+        this.#emitError(closeError);
+      }
+    }, interval);
+  }
+  #message(ws, raw) {
+    let payload;
+    try {
+      payload = JSON.parse(raw);
+    } catch (error) {
+      this.#emitError(new GatewayError("Gateway returned invalid JSON", undefined, {
+        cause: error
+      }));
+      ws.close(1002, "Invalid JSON");
+      return;
+    }
+    if (!payload || typeof payload.op !== "number") {
+      this.#emitError(new GatewayError("Gateway returned an invalid payload"));
+      ws.close(1002, "Invalid payload");
+      return;
+    }
+    if (typeof payload.s === "number")
+      this.#sequence = payload.s;
+    switch (payload.op) {
+      case GatewayOpcodes.Dispatch:
+        this.#setState("DISPATCH" /* Dispatch */);
+        this.#handleDispatch(payload.t, payload.d);
+        break;
+      case GatewayOpcodes.Hello:
+        this.#handleHello(payload.d);
+        break;
+      case GatewayOpcodes.Heartbeat:
+        this.#setState("HEARTBEAT" /* Heartbeat */);
+        this.#sendHeartbeat();
+        break;
+      case GatewayOpcodes.HeartbeatAck:
+        this.#heartbeatACK = true;
+        this.#clearHeartbeatAckTimer();
+        this.#emit("heartbeatAck", payload.d);
+        break;
+      case GatewayOpcodes.Reconnect:
+        ws.close(1001, "Server requested reconnect");
+        break;
+      case GatewayOpcodes.InvalidSession:
+        this.#sessionId = undefined;
+        this.#sequence = null;
+        this.#emit("invalidSession", payload.d);
+        ws.close(1000, "Invalid session");
+        break;
+    }
+  }
+  #handleDispatch(event, data) {
+    if (event === "READY") {
+      const ready = data;
+      if (typeof ready?.session_id !== "string" || typeof ready?.resume_gateway_url !== "string") {
+        this.#emitError(new GatewayError("Gateway READY payload is missing session information"));
+        return;
+      }
+      this.#sessionId = ready.session_id;
+      this.#resumeURL = ready.resume_gateway_url;
+      this.#attempt = 0;
+      this.#setState("READY" /* Ready */);
+      this.#emit("ready", data);
+    }
+    this.#emit(event ?? "dispatch", data);
+  }
+  #handleHello(data) {
+    const interval = data?.heartbeat_interval;
+    if (typeof interval !== "number" || !Number.isFinite(interval) || interval <= 0) {
+      this.#emitError(new GatewayError("Gateway HELLO payload contains an invalid heartbeat interval"));
+      this.#ws?.close(1002, "Invalid heartbeat interval");
+      return;
+    }
+    this.#heartbeatInterval = interval;
+    this.#setState(this.#sessionId && this.#sequence !== null && this.#resumeURL ? "RESUME" /* Resume */ : "HELLO" /* Hello */);
+    this.#startHeartbeat(interval);
+    this.#identifyOrResume();
+  }
+  #identifyOrResume() {
+    if (this.#sessionId && this.#sequence !== null && this.#resumeURL) {
+      this.#setState("RESUME" /* Resume */);
+      this.send({
+        op: GatewayOpcodes.Resume,
+        d: {
+          token: this.#options.token,
+          session_id: this.#sessionId,
+          seq: this.#sequence
+        },
+        s: null,
+        t: null
+      });
+      return;
+    }
+    this.#setState("IDENTIFY" /* Identify */);
+    const props = this.#options.properties ?? {
+      os: "Android",
+      browser: "Discord Android",
+      device: "Discord Android"
+    };
+    const os = props.os ?? "Android";
+    const browser = props.browser ?? "Discord Android";
+    const device = props.device ?? "Discord Android";
+    this.send({
+      op: GatewayOpcodes.Identify,
+      d: {
+        token: this.#options.token,
+        intents: this.#options.intents,
+        properties: {
+          os,
+          browser,
+          device,
+          $os: os,
+          $browser: browser,
+          $device: device,
+          ...props
+        },
+        presence: {
+          since: this.#options.presence?.since ?? null,
+          activities: this.#options.presence?.activities ?? [],
+          status: this.#options.presence?.status ?? "online",
+          afk: Boolean(this.#options.presence?.afk)
+        },
+        shard: [this.#options.shardId, this.#options.shardCount]
+      },
+      s: null,
+      t: null
+    });
+  }
+  #startHeartbeat(interval) {
+    this.#clearHeartbeatTimers();
+    this.#heartbeatACK = true;
+    this.#initialHeartbeat = setTimeout(() => this.#sendHeartbeat(), Math.random() * interval);
+    this.#heartbeatTimer = setInterval(() => this.#sendHeartbeat(), interval);
+  }
+  #sendHeartbeat() {
+    this.#heartbeatACK = false;
+    this.#heartbeatSentAt = Date.now();
+    if (!this.send({
+      op: GatewayOpcodes.Heartbeat,
+      d: this.#sequence,
+      s: null,
+      t: null
+    })) {
+      this.#emitError(new GatewayError("Unable to send Gateway heartbeat because the WebSocket is not open."));
+      return;
+    }
+    this.#clearHeartbeatAckTimer();
+    this.#heartbeatAckTimer = setTimeout(() => {
+      if (!this.#heartbeatACK) {
+        const elapsed = Date.now() - this.#heartbeatSentAt;
+        this.#emitError(new GatewayError(`Gateway heartbeat acknowledgement timed out after ${elapsed}ms.`));
+        try {
+          this.#ws?.close(1001, "Heartbeat timeout");
+        } catch (error) {
+          this.#emitError(error);
+        }
+      }
+    }, this.#options.heartbeatAckTimeout);
+  }
+  #close(ws, code) {
+    if (this.#ws !== ws)
+      return;
+    this.#ws = undefined;
+    this.#clearTimers();
+    const action = this.#closeAction(code);
+    this.#emit("close", { code, action });
+    if (this.#closed || !this.#options.reconnect || action === "stop") {
+      this.#setState(this.#closed ? "CLOSED" /* Closed */ : "CONNECT" /* Connect */);
+      if (this.#connectPromise)
+        this.#settleConnect(new GatewayError(`Gateway closed before READY (code ${code}).`, code));
+      return;
+    }
+    if (action === "identify") {
+      this.#sequence = null;
+      this.#sessionId = undefined;
+    }
+    this.#setState("RECONNECT" /* Reconnect */);
+    this.#scheduleReconnect();
+  }
+  #closeAction(code) {
+    if ([4004, 4010, 4011, 4012, 4013, 4014].includes(code))
+      return "stop";
+    if ([4007, 4009].includes(code))
+      return "identify";
+    return this.#sessionId && this.#sequence !== null ? "resume" : "identify";
+  }
+  #scheduleReconnect() {
+    if (this.#closed || !this.#options.reconnect || this.#reconnectTimer)
+      return;
+    if (this.#attempt >= this.#options.maxReconnectAttempts) {
+      this.#settleConnect(new GatewayError("Gateway reconnect attempts exhausted."));
+      this.#setState("CONNECT" /* Connect */);
+      return;
+    }
+    const delay = Math.min(this.#options.reconnectMaxDelay, this.#options.reconnectBaseDelay * 2 ** this.#attempt++);
+    const jitter = Math.random() * Math.min(250, Math.max(1, delay * 0.25));
+    this.#reconnectTimer = setTimeout(() => {
+      this.#reconnectTimer = undefined;
+      this.#open(this.#resumeURL ?? "wss://gateway.discord.gg/?v=10&encoding=json");
+    }, delay + jitter);
+  }
+  #settleConnect(error) {
+    const resolve = this.#resolveConnect;
+    const reject = this.#rejectConnect;
+    this.#resolveConnect = undefined;
+    this.#rejectConnect = undefined;
+    this.#connectPromise = undefined;
+    if (error)
+      reject?.(error);
+    else
+      resolve?.();
+  }
+  #clearHeartbeatAckTimer() {
+    if (this.#heartbeatAckTimer)
+      clearTimeout(this.#heartbeatAckTimer);
+    this.#heartbeatAckTimer = undefined;
+  }
+  #clearHeartbeatTimers() {
+    if (this.#heartbeatTimer)
+      clearInterval(this.#heartbeatTimer);
+    if (this.#initialHeartbeat)
+      clearTimeout(this.#initialHeartbeat);
+    this.#clearHeartbeatAckTimer();
+    this.#heartbeatTimer = undefined;
+    this.#initialHeartbeat = undefined;
+  }
+  #clearTimers() {
+    if (this.#reconnectTimer)
+      clearTimeout(this.#reconnectTimer);
+    if (this.#zombieTimer)
+      clearInterval(this.#zombieTimer);
+    this.#reconnectTimer = undefined;
+    this.#zombieTimer = undefined;
+    this.#clearHeartbeatTimers();
+  }
+  #normalizeError(error) {
+    return error instanceof GatewayError ? error : new GatewayError("Gateway connection failed.", undefined, {
+      cause: error
+    });
+  }
+  #emitError(error) {
+    this.#emit("error", error instanceof Error ? error : new GatewayError(String(error)));
+  }
+  #emit(event, data) {
+    for (const listener of this.#listeners.get(event) ?? []) {
+      try {
+        const result = listener(data);
+        if (result && typeof result.then === "function")
+          Promise.resolve(result).catch((error) => {
+            if (event !== "error")
+              this.#emitError(error);
+          });
+      } catch (error) {
+        if (event !== "error")
+          this.#emitError(error);
+      }
+    }
+  }
+  #setState(next) {
+    if (this.state === next)
+      return;
+    const previous = this.state;
+    this.state = next;
+    this.#emit("stateChange", { previous, next });
+  }
+}
+export {
+  Gateway,
+  GatewayError,
+  GatewayOpcodes,
+  GatewayState
+};
+
+//# debugId=8F34B5651398B33564756E2164756E21