discordeno/src/structures/guild.ts

import { botId, eventHandlers } from "../bot.ts";
import { cache, cacheHandlers } from "../cache.ts";
import { deleteServer } from "../helpers/guilds/delete_server.ts";
import { editGuild } from "../helpers/guilds/edit_guild.ts";
import { getAuditLogs } from "../helpers/guilds/get_audit_logs.ts";
import { getBan } from "../helpers/guilds/get_ban.ts";
import { getBans } from "../helpers/guilds/get_bans.ts";
import { guildBannerURL } from "../helpers/guilds/guild_banner_url.ts";
import { guildIconURL } from "../helpers/guilds/guild_icon_url.ts";
import { guildSplashURL } from "../helpers/guilds/guild_splash_url.ts";
import { leaveGuild } from "../helpers/guilds/leave_guild.ts";
import { getInvites } from "../helpers/invites/get_invites.ts";
import { banMember } from "../helpers/members/ban_member.ts";
import { unbanMember } from "../helpers/members/unban_member.ts";
import { GetGuildAuditLog } from "../types/audit_log/get_guild_audit_log.ts";
import { Emoji } from "../types/emojis/emoji.ts";
import { CreateGuildBan } from "../types/guilds/create_guild_ban.ts";
import { Guild } from "../types/guilds/guild.ts";
import { DiscordGuildFeatures } from "../types/guilds/guild_features.ts";
import {
  GuildMember,
  GuildMemberWithUser,
} from "../types/guilds/guild_member.ts";
import { ModifyGuild } from "../types/guilds/modify_guild.ts";
import { DiscordImageFormat } from "../types/misc/image_format.ts";
import { DiscordImageSize } from "../types/misc/image_size.ts";
import { PresenceUpdate } from "../types/misc/presence_update.ts";
import { VoiceState } from "../types/voice/voice_state.ts";
import { Collection } from "../util/collection.ts";
import { createNewProp } from "../util/utils.ts";
import { DiscordenoChannel } from "./channel.ts";
import { DiscordenoMember } from "./member.ts";
import { structures } from "./mod.ts";
import { DiscordenoRole } from "./role.ts";

export const initialMemberLoadQueue = new Map<string, GuildMember[]>();

const baseGuild: Partial<DiscordenoGuild> = {
  get members() {
    return cache.members.filter((member) => member.guilds.has(this.id!));
  },
  get channels() {
    return cache.channels.filter((channel) => channel.guildId === this.id);
  },
  get afkChannel() {
    return cache.channels.get(this.afkChannelId!);
  },
  get publicUpdatesChannel() {
    return cache.channels.get(this.publicUpdatesChannelId!);
  },
  get rulesChannel() {
    return cache.channels.get(this.rulesChannelId!);
  },
  get systemChannel() {
    return cache.channels.get(this.systemChannelId!);
  },
  get bot() {
    return cache.members.get(botId);
  },
  get botMember() {
    return this.bot?.guilds.get(this.id!);
  },
  get botVoice() {
    return this.voiceStates?.get(botId);
  },
  get owner() {
    return cache.members.get(this.ownerId!);
  },
  get partnered() {
    return Boolean(this.features?.includes(DiscordGuildFeatures.PARTNERED));
  },
  get verified() {
    return Boolean(this.features?.includes(DiscordGuildFeatures.VERIFIED));
  },
  bannerURL(size, format) {
    return guildBannerURL(this.id!, this.banner!, size, format);
  },
  splashURL(size, format) {
    return guildSplashURL(this.id!, this.splash!, size, format);
  },
  delete() {
    return deleteServer(this.id!);
  },
  edit(options) {
    return editGuild(this.id!, options);
  },
  auditLogs(options) {
    return getAuditLogs(this.id!, options);
  },
  getBan(memberId) {
    return getBan(this.id!, memberId);
  },
  bans() {
    return getBans(this.id!);
  },
  ban(memberId, options) {
    return banMember(this.id!, memberId, options);
  },
  unban(memberId) {
    return unbanMember(this.id!, memberId);
  },
  invites() {
    return getInvites(this.id!);
  },
  iconURL(size, format) {
    return guildIconURL(this.id!, this.icon!, size, format);
  },
  leave() {
    return leaveGuild(this.id!);
  },
};

