export function registerBeforeCombinePromptsController(runtime, listener) { const makeFirst = runtime.getEventMakeFirst(); if (typeof makeFirst === "function") { return makeFirst( runtime.eventTypes.GENERATE_BEFORE_COMBINE_PROMPTS, listener, ); } runtime.console.warn("[ST-BME] eventMakeFirst 不可用,回退到普通事件注册"); runtime.eventSource.on(runtime.eventTypes.GENERATE_BEFORE_COMBINE_PROMPTS, listener); return null; } export function registerGenerationAfterCommandsController(runtime, listener) { const makeFirst = runtime.getEventMakeFirst(); if (typeof makeFirst === "function") { return makeFirst(runtime.eventTypes.GENERATION_AFTER_COMMANDS, listener); } runtime.console.warn( "[ST-BME] eventMakeFirst 不可用,GENERATION_AFTER_COMMANDS 回退到普通事件注册", ); runtime.eventSource.on(runtime.eventTypes.GENERATION_AFTER_COMMANDS, listener); return null; } export function scheduleSendIntentHookRetryController(runtime, delayMs = 400) { runtime.clearTimeout(runtime.getSendIntentHookRetryTimer()); const timer = runtime.setTimeout(() => { runtime.setSendIntentHookRetryTimer(null); runtime.installSendIntentHooks(); }, delayMs); runtime.setSendIntentHookRetryTimer(timer); } export function installSendIntentHooksController(runtime) { for (const cleanup of runtime.consumeSendIntentHookCleanup()) { try { cleanup(); } catch (error) { runtime.console.warn("[ST-BME] 清理发送意图钩子失败:", error); } } const sendButton = runtime.document.getElementById("send_but"); const sendTextarea = runtime.document.getElementById("send_textarea"); if (sendButton) { const captureSendIntent = () => { runtime.recordRecallSendIntent(runtime.getSendTextareaValue(), "send-button"); }; sendButton.addEventListener("click", captureSendIntent, true); sendButton.addEventListener("pointerup", captureSendIntent, true); sendButton.addEventListener("touchend", captureSendIntent, true); runtime.pushSendIntentHookCleanup(() => { sendButton.removeEventListener("click", captureSendIntent, true); sendButton.removeEventListener("pointerup", captureSendIntent, true); sendButton.removeEventListener("touchend", captureSendIntent, true); }); } if (sendTextarea) { const captureEnterIntent = (event) => { if ( (event.key === "Enter" || event.key === "NumpadEnter") && !event.shiftKey ) { runtime.recordRecallSendIntent( runtime.getSendTextareaValue(), "textarea-enter", ); } }; sendTextarea.addEventListener("keydown", captureEnterIntent, true); runtime.pushSendIntentHookCleanup(() => { sendTextarea.removeEventListener("keydown", captureEnterIntent, true); }); } if (!sendButton || !sendTextarea) { runtime.scheduleSendIntentHookRetry(); } } export function registerCoreEventHooksController(runtime) { const { eventSource, eventTypes, handlers } = runtime; eventSource.on(eventTypes.CHAT_CHANGED, handlers.onChatChanged); if (eventTypes.CHAT_LOADED) { eventSource.on(eventTypes.CHAT_LOADED, handlers.onChatLoaded); } if (eventTypes.MESSAGE_SENT) { eventSource.on(eventTypes.MESSAGE_SENT, handlers.onMessageSent); } runtime.registerGenerationAfterCommands(handlers.onGenerationAfterCommands); runtime.registerBeforeCombinePrompts(handlers.onBeforeCombinePrompts); eventSource.on(eventTypes.MESSAGE_RECEIVED, handlers.onMessageReceived); eventSource.on(eventTypes.MESSAGE_DELETED, handlers.onMessageDeleted); eventSource.on(eventTypes.MESSAGE_EDITED, handlers.onMessageEdited); eventSource.on(eventTypes.MESSAGE_SWIPED, handlers.onMessageSwiped); if (eventTypes.MESSAGE_UPDATED) { eventSource.on(eventTypes.MESSAGE_UPDATED, handlers.onMessageEdited); } }