piixishstream 0.0.0-stage → 1.0.2

package/src/utils.ts

@@ -0,0 +1,101 @@
+import type {
+  AnyChannel,
+  DMChannel,
+  GroupDMChannel,
+  VoiceBasedChannel,
+} from "discord.js-selfbot-v13";
+
+export function normalizeVideoCodec(
+  codec: string,
+): "H264" | "H265" | "VP8" | "VP9" | "AV1" {
+  if (/H\.?264|AVC/i.test(codec)) return "H264";
+  if (/H\.?265|HEVC/i.test(codec)) return "H265";
+  if (/VP(8|9)/i.test(codec)) return codec.toUpperCase() as "VP8" | "VP9";
+  if (/AV1/i.test(codec)) return "AV1";
+  throw new Error(`Unknown codec: ${codec}`);
+}
+
+// The available video streams are sent by client on connection to voice gateway using OpCode Identify (0)
+// The server then replies with the ssrc and rtxssrc for each available stream using OpCode Ready (2)
+// RID is used specifically to distinguish between different simulcast streams of the same video source,
+// but we don't really care about sending multiple quality streams, so we hardcode a single one
+export const STREAMS_SIMULCAST = [{ type: "screen", rid: "100", quality: 100 }];
+
+export enum SupportedEncryptionModes {
+  AES256 = "aead_aes256_gcm_rtpsize",
+  XCHACHA20 = "aead_xchacha20_poly1305_rtpsize",
+}
+
+export type SupportedVideoCodec = "H264" | "H265" | "VP8" | "VP9" | "AV1";
+
+export const max_int16bit = 2 ** 16;
+export const max_int32bit = 2 ** 32;
+
+export function isFiniteNonZero(n: unknown): n is number {
+  return !!n && Number.isFinite(n);
+}
+
+export function parseStreamKey(streamKey: string): {
+  type: "guild" | "call";
+  channelId: string;
+  guildId: string | null;
+  userId: string;
+} {
+  const streamKeyArray = streamKey.split(":");
+
+  const type = streamKeyArray.shift();
+
+  if (type !== "guild" && type !== "call") {
+    throw new Error(`Invalid stream key type: ${type}`);
+  }
+
+  if (
+    (type === "guild" && streamKeyArray.length < 3) ||
+    (type === "call" && streamKey.length < 2)
+  )
+    throw new Error(`Invalid stream key: ${streamKey}`); // invalid stream key
+
+  let guildId: string | null = null;
+  if (type === "guild") {
+    guildId = streamKeyArray.shift() ?? null;
+  }
+  const channelId = streamKeyArray.shift();
+  const userId = streamKeyArray.shift();
+
+  if (!channelId || !userId) {
+    throw new Error(`Invalid stream key: ${streamKey}`);
+  }
+  return { type, channelId, guildId, userId };
+}
+
+export function generateStreamKey(
+  type: "guild" | "call",
+  guildId: string | null,
+  channelId: string,
+  userId: string,
+): string {
+  const streamKey = `${type}${type === "guild" ? `:${guildId}` : ""}:${channelId}:${userId}`;
+
+  return streamKey;
+}
+
+export function isVoiceChannel(
+  channel: AnyChannel,
+): channel is DMChannel | GroupDMChannel | VoiceBasedChannel {
+  return (
+    channel.type === "DM" ||
+    channel.type === "GROUP_DM" ||
+    channel.type === "GUILD_STAGE_VOICE" ||
+    channel.type === "GUILD_VOICE"
+  );
+}
+
+export function isDeno() {
+  // @ts-expect-error
+  return typeof Deno !== "undefined";
+}
+
+export function isBun() {
+  // @ts-expect-error
+  return typeof Bun !== "undefined";
+}