export async function createDiscordenoGuild(
  data: Guild,
  shardId: number,
) {
  const {
    memberCount = 0,
    voiceStates = [],
    channels = [],
    presences = [],
    joinedAt = "",
    emojis,
    members = [],
    ...rest
  } = data;

  const roles = await Promise.all(
    (data.roles || []).map((role) =>
      structures.createDiscordenoRole({ role, guildId: rest.id })
    ),
  );

  await Promise.all(channels.map(async (channel) => {
    const discordenoChannel = await structures.createDiscordenoChannel(
      channel,
      rest.id,
    );

    return cacheHandlers.set(
      "channels",
      discordenoChannel.id,
      discordenoChannel,
    );
  }));

  const props: Record<string, ReturnType<typeof createNewProp>> = {};
  for (const key of Object.keys(rest)) {
    eventHandlers.debug?.(
      "loop",
      `Running for of loop in createDiscordenoGuild function.`,
    );
    // @ts-ignore index signature
    props[key] = createNewProp(rest[key]);
  }

  const guild: DiscordenoGuild = Object.create(baseGuild, {
    ...props,
    shardId: createNewProp(shardId),
    roles: createNewProp(
      new Collection(roles.map((r: DiscordenoRole) => [r.id, r])),
    ),
    joinedAt: createNewProp(Date.parse(joinedAt)),
    presences: createNewProp(
      new Collection(presences.map((p) => [p.user?.id, p])),
    ),
    memberCount: createNewProp(memberCount),
    emojis: createNewProp(
      new Collection(
        (emojis || []).map((emoji) => [emoji.id ?? emoji.name, emoji]),
      ),
    ),
    voiceStates: createNewProp(
      new Collection(
        voiceStates.map((vs) => [vs.userId, vs]),
      ),
    ),
  });

  // ONLY ADD TO QUEUE WHEN BOT IS NOT FULLY ONLINE
  if (!cache.isReady) initialMemberLoadQueue.set(guild.id, members);
  // BOT IS ONLINE, JUST DIRECTLY ADD MEMBERS
  else {
    await Promise.allSettled(
      members.map(async (member) => {
        const discordenoMember = await structures.createDiscordenoMember(
          member as GuildMemberWithUser,
          guild.id,
        );

        return cacheHandlers.set(
          "members",
          discordenoMember.id,
          discordenoMember,
        );
      }),
    );
  }

  return guild;
}

export interface DiscordenoGuild extends
  Omit<
    Guild,
    | "roles"
    | "presences"
    | "voiceStates"
    | "members"
    | "channels"
    | "memberCount"
    | "owner"
    | "emojis"
  > {
  /** The id of the shard this guild is bound to */
  shardId: number;
  /** Total number of members in this guild */
  memberCount: number;
  /** The roles in the guild */
  roles: Collection<string, DiscordenoRole>;
  /** The presences of all the users in the guild. */
  presences: Collection<string, PresenceUpdate>;
  /** The Voice State data for each user in a voice channel in this server. */
  voiceStates: Collection<string, VoiceState>;
  /** Custom guild emojis */
  emojis: Collection<string, Emoji>;

  // GETTERS
  /** Members in this guild. */
  members: Collection<string, DiscordenoMember>;
  /** Channels in this guild. */
  channels: Collection<string, DiscordenoChannel>;
  /** The afk channel if one is set */
  afkChannel?: DiscordenoChannel;
  /** The public update channel if one is set */
  publicUpdatesChannel?: DiscordenoChannel;
  /** The rules channel in this guild if one is set */
  rulesChannel?: DiscordenoChannel;
  /** The system channel in this guild if one is set */
  systemChannel?: DiscordenoChannel;
  /** The bot member in this guild if cached */
  bot?: DiscordenoMember;
  /** The bot guild member in this guild if cached */
  botMember?: Omit<GuildMember, "joinedAt" | "premiumSince"> & {
    joinedAt: number;
    premiumSince?: number;
  };
  /** The bots voice state if there is one in this guild */
  botVoice?: VoiceState;
  /** The owner member of this guild */
  owner?: DiscordenoMember;
  /** Whether or not this guild is partnered */
  partnered: boolean;
  /** Whether or not this guild is verified */
  verified: boolean;

  // METHODS

  /** The banner url for this server */
  bannerURL(
    size?: DiscordImageSize,
    format?: DiscordImageFormat,
  ): string | undefined;
  /** The splash url for this server */
  splashURL(
    size?: DiscordImageSize,
    format?: DiscordImageFormat,
  ): string | undefined;
  /** The full URL of the icon from Discords CDN. Undefined when no icon is set. */
  iconURL(
    size?: DiscordImageSize,
    format?: DiscordImageFormat,
  ): string | undefined;
  /** Delete a guild permanently. User must be owner. Returns 204 No Content on success. Fires a Guild Delete Gateway event. */
  delete(): ReturnType<typeof deleteServer>;
  /** Leave a guild */
  leave(): ReturnType<typeof leaveGuild>;
  /** Edit the server. Requires the MANAGE_GUILD permission. */
  edit(options: ModifyGuild): ReturnType<typeof editGuild>;
  /** Returns the audit logs for the guild. Requires VIEW AUDIT LOGS permission */
  auditLogs(options: GetGuildAuditLog): ReturnType<typeof getAuditLogs>;
  /** Returns a ban object for the given user or a 404 not found if the ban cannot be found. Requires the BAN_MEMBERS permission. */
  getBan(memberId: string): ReturnType<typeof getBan>;
  /** Returns a list of ban objects for the users banned from this guild. Requires the BAN_MEMBERS permission. */
  bans(): ReturnType<typeof getBans>;
  /** Ban a user from the guild and optionally delete previous messages sent by the user. Requires the BAN_MEMBERS permission. */
  ban(memberId: string, options: CreateGuildBan): ReturnType<typeof banMember>;
  /** Remove the ban for a user. Requires BAN_MEMBERS permission */
  unban(memberId: string): ReturnType<typeof unbanMember>;
  /** Get all the invites for this guild. Requires MANAGE_GUILD permission */
  invites(): ReturnType<typeof getInvites>;
}