@amityco/ts-sdk-react-native/src/userRepository/observers/searchUserByDisplayName/SearchUserLiveCollectionController.ts

Amity Social Cloud Typescript SDK

import hash from 'object-hash';
import { pullFromCache, pushToCache } from '~/cache/api';
import { SearchUserPaginationController } from './SearchUserPaginationController';
import { SearchUserQueryStreamController } from './SearchUserQueryStreamController';
import { LiveCollectionController } from '~/core/liveCollection/LiveCollectionController';
import {
  onUserDeleted,
  onUserFlagCleared,
  onUserFlagged,
  onUserUnflagged,
  onUserUpdated,
} from '~/userRepository/events';
import { prepareUserPayload } from '~/userRepository/utils/prepareUserPayload';
import { EnumUserActions } from '~/userRepository/observers/enums';
import { isNonNullable } from '~/utils';
import { LinkedObject } from '~/utils/linkedObject';

export class SearchUserLiveCollectionController extends LiveCollectionController<
  'user',
  Amity.SearchUserLiveCollection,
  Amity.User,
  SearchUserPaginationController
> {
  private queryStreamController: SearchUserQueryStreamController;

  private query: Amity.SearchUserLiveCollection;

  constructor(
    query: Amity.SearchUserLiveCollection,
    callback: Amity.LiveCollectionCallback<Amity.User>,
  ) {
    const queryStreamId = hash(query);
    const cacheKey = ['user', 'collection', queryStreamId];
    const paginationController = new SearchUserPaginationController(query);

    super(paginationController, queryStreamId, cacheKey, callback);

    this.query = { ...query, filter: query.filter ?? 'all' };
    this.queryStreamController = new SearchUserQueryStreamController(
      this.query,
      this.cacheKey,
      this.notifyChange.bind(this),
      prepareUserPayload,
    );

    this.callback = callback.bind(this);
    this.loadPage({ initial: true });
  }

  protected setup() {
    const collection = pullFromCache<Amity.UserLiveCollectionCache>(this.cacheKey)?.data;
    if (!collection) {
      pushToCache(this.cacheKey, {
        data: [],
        params: {},
      });
    }
  }

  protected async persistModel(queryPayload: Amity.UserPayload & Amity.Pagination) {
    await this.queryStreamController.saveToMainDB(queryPayload);
  }

  protected persistQueryStream({
    response,
    direction,
    refresh,
  }: Amity.LiveCollectionPersistQueryStreamParams<'user'>) {
    this.queryStreamController.appendToQueryStream(response, direction, refresh);
  }

  startSubscription() {
    return this.queryStreamController.subscribeRTE([
      { fn: onUserDeleted, action: EnumUserActions.OnUserDeleted },
      { fn: onUserUpdated, action: EnumUserActions.OnUserUpdated },
      { fn: onUserFlagged, action: EnumUserActions.OnUserFlagged },
      { fn: onUserUnflagged, action: EnumUserActions.OnUserUnflagged },
      { fn: onUserFlagCleared, action: EnumUserActions.OnUserFlagCleared },
    ]);
  }

  notifyChange({ origin, loading, error }: Amity.LiveCollectionNotifyParams) {
    const collection = pullFromCache<Amity.UserLiveCollectionCache>(this.cacheKey)?.data;
    if (!collection) return;

    const data = this.applyFilter(
      collection.data
        .map(id => pullFromCache<Amity.InternalUser>(['user', 'get', id])!)
        .filter(isNonNullable)
        .map(({ data }) => data)
        .map(LinkedObject.user) ?? [],
    );

    if (!this.shouldNotify(data) && origin === 'event') return;

    this.callback({
      onNextPage: () => this.loadPage({ direction: Amity.LiveCollectionPageDirection.NEXT }),
      data,
      hasNextPage: !!this.paginationController.getNextToken(),
      loading,
      error,
    });
  }

  applyFilter(data: Amity.InternalUser[]) {
    let users = data;

    if (this.query.filter === 'flagged') {
      users = users.filter(user => !!user.hashFlag);
    }

    users = users.filter(user => user.isDeleted == null || user.isDeleted === false);

    return users;
  }
}