@easyv/biz-components 1.0.29 → 1.0.30

package/dist/lib/easyv-biz-components/src/utils/vosk-browser-manager/vosk-browser-manager.d.ts

@@ -0,0 +1,38 @@
+import { AudiManagerConfig, AudioManager } from './audio-manager/audio-manager';
+import { SampleRate } from './audio-manager/types';
+import type * as Vosk from 'vosk-browser';
+interface VoskBrowserManagerConfig extends AudiManagerConfig {
+    vosk: typeof Vosk;
+    modelUrl: string;
+    /** 加载模型等初始化配置完成后，自动开始监听 */
+    autoStart?: boolean;
+    onInitialized?: () => void;
+    onResult?: (text: string) => void;
+    onPartialResult?: (text: string) => void;
+    /** 热词识别，准确度不高, 而且会以影响原有的文字识别，慎用 */
+    hotWords?: string[];
+}
+/** 本地语音识别
+ * - 依赖 vosk-browser 库，vosk-browser 在 easyv 编辑页已安装 */
+export declare class VoskBrowserManager {
+    config: VoskBrowserManagerConfig;
+    /** 采样率（每秒采样数）的浮点数，常用值：16000, 32000, 44100, 48000 */
+    sampleRate: SampleRate;
+    audioManager?: AudioManager;
+    model?: Vosk.Model;
+    autoStart: boolean;
+    startResolve?: (value?: unknown) => void;
+    status?: 'initializing' | 'initialized';
+    recognizer?: Vosk.KaldiRecognizer;
+    constructor(config: VoskBrowserManagerConfig);
+    setConfig(config: Partial<Pick<VoskBrowserManagerConfig, 'onResult' | 'onPartialResult'>>): void;
+    init(vosk: typeof Vosk): Promise<void>;
+    /** 设置热词会重启识别器
+     * - 唤醒词识别阈值非常大，容易误唤醒，慎用！！！
+     */
+    setHotWords(hotWords: string[]): Promise<void>;
+    start(): Promise<void>;
+    stop(): void;
+    destroy(): void;
+}
+export {};

package/dist/lib/easyv-biz-components/src/utils/wait-something/index.d.ts

@@ -0,0 +1,9 @@
+/** 等待一个时间，单位是毫秒 */
+export declare function wait(time: number): Promise<unknown>;
+/** 等待某个条件成立
+ * @param waitFn 等待条件
+ * @param everyWaitTime 每次等待时间，单位是毫秒
+ * @param maxWaitTime 最大等待时间，单位是毫秒, 默认是 3 秒
+ * @returns 'success' | 'timeout'
+ */
+export declare function waitSomeThingBeTrue(waitFn: () => boolean, everyWaitTime: number, maxWaitTime?: number): Promise<'success' | 'timeout'>;

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/XF-stream-voice-manager.d.ts

@@ -0,0 +1,90 @@
+import { RecorderManager } from './recorder-manager';
+import { OnMessageInfo, RecordFrameInfo } from './types';
+
+type errorType = 'socketConnect' | 'message' | 'record' | 'recordPermission';
+export interface XFStreamVoiceManagerConfig {
+    onError: (errInfo: {
+        type: errorType;
+        message: string;
+    }) => void;
+    /** ws 收到消息的回调
+     * @param info.message 消息内容
+     * @param info.isLatest 是否是最后一条消息。 isLatest 为 true 时， message 为完整的语音识别结果。
+     */
+    onMessage: (info: OnMessageInfo) => void;
+    /** webSocket 关闭的回调。
+     * - 每次录音都会新建一个 webSocket 连接。录音结束后关闭 webSocket
+     */
+    onWsClose: () => void;
+    /**
+     * 录音开始的回调
+     */
+    onRecordStart?: () => void;
+    /**
+     * 录音开始的回调
+     */
+    onRecordStop?: () => void;
+    /**
+     * 录音最后一帧的回调。
+     * @param totalData 所有的录音数据, ArrayBuffer[]。
+     */
+    onLastFrame?: (totalData: ArrayBuffer[]) => void;
+    /**
+     * 获取链接讯飞语音识别地址的 url
+     * @returns websocket url。详情查看讯飞文档里的鉴权方法：https://www.xfyun.cn/doc/asr/voicedictation/API.html#%E6%8E%A5%E5%8F%A3%E8%B0%83%E7%94%A8%E6%B5%81%E7%A8%8B
+     */
+    getWebSocketUrl: () => string;
+    /**
+     * 获取录音权限的回调, 用户初次授权或者没有给录音权限时，会调用此回调。
+     * @param status 权限状态。
+     * - denied: 已拒绝
+     * - granted: 已授权
+     */
+    onGetMicrophonePermission?: (status: 'denied' | 'granted') => void;
+    /**
+     * 讯飞语音的 appid
+     */
+    xunFeiAppId: string;
+}
+/**
+ * 讯飞流式语音识别
+ * 大致流程：
+ * 1. recorder start 后开始接受录音，通过 recorder.onFrameRecorded 把数据透出给上层。数据中 isLastFrame 表示是否最后一次数据, frameBuffer 表示真实数据。
+ * 2. 上层拿到录音数据后，通过 ws 发送给讯飞。如果 isLastFrame 为 true 时，给讯飞的数据中 status 为 2。如果 isLastFrame 为 false 时，给讯飞的数据中 status 为 1。
+ * 3. ws 返回的数据，通过 renderResult 处理。数据中 status 为 2 表示本次录音数据识别结束了，这时候需要关闭 ws，避免消耗资源。且这两个情况服务端会主动断开连接（整个会话时长最多持续60s，或者超过10s未发送数据）。
+ *     讯飞流式语音文档：https://www.xfyun.cn/doc/asr/voicedictation/API.html#%E6%8E%A5%E5%8F%A3%E8%B0%83%E7%94%A8%E6%B5%81%E7%A8%8B
+ * 4. 断开链接后会关闭 recorder ，重置录音等数据，方便下次使用。
+ * 备注：使用时，建议在收到 onWsClose 事件后，才让前端恢复到可以再次输入录音的状态
+ */
+export declare class XFStreamVoiceManager {
+    config: XFStreamVoiceManagerConfig;
+    wsInstance: WebSocket | null;
+    recorder: RecorderManager | null;
+    isDestroy: boolean;
+    resultText: string;
+    resultTextTemp: string;
+    continueRecord: boolean;
+    startTimeStamp: number;
+    recordStatus: 'start' | 'stop';
+    readyToSend: boolean;
+    microphonePermissionStatus: 'denied' | 'granted' | 'prompt';
+    private inStopping;
+    private stopResolve;
+    constructor(config: XFStreamVoiceManagerConfig);
+    renderResult(resultData: string): void;
+    setConfig(newConfig: Partial<XFStreamVoiceManagerConfig>): void;
+    sendMessage(message: string): void;
+    recordConfigSend(): Promise<void>;
+    startNewSocket(): void;
+    onFrameRecorded({ isLastFrame, frameBuffer }: RecordFrameInfo): void;
+    initializeRecorder(): void;
+    startRecord(): Promise<void>;
+    stopRecord(): Promise<void>;
+    closeWs(): void;
+    /** 开启持续监听 */
+    startContinueRecord(): Promise<void>;
+    stopContinueRecord(): Promise<void>;
+    resetStatus(): Promise<void>;
+    destroy(): void;
+}
+export {};

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/index.d.ts

