weseek__growi/apps/app/src/stores/in-app-notification.ts

import type { SWRConfiguration, SWRResponse } from 'swr';
import useSWR from 'swr';

import { SupportedTargetModel } from '~/interfaces/activity';
import type {
  IInAppNotification,
  InAppNotificationStatuses,
  PaginateResult,
} from '~/interfaces/in-app-notification';
import * as userSerializers from '~/models/serializers/in-app-notification-snapshot/user';
import loggerFactory from '~/utils/logger';

import { apiv3Get } from '../client/util/apiv3-client';

const logger = loggerFactory('growi:cli:InAppNotification');

type inAppNotificationPaginateResult = PaginateResult<IInAppNotification>;

export const useSWRxInAppNotifications = (
  limit: number,
  offset?: number,
  status?: InAppNotificationStatuses,
  config?: SWRConfiguration,
): SWRResponse<PaginateResult<IInAppNotification>, Error> => {
  return useSWR(
    ['/in-app-notification/list', limit, offset, status],
    ([endpoint]) =>
      apiv3Get(endpoint, { limit, offset, status }).then((response) => {
        const inAppNotificationPaginateResult =
          response.data as inAppNotificationPaginateResult;
        inAppNotificationPaginateResult.docs.forEach((doc) => {
          try {
            if (doc.targetModel === SupportedTargetModel.MODEL_USER) {
              doc.parsedSnapshot = userSerializers.parseSnapshot(doc.snapshot);
            }
          } catch (err) {
            logger.warn('Failed to parse snapshot', err);
          }
        });
        return inAppNotificationPaginateResult;
      }),
    config,
  );
};

export const useSWRxInAppNotificationStatus = (): SWRResponse<
  number,
  Error
> => {
  return useSWR('/in-app-notification/status', (endpoint) =>
    apiv3Get(endpoint).then((response) => response.data.count),
  );
};