import { getActiveClient } from '~/client/api/activeClient';
import { getResolver } from '~/core/model';
import { pullFromCache, pushToCache } from '~/cache/api';
import { ingestInCache } from '~/cache/api/ingestInCache';
import { QueryStreamController } from '~/core/liveCollection/QueryStreamController';
import { InvitationActionsEnum } from '../getInvitations/enums';

export class MyInvitationsQueryStreamController extends QueryStreamController<
  Amity.MyInvitationsPayload,
  Amity.MyInvitationsLiveCollection
> {
  private notifyChange: (params: Amity.LiveCollectionNotifyParams) => void;

  private preparePayload: (
    response: Amity.MyInvitationsPayload,
  ) => Amity.ProcessedMyInvitationsPayload;

  constructor(
    query: Amity.MyInvitationsLiveCollection,
    cacheKey: string[],
    notifyChange: (params: Amity.LiveCollectionNotifyParams) => void,
    preparePayload: (response: Amity.MyInvitationsPayload) => Amity.ProcessedMyInvitationsPayload,
  ) {
    super(query, cacheKey);
    this.notifyChange = notifyChange;
    this.preparePayload = preparePayload;
  }

  async saveToMainDB(response: Amity.MyInvitationsPayload) {
    const processedPayload = await this.preparePayload(response);

    const client = getActiveClient();
    const cachedAt = client.cache && Date.now();

    if (client.cache) {
      ingestInCache(processedPayload, { cachedAt });
    }
  }

  appendToQueryStream(
    response: Amity.MyInvitationsPayload & Partial<Amity.Pagination>,
    direction: Amity.LiveCollectionPageDirection,
    refresh = false,
  ) {
    if (refresh) {
      pushToCache(this.cacheKey, {
        data: response.invitations.map(getResolver('invitation')),
      });
    } else {
      const collection = pullFromCache<Amity.MyInvitationsLiveCollectionCache>(this.cacheKey)?.data;

      const invitations = collection?.data ?? [];

      pushToCache(this.cacheKey, {
        ...collection,
        data: [
          ...new Set([...invitations, ...response.invitations.map(getResolver('invitation'))]),
        ],
      });
    }
  }

  reactor(action: InvitationActionsEnum) {
    return (invitations: Amity.InternalInvitation[]) => {
      const collection = pullFromCache<Amity.InvitationLiveCollectionCache>(this.cacheKey)?.data;
      if (!collection) return;

      if (action === InvitationActionsEnum.OnLocalInvitationUpdated) {
        const isExist = collection.data.find(id => id === invitations[0].invitationId);
        if (!isExist) return;
      }

      if (action === InvitationActionsEnum.OnLocalInvitationCreated) {
        const client = getActiveClient();

        const myInvitations = invitations.filter(
          invitation => invitation.invitedUserId === client.userId,
        );

        collection.data = [
          ...new Set([
            ...myInvitations.map(invitation => invitation.invitationId),
            ...collection.data,
          ]),
        ];
      }

      if (action === InvitationActionsEnum.OnLocalInvitationDeleted) {
        collection.data = collection.data.filter(id => id !== invitations[0].invitationId);
      }

      pushToCache(this.cacheKey, collection);
      this.notifyChange({ origin: Amity.LiveDataOrigin.EVENT, loading: false });
    };
  }

  subscribeRTE(
    createSubscriber: {
      fn: (reactor: Amity.Listener<Amity.InternalInvitation[]>) => Amity.Unsubscriber;
      action: InvitationActionsEnum;
    }[],
  ) {
    return createSubscriber.map(subscriber => subscriber.fn(this.reactor(subscriber.action)));
  }
}