fix(chat): preserve workspace links when copying

Author: j15z

apps/sim/app/workspace/[workspaceId]/home/components/message-content/components/chat-content/chat-content.tsx

@@ -15,6 +15,10 @@ import { Checkbox, CopyCodeButton, cn, languages, highlight as prismHighlight }
 import { decodeVfsSegmentSafe } from '@/lib/copilot/vfs/path-utils'
 import { extractTextContent } from '@/lib/core/utils/react-node-text'
 import { ContextMentionIcon } from '@/app/workspace/[workspaceId]/home/components/context-mention-icon'
+import {
+  appendInlineReferenceMarkdown,
+  workspaceResourceReferenceMarkdown,
+} from '@/app/workspace/[workspaceId]/home/components/message-content/components/chat-content/workspace-resource-markdown'
 import {
   type ContentSegment,
   type CredentialSubmissionPayload,
@@ -95,47 +99,6 @@ const ANIMATION_DRAIN_MS = 300
  */
 const FADE_MAX_REVEALED_CHARS = 6000
 
-function startsInlineWord(value: string): boolean {
-  return /^[A-Za-z0-9_(]/.test(value)
-}
-
-function endsInlineWord(value: string): boolean {
-  return /[A-Za-z0-9_)]$/.test(value)
-}
-
-function nextInlineSegmentLabel(segment?: ContentSegment): string {
-  if (!segment) return ''
-  // Thinking segments are never rendered, so they contribute no following text.
-  if (segment.type === 'text') return segment.content
-  if (segment.type === 'workspace_resource') return segment.data.title || segment.data.id || ''
-  return ''
-}
-
-function appendInlineReferenceMarkdown(
-  currentMarkdown: string,
-  referenceMarkdown: string,
-  nextSegment?: ContentSegment
-): string {
-  let nextMarkdown = currentMarkdown
-  if (currentMarkdown && endsInlineWord(currentMarkdown) && !/\s$/.test(currentMarkdown)) {
-    nextMarkdown += ' '
-  }
-
-  nextMarkdown += referenceMarkdown
-
-  const followingText = nextInlineSegmentLabel(nextSegment)
-  if (
-    followingText &&
-    startsInlineWord(followingText) &&
-    !/^\s/.test(followingText) &&
-    !/\s$/.test(nextMarkdown)
-  ) {
-    nextMarkdown += ' '
-  }
-
-  return nextMarkdown
-}
-
 type TdProps = ComponentPropsWithoutRef<'td'>
 type ThProps = ComponentPropsWithoutRef<'th'>
 
@@ -586,14 +549,9 @@ function ChatContentInner({
     const s = parsed.segments[i]
     const nextSegment = parsed.segments[i + 1]
     if (s.type === 'workspace_resource') {
-      // Files are addressed by their encoded VFS path (copied verbatim from the tag);
-      // workflows/tables/KBs by id. The angle-bracket link destination keeps the path
-      // intact through markdown parsing (tolerates parens) without re-encoding it.
-      const ref = s.data.type === 'file' ? (s.data.path ?? s.data.id ?? '') : (s.data.id ?? '')
-      const label = s.data.title || ref
       pendingMarkdown = appendInlineReferenceMarkdown(
         pendingMarkdown,
-        `[${label}](<#wsres-${s.data.type}-${ref}>)`,
+        workspaceResourceReferenceMarkdown(s.data),
         nextSegment
       )
     } else if (s.type === 'thinking') {

apps/sim/app/workspace/[workspaceId]/home/components/message-content/components/chat-content/workspace-resource-markdown.ts

@@ -0,0 +1,54 @@
+import type {
+  ContentSegment,
+  WorkspaceResourceTagData,
+} from '@/app/workspace/[workspaceId]/home/components/message-content/components/special-tags'
+
+function startsInlineWord(value: string): boolean {
+  return /^[A-Za-z0-9_(]/.test(value)
+}
+
+function endsInlineWord(value: string): boolean {
+  return /[A-Za-z0-9_)]$/.test(value)
+}
+
+function workspaceResourceLabel(data: WorkspaceResourceTagData): string {
+  if (data.title) return data.title
+  return data.type === 'file' ? (data.path ?? data.id ?? '') : (data.id ?? '')
+}
+
+function nextInlineSegmentLabel(segment?: ContentSegment): string {
+  if (!segment) return ''
+  if (segment.type === 'text') return segment.content
+  if (segment.type === 'workspace_resource') return segment.data.title || segment.data.id || ''
+  return ''
+}
+
+export function workspaceResourceReferenceMarkdown(data: WorkspaceResourceTagData): string {
+  const ref = data.type === 'file' ? (data.path ?? data.id ?? '') : (data.id ?? '')
+  return `[${workspaceResourceLabel(data)}](<#wsres-${data.type}-${ref}>)`
+}
+
+export function appendInlineReferenceMarkdown(
+  currentMarkdown: string,
+  referenceMarkdown: string,
+  nextSegment?: ContentSegment
+): string {
+  let nextMarkdown = currentMarkdown
+  if (currentMarkdown && endsInlineWord(currentMarkdown) && !/\s$/.test(currentMarkdown)) {
+    nextMarkdown += ' '
+  }
+
+  nextMarkdown += referenceMarkdown
+
+  const followingText = nextInlineSegmentLabel(nextSegment)
+  if (
+    followingText &&
+    startsInlineWord(followingText) &&
+    !/^\s/.test(followingText) &&
+    !/\s$/.test(nextMarkdown)
+  ) {
+    nextMarkdown += ' '
+  }
+
+  return nextMarkdown
+}

apps/sim/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown.test.ts

@@ -44,4 +44,16 @@ describe('toCopyableMarkdown', () => {
 
     expect(toCopyableMarkdown(message)).toBe(message)
   })
+
+  it('preserves visible workspace resources as Markdown links', () => {
+    const message = [
+      'Read',
+      '<workspace_resource>{"type":"file","path":"files/notes.md","title":"notes.md"}</workspace_resource>',
+      'for details.',
+    ].join(' ')
+
+    expect(toCopyableMarkdown(message)).toBe(
+      'Read [notes.md](<#wsres-file-files/notes.md>) for details.'
+    )
+  })
 })

apps/sim/app/workspace/[workspaceId]/home/components/mothership-chat/copyable-markdown.ts

@@ -1,13 +1,25 @@
 import { sanitizeChatDisplayContent } from '@/app/workspace/[workspaceId]/home/components/message-content/components/chat-content/chat-sanitize'
+import {
+  appendInlineReferenceMarkdown,
+  workspaceResourceReferenceMarkdown,
+} from '@/app/workspace/[workspaceId]/home/components/message-content/components/chat-content/workspace-resource-markdown'
 import { parseSpecialTags } from '@/app/workspace/[workspaceId]/home/components/message-content/components/special-tags'
 
 export function toCopyableMarkdown(raw: string): string {
   const displayContent = sanitizeChatDisplayContent(raw)
   const { segments } = parseSpecialTags(displayContent, false)
 
   return segments
-    .reduce((markdown, segment) => {
-      return segment.type === 'text' ? markdown + segment.content : markdown
+    .reduce((markdown, segment, index) => {
+      if (segment.type === 'text') return markdown + segment.content
+      if (segment.type === 'workspace_resource') {
+        return appendInlineReferenceMarkdown(
+          markdown,
+          workspaceResourceReferenceMarkdown(segment.data),
+          segments[index + 1]
+        )
+      }
+      return markdown
     }, '')
     .trim()
 }