@amityco/ts-sdk/src/invitationRepository/observers/getInvitations/InvitationsLiveCollectionController.ts

Amity Social Cloud Typescript SDK

import hash from 'object-hash';
import { isNonNullable } from '~/utils';
import { InvitationActionsEnum } from './enums';
import { invitationLinkedObject } from '~/utils/linkedObject/invitationLinkedObject';
import { pullFromCache, pushToCache } from '~/cache/api';
import { sortByFirstCreated, sortByLastCreated } from '~/core/query';
import { InvitationsPaginationController } from './InvitationsPaginationController';
import { InvitationsQueryStreamController } from './InvitationsQueryStreamController';
import { LiveCollectionController } from '~/core/liveCollection/LiveCollectionController';
import { prepareInvitationPayload } from '~/invitationRepository/utils';
import {
  onLocalInvitationCreated,
  onLocalInvitationDeleted,
  onLocalInvitationUpdated,
} from '~/invitationRepository/events';

export class InvitationsLiveCollectionController extends LiveCollectionController<
  'invitation',
  Amity.InvitationLiveCollection,
  Amity.Invitation,
  InvitationsPaginationController
> {
  private queryStreamController: InvitationsQueryStreamController;

  private query: Amity.InvitationLiveCollection;

  constructor(
    query: Amity.InvitationLiveCollection,
    callback: Amity.LiveCollectionCallback<Amity.Invitation>,
  ) {
    const queryStreamId = hash(query);
    const cacheKey = ['invitation', 'collection', queryStreamId];
    const paginationController = new InvitationsPaginationController(query);

    super(paginationController, queryStreamId, cacheKey, callback);

    this.query = query;
    this.queryStreamController = new InvitationsQueryStreamController(
      this.query,
      this.cacheKey,
      this.notifyChange.bind(this),
      prepareInvitationPayload,
    );

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

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

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

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

  startSubscription() {
    return this.queryStreamController.subscribeRTE([
      {
        fn: onLocalInvitationCreated,
        action: InvitationActionsEnum.OnLocalInvitationCreated,
      },
      {
        fn: onLocalInvitationUpdated,
        action: InvitationActionsEnum.OnLocalInvitationUpdated,
      },
      {
        fn: onLocalInvitationDeleted,
        action: InvitationActionsEnum.OnLocalInvitationDeleted,
      },
    ]);
  }

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

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

    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.Invitation[]) {
    let invitations = data;

    if (this.query.targetId) {
      invitations = invitations.filter(invitation => invitation.targetId === this.query.targetId);
    }

    if (this.query.statuses) {
      invitations = invitations.filter(invitation =>
        this.query.statuses?.includes(invitation.status),
      );
    }

    if (this.query.targetType) {
      invitations = invitations.filter(
        invitation => invitation.targetType === this.query.targetType,
      );
    }

    if (this.query.type) {
      invitations = invitations.filter(invitation => invitation.type === this.query.type);
    }

    const sortFn = (() => {
      switch (this.query.sortBy) {
        case 'firstCreated':
          return sortByFirstCreated;
        case 'lastCreated':
          return sortByLastCreated;
        default:
          return sortByLastCreated;
      }
    })();

    invitations = invitations.sort(sortFn);

    return invitations;
  }
}