discordeno/src/helpers/members/get_members.ts

import { eventHandlers, identifyPayload } from "../../bot.ts";
import { cacheHandlers } from "../../cache.ts";
import { rest } from "../../rest/rest.ts";
import { DiscordenoMember } from "../../structures/member.ts";
import { structures } from "../../structures/mod.ts";
import { DiscordGatewayIntents } from "../../types/gateway/gateway_intents.ts";
import {
  DiscordGuildMemberWithUser,
  GuildMember,
} from "../../types/guilds/guild_member.ts";
import { ListGuildMembers } from "../../types/guilds/list_guild_members.ts";
import { Errors } from "../../types/misc/errors.ts";
import { Collection } from "../../util/collection.ts";
import { endpoints } from "../../util/constants.ts";

/**
 * ⚠️ BEGINNER DEVS!! YOU SHOULD ALMOST NEVER NEED THIS AND YOU CAN GET FROM cache.members.get()
 *
 * ADVANCED:
 * Highly recommended to **NOT** use this function to get members instead use fetchMembers().
 * REST(this function): 50/s global(across all shards) rate limit with ALL requests this included
 * GW(fetchMembers): 120/m(PER shard) rate limit. Meaning if you have 8 shards your limit is 960/m.
 */
export async function getMembers(guildId: string, options?: ListGuildMembers) {
  if (!(identifyPayload.intents && DiscordGatewayIntents.GUILD_MEMBERS)) {
    throw new Error(Errors.MISSING_INTENT_GUILD_MEMBERS);
  }

  const guild = await cacheHandlers.get("guilds", guildId);
  if (!guild) throw new Error(Errors.GUILD_NOT_FOUND);

  const members = new Collection<string, DiscordenoMember>();

  let membersLeft = options?.limit ?? guild.memberCount;
  let loops = 1;
  while (
    (options?.limit ?? guild.memberCount) > members.size &&
    membersLeft > 0
  ) {
    eventHandlers.debug?.("loop", "Running while loop in getMembers function.");

    if (options?.limit && options.limit > 1000) {
      console.log(
        `Paginating get members from REST. #${loops} / ${
          Math.ceil(
            (options?.limit ?? 1) / 1000,
          )
        }`,
      );
    }

    const result: GuildMember[] = (await rest.runMethod(
      "get",
      `${endpoints.GUILD_MEMBERS(guildId)}?limit=${
        membersLeft > 1000 ? 1000 : membersLeft
      }${options?.after ? `&after=${options.after}` : ""}`,
    ));

    const discordenoMembers = await Promise.all(
      result.map(async (member) => {
        const discordenoMember = await structures.createDiscordenoMember(
          member as DiscordGuildMemberWithUser,
          guildId,
        );

        await cacheHandlers.set(
          "members",
          discordenoMember.id,
          discordenoMember,
        );

        return discordenoMember;
      }),
    ) as DiscordenoMember[];

    if (!discordenoMembers.length) break;

    discordenoMembers.forEach((member) => {
      eventHandlers.debug?.(
        "loop",
        `Running forEach loop in get_members file.`,
      );
      members.set(member.id, member);
    });

    options = {
      limit: options?.limit,
      after: discordenoMembers[discordenoMembers.length - 1].id,
    };

    membersLeft -= 1000;

    loops++;
  }

  return members;
}