@@ -0,0 +1,3 @@
+export * from './XF-stream-voice-manager';
+export * from './recorder-manager';
+export * from './types';

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/recorder-manager/RecorderManager-draft.d.ts

@@ -0,0 +1,38 @@
+import { RecordFrameInfo } from '../types';
+
+export interface RecorderManagerConfig {
+    /** 需要传入 一个 AudioWorklet 脚本，应用到 audioContext.audioWorklet.addModule(workletJsPath).
+     *  AudioWorklet 运行在 Web Audio API 的 AudioWorklet 环境中，延迟更低，更适合实时音频处理。   */
+    workletJsPath: string;
+    /** 传入一个 一个 Web Worker 脚本。
+     * 用于后台对音频数据的处理。
+     */
+    workerJsPath?: string;
+    onStop?: (audioBuffers: ArrayBuffer[]) => void;
+    onFrameRecorded?: (params: RecordFrameInfo) => void;
+    /**
+     * 监听录音开始事件
+     */
+    onStart?: () => void;
+    onError?: (error: string) => void;
+}
+export interface startConfig {
+    toSampleRate?: number;
+    fromSampleRate?: number;
+    frameSize?: number;
+    arrayBufferType?: 'short16' | 'float32';
+}
+export declare class RecorderManager {
+    config: RecorderManagerConfig;
+    audioContext: AudioContext | null;
+    audioSource: MediaStreamAudioSourceNode | undefined;
+    workletNode: AudioWorkletNode | undefined;
+    status: 'beforeInit' | 'initialized' | 'start' | 'stop';
+    audioChunks: any[];
+    constructor(config: RecorderManagerConfig);
+    setConfig: (newConfig: Partial<RecorderManagerConfig>) => void;
+    resetAudioContext: () => Promise<void>;
+    start: (config: startConfig) => Promise<void>;
+    stop: () => void;
+    getAudioUrl: () => string;
+}

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/recorder-manager/index.d.ts

@@ -0,0 +1 @@
+export * from './RecorderManager';

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/test/RecorderManager.cy.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/test/XunFeiVoiceManager.cy.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/types.d.ts

@@ -0,0 +1,39 @@
+export interface RecordFrameInfo {
+    isLastFrame: boolean;
+    frameBuffer: ArrayBuffer;
+}
+declare class RecorderManagerClass {
+    /**
+     * 构造函数
+     * @param processorPath processor的文件路径，如果processor.worker.js的访问地址为`/a/b/processor.worker.js`,则processorPath 为`/a/b`
+     *
+     */
+    constructor(processorPath: string);
+    private audioBuffers;
+    private processorPath;
+    private audioContext?;
+    private audioTracks?;
+    private audioWorklet?;
+    onStop?: (audioBuffers: ArrayBuffer[]) => void;
+    onFrameRecorded?: (params: RecordFrameInfo) => void;
+    /**
+     * 监听录音开始事件
+     */
+    onStart?: () => void;
+    start({ sampleRate, frameSize, arrayBufferType, }: {
+        sampleRate?: number;
+        frameSize?: number;
+        arrayBufferType?: 'short16' | 'float32';
+    }): Promise<void>;
+    stop(): void;
+}
+export { RecorderManagerClass };
+export interface OnMessageInfo {
+    /** 优化后的结果。最终结果 */
+    message: string;
+    isLatest?: boolean;
+    /** 临时结果
+     * 区别： 如用户再说 “一二三四五”， 临时结果为 “一、二、三、四、五”，最终结果会优化成阿拉伯数字 “12345
+     */
+    tempMessage: string;
+}

package/dist/lib/easyv-biz-components/src/utils/xun-fei-voice/utils.d.ts

@@ -0,0 +1,17 @@
+export declare function toBase64(buffer: ArrayBuffer): string;
+export declare function isWsClosed(ws?: WebSocket | null): boolean;
+/**
+ *
+ * @returns
+ * granted: 已授权
+ * denied: 已拒绝
+ * prompt: 未决定
+ * error: 错误
+ */
+export declare function checkMicrophonePermission(): Promise<PermissionState>;
+/**
+ * 获取用户媒体设备权限。
+ * - 如果用户拒绝授权，抛出 name 为 NotAllowedError 的 DOMException
+ * @returns {Promise<MediaStream>}
+ */
+export declare function requestMicrophonePermission(): Promise<MediaStream>;

package/dist/lib/hooks/use-ai-agent-message/constants.es.js

@@ -0,0 +1,10 @@
+import { AiAgentMessageType } from "../../utils/ai-agent-message-manager/constants.es.js";
+const aiAnswerContentTypeList = [
+  AiAgentMessageType.TEXT,
+  AiAgentMessageType.DASHBOARD,
+  AiAgentMessageType.EXECUTION_STEP
+];
+export {
+  aiAnswerContentTypeList
+};
+//# sourceMappingURL=constants.es.js.map

package/dist/lib/hooks/use-ai-agent-message/constants.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.es.js","sources":["../../../../src/hooks/use-ai-agent-message/constants.ts"],"sourcesContent":["import { AiAgentMessageType } from '@/utils/ai-agent-message-manager';\n\n/** ai 回答的消息类型的合集 */\nexport const aiAnswerContentTypeList = [\n  AiAgentMessageType.TEXT,\n  AiAgentMessageType.DASHBOARD,\n  AiAgentMessageType.EXECUTION_STEP,\n];\n"],"names":[],"mappings":";AAGO,MAAM,0BAA0B;AAAA,EACrC,mBAAmB;AAAA,EACnB,mBAAmB;AAAA,EACnB,mBAAmB;AACrB;"}

package/dist/lib/hooks/use-ai-agent-message/use-ai-agent-message.es.js

