@athenaintel/react 0.57.2 → 0.59.0

package/dist/chat-ui/composer/tiptap-composer.js

@@ -17,11 +17,17 @@ import { MentionSuggestionList } from "../mentions/suggestions/components/mentio
 import { useMentionSuggestions } from "../mentions/suggestions/data/use-mention-suggestions.js";
 import { Selection } from "@tiptap/pm/state";
 import { attachedMessageBehindWaiting, attachedMessageNeedsConnection } from "../../agent-runtime/src/offline-status.js";
+import { readDeepAgentNamedAgents, deepAgentOutboxAddressedAgents, restoreDeepAgentSentNamedAgents } from "../../agent-runtime/src/named-agents.js";
 import { captureComposerSend, prepareComposerSend, ComposerDraftWhitespace, draftForReturnedSend, serializeComposerDraft } from "./composer-message.js";
+import { stageComposerNamedAgents } from "./composer-named-agents.js";
 import { PendingAttachmentUploads } from "./pending-attachment-uploads.js";
 import { runInputRulesOnEnter } from "./run-input-rules-on-enter.js";
+import { matchSlashCommands } from "./slash-commands.js";
+import { slashCommandRuns, SlashCommandMenu } from "./slash-command-menu.js";
+import { SlashCommandsExtension } from "./slash-commands-extension.js";
 import { useDeepAgentChatOutboxContext } from "../../agent-runtime/src/use-deep-agent-chat-outbox.js";
 import { withDeepAgentComposerMarkdown } from "../../agent-runtime/src/composer-markdown.js";
+import { autoComposerSendOptions } from "../../agent-runtime/src/run-send.js";
 import { readAthenaAssetDrop, isAthenaAssetDrag } from "../../agent-runtime/src/athena-asset-drag.js";
 import { registerChatAssetMentionHandler } from "../../agent-runtime/src/asset-attachment.js";
 import { cn } from "cnfast";
@@ -119,6 +125,7 @@ function chatOutboxRead(outbox) {
 }
 const CHAT_OUTBOX_READ_WAIT_MS = 2e3;
 const NO_WEBPAGES = [];
+const NO_SLASH_COMMANDS = [];
 function revokeImagePreview(src) {
   if (typeof URL.revokeObjectURL === "function") URL.revokeObjectURL(src);
 }
@@ -213,6 +220,8 @@ const TiptapComposer = forwardRef(
     webpages = NO_WEBPAGES,
     tools,
     rootCategories,
+    chatAgents,
+    viewerIsAdmin = false,
     onDraftStateChange,
     showImagePreviews = false,
     attachAnyFile = false,
@@ -220,9 +229,13 @@ const TiptapComposer = forwardRef(
     fileIntake,
     waitForHostUploads,
     interceptSubmit,
+    beforeSubmit,
     hasPendingExtras,
     holdPendingExtras,
-    onSubmitted
+    prepareSend,
+    onSendReturned,
+    onSubmitted,
+    slashCommands = NO_SLASH_COMMANDS
   }, outerRef) => {
     const aui = useAui();
     const outbox = useDeepAgentChatOutboxContext();
@@ -286,12 +299,24 @@ const TiptapComposer = forwardRef(
       webpages,
       tools,
       rootCategories,
-      athenaUnavailable: mentionSearchUnavailable
+      athenaUnavailable: mentionSearchUnavailable,
+      chatAgents,
+      viewerIsAdmin
     });
