shlinkio / shlink-web-client

import { FC, useEffect } from 'react';

import { pipe } from 'ramda';

import { CreateVisit } from '../../visits/types';

import { MercureInfo } from '../reducers/mercureInfo';

import { bindToMercureTopic } from './index';

export interface MercureBoundProps {

  createNewVisits: (createdVisits: CreateVisit[]) => void;

  loadMercureInfo: Function;

  mercureInfo: MercureInfo;

}

export function boundToMercureHub<T = {}>(

  WrappedComponent: FC<MercureBoundProps & T>,

  getTopicForProps: (props: T) => string,

) {

  const pendingUpdates = new Set<CreateVisit>();

  return (props: MercureBoundProps & T) => {

    const { createNewVisits, loadMercureInfo, mercureInfo } = props;

    const { interval } = mercureInfo;

    useEffect(() => {

      const onMessage = (visit: CreateVisit) => interval ? pendingUpdates.add(visit) : createNewVisits([ visit ]);

      const closeEventSource = bindToMercureTopic(mercureInfo, getTopicForProps(props), onMessage, loadMercureInfo);

      if (!interval) {

        return closeEventSource;

      }

      const timer = setInterval(() => {

        createNewVisits([ ...pendingUpdates ]);

        pendingUpdates.clear();

      }, interval * 1000 * 60);

      return pipe(() => clearInterval(timer), () => closeEventSource?.());

    }, [ mercureInfo ]);

    return <WrappedComponent {...props} />;

  };

}