fix(chat): keep the draft when the plan gate refuses a send

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/home/components/user-input/user-input.tsx

@@ -35,6 +35,7 @@ import { InputToolbar } from '@/app/workspace/[workspaceId]/home/components/user
 import type { ModelSelectorOptions } from '@/app/workspace/[workspaceId]/home/components/user-input/components/model-selector'
 import { usePromptHistory } from '@/app/workspace/[workspaceId]/home/components/user-input/hooks/use-prompt-history'
 import { handleMothershipAddContextEvent } from '@/app/workspace/[workspaceId]/home/components/user-input/mothership-context-event'
+import { useChatPlanGate } from '@/app/workspace/[workspaceId]/home/hooks/use-chat-plan-gate'
 import type {
   ChatMessage,
   FileAttachmentForApi,
@@ -507,6 +508,9 @@ const UserInputImpl = forwardRef<UserInputHandle, UserInputProps>(function UserI
     textareaRef.current?.focus()
   }
 
+  /** Refuses a send the workspace plan excludes here, before `onSubmit`, so the draft survives. */
+  const stopSendWithoutChatPlan = useChatPlanGate()
+
   /** Empties the text, chips, attachments, transcript, and the saved draft in one step. */
   const clearComposer = useCallback(() => {
     history.reset()
@@ -527,6 +531,7 @@ const UserInputImpl = forwardRef<UserInputHandle, UserInputProps>(function UserI
   }, [resetTranscript, history.reset])
 
   const handleSubmit = useCallback(() => {
+    if (stopSendWithoutChatPlan()) return
     const currentFiles = filesRef.current
     const currentEditor = editorRef.current
 
@@ -552,7 +557,7 @@ const UserInputImpl = forwardRef<UserInputHandle, UserInputProps>(function UserI
     )
     if (accepted === false) return
     clearComposer()
-  }, [onSubmit, clearComposer])
+  }, [stopSendWithoutChatPlan, onSubmit, clearComposer])
 
   /**
    * Enter policy for the editor: mirror canSubmit's uploading guard (Enter

apps/sim/app/workspace/[workspaceId]/home/hooks/use-chat-plan-gate.ts

@@ -1,7 +1,7 @@
 'use client'
 
 import { useCallback } from 'react'
-import { useToast } from '@sim/emcn'
+import { toast } from '@sim/emcn'
 import { useRouter } from 'next/navigation'
 import { useSession } from '@/lib/auth/client'
 import { CHAT_PLAN_REQUIRED_MESSAGE } from '@/lib/billing/chat-plan'
@@ -25,7 +25,6 @@ export function useChatPlanGate(): () => boolean {
   const { data: session } = useSession()
   const viewerId = session?.user?.id
   const router = useRouter()
-  const { toast } = useToast()
 
   return useCallback(() => {
     if (!hostContext || hostContext.features?.chat !== false) return false
@@ -40,5 +39,5 @@ export function useChatPlanGate(): () => boolean {
       )
     }
     return true
-  }, [hostContext, viewerId, router, toast])
+  }, [hostContext, viewerId, router])
 }