+    const storeRef = useRef(store);
+    storeRef.current = store;
+    const namesAgentsRef = useRef(chatAgents !== void 0);
+    namesAgentsRef.current = chatAgents !== void 0;
+    const onSendReturnedRef = useRef(onSendReturned);
+    onSendReturnedRef.current = onSendReturned;
     const [popupOpen, setPopupOpen] = useState(false);
     const popupOpenRef = useRef(false);
     const [popupProps, setPopupProps] = useState(null);
     const callbacksRef = useRef({
+      nameAgents: (items) => {
+        var _a, _b;
+        return ((_b = (_a = storeRef.current).nameAgents) == null ? void 0 : _b.call(_a, items)) ?? false;
+      },
       onStart: (props) => {
         setPopupOpen(true);
         popupOpenRef.current = true;
@@ -310,6 +335,60 @@ const TiptapComposer = forwardRef(
         setPopupProps(null);
       }
     });
+    const slashCommandsRef = useRef(slashCommands);
+    slashCommandsRef.current = slashCommands;
+    const slashMenuRef = useRef(null);
+    const [slashMenu, setSlashMenu] = useState(null);
+    const slashQueryRef = useRef(null);
+    const slashCallbacksRef = useRef({
+      onStart: (props) => {
+        slashQueryRef.current = props.query;
+        setSlashMenu(props);
+      },
+      onUpdate: (props) => {
+        slashQueryRef.current = props.query;
+        setSlashMenu(props);
+      },
+      onKeyDown: (event) => {
+        var _a;
+        return ((_a = slashMenuRef.current) == null ? void 0 : _a.onKeyDown(event)) ?? false;
+      },
+      onExit: () => {
+        slashQueryRef.current = null;
+        setSlashMenu(null);
+      }
+    });
+    const slashMenuListsCommands = useCallback(() => {
+      const query = slashQueryRef.current;
+      return query !== null && matchSlashCommands({ commands: slashCommandsRef.current, query }).length > 0;
+    }, []);
+    const runSlashCommand = useCallback(
+      ({ editor: commandEditor, range, match }) => {
+        if (!slashCommandRuns(match.command) || commandEditor.isDestroyed) return;
+        commandEditor.chain().focus().deleteRange(range).run();
+        const report = (error) => {
+          athenaDiagnostics.error(
+            "composer.slash_command",
+            error,
+            { code: "slash_command_failed", message: "A composer command failed" },
+            { command: match.command.name }
+          );
+        };
+        try {
+          const ran = match.command.run({
+            args: match.args,
+            insertText: (text) => {
+              if (!text || commandEditor.isDestroyed) return;
+              commandEditor.chain().focus().insertContent({ type: "text", text }).run();
+            }
+          });
+          if (ran) void Promise.resolve(ran).catch(report);
+        } catch (error) {
+          report(error);
+        }
+      },
+      []
+    );
     const editorRef = useRef(null);
     const handleSubmit = useCallback(async () => {
       var _a, _b;
@@ -325,6 +404,8 @@ const TiptapComposer = forwardRef(
         const ed = editorRef.current;
         if (!ed) return;
         const { text, draftMarkdown } = captureComposerSend(ed);
+        const namesAgents = namesAgentsRef.current;
+        const namedAgents = namesAgents ? readDeepAgentNamedAgents(threadId) : [];
         const submittedDocument = ed.state.doc;
         const consumeSubmittedDraft = () => {
           if (ed.state.doc.eq(submittedDocument)) {
@@ -357,11 +438,13 @@ const TiptapComposer = forwardRef(
             threadId,
             assetIdsInDraft: assetMentionIds(ed)
           });
+          const addressedAgents = deepAgentOutboxAddressedAgents(namedAgents);
           if (hostExtras && holdPendingExtras) {
             if (!holdPendingExtras({
               outbox: waitingOutbox,
               send: { text, draftMarkdown },
-              attachments
+              attachments,
+              ...addressedAgents && { addressedAgents }
             })) {
               return;
             }
@@ -374,12 +457,31 @@ const TiptapComposer = forwardRef(
                 text,
                 markdown: draftMarkdown
               }),
-              ...attachments.length > 0 && { attachments }
+              ...attachments.length > 0 && { attachments },
+              ...addressedAgents && { addressedAgents }
             });
           }
           consumeSubmittedDraft();
           return;
         }
+        const goesNow = Boolean(text) || hostExtras && !runtimeAttachments && !!interceptSubmit;
+        if (beforeSubmit && goesNow) {
+          interceptionPendingRef.current = true;
+          setEditorEditable(ed);
+          try {
+            await beforeSubmit({ text });
+          } catch (error) {
+            athenaDiagnostics.emit("composer.before_send.rejected", {
+              level: "warn",
+              data: { error: error instanceof Error ? error.message : String(error) }
+            });
+            return;
+          } finally {
+            interceptionPendingRef.current = false;
+            if (!ed.isDestroyed) setEditorEditable(ed);
+          }
+          if (ed.isDestroyed) return;
+        }
         if (interceptSubmit) {
           const hasAttachments = (((_b = aui.composer.getState().attachments) == null ? void 0 : _b.length) ?? 0) > 0;
           if (!hasAttachments) {
@@ -402,13 +504,18 @@ const TiptapComposer = forwardRef(
         if (!text) return;
         if (!composerInput.canSend) return;
         prepareComposerSend({ threadId, text, draftMarkdown });
+        if (namesAgents) stageComposerNamedAgents({ threadId, text, agents: namedAgents });
+        const prepared = prepareSend == null ? void 0 : prepareSend({ text });
         composerInput.setText(text);
-        composerInput.send({ steer: false });
+        composerInput.send(
+          autoComposerSendOptions(aui.thread().getState().isRunning && !(prepared == null ? void 0 : prepared.queue))
+        );
         consumeSubmittedDraft();
       } finally {
         submitInFlightRef.current = false;
       }
     }, [
+      beforeSubmit,
       clearImagePreviews,
       composerInput,
       disabled,
@@ -419,6 +526,7 @@ const TiptapComposer = forwardRef(
       onAttachmentError,
       onDraftStateChange,
       onSubmitted,
+      prepareSend,
       setEditorEditable,
       threadId,
       waitForHostUploads
@@ -574,9 +682,14 @@ const TiptapComposer = forwardRef(
           store,
           callbacks: callbacksRef.current
         }),
+        SlashCommandsExtension.configure({
+          hasCommands: () => slashCommandsRef.current.length > 0,
+          callbacks: slashCallbacksRef.current,
+          onPick: runSlashCommand
+        }),
         SubmitKeybind.configure({
           onSubmit: () => void handleSubmitRef.current(),
-          isMentionOpen: () => popupOpenRef.current
+          isMentionOpen: () => popupOpenRef.current || slashMenuListsCommands()
         })
       ],
       editorProps: {
@@ -629,12 +742,17 @@ const TiptapComposer = forwardRef(
       if (restored === currentMarkdown) return;
       let cancelled = false;
       queueMicrotask(() => {
+        var _a;
         if (cancelled || editor.isDestroyed) return;
         applyingRuntimeTextRef.current = true;
         try {
           editor.commands.setContent(restored, { emitUpdate: false });
           lastEmittedMarkdownRef.current = serializeDraft(editor);
-          if (returnedDraft !== null) restoredSendRef.current = value;
+          if (returnedDraft !== null) {
+            restoredSendRef.current = value;
+            if (namesAgentsRef.current) restoreDeepAgentSentNamedAgents({ threadId, text: value });
+            (_a = onSendReturnedRef.current) == null ? void 0 : _a.call(onSendReturnedRef, { text: value });
+          }
         } finally {
           applyingRuntimeTextRef.current = false;
         }
@@ -748,6 +866,10 @@ const TiptapComposer = forwardRef(
       const currentEditor = editorRef.current;
       return currentEditor && !currentEditor.isDestroyed ? captureComposerSend(currentEditor) : null;
     }, []);
+    const focusEditor = useCallback(() => {
+      const currentEditor = editorRef.current;
+      if (currentEditor && !currentEditor.isDestroyed) currentEditor.commands.focus("end");
+    }, []);
     useImperativeHandle(
       outerRef,
       () => ({
@@ -756,17 +878,20 @@ const TiptapComposer = forwardRef(
         insertAssetMention,
         openMentionMenu,
         appendText,
-        captureSend
+        captureSend,
+        focus: focusEditor
       }),
       [
         addAttachment,
         appendText,
         captureSend,
+        focusEditor,
         handleSubmit,
         insertAssetMention,
         openMentionMenu
       ]
     );
+    const slashSections = slashMenu ? matchSlashCommands({ commands: slashCommands, query: slashMenu.query }) : [];
     return /* @__PURE__ */ jsxs(Fragment, { children: [
       /* @__PURE__ */ jsx("style", { children: MENTION_STYLES }),
       /* @__PURE__ */ jsxs(
@@ -812,7 +937,8 @@ const TiptapComposer = forwardRef(
             editor.commands.focus();
           }
         }
-      ) })
+      ) }),
+      slashMenu && editor && slashSections.length > 0 && /* @__PURE__ */ jsx(MentionSuggestionPopup, { clientRect: slashMenu.clientRect, children: /* @__PURE__ */ jsx(SlashCommandMenu, { ref: slashMenuRef, sections: slashSections, onPick: slashMenu.pick }) })
     ] });
   }
 );