think/packages/client/src/data/message.ts

import type { IMessage } from '@think/domains';
import React, { useState, useCallback } from 'react';
import useSWR from 'swr';
import { HttpClient } from 'services/http-client';

/**
 * 所有消息
 * @returns
 */
export const useAllMessages = () => {
  const [page, setPage] = useState(1);
  const { data, error, mutate } = useSWR<{
    data: Array<IMessage>;
    total: number;
  }>(`/message/all?page=${page}`, (url) => HttpClient.get(url), {
    refreshInterval: 200,
  });
  const loading = !data && !error;

  return {
    data,
    loading,
    error,
    page,
    setPage,
  };
};

/**
 * 所有已读消息
 * @returns
 */
export const useReadMessages = () => {
  const [page, setPage] = useState(1);
  const { data, error, mutate } = useSWR<{
    data: Array<IMessage>;
    total: number;
  }>(`/message/read?page=${page}`, (url) => HttpClient.get(url), {
    refreshInterval: 200,
  });
  const loading = !data && !error;

  return {
    data,
    loading,
    error,
    page,
    setPage,
  };
};

/**
 * 所有未读消息
 * @returns
 */
export const useUnreadMessages = () => {
  const [page, setPage] = useState(1);
  const { data, error, mutate } = useSWR<{
    data: Array<IMessage>;
    total: number;
  }>(`/message/unread?page=${page}`, (url) => HttpClient.get(url), {
    refreshInterval: 200,
  });
  const loading = !data && !error;

  const readMessage = useCallback(
    async (messageId) => {
      const ret = await HttpClient.post(`/message/read/${messageId}`);
      mutate();
      return ret;
    },
    [mutate]
  );

  return {
    data,
    loading,
    error,
    page,
    setPage,
    readMessage,
  };
};