优化

client/shared/model/message.ts

@@ -0,0 +1,204 @@
+import { request } from '../api/request';
+import type { ChatMessageReaction, ChatMessage } from 'tailchat-types';
+import {
+  createAutoMergedRequest,
+  createAutoSplitRequest,
+} from '../utils/request';
+import _uniq from 'lodash/uniq';
+import _flatten from 'lodash/flatten';
+import _zipObject from 'lodash/zipObject';
+
+export { ChatMessageReaction, ChatMessage };
+
+export interface LocalChatMessage extends ChatMessage {
+  /**
+   * 本地添加消息的标识，用于标记该条消息尚未确定已经发送到服务端
+   */
+  isLocal?: boolean;
+  /**
+   * 判断是否发送失败
+   */
+  sendFailed?: boolean;
+}
+
+export interface SimpleMessagePayload {
+  groupId?: string;
+  converseId: string;
+  content: string;
+}
+
+export interface SendMessagePayloadMeta {
+  mentions?: string[];
+}
+
+export interface SendMessagePayload extends SimpleMessagePayload {
+  /**
+   * content的plain内容
+   * 用于inbox
+   */
+  plain?: string;
+  meta?: SendMessagePayloadMeta;
+}
+
+/**
+ * 获取会话消息
+ * @param converseId 会话ID
+ * @param startId 开始ID
+ */
+export async function fetchConverseMessage(
+  converseId: string,
+  startId?: string
+): Promise<ChatMessage[]> {
+  const { data } = await request.get('/api/chat/message/fetchConverseMessage', {
+    params: {
+      converseId,
+      startId,
+    },
+  });
+
+  return data;
+}
+
+/**
+ * 发送消息
+ * @param payload 消息体
+ */
+export async function sendMessage(
+  payload: SendMessagePayload
+): Promise<ChatMessage> {
+  const { data } = await request.post('/api/chat/message/sendMessage', payload);
+
+  return data;
+}
+
+/**
+ * 撤回消息
+ * @param messageId 消息ID
+ */
+export async function recallMessage(messageId: string): Promise<ChatMessage> {
+  const { data } = await request.post('/api/chat/message/recallMessage', {
+    messageId,
+  });
+
+  return data;
+}
+
+export async function deleteMessage(messageId: string): Promise<boolean> {
+  const { data } = await request.post('/api/chat/message/deleteMessage', {
+    messageId,
+  });
+
+  return data;
+}
+
+/**
+ * 搜索聊天记录
+ * @param converseId 会话id
+ * @param messageText 聊天文本
+ */
+export async function searchMessage(
+  text: string,
+  converseId: string,
+  groupId?: string
+): Promise<ChatMessage[]> {
+  const { data } = await request.post('/api/chat/message/searchMessage', {
+    text,
+    converseId,
+    groupId,
+  });
+
+  return data;
+}
+
+interface LastMessageInfo {
+  converseId: string;
+  lastMessageId: string;
+}
+
+/**
+ * 基于会话id获取会话最后一条消息的id
+ */
+async function fetchConverseLastMessages(
+  converseIds: string[]
+): Promise<{ converseId: string; lastMessageId: string }[]> {
+  const { data } = await request.post(
+    '/api/chat/message/fetchConverseLastMessages',
+    {
+      converseIds,
+    }
+  );
+
+  return data;
+}
+
+export const _fetchConverseLastMessageInfo = createAutoMergedRequest<
+  string[],
+  (LastMessageInfo | null)[]
+>(
+  createAutoSplitRequest(
+    async (converseIdsList) => {
+      const uniqList = _uniq(_flatten(converseIdsList));
+      const infoList = await fetchConverseLastMessages(uniqList);
+
+      const map = _zipObject<LastMessageInfo | null>(uniqList, infoList);
+
+      // 将请求结果根据传输来源重新分组
+      return converseIdsList.map((converseIds) =>
+        converseIds.map((converseId) => map[converseId] ?? null)
+      );
+    },
+    'serial',
+    100
+  )
+);
+export function getConverseLastMessageInfo(converseIds: string[]) {
+  return _fetchConverseLastMessageInfo(converseIds);
+}
+
+/**
+ * @param converseId 会话ID
+ * @param messageId 消息ID
+ * @returns 消息附近的信息
+ */
+export async function fetchNearbyMessage(params: {
+  groupId?: string;
+  converseId: string;
+  messageId: string;
+}): Promise<ChatMessage[]> {
+  const { data } = await request.post(
+    '/api/chat/message/fetchNearbyMessage',
+    params
+  );
+
+  return data;
+}
+
+/**
+ * 增加表情行为
+ */
+export async function addReaction(
+  messageId: string,
+  emoji: string
+): Promise<boolean> {
+  const { data } = await request.post('/api/chat/message/addReaction', {
+    messageId,
+    emoji,
+  });
+
+  return data;
+}
+
+/**
+ * 移除表情行为
+ */
+export async function removeReaction(
+  messageId: string,
+  emoji: string
+): Promise<boolean> {
+  const { data } = await request.post('/api/chat/message/removeReaction', {
+    messageId,
+    emoji,
+  });
+
+  return data;
+}