@@ -0,0 +1,243 @@
+import { useState, useRef } from "react";
+import useMemoizedFn from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useMemoizedFn/index.es.js";
+import useThrottleFn from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useThrottleFn/index.es.js";
+import useUnmount from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useUnmount/index.es.js";
+import { message } from "antd";
+import { AiAgentMessageManager } from "../../utils/ai-agent-message-manager/ai-agent-message-manager.es.js";
+import { useInitialized } from "../use-initialized.es.js";
+import { useWatchValue } from "../use-watch-value/use-watch-value.es.js";
+import { aiAnswerContentTypeList } from "./constants.es.js";
+import { generateUserMsgItem } from "./utils.es.js";
+import { RoomType } from "../use-socket-events/constant.es.js";
+import { fetchKnowledge } from "../../utils/fetchKnowledge/fetchKnowledge.es.js";
+import v4 from "../../node_modules/.pnpm/uuid@11.1.0/node_modules/uuid/dist/esm-browser/v4.es.js";
+import { wait } from "../../utils/wait-something/index.es.js";
+import { generateAiChatMessageFromSocketMessage } from "../../components/ai-message-render/utils.es.js";
+import isNil from "../../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/isNil.es.js";
+import { AiAgentMessageType } from "../../utils/ai-agent-message-manager/constants.es.js";
+const useAiAgentMessage = (props) => {
+  const { aiAgentManagerConfig, roomType, agentId, agentSessionId, onReceiveConversationTitle } = props;
+  const [isAnswer, setIsAnswer] = useState(false);
+  const [chatLoading, setChatLoading] = useState(false);
+  const socketMsgListRef = useRef([]);
+  const [messageList, setMessageList] = useState([]);
+  const [stepHintAnimation, setStepHintAnimation] = useState(false);
+  const stepHintContentRef = useRef("");
+  const setStepHintContent = (content) => {
+    stepHintContentRef.current = content;
+  };
+  const currOneChatIdRef = useRef("");
+  const setCurrOneChatId = (id) => {
+    currOneChatIdRef.current = id;
+  };
+  const messageStatusRef = useRef("init");
+  const setMessageStatus = (status) => {
+    messageStatusRef.current = status;
+  };
+  const lastQuestionRef = useRef("");
+  const onStopChat = useMemoizedFn((id) => {
+    var _a;
+    (_a = aiAgentManagerConfig.onStopChat) == null ? void 0 : _a.call(aiAgentManagerConfig, id);
+    setIsAnswer(false);
+    setChatLoading(false);
+  });
+  const onAiAgentMessage = useMemoizedFn(async (payload) => {
+    var _a;
+    (_a = aiAgentManagerConfig.onAiAgentMessage) == null ? void 0 : _a.call(aiAgentManagerConfig, payload);
+    socketMsgListRef.current.push(payload);
+    if (payload.type === AiAgentMessageType.TITLE) {
+      onReceiveConversationTitle == null ? void 0 : onReceiveConversationTitle(payload.content || "");
+    }
+    if (payload.type === AiAgentMessageType.CONVERSION_START) {
+      setMessageStatus("start");
+      socketMsgListRef.current = [];
+      setIsAnswer(true);
+      setChatLoading(true);
+      setCurrOneChatId(v4());
+    }
+    if (aiAnswerContentTypeList.includes(payload.type)) {
+      if (!currOneChatIdRef.current) {
+        setCurrOneChatId(v4());
+      }
+      if (!isAnswer) {
+        setIsAnswer(true);
+      }
+      if (chatLoading) {
+        setChatLoading(false);
+      }
+      if (stepHintAnimation) {
+        setStepHintAnimation(false);
+      }
+    }
+    if (payload.type === AiAgentMessageType.ERROR) {
+      setIsAnswer(false);
+      setChatLoading(false);
+      setStepHintAnimation(false);
+      setMessageStatus("error");
+      if (messageStatusRef.current !== "start") {
+        message.error(payload.content || "智能体消息出现错误！");
+      } else {
+        if (!currOneChatIdRef.current) {
+          setCurrOneChatId(v4());
+        }
+        renderMessageList();
+      }
+      wait(0).then(() => {
+        setCurrOneChatId("");
+      });
+    }
+    if (payload.type === AiAgentMessageType.CONVERSION_END) {
+      setChatLoading(false);
+      setIsAnswer(false);
+      setMessageStatus("end");
+      setStepHintContent("结果已生成");
+      setStepHintAnimation(false);
+      await wait(150);
+      renderMessageList();
+      socketMsgListRef.current = [];
+      await wait(0);
+      setCurrOneChatId("");
+    }
+    if (payload.type === AiAgentMessageType.EXECUTION_STEP) {
+      setStepHintContent(payload.content || "");
+      setStepHintAnimation(true);
+    }
+    renderMessageListThrottled();
+  });
+  const renderMessageList = useMemoizedFn(() => {
+    const socketMsgList = [...socketMsgListRef.current];
+    const currOneChatId = currOneChatIdRef.current;
+    if (!currOneChatId) {
+      return;
+    }
+    if (socketMsgListRef.current.some((msg) => msg.type === AiAgentMessageType.CONVERSION_END)) {
+      socketMsgListRef.current = [];
+    }
+    const newMsgList = [...messageList];
+    const { aiMsg, userMsg } = generateAiChatMessageFromSocketMessage({
+      socketMsgList,
+      currMsgId: currOneChatId
+    });
+    const isExistUserQuestion = newMsgList.some(
+      (msg) => msg.id === currOneChatId && msg.role === "user"
+    );
+    if (!isExistUserQuestion && currOneChatId) {
+      if (userMsg) {
+        newMsgList.push(userMsg);
+      } else if (socketManager == null ? void 0 : socketManager.currChatInfo.question) {
+        newMsgList.push(generateUserMsgItem(currOneChatId, socketManager.currChatInfo.question));
+      }
+    }
+    const index = newMsgList.findIndex((msg) => msg.id === currOneChatId && msg.role === "ai");
+    if (aiMsg) {
+      if (stepHintContentRef.current) {
+        const stepItem = aiMsg.contents.find((i) => Boolean(i.aiAgentStepMessage));
+        if (stepItem) {
+          stepItem.aiAgentStepMessage = stepHintContentRef.current;
+        } else {
+          aiMsg.contents.push({ aiAgentStepMessage: stepHintContentRef.current });
+        }
+      } else {
+        aiMsg.contents = aiMsg.contents.filter((i) => isNil(i.aiAgentStepMessage));
+      }
+      if (messageStatusRef.current === "end" && aiMsg.contents.every((i) => Boolean(i.aiAgentStepMessage))) {
+        aiMsg.contents = [];
+      }
+      if (index !== -1) {
+        newMsgList[index] = aiMsg;
+      } else {
+        newMsgList.push(aiMsg);
+      }
+    }
+    setMessageList(newMsgList);
+  });
+  const { run: renderMessageListThrottled } = useThrottleFn(renderMessageList, {
+    wait: 50,
+    trailing: false
+  });
+  const [socketManager, setSocketManager] = useState();
+  const onSendMsgToAi = useMemoizedFn((params) => {
+    setMessageStatus("init");
+    lastQuestionRef.current = params.question;
+  });
+  const onAiAgentToolCall = useMemoizedFn(async (data) => {
+    var _a;
+    let result = "";
+    try {
+      const toolInfo = data.toolInfo;
+      if (!toolInfo) {
+        return;
+      }
+      result = await fetchKnowledge(toolInfo.query, {
+        baseURL: toolInfo.baseURL,
+        requestHandler: toolInfo.requestHandler,
+        requestBody: toolInfo.requestBody,
+        requestHeader: toolInfo.requestHeader
+      });
+    } catch (error) {
+      result = error.message;
+    }
+    socketManager == null ? void 0 : socketManager.sendToolResultToAiAgent(result || "知识库请求错误");
+    (_a = aiAgentManagerConfig.onAiAgentToolCall) == null ? void 0 : _a.call(aiAgentManagerConfig, data);
+  });
+  useInitialized(() => {
+    const _socketManager = new AiAgentMessageManager({
+      ...aiAgentManagerConfig,
+      onAiAgentMessage,
+      onStopChat,
+      onSendMsgToAi,
+      onAiAgentToolCall
+    });
+    _socketManager.init();
+    setSocketManager(_socketManager);
+  });
+  useWatchValue({
+    value: { agentId, agentSessionId },
+    callback: (newInfo, oldInfo) => {
+      if (roomType === RoomType.AI_AGENT && !(newInfo == null ? void 0 : newInfo.agentSessionId)) {
+        return;
+      }
+      if (oldInfo == null ? void 0 : oldInfo.agentId) {
+        socketManager == null ? void 0 : socketManager.leaveRoom({
+          roomType,
+          id: oldInfo.agentId,
+          subIds: oldInfo.agentSessionId ? [oldInfo.agentSessionId] : void 0
+        });
+      }
+      if (newInfo.agentId) {
+        socketManager == null ? void 0 : socketManager.joinRoom({
+          roomType,
+          id: newInfo.agentId,
+          subIds: newInfo.agentSessionId ? [newInfo.agentSessionId] : void 0
+        });
+      }
+    }
+  });
+  useUnmount(() => {
+    if (!agentId) {
+      return;
+    }
+    socketManager == null ? void 0 : socketManager.leaveRoom({
+      roomType,
+      id: agentId,
+      subIds: agentSessionId ? [agentSessionId] : void 0
+    });
+    socketManager == null ? void 0 : socketManager.closeWs();
+  });
+  return {
+    socketManager,
+    isAnswer,
+    setIsAnswer,
+    chatLoading,
+    setChatLoading,
+    messageList,
+    setMessageList,
+    currOneChatId: currOneChatIdRef.current,
+    stepHintContent: stepHintContentRef.current,
+    stepHintAnimation
+  };
+};
+export {
+  useAiAgentMessage
+};
+//# sourceMappingURL=use-ai-agent-message.es.js.map

