stream-chat/src/client_state.ts

JS SDK for the Stream Chat API

import type { UserResponse } from './types';
import type { StreamChat } from './client';

/**
 * ClientState - A container class for the client state.
 */
export class ClientState {
  private client: StreamChat;
  users: {
    [key: string]: UserResponse;
  };
  userChannelReferences: { [key: string]: { [key: string]: boolean } };
  constructor({ client }: { client: StreamChat }) {
    // show the status for a certain user...
    // ie online, offline etc
    this.client = client;
    this.users = {};
    // store which channels contain references to the specified user...
    this.userChannelReferences = {};
  }

  updateUsers(users: UserResponse[]) {
    for (const user of users) {
      this.updateUser(user);
    }
  }

  updateUser(user?: UserResponse) {
    if (user != null && this.client._cacheEnabled()) {
      this.users[user.id] = user;
    }
  }

  updateUserReference(user: UserResponse, channelID: string) {
    if (user == null || !this.client._cacheEnabled()) {
      return;
    }
    this.updateUser(user);
    if (!this.userChannelReferences[user.id]) {
      this.userChannelReferences[user.id] = {};
    }
    this.userChannelReferences[user.id][channelID] = true;
  }

  deleteAllChannelReference(channelID: string) {
    for (const userID in this.userChannelReferences) {
      delete this.userChannelReferences[userID][channelID];
    }
  }
}