javascript-client-sdk/src/collections/ServerMemberCollection.ts

import type { Member, MemberCompositeKey } from "revolt-api";

import { ServerMember } from "../classes/ServerMember.js";
import type { HydratedServerMember } from "../hydration/serverMember.js";

import { ClassCollection } from "./Collection.js";

/**
 * Collection of Server Members
 */
export class ServerMemberCollection extends ClassCollection<
  ServerMember,
  HydratedServerMember
> {
  /**
   * Check if member exists by composite key
   * @param id Id
   * @returns Whether it exists
   */
  hasByKey(id: MemberCompositeKey): boolean {
    return super.has(id.server + id.user);
  }

  /**
   * Get member by composite key
   * @param id Id
   * @returns Member
   */
  getByKey(id: MemberCompositeKey): ServerMember | undefined {
    return super.get(id.server + id.user);
  }

  /**
   * check partial status by composite key
   * @param id Id
   * @returns Member
   */
  isPartialByKey(id: MemberCompositeKey): boolean {
    return super.isPartial(id.server + id.user);
  }

  /**
   * Fetch server member by Id
   * @param serverId Server Id
   * @param userId User Id
   * @returns Message
   */
  async fetch(serverId: string, userId: string): Promise<ServerMember> {
    const member = this.get(serverId + userId);
    if (member && !this.isPartial(serverId + userId)) return member;

    const data = (await this.client.api.get(
      `/servers/${serverId as ""}/members/${userId as ""}`,
      {
        roles: false,
      },
      // TODO: fix typings in revolt-api
    )) as Member;

    return this.getOrCreate(data._id, data);
  }

  /**
   * Get or create
   * @param id Id
   * @param data Data
   */
  getOrCreate(id: MemberCompositeKey, data: Member): ServerMember {
    if (this.hasByKey(id) && !this.isPartialByKey(id)) {
      return this.getByKey(id)!;
    } else {
      const instance = new ServerMember(this, id);
      this.create(
        id.server + id.user,
        "serverMember",
        instance,
        this.client,
        data,
      );
      return instance;
    }
  }

  /**
   * Get or return partial
   * @param id Id
   */
  getOrPartial(id: MemberCompositeKey): ServerMember | undefined {
    if (this.hasByKey(id)) {
      return this.getByKey(id)!;
    } else if (this.client.options.partials) {
      const instance = new ServerMember(this, id);
      this.create(id.server + id.user, "serverMember", instance, this.client, {
        id,
        partial: true,
      });
      return instance;
    }
  }
}