package/dist/lib/hooks/use-ai-agent-message/use-ai-agent-message.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-ai-agent-message.es.js","sources":["../../../../src/hooks/use-ai-agent-message/use-ai-agent-message.ts"],"sourcesContent":["import { useRef, useState } from 'react';\nimport { useMemoizedFn, useThrottleFn, useUnmount } from 'ahooks';\nimport { message } from 'antd';\nimport { isNil } from 'lodash-es';\nimport { v4 as uuid } from 'uuid';\nimport {\n  AiChatMessageItem,\n  generateAiChatMessageFromSocketMessage,\n} from '@/components/ai-message-render';\nimport { AiAgentMessageType, fetchKnowledge, ToolCallInfo, wait } from '@/utils';\nimport { AiAgentMessageManager } from '@/utils/ai-agent-message-manager/ai-agent-message-manager';\nimport {\n  AiAgentMessageManagerConfig,\n  AiAgentSocketMessagePayload,\n} from '@/utils/ai-agent-message-manager/types';\nimport { useInitialized } from '../use-initialized';\nimport { RoomType } from '../use-socket-events';\nimport { useWatchValue } from '../use-watch-value/use-watch-value';\nimport { aiAnswerContentTypeList } from './constants';\nimport { KnowledgeBaseDetailForRequestStandard } from './types';\nimport { generateUserMsgItem } from './utils';\n\nexport interface UseAiAgentMessageProps {\n  /** 这个 config 里的函数因为只在初始化调用，所以函数内部拿到的 state 是初始化时的 state，需要用 ref 透传才能拿到最新的 */\n  aiAgentManagerConfig: AiAgentMessageManagerConfig;\n  roomType: RoomType;\n  /** 智能体组件的话，传智能体组件 id */\n  agentId?: number;\n  /** 智能体整个会话的 id，如果是独立对话，那么需要，共享会话不需要，在智能体页面都是独立对话模式\n   * - 用 uuid 生成\n   */\n  agentSessionId?: string;\n  /** 后端返回对话标题 */\n  onReceiveConversationTitle?: (title: string) => void;\n}\n\nexport const useAiAgentMessage = (props: UseAiAgentMessageProps) => {\n  const { aiAgentManagerConfig, roomType, agentId, agentSessionId, onReceiveConversationTitle } =\n    props;\n  const [isAnswer, setIsAnswer] = useState(false);\n  const [chatLoading, setChatLoading] = useState(false);\n  const socketMsgListRef = useRef<AiAgentSocketMessagePayload[]>([]);\n  const [messageList, setMessageList] = useState<AiChatMessageItem[]>([]);\n  const [stepHintAnimation, setStepHintAnimation] = useState<boolean>(false);\n  const stepHintContentRef = useRef<string>('');\n  const setStepHintContent = (content: string) => {\n    stepHintContentRef.current = content;\n  };\n  /** 当前这个对话的 id，为空时不会在渲染消息。 */\n  const currOneChatIdRef = useRef<string>('');\n  const setCurrOneChatId = (id: string) => {\n    currOneChatIdRef.current = id;\n  };\n  const messageStatusRef = useRef<'start' | 'end' | 'error' | 'init'>('init');\n  const setMessageStatus = (status: 'start' | 'end' | 'error' | 'init') => {\n    messageStatusRef.current = status;\n  };\n  const lastQuestionRef = useRef<string>('');\n\n  const onStopChat = useMemoizedFn((id?: string) => {\n    aiAgentManagerConfig.onStopChat?.(id);\n    setIsAnswer(false);\n    setChatLoading(false);\n  });\n\n  const onAiAgentMessage = useMemoizedFn(async (payload: AiAgentSocketMessagePayload) => {\n    aiAgentManagerConfig.onAiAgentMessage?.(payload);\n    socketMsgListRef.current.push(payload);\n    if (payload.type === AiAgentMessageType.TITLE) {\n      onReceiveConversationTitle?.(payload.content || '');\n    }\n    if (payload.type === AiAgentMessageType.CONVERSION_START) {\n      setMessageStatus('start');\n      socketMsgListRef.current = [];\n      setIsAnswer(true);\n      setChatLoading(true);\n      setCurrOneChatId(uuid());\n    }\n    // 问答消息开始 或者 图表消息开始时 结束 loading\n    if (aiAnswerContentTypeList.includes(payload.type)) {\n      if (!currOneChatIdRef.current) {\n        setCurrOneChatId(uuid());\n      }\n      if (!isAnswer) {\n        setIsAnswer(true);\n      }\n      if (chatLoading) {\n        setChatLoading(false);\n      }\n      if (stepHintAnimation) {\n        setStepHintAnimation(false);\n      }\n    }\n    if (payload.type === AiAgentMessageType.ERROR) {\n      setIsAnswer(false);\n      setChatLoading(false);\n      setStepHintAnimation(false);\n      setMessageStatus('error');\n      // 在 start 和 end 之外的 error 直接 message 提示，不需要渲染。\n      if (messageStatusRef.current !== 'start') {\n        message.error(payload.content || '智能体消息出现错误！');\n      } else {\n        // error 的时候没有 start 的消息，所以手动设置一个\n        if (!currOneChatIdRef.current) {\n          setCurrOneChatId(uuid());\n        }\n        renderMessageList();\n      }\n      wait(0).then(() => {\n        setCurrOneChatId('');\n      });\n    }\n    if (payload.type === AiAgentMessageType.CONVERSION_END) {\n      setChatLoading(false);\n      setIsAnswer(false);\n      setMessageStatus('end');\n      setStepHintContent('结果已生成');\n      setStepHintAnimation(false);\n      await wait(150);\n      renderMessageList();\n      socketMsgListRef.current = [];\n      await wait(0);\n      setCurrOneChatId('');\n    }\n    if (payload.type === AiAgentMessageType.EXECUTION_STEP) {\n      setStepHintContent(payload.content || '');\n      setStepHintAnimation(true);\n    }\n    renderMessageListThrottled();\n  });\n\n  const renderMessageList = useMemoizedFn(() => {\n    const socketMsgList = [...socketMsgListRef.current];\n    const currOneChatId = currOneChatIdRef.current;\n    if (!currOneChatId) {\n      return;\n    }\n    if (socketMsgListRef.current.some((msg) => msg.type === AiAgentMessageType.CONVERSION_END)) {\n      socketMsgListRef.current = [];\n    }\n    const newMsgList = [...messageList];\n    // 如果消息列表中能 找不到 当前消息id，那么添加用户和ai的消息\n    const { aiMsg, userMsg } = generateAiChatMessageFromSocketMessage({\n      socketMsgList,\n      currMsgId: currOneChatId,\n    });\n    const isExistUserQuestion = newMsgList.some(\n      (msg) => msg.id === currOneChatId && msg.role === 'user',\n    );\n    if (!isExistUserQuestion && currOneChatId) {\n      if (userMsg) {\n        newMsgList.push(userMsg);\n      } else if (socketManager?.currChatInfo.question) {\n        newMsgList.push(generateUserMsgItem(currOneChatId, socketManager.currChatInfo.question));\n      }\n    }\n    const index = newMsgList.findIndex((msg) => msg.id === currOneChatId && msg.role === 'ai');\n    if (aiMsg) {\n      if (stepHintContentRef.current) {\n        const stepItem = aiMsg.contents.find((i) => Boolean(i.aiAgentStepMessage));\n        if (stepItem) {\n          stepItem.aiAgentStepMessage = stepHintContentRef.current;\n        } else {\n          aiMsg.contents.push({ aiAgentStepMessage: stepHintContentRef.current });\n        }\n      } else {\n        aiMsg.contents = aiMsg.contents.filter((i) => isNil(i.aiAgentStepMessage));\n      }\n      if (\n        messageStatusRef.current === 'end' &&\n        aiMsg.contents.every((i) => Boolean(i.aiAgentStepMessage))\n      ) {\n        aiMsg.contents = [];\n      }\n      if (index !== -1) {\n        newMsgList[index] = aiMsg;\n      } else {\n        newMsgList.push(aiMsg);\n      }\n    }\n    setMessageList(newMsgList);\n  });\n  const { run: renderMessageListThrottled } = useThrottleFn(renderMessageList, {\n    wait: 50,\n    trailing: false,\n  });\n\n  const [socketManager, setSocketManager] = useState<AiAgentMessageManager>();\n\n  const onSendMsgToAi = useMemoizedFn((params) => {\n    setMessageStatus('init');\n    lastQuestionRef.current = params.question;\n  });\n  const onAiAgentToolCall = useMemoizedFn(async (data: ToolCallInfo) => {\n    let result = '';\n    try {\n      const toolInfo = data.toolInfo as unknown as KnowledgeBaseDetailForRequestStandard & {\n        query: string;\n      };\n      if (!toolInfo) {\n        return;\n      }\n      result = await fetchKnowledge(toolInfo.query, {\n        baseURL: toolInfo.baseURL,\n        requestHandler: toolInfo.requestHandler,\n        requestBody: toolInfo.requestBody,\n        requestHeader: toolInfo.requestHeader,\n      });\n    } catch (error: any) {\n      result = error.message;\n    }\n    socketManager?.sendToolResultToAiAgent(result || '知识库请求错误');\n    aiAgentManagerConfig.onAiAgentToolCall?.(data);\n  });\n\n  useInitialized(() => {\n    // 初始化时加入房间\n    const _socketManager = new AiAgentMessageManager({\n      ...aiAgentManagerConfig,\n      onAiAgentMessage,\n      onStopChat,\n      onSendMsgToAi,\n      onAiAgentToolCall,\n    });\n    _socketManager.init();\n    setSocketManager(_socketManager);\n  });\n\n  useWatchValue({\n    value: { agentId, agentSessionId },\n    callback: (newInfo, oldInfo) => {\n      if (roomType === RoomType.AI_AGENT && !newInfo?.agentSessionId) {\n        return;\n      }\n      if (oldInfo?.agentId) {\n        socketManager?.leaveRoom({\n          roomType,\n          id: oldInfo.agentId,\n          subIds: oldInfo.agentSessionId ? [oldInfo.agentSessionId] : undefined,\n        });\n      }\n      if (newInfo.agentId) {\n        socketManager?.joinRoom({\n          roomType,\n          id: newInfo.agentId,\n          subIds: newInfo.agentSessionId ? [newInfo.agentSessionId] : undefined,\n        });\n      }\n    },\n  });\n\n  /** 卸载时，离开房间 */\n  useUnmount(() => {\n    if (!agentId) {\n      return;\n    }\n    socketManager?.leaveRoom({\n      roomType,\n      id: agentId,\n      subIds: agentSessionId ? [agentSessionId] : undefined,\n    });\n    socketManager?.closeWs();\n  });\n\n  return {\n    socketManager,\n    isAnswer,\n    setIsAnswer,\n    chatLoading,\n    setChatLoading,\n    messageList,\n    setMessageList,\n    currOneChatId: currOneChatIdRef.current,\n    stepHintContent: stepHintContentRef.current,\n    stepHintAnimation,\n  };\n};\n"],"names":["uuid"],"mappings":";;;;;;;;;;;;;;;;;AAoCa,MAAA,oBAAoB,CAAC,UAAkC;AAClE,QAAM,EAAE,sBAAsB,UAAU,SAAS,gBAAgB,+BAC/D;AACF,QAAM,CAAC,UAAU,WAAW,IAAI,SAAS,KAAK;AAC9C,QAAM,CAAC,aAAa,cAAc,IAAI,SAAS,KAAK;AAC9C,QAAA,mBAAmB,OAAsC,EAAE;AACjE,QAAM,CAAC,aAAa,cAAc,IAAI,SAA8B,CAAA,CAAE;AACtE,QAAM,CAAC,mBAAmB,oBAAoB,IAAI,SAAkB,KAAK;AACnE,QAAA,qBAAqB,OAAe,EAAE;AACtC,QAAA,qBAAqB,CAAC,YAAoB;AAC9C,uBAAmB,UAAU;AAAA,EAC/B;AAEM,QAAA,mBAAmB,OAAe,EAAE;AACpC,QAAA,mBAAmB,CAAC,OAAe;AACvC,qBAAiB,UAAU;AAAA,EAC7B;AACM,QAAA,mBAAmB,OAA2C,MAAM;AACpE,QAAA,mBAAmB,CAAC,WAA+C;AACvE,qBAAiB,UAAU;AAAA,EAC7B;AACM,QAAA,kBAAkB,OAAe,EAAE;AAEnC,QAAA,aAAa,cAAc,CAAC,OAAgB;;AAChD,+BAAqB,eAArB,8CAAkC;AAClC,gBAAY,KAAK;AACjB,mBAAe,KAAK;AAAA,EAAA,CACrB;AAEK,QAAA,mBAAmB,cAAc,OAAO,YAAyC;;AACrF,+BAAqB,qBAArB,8CAAwC;AACvB,qBAAA,QAAQ,KAAK,OAAO;AACjC,QAAA,QAAQ,SAAS,mBAAmB,OAAO;AAChB,+EAAA,QAAQ,WAAW;AAAA,IAAE;AAEhD,QAAA,QAAQ,SAAS,mBAAmB,kBAAkB;AACxD,uBAAiB,OAAO;AACxB,uBAAiB,UAAU,CAAC;AAC5B,kBAAY,IAAI;AAChB,qBAAe,IAAI;AACnB,uBAAiBA,IAAM;AAAA,IAAA;AAGzB,QAAI,wBAAwB,SAAS,QAAQ,IAAI,GAAG;AAC9C,UAAA,CAAC,iBAAiB,SAAS;AAC7B,yBAAiBA,IAAM;AAAA,MAAA;AAEzB,UAAI,CAAC,UAAU;AACb,oBAAY,IAAI;AAAA,MAAA;AAElB,UAAI,aAAa;AACf,uBAAe,KAAK;AAAA,MAAA;AAEtB,UAAI,mBAAmB;AACrB,6BAAqB,KAAK;AAAA,MAAA;AAAA,IAC5B;AAEE,QAAA,QAAQ,SAAS,mBAAmB,OAAO;AAC7C,kBAAY,KAAK;AACjB,qBAAe,KAAK;AACpB,2BAAqB,KAAK;AAC1B,uBAAiB,OAAO;AAEpB,UAAA,iBAAiB,YAAY,SAAS;AAChC,gBAAA,MAAM,QAAQ,WAAW,YAAY;AAAA,MAAA,OACxC;AAED,YAAA,CAAC,iBAAiB,SAAS;AAC7B,2BAAiBA,IAAM;AAAA,QAAA;AAEP,0BAAA;AAAA,MAAA;AAEf,WAAA,CAAC,EAAE,KAAK,MAAM;AACjB,yBAAiB,EAAE;AAAA,MAAA,CACpB;AAAA,IAAA;AAEC,QAAA,QAAQ,SAAS,mBAAmB,gBAAgB;AACtD,qBAAe,KAAK;AACpB,kBAAY,KAAK;AACjB,uBAAiB,KAAK;AACtB,yBAAmB,OAAO;AAC1B,2BAAqB,KAAK;AAC1B,YAAM,KAAK,GAAG;AACI,wBAAA;AAClB,uBAAiB,UAAU,CAAC;AAC5B,YAAM,KAAK,CAAC;AACZ,uBAAiB,EAAE;AAAA,IAAA;AAEjB,QAAA,QAAQ,SAAS,mBAAmB,gBAAgB;AACnC,yBAAA,QAAQ,WAAW,EAAE;AACxC,2BAAqB,IAAI;AAAA,IAAA;AAEA,+BAAA;AAAA,EAAA,CAC5B;AAEK,QAAA,oBAAoB,cAAc,MAAM;AAC5C,UAAM,gBAAgB,CAAC,GAAG,iBAAiB,OAAO;AAClD,UAAM,gBAAgB,iBAAiB;AACvC,QAAI,CAAC,eAAe;AAClB;AAAA,IAAA;AAEE,QAAA,iBAAiB,QAAQ,KAAK,CAAC,QAAQ,IAAI,SAAS,mBAAmB,cAAc,GAAG;AAC1F,uBAAiB,UAAU,CAAC;AAAA,IAAA;AAExB,UAAA,aAAa,CAAC,GAAG,WAAW;AAElC,UAAM,EAAE,OAAO,QAAQ,IAAI,uCAAuC;AAAA,MAChE;AAAA,MACA,WAAW;AAAA,IAAA,CACZ;AACD,UAAM,sBAAsB,WAAW;AAAA,MACrC,CAAC,QAAQ,IAAI,OAAO,iBAAiB,IAAI,SAAS;AAAA,IACpD;AACI,QAAA,CAAC,uBAAuB,eAAe;AACzC,UAAI,SAAS;AACX,mBAAW,KAAK,OAAO;AAAA,MAAA,WACd,+CAAe,aAAa,UAAU;AAC/C,mBAAW,KAAK,oBAAoB,eAAe,cAAc,aAAa,QAAQ,CAAC;AAAA,MAAA;AAAA,IACzF;AAEI,UAAA,QAAQ,WAAW,UAAU,CAAC,QAAQ,IAAI,OAAO,iBAAiB,IAAI,SAAS,IAAI;AACzF,QAAI,OAAO;AACT,UAAI,mBAAmB,SAAS;AACxB,cAAA,WAAW,MAAM,SAAS,KAAK,CAAC,MAAM,QAAQ,EAAE,kBAAkB,CAAC;AACzE,YAAI,UAAU;AACZ,mBAAS,qBAAqB,mBAAmB;AAAA,QAAA,OAC5C;AACL,gBAAM,SAAS,KAAK,EAAE,oBAAoB,mBAAmB,SAAS;AAAA,QAAA;AAAA,MACxE,OACK;AACC,cAAA,WAAW,MAAM,SAAS,OAAO,CAAC,MAAM,MAAM,EAAE,kBAAkB,CAAC;AAAA,MAAA;AAE3E,UACE,iBAAiB,YAAY,SAC7B,MAAM,SAAS,MAAM,CAAC,MAAM,QAAQ,EAAE,kBAAkB,CAAC,GACzD;AACA,cAAM,WAAW,CAAC;AAAA,MAAA;AAEpB,UAAI,UAAU,IAAI;AAChB,mBAAW,KAAK,IAAI;AAAA,MAAA,OACf;AACL,mBAAW,KAAK,KAAK;AAAA,MAAA;AAAA,IACvB;AAEF,mBAAe,UAAU;AAAA,EAAA,CAC1B;AACD,QAAM,EAAE,KAAK,+BAA+B,cAAc,mBAAmB;AAAA,IAC3E,MAAM;AAAA,IACN,UAAU;AAAA,EAAA,CACX;AAED,QAAM,CAAC,eAAe,gBAAgB,IAAI,SAAgC;AAEpE,QAAA,gBAAgB,cAAc,CAAC,WAAW;AAC9C,qBAAiB,MAAM;AACvB,oBAAgB,UAAU,OAAO;AAAA,EAAA,CAClC;AACK,QAAA,oBAAoB,cAAc,OAAO,SAAuB;;AACpE,QAAI,SAAS;AACT,QAAA;AACF,YAAM,WAAW,KAAK;AAGtB,UAAI,CAAC,UAAU;AACb;AAAA,MAAA;AAEO,eAAA,MAAM,eAAe,SAAS,OAAO;AAAA,QAC5C,SAAS,SAAS;AAAA,QAClB,gBAAgB,SAAS;AAAA,QACzB,aAAa,SAAS;AAAA,QACtB,eAAe,SAAS;AAAA,MAAA,CACzB;AAAA,aACM,OAAY;AACnB,eAAS,MAAM;AAAA,IAAA;AAEF,mDAAA,wBAAwB,UAAU;AACjD,+BAAqB,sBAArB,8CAAyC;AAAA,EAAI,CAC9C;AAED,iBAAe,MAAM;AAEb,UAAA,iBAAiB,IAAI,sBAAsB;AAAA,MAC/C,GAAG;AAAA,MACH;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA,CACD;AACD,mBAAe,KAAK;AACpB,qBAAiB,cAAc;AAAA,EAAA,CAChC;AAEa,gBAAA;AAAA,IACZ,OAAO,EAAE,SAAS,eAAe;AAAA,IACjC,UAAU,CAAC,SAAS,YAAY;AAC9B,UAAI,aAAa,SAAS,YAAY,EAAC,mCAAS,iBAAgB;AAC9D;AAAA,MAAA;AAEF,UAAI,mCAAS,SAAS;AACpB,uDAAe,UAAU;AAAA,UACvB;AAAA,UACA,IAAI,QAAQ;AAAA,UACZ,QAAQ,QAAQ,iBAAiB,CAAC,QAAQ,cAAc,IAAI;AAAA,QAAA;AAAA,MAC7D;AAEH,UAAI,QAAQ,SAAS;AACnB,uDAAe,SAAS;AAAA,UACtB;AAAA,UACA,IAAI,QAAQ;AAAA,UACZ,QAAQ,QAAQ,iBAAiB,CAAC,QAAQ,cAAc,IAAI;AAAA,QAAA;AAAA,MAC7D;AAAA,IACH;AAAA,EACF,CACD;AAGD,aAAW,MAAM;AACf,QAAI,CAAC,SAAS;AACZ;AAAA,IAAA;AAEF,mDAAe,UAAU;AAAA,MACvB;AAAA,MACA,IAAI;AAAA,MACJ,QAAQ,iBAAiB,CAAC,cAAc,IAAI;AAAA,IAAA;AAE9C,mDAAe;AAAA,EAAQ,CACxB;AAEM,SAAA;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,eAAe,iBAAiB;AAAA,IAChC,iBAAiB,mBAAmB;AAAA,IACpC;AAAA,EACF;AACF;"}

package/dist/lib/hooks/use-ai-agent-message/utils.es.js

@@ -0,0 +1,25 @@
+import dayjs from "../../node_modules/.pnpm/dayjs@1.11.13/node_modules/dayjs/dayjs.min.es.js";
+import { generateRandomId } from "../../utils/common/common.es.js";
+const generateUserMsgItem = (id, content) => {
+  return {
+    id,
+    contents: [{ text: content }],
+    role: "user",
+    createTime: dayjs().toString(),
+    createTimeStamp: dayjs().unix()
+  };
+};
+const generateAiMsgItem = (content, showVoiceBtn = false) => {
+  return {
+    id: generateRandomId(),
+    role: "ai",
+    contents: [{ text: content }],
+    createTime: (/* @__PURE__ */ new Date()).toLocaleString(),
+    showVoiceBtn
+  };
+};
+export {
+  generateAiMsgItem,
+  generateUserMsgItem
+};
+//# sourceMappingURL=utils.es.js.map

package/dist/lib/hooks/use-ai-agent-message/utils.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.es.js","sources":["../../../../src/hooks/use-ai-agent-message/utils.ts"],"sourcesContent":["import dayjs from 'dayjs';\nimport { AiChatMessageItem } from '@/components';\nimport { generateRandomId } from '@/utils';\n\nexport const generateUserMsgItem = (id: string, content: string): AiChatMessageItem => {\n  return {\n    id,\n    contents: [{ text: content }],\n    role: 'user',\n    createTime: dayjs().toString(),\n    createTimeStamp: dayjs().unix(),\n  };\n};\nexport const generateAiMsgItem = (content: string, showVoiceBtn = false): AiChatMessageItem => {\n  return {\n    id: generateRandomId(),\n    role: 'ai',\n    contents: [{ text: content }],\n    createTime: new Date().toLocaleString(),\n    showVoiceBtn,\n  };\n};\n"],"names":[],"mappings":";;AAIa,MAAA,sBAAsB,CAAC,IAAY,YAAuC;AAC9E,SAAA;AAAA,IACL;AAAA,IACA,UAAU,CAAC,EAAE,MAAM,SAAS;AAAA,IAC5B,MAAM;AAAA,IACN,YAAY,MAAM,EAAE,SAAS;AAAA,IAC7B,iBAAiB,MAAM,EAAE,KAAK;AAAA,EAChC;AACF;AACO,MAAM,oBAAoB,CAAC,SAAiB,eAAe,UAA6B;AACtF,SAAA;AAAA,IACL,IAAI,iBAAiB;AAAA,IACrB,MAAM;AAAA,IACN,UAAU,CAAC,EAAE,MAAM,SAAS;AAAA,IAC5B,aAAY,oBAAI,KAAK,GAAE,eAAe;AAAA,IACtC;AAAA,EACF;AACF;"}

package/dist/lib/hooks/use-div-auto-scroll.es.js

@@ -0,0 +1,51 @@
+import { useState, useRef, useCallback, useEffect } from "react";
+function useDivAutoScroll(params) {
+  const { scrollBuffer = 50, scrollBehavior = "smooth", containerRef } = params;
+  const [isAutoScroll, setIsAutoScroll] = useState(true);
+  const isUserScrolling = useRef(false);
+  const scrollTimeout = useRef();
+  const isNearBottom = useCallback(() => {
+    if (!containerRef.current) return false;
+    const { scrollHeight, scrollTop, clientHeight } = containerRef.current;
+    return scrollHeight - scrollTop - clientHeight < scrollBuffer;
+  }, [containerRef, scrollBuffer]);
+  const autoScroll = useCallback(() => {
+    if (isAutoScroll && containerRef.current) {
+      containerRef.current.scrollTo({
+        top: containerRef.current.scrollHeight,
+        behavior: scrollBehavior
+      });
+    }
+  }, [containerRef, isAutoScroll, scrollBehavior]);
+  const handleScroll = useCallback(() => {
+    if (!containerRef.current) return;
+    if (scrollTimeout.current) clearTimeout(scrollTimeout.current);
+    if (!isUserScrolling.current) {
+      isUserScrolling.current = true;
+    }
+    scrollTimeout.current = setTimeout(() => {
+      isUserScrolling.current = false;
+    }, 100);
+    if (isNearBottom()) {
+      setIsAutoScroll(true);
+    } else {
+      setIsAutoScroll(false);
+    }
+  }, [containerRef, isNearBottom]);
+  useEffect(() => {
+    const container = containerRef.current;
+    if (!container) return;
+    container.addEventListener("scroll", handleScroll);
+    return () => container.removeEventListener("scroll", handleScroll);
+  }, [containerRef, handleScroll]);
+  const triggerAutoScroll = useCallback(() => {
+    if (!isUserScrolling.current) {
+      autoScroll();
+    }
+  }, [autoScroll, isUserScrolling]);
+  return triggerAutoScroll;
+}
+export {
+  useDivAutoScroll
+};
+//# sourceMappingURL=use-div-auto-scroll.es.js.map

package/dist/lib/hooks/use-div-auto-scroll.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-div-auto-scroll.es.js","sources":["../../../src/hooks/use-div-auto-scroll.ts"],"sourcesContent":["import React, { useCallback, useEffect, useRef, useState } from 'react';\n\ninterface ScrollControllerProps {\n  scrollBuffer?: number;\n  scrollBehavior?: 'auto' | 'smooth';\n  containerRef: React.RefObject<HTMLDivElement | null>;\n}\n\n/**\n * 自动滚动 hook，用法是你传入一个容器的 ref，当你调用返回函数时，那么容器内的内容变化时，会自动滚动到最底部。\n * @param params 传入一个对象，包含以下属性：\n * scrollBuffer：滚动缓冲距离，默认为 50px。\n * scrollBehavior：滚动行为，默认为 'smooth'。\n * containerRef：容器的 React ref 引用对象。\n * @returns 一个触发自动滚动的函数\n */\nexport function useDivAutoScroll(params: ScrollControllerProps) {\n  const { scrollBuffer = 50, scrollBehavior = 'smooth', containerRef } = params;\n  const [isAutoScroll, setIsAutoScroll] = useState(true);\n  const isUserScrolling = useRef(false);\n  const scrollTimeout = useRef<number>();\n\n  // 检测是否接近底部\n  const isNearBottom = useCallback(() => {\n    if (!containerRef.current) return false;\n    const { scrollHeight, scrollTop, clientHeight } = containerRef.current;\n    return scrollHeight - scrollTop - clientHeight < scrollBuffer;\n  }, [containerRef, scrollBuffer]);\n\n  // 自动滚动逻辑\n  const autoScroll = useCallback(() => {\n    if (isAutoScroll && containerRef.current) {\n      containerRef.current.scrollTo({\n        top: containerRef.current.scrollHeight,\n        behavior: scrollBehavior,\n      });\n    }\n  }, [containerRef, isAutoScroll, scrollBehavior]);\n\n  // 处理滚动事件\n  const handleScroll = useCallback(() => {\n    if (!containerRef.current) return;\n\n    // 清除之前的定时器\n    if (scrollTimeout.current) clearTimeout(scrollTimeout.current);\n\n    // 标记用户滚动行为\n    if (!isUserScrolling.current) {\n      isUserScrolling.current = true;\n    }\n\n    // 设置滚动状态重置\n    scrollTimeout.current = setTimeout(() => {\n      isUserScrolling.current = false;\n    }, 100);\n\n    // 判断是否接近底部\n    if (isNearBottom()) {\n      setIsAutoScroll(true);\n    } else {\n      setIsAutoScroll(false);\n    }\n  }, [containerRef, isNearBottom]);\n\n  // 初始化事件监听\n  useEffect(() => {\n    const container = containerRef.current;\n    if (!container) return;\n\n    container.addEventListener('scroll', handleScroll);\n    return () => container.removeEventListener('scroll', handleScroll);\n  }, [containerRef, handleScroll]);\n\n  // 内容变化时自动滚动\n  const triggerAutoScroll = useCallback(() => {\n    if (!isUserScrolling.current) {\n      autoScroll();\n    }\n  }, [autoScroll, isUserScrolling]);\n\n  return triggerAutoScroll;\n}\n"],"names":[],"mappings":";AAgBO,SAAS,iBAAiB,QAA+B;AAC9D,QAAM,EAAE,eAAe,IAAI,iBAAiB,UAAU,iBAAiB;AACvE,QAAM,CAAC,cAAc,eAAe,IAAI,SAAS,IAAI;AAC/C,QAAA,kBAAkB,OAAO,KAAK;AACpC,QAAM,gBAAgB,OAAe;AAG/B,QAAA,eAAe,YAAY,MAAM;AACjC,QAAA,CAAC,aAAa,QAAgB,QAAA;AAClC,UAAM,EAAE,cAAc,WAAW,iBAAiB,aAAa;AACxD,WAAA,eAAe,YAAY,eAAe;AAAA,EAAA,GAChD,CAAC,cAAc,YAAY,CAAC;AAGzB,QAAA,aAAa,YAAY,MAAM;AAC/B,QAAA,gBAAgB,aAAa,SAAS;AACxC,mBAAa,QAAQ,SAAS;AAAA,QAC5B,KAAK,aAAa,QAAQ;AAAA,QAC1B,UAAU;AAAA,MAAA,CACX;AAAA,IAAA;AAAA,EAEF,GAAA,CAAC,cAAc,cAAc,cAAc,CAAC;AAGzC,QAAA,eAAe,YAAY,MAAM;AACjC,QAAA,CAAC,aAAa,QAAS;AAG3B,QAAI,cAAc,QAAsB,cAAA,cAAc,OAAO;AAGzD,QAAA,CAAC,gBAAgB,SAAS;AAC5B,sBAAgB,UAAU;AAAA,IAAA;AAId,kBAAA,UAAU,WAAW,MAAM;AACvC,sBAAgB,UAAU;AAAA,OACzB,GAAG;AAGN,QAAI,gBAAgB;AAClB,sBAAgB,IAAI;AAAA,IAAA,OACf;AACL,sBAAgB,KAAK;AAAA,IAAA;AAAA,EACvB,GACC,CAAC,cAAc,YAAY,CAAC;AAG/B,YAAU,MAAM;AACd,UAAM,YAAY,aAAa;AAC/B,QAAI,CAAC,UAAW;AAEN,cAAA,iBAAiB,UAAU,YAAY;AACjD,WAAO,MAAM,UAAU,oBAAoB,UAAU,YAAY;AAAA,EAAA,GAChE,CAAC,cAAc,YAAY,CAAC;AAGzB,QAAA,oBAAoB,YAAY,MAAM;AACtC,QAAA,CAAC,gBAAgB,SAAS;AACjB,iBAAA;AAAA,IAAA;AAAA,EACb,GACC,CAAC,YAAY,eAAe,CAAC;AAEzB,SAAA;AACT;"}

package/dist/lib/hooks/use-document-unmount/use-document-unmount.es.js

@@ -0,0 +1,18 @@
+import { useEffect } from "react";
+import useMemoizedFn from "../../node_modules/.pnpm/ahooks@3.8.5_react@18.2.0/node_modules/ahooks/es/useMemoizedFn/index.es.js";
+function useDocumentUnmount(fn) {
+  const finalFn = useMemoizedFn(fn);
+  useEffect(() => {
+    const handleBeforeUnload = () => {
+      finalFn();
+    };
+    window.addEventListener("beforeunload", handleBeforeUnload);
+    return () => {
+      window.removeEventListener("beforeunload", handleBeforeUnload);
+    };
+  }, [finalFn]);
+}
+export {
+  useDocumentUnmount
+};
+//# sourceMappingURL=use-document-unmount.es.js.map