Merge branch 'feat/mothership-copilot' of github.com:simstudioai/sim into feat/mothership-copilot

Author: icecrasher321

apps/sim/app/workspace/[workspaceId]/home/components/mothership-view/mothership-view.tsx

@@ -1,14 +1,90 @@
 'use client'
 
-export function MothershipView() {
+import { useMemo } from 'react'
+import { Skeleton } from '@/components/emcn'
+import { cn } from '@/lib/core/utils/cn'
+import { FileViewer } from '@/app/workspace/[workspaceId]/files/components/file-viewer'
+import type { MothershipResource } from '@/app/workspace/[workspaceId]/home/types'
+import { Table } from '@/app/workspace/[workspaceId]/tables/[tableId]/components'
+import { useWorkspaceFiles } from '@/hooks/queries/workspace-files'
+
+interface MothershipViewProps {
+  workspaceId: string
+  resources: MothershipResource[]
+  activeResourceId: string | null
+  onSelectResource: (id: string) => void
+}
+
+export function MothershipView({
+  workspaceId,
+  resources,
+  activeResourceId,
+  onSelectResource,
+}: MothershipViewProps) {
+  const active = resources.find((r) => r.id === activeResourceId) ?? resources[0] ?? null
+
   return (
-    <div className='flex h-full w-[480px] flex-shrink-0 flex-col border-[var(--border)] border-l'>
-      <div className='flex items-center border-[var(--border)] border-b px-[16px] py-[12px]'>
-        <span className='font-medium text-[13px] text-[var(--text-secondary)]'>Mothership</span>
+    <div className='flex h-full w-[50%] min-w-[400px] flex-col border-[var(--border)] border-l'>
+      <div className='flex shrink-0 gap-[2px] overflow-x-auto border-[var(--border)] border-b px-[12px]'>
+        {resources.map((r) => (
+          <button
+            key={r.id}
+            type='button'
+            onClick={() => onSelectResource(r.id)}
+            className={cn(
+              'shrink-0 cursor-pointer border-b-[2px] px-[12px] py-[10px] text-[13px] transition-colors',
+              active?.id === r.id
+                ? 'border-[var(--text-primary)] font-medium text-[var(--text-primary)]'
+                : 'border-transparent text-[var(--text-secondary)] hover:text-[var(--text-primary)]'
+            )}
+          >
+            {r.title}
+          </button>
+        ))}
       </div>
-      <div className='flex flex-1 items-center justify-center'>
-        <span className='text-[13px] text-[var(--text-muted)]'>No artifacts yet</span>
+
+      <div className='min-h-0 flex-1 overflow-hidden'>
+        {active?.type === 'table' && (
+          <Table key={active.id} workspaceId={workspaceId} tableId={active.id} embedded />
+        )}
+        {active?.type === 'file' && (
+          <EmbeddedFile key={active.id} workspaceId={workspaceId} fileId={active.id} />
+        )}
       </div>
     </div>
   )
 }
+
+interface EmbeddedFileProps {
+  workspaceId: string
+  fileId: string
+}
+
+function EmbeddedFile({ workspaceId, fileId }: EmbeddedFileProps) {
+  const { data: files = [], isLoading } = useWorkspaceFiles(workspaceId)
+  const file = useMemo(() => files.find((f) => f.id === fileId), [files, fileId])
+
+  if (isLoading) {
+    return (
+      <div className='flex h-full flex-col gap-[8px] p-[24px]'>
+        <Skeleton className='h-[16px] w-[60%]' />
+        <Skeleton className='h-[16px] w-[80%]' />
+        <Skeleton className='h-[16px] w-[40%]' />
+      </div>
+    )
+  }
+
+  if (!file) {
+    return (
+      <div className='flex h-full items-center justify-center'>
+        <span className='text-[13px] text-[var(--text-muted)]'>File not found</span>
+      </div>
+    )
+  }
+
+  return (
+    <div className='flex h-full flex-col overflow-hidden'>
+      <FileViewer key={file.id} file={file} workspaceId={workspaceId} canEdit={true} />
+    </div>
+  )
+}

apps/sim/app/workspace/[workspaceId]/home/home.tsx

@@ -4,7 +4,7 @@ import { useCallback, useEffect, useRef, useState } from 'react'
 import { createLogger } from '@sim/logger'
 import { useParams } from 'next/navigation'
 import { LandingPromptStorage } from '@/lib/core/utils/browser-storage'
-import { MessageContent, UserInput } from './components'
+import { MessageContent, MothershipView, UserInput } from './components'
 import { useChat } from './hooks'
 
 const logger = createLogger('Home')
@@ -28,10 +28,17 @@ export function Home({ chatId }: HomeProps = {}) {
       setInputValue(prompt)
     }
   }, [])
-  const { messages, isSending, sendMessage, stopGeneration, chatBottomRef } = useChat(
-    workspaceId,
-    chatId
-  )
+
+  const {
+    messages,
+    isSending,
+    sendMessage,
+    stopGeneration,
+    chatBottomRef,
+    resources,
+    activeResourceId,
+    setActiveResourceId,
+  } = useChat(workspaceId, chatId)
 
   const handleSubmit = useCallback(() => {
     const trimmed = inputValue.trim()
@@ -118,6 +125,15 @@ export function Home({ chatId }: HomeProps = {}) {
           />
         </div>
       </div>
+
+      {resources.length > 0 && (
+        <MothershipView
+          workspaceId={workspaceId}
+          resources={resources}
+          activeResourceId={activeResourceId}
+          onSelectResource={setActiveResourceId}
+        />
+      )}
     </div>
   )
 }

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

@@ -1,18 +1,22 @@
 import { useCallback, useEffect, useRef, useState } from 'react'
+import { createLogger } from '@sim/logger'
 import { useQueryClient } from '@tanstack/react-query'
 import { usePathname, useRouter } from 'next/navigation'
 import { MOTHERSHIP_CHAT_API_PATH } from '@/lib/copilot/constants'
+import { tableKeys } from '@/hooks/queries/tables'
 import {
   type TaskChatHistory,
   type TaskStoredContentBlock,
   type TaskStoredMessage,
   taskKeys,
   useChatHistory,
 } from '@/hooks/queries/tasks'
+import { workspaceFilesKeys } from '@/hooks/queries/workspace-files'
 import type {
   ChatMessage,
   ContentBlock,
   ContentBlockType,
+  MothershipResource,
   SSEPayload,
   SSEPayloadData,
   ToolCallStatus,
@@ -26,6 +30,9 @@ export interface UseChatReturn {
   sendMessage: (message: string) => Promise<void>
   stopGeneration: () => void
   chatBottomRef: React.RefObject<HTMLDivElement | null>
+  resources: MothershipResource[]
+  activeResourceId: string | null
+  setActiveResourceId: (id: string | null) => void
 }
 
 const STATE_TO_STATUS: Record<string, ToolCallStatus> = {
@@ -65,10 +72,66 @@ function mapStoredMessage(msg: TaskStoredMessage): ChatMessage {
   return mapped
 }
 
+const logger = createLogger('useChat')
+
 function getPayloadData(payload: SSEPayload): SSEPayloadData | undefined {
   return typeof payload.data === 'object' ? payload.data : undefined
 }
 
+const RESOURCE_TOOL_NAMES = new Set(['user_table', 'workspace_file'])
+
+function getResultData(parsed: SSEPayload): Record<string, unknown> | undefined {
+  const topResult = parsed.result as Record<string, unknown> | undefined
+  const nestedResult =
+    typeof parsed.data === 'object' ? (parsed.data?.result as Record<string, unknown>) : undefined
+  const result = topResult ?? nestedResult
+  return result?.data as Record<string, unknown> | undefined
+}
+
+function extractTableResource(
+  parsed: SSEPayload,
+  storedArgs: Record<string, unknown> | undefined,
+  fallbackTableId: string | null
+): MothershipResource | null {
+  const data = getResultData(parsed)
+  const storedInnerArgs = storedArgs?.args as Record<string, unknown> | undefined
+
+  const table = data?.table as Record<string, unknown> | undefined
+  if (table?.id) {
+    return { type: 'table', id: table.id as string, title: (table.name as string) || 'Table' }
+  }
+
+  const tableId =
+    (data?.tableId as string) ?? storedInnerArgs?.tableId ?? storedArgs?.tableId ?? fallbackTableId
+  const tableName = (data?.tableName as string) || (table?.name as string) || 'Table'
+  if (tableId) return { type: 'table', id: tableId as string, title: tableName }
+
+  return null
+}
+
+function extractFileResource(
+  parsed: SSEPayload,
+  storedArgs: Record<string, unknown> | undefined
+): MothershipResource | null {
+  const data = getResultData(parsed)
+  const storedInnerArgs = storedArgs?.args as Record<string, unknown> | undefined
+
+  const file = data?.file as Record<string, unknown> | undefined
+  if (file?.id) {
+    return { type: 'file', id: file.id as string, title: (file.name as string) || 'File' }
+  }
+
+  const fileId = (data?.fileId as string) ?? (data?.id as string)
+  const fileName =
+    (data?.fileName as string) ||
+    (data?.name as string) ||
+    (storedInnerArgs?.fileName as string) ||
+    'File'
+  if (fileId && typeof fileId === 'string') return { type: 'file', id: fileId, title: fileName }
+
+  return null
+}
+
 export function useChat(workspaceId: string, initialChatId?: string): UseChatReturn {
   const pathname = usePathname()
   const queryClient = useQueryClient()
@@ -77,13 +140,16 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
   const [messages, setMessages] = useState<ChatMessage[]>([])
   const [isSending, setIsSending] = useState(false)
   const [error, setError] = useState<string | null>(null)
+  const [resources, setResources] = useState<MothershipResource[]>([])
+  const [activeResourceId, setActiveResourceId] = useState<string | null>(null)
   const abortControllerRef = useRef<AbortController | null>(null)
   const chatIdRef = useRef<string | undefined>(initialChatId)
   const chatBottomRef = useRef<HTMLDivElement>(null)
   const appliedChatIdRef = useRef<string | undefined>(undefined)
   const pendingUserMsgRef = useRef<{ id: string; content: string } | null>(null)
   const streamIdRef = useRef<string | undefined>(undefined)
   const sendingRef = useRef(false)
+  const toolArgsMapRef = useRef<Map<string, Record<string, unknown>>>(new Map())
 
   useEffect(() => {
     routerRef.current = router
@@ -93,12 +159,30 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
 
   const { data: chatHistory } = useChatHistory(initialChatId)
 
+  const addResource = useCallback((resource: MothershipResource) => {
+    setResources((prev) => {
+      const existing = prev.find((r) => r.type === resource.type && r.id === resource.id)
+      if (existing) {
+        const keepOldTitle = existing.title !== 'Table' && existing.title !== 'File'
+        const title = keepOldTitle ? existing.title : resource.title
+        if (title === existing.title) return prev
+        return prev.map((r) =>
+          r.id === existing.id && r.type === existing.type ? { ...r, title } : r
+        )
+      }
+      return [...prev, resource]
+    })
+    setActiveResourceId(resource.id)
+  }, [])
+
   useEffect(() => {
     chatIdRef.current = initialChatId
     appliedChatIdRef.current = undefined
     setMessages([])
     setError(null)
     setIsSending(false)
+    setResources([])
+    setActiveResourceId(null)
   }, [initialChatId])
 
   useEffect(() => {
@@ -110,6 +194,8 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
     setMessages([])
     setError(null)
     setIsSending(false)
+    setResources([])
+    setActiveResourceId(null)
   }, [isHomePage])
 
   useEffect(() => {
@@ -124,6 +210,7 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
       let buffer = ''
       const blocks: ContentBlock[] = []
       const toolMap = new Map<string, number>()
+      let lastTableId: string | null = null
 
       const ensureTextBlock = (): ContentBlock => {
         const last = blocks[blocks.length - 1]
@@ -164,6 +251,8 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
             continue
           }
 
+          logger.debug('SSE event received', parsed)
+
           switch (parsed.type) {
             case 'chat_id': {
               if (parsed.chatId) {
@@ -201,6 +290,14 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
               const data = getPayloadData(parsed)
               const name = parsed.toolName || data?.name || 'unknown'
               if (!id) break
+
+              if (RESOURCE_TOOL_NAMES.has(name)) {
+                const args = data?.arguments ?? data?.input
+                if (args) {
+                  toolArgsMapRef.current.set(id, args)
+                }
+              }
+
               const ui = parsed.ui || data?.ui
               if (ui?.hidden) break
               const displayTitle = ui?.title || ui?.phaseLabel
@@ -229,6 +326,33 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
                 blocks[idx].toolCall!.status = parsed.success ? 'success' : 'error'
                 flush()
               }
+
+              const toolName = parsed.toolName || getPayloadData(parsed)?.name
+              if (toolName && parsed.success && RESOURCE_TOOL_NAMES.has(toolName)) {
+                const storedArgs = toolArgsMapRef.current.get(id)
+                let resource: MothershipResource | null = null
+
+                if (toolName === 'user_table') {
+                  resource = extractTableResource(parsed, storedArgs, lastTableId)
+                  if (resource) {
+                    lastTableId = resource.id
+                    queryClient.invalidateQueries({ queryKey: tableKeys.detail(resource.id) })
+                    queryClient.invalidateQueries({ queryKey: tableKeys.rowsRoot(resource.id) })
+                  }
+                } else if (toolName === 'workspace_file') {
+                  resource = extractFileResource(parsed, storedArgs)
+                  if (resource) {
+                    queryClient.invalidateQueries({
+                      queryKey: workspaceFilesKeys.list(workspaceId),
+                    })
+                    queryClient.invalidateQueries({
+                      queryKey: workspaceFilesKeys.content(workspaceId, resource.id),
+                    })
+                  }
+                }
+
+                if (resource) addResource(resource)
+              }
               break
             }
             case 'tool_error': {
@@ -265,7 +389,7 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
         }
       }
     },
-    [workspaceId, queryClient]
+    [workspaceId, queryClient, addResource]
   )
 
   const finalize = useCallback(() => {
@@ -401,5 +525,8 @@ export function useChat(workspaceId: string, initialChatId?: string): UseChatRet
     sendMessage,
     stopGeneration,
     chatBottomRef,
+    resources,
+    activeResourceId,
+    setActiveResourceId,
   }
 }