@agent-native/toolkit 0.198.4 → 0.198.5

package/src/composer/TiptapComposer.tsx

@@ -888,6 +888,8 @@ export interface TiptapComposerProps {
   contextControlsDisabled?: boolean;
   /** Prevent submission without making the editable surface lose focus. */
   submissionDisabled?: boolean;
+  /** Disable only the send control while the submission is being accepted. */
+  sendButtonDisabled?: boolean;
   /** Prevent submission while a host request is in flight. */
   submitting?: boolean;
   /** Override the generic document attachment cap for a multipart host. */
@@ -916,6 +918,7 @@ export interface TiptapComposerProps {
   ) => void | Promise<void>;
   /** Return false to stop a submit before it enters the chat runtime. */
   onBeforeSubmit?: () => boolean | Promise<boolean>;
+  onSubmissionPendingChange?: (pending: boolean) => void;
   /** Scope where a failed submission should be recovered after the host forks. */
   getSubmitFailureDraftScope?: () => string | null;
   /**
@@ -2547,6 +2550,7 @@ export function TiptapComposer({
   disabled = false,
   contextControlsDisabled = false,
   submissionDisabled = false,
+  sendButtonDisabled = false,
   submitting = false,
   maxDocumentAttachmentBytes = MAX_DOCUMENT_ATTACHMENT_BYTES,
   documentAttachmentLimitLabel = "PDFs",
@@ -2558,6 +2562,7 @@ export function TiptapComposer({
   initialTextKey,
   onSubmit,
   onBeforeSubmit,
+  onSubmissionPendingChange,
   getSubmitFailureDraftScope,
   clearOnSubmit = true,
   clearOnSubmitImmediately = false,
@@ -4243,7 +4248,6 @@ export function TiptapComposer({
             slotReferencesRef.current.length > 0);
         const canPreserveFollowUp =
           preserveFollowUp &&
-          clearedBeforePreflight &&
           !isForkRecovery &&
           hasFollowUp &&
           (currentDraft === null ||
@@ -4330,6 +4334,7 @@ export function TiptapComposer({
 
       if (onBeforeSubmit) {
         submitInFlightRef.current = true;
+        onSubmissionPendingChange?.(true);
         try {
           const shouldSubmit = await onBeforeSubmit();
           if (!shouldSubmit) {
@@ -4351,6 +4356,7 @@ export function TiptapComposer({
           return false;
         } finally {
           submitInFlightRef.current = false;
+          onSubmissionPendingChange?.(false);
         }
       }
       if (
@@ -4498,6 +4504,7 @@ export function TiptapComposer({
         submitInFlightRef.current = true;
         let locallySubmitted = false;
         let settled = false;
+        onSubmissionPendingChange?.(true);
         const clearSubmittedComposer = () => {
           if (clearOnSubmit) {
             clearComposerDraft(submittingDraftKey, submittingDraftSnapshot);
@@ -4575,7 +4582,7 @@ export function TiptapComposer({
           );
         } catch (error) {
           if (locallySubmitted) {
-            restoreSubmittedDraft();
+            restoreSubmittedDraft(true);
             return true;
           }
           restoreSubmittedDraft(true);
@@ -4593,6 +4600,7 @@ export function TiptapComposer({
         } finally {
           settled = true;
           submitInFlightRef.current = false;
+          onSubmissionPendingChange?.(false);
         }
 
         if (!isCurrentDraftScope()) {
@@ -4685,6 +4693,7 @@ export function TiptapComposer({
       clearOnSubmitImmediately,
       getSubmitFailureDraftScope,
       onBeforeSubmit,
+      onSubmissionPendingChange,
       extractComposerPayload,
       syncComposerState,
       updateSlotReferences,
@@ -5391,7 +5400,7 @@ export function TiptapComposer({
                   <button
                     type="button"
                     onClick={() => void submitComposer("immediate")}
-                    disabled={!canSend}
+                    disabled={!canSend || sendButtonDisabled}
                     aria-label={
                       submitting ? t("common.loading") : sendButtonTooltip
                     }