chore: import upstream snapshot with attribution

apps/sim/ee/workspace-forking/lib/remap/remap-content-refs.test.ts

@@ -0,0 +1,194 @@
+/**
+ * @vitest-environment node
+ */
+import { describe, expect, it } from 'vitest'
+import {
+  type ForkContentRefMaps,
+  rewriteForkContentRefs,
+  rewriteForkResourceUrls,
+} from '@/ee/workspace-forking/lib/remap/remap-content-refs'
+
+const SRC_KEY = 'workspace/SRC/1700000000000-deadbeef-photo.png'
+const DST_KEY = 'workspace/DST/1700000000001-cafebabe-photo.png'
+
+const maps = (): ForkContentRefMaps => ({
+  workspaceId: { from: 'SRC', to: 'DST' },
+  fileKeys: new Map([[SRC_KEY, DST_KEY]]),
+  fileIds: new Map([['file-src', 'file-dst']]),
+  workflows: new Map([['wf-src', 'wf-dst']]),
+  knowledgeBases: new Map([['kb-src', 'kb-dst']]),
+  tables: new Map([['tbl-src', 'tbl-dst']]),
+  skills: new Map([['skill-src', 'skill-dst']]),
+  folders: new Map([['fld-src', 'fld-dst']]),
+})
+
+describe('rewriteForkContentRefs - sim: links', () => {
+  it('remaps each mapped sim: link kind by its id map', () => {
+    const input = [
+      'see [F](sim:file/file-src)',
+      '[W](sim:workflow/wf-src)',
+      '[K](sim:knowledge/kb-src)',
+      '[T](sim:table/tbl-src)',
+      '[S](sim:skill/skill-src)',
+      '[D](sim:folder/fld-src)',
+    ].join(' ')
+    expect(rewriteForkContentRefs(input, maps())).toBe(
+      [
+        'see [F](sim:file/file-dst)',
+        '[W](sim:workflow/wf-dst)',
+        '[K](sim:knowledge/kb-dst)',
+        '[T](sim:table/tbl-dst)',
+        '[S](sim:skill/skill-dst)',
+        '[D](sim:folder/fld-dst)',
+      ].join(' ')
+    )
+  })
+
+  it('leaves an unmapped id unchanged (graceful broken link)', () => {
+    const input = '[F](sim:file/unknown-file) and [I](sim:integration/gmail_v2)'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+
+  it('leaves a kind with no supplied map unchanged', () => {
+    const input = '[W](sim:workflow/wf-src)'
+    expect(rewriteForkContentRefs(input, { fileIds: new Map() })).toBe(input)
+  })
+})
+
+describe('rewriteForkContentRefs - embedded urls', () => {
+  it('remaps a serve-url storage key (encoded form output)', () => {
+    const input = `![a](/api/files/serve/${encodeURIComponent(SRC_KEY)}?context=workspace)`
+    expect(rewriteForkContentRefs(input, maps())).toBe(
+      `![a](/api/files/serve/${encodeURIComponent(DST_KEY)}?context=workspace)`
+    )
+  })
+
+  it('remaps a serve-url key given in raw (unencoded) form', () => {
+    const input = `![a](/api/files/serve/${SRC_KEY})`
+    expect(rewriteForkContentRefs(input, maps())).toBe(
+      `![a](/api/files/serve/${encodeURIComponent(DST_KEY)})`
+    )
+  })
+
+  it('remaps an s3/blob-prefixed serve url, preserving the prefix', () => {
+    const input = `![a](/api/files/serve/s3/${encodeURIComponent(SRC_KEY)})`
+    expect(rewriteForkContentRefs(input, maps())).toBe(
+      `![a](/api/files/serve/s3/${encodeURIComponent(DST_KEY)})`
+    )
+  })
+
+  it('remaps a view-url file id', () => {
+    const input = '![a](/api/files/view/file-src)'
+    expect(rewriteForkContentRefs(input, maps())).toBe('![a](/api/files/view/file-dst)')
+  })
+
+  it('remaps both the workspace id and file id in an in-app files path', () => {
+    const input = '![a](/workspace/SRC/files/file-src)'
+    expect(rewriteForkContentRefs(input, maps())).toBe('![a](/workspace/DST/files/file-dst)')
+  })
+
+  it('leaves a foreign-workspace in-app file path unchanged (both-or-nothing)', () => {
+    // The ws id is not the mapped source, so emitting the child file id under OTHER would 404.
+    const input = '![a](/workspace/OTHER/files/file-src)'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+
+  it('leaves an in-app file path unchanged when the file id is unmapped (both-or-nothing)', () => {
+    const input = '![a](/workspace/SRC/files/unknown-file)'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+
+  it('leaves an unmapped storage key / file id unchanged', () => {
+    const input =
+      '![a](/api/files/serve/workspace%2FSRC%2Funknown.png) ![b](/api/files/view/unknown-id)'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+
+  it('leaves an external / data url unchanged', () => {
+    const input = '![a](https://cdn.example.com/x.png) ![b](data:image/png;base64,AAAA)'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+})
+
+describe('rewriteForkContentRefs - mixed and edge cases', () => {
+  it('rewrites multiple references of different shapes in one string', () => {
+    const input = [
+      'intro [S](sim:skill/skill-src)',
+      `![img](/api/files/serve/${encodeURIComponent(SRC_KEY)})`,
+      'link [W](sim:workflow/wf-src)',
+      '![v](/api/files/view/file-src)',
+    ].join('\n')
+    const output = rewriteForkContentRefs(input, maps())
+    expect(output).toContain('sim:skill/skill-dst')
+    expect(output).toContain('sim:workflow/wf-dst')
+    expect(output).toContain(encodeURIComponent(DST_KEY))
+    expect(output).toContain('/api/files/view/file-dst')
+  })
+
+  it('returns the input unchanged when there are no references', () => {
+    const input = '# Heading\n\nNo references here, just text.'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+
+  it('returns the input unchanged for an empty string', () => {
+    expect(rewriteForkContentRefs('', maps())).toBe('')
+  })
+
+  it('leaves a malformed (un-decodable) serve key unchanged', () => {
+    const input = '![a](/api/files/serve/%E0%A4%A)'
+    expect(rewriteForkContentRefs(input, maps())).toBe(input)
+  })
+
+  it('does nothing when no maps are supplied', () => {
+    const input = `[S](sim:skill/skill-src) ![a](/api/files/serve/${SRC_KEY})`
+    expect(rewriteForkContentRefs(input, {})).toBe(input)
+  })
+})
+
+describe('rewriteForkResourceUrls - table cell resource chip urls', () => {
+  it('rewrites the workspace id + resource id for each section when the resource is mapped', () => {
+    expect(rewriteForkResourceUrls('/workspace/SRC/w/wf-src', maps())).toBe(
+      '/workspace/DST/w/wf-dst'
+    )
+    expect(rewriteForkResourceUrls('/workspace/SRC/tables/tbl-src', maps())).toBe(
+      '/workspace/DST/tables/tbl-dst'
+    )
+    expect(rewriteForkResourceUrls('/workspace/SRC/knowledge/kb-src', maps())).toBe(
+      '/workspace/DST/knowledge/kb-dst'
+    )
+    expect(rewriteForkResourceUrls('/workspace/SRC/files/file-src', maps())).toBe(
+      '/workspace/DST/files/file-dst'
+    )
+  })
+
+  it('leaves an unmapped resource id unchanged (both-or-nothing -> graceful plain link)', () => {
+    expect(rewriteForkResourceUrls('/workspace/SRC/w/wf-unknown', maps())).toBe(
+      '/workspace/SRC/w/wf-unknown'
+    )
+  })
+
+  it('leaves a foreign / unknown workspace id unchanged', () => {
+    expect(rewriteForkResourceUrls('/workspace/OTHER/w/wf-src', maps())).toBe(
+      '/workspace/OTHER/w/wf-src'
+    )
+  })
+
+  it('leaves a non-matching string (unknown section / no resource path) unchanged', () => {
+    const input = 'text /workspace/ and /workspace/SRC/settings/x and /workspace/SRC/w/'
+    expect(rewriteForkResourceUrls(input, maps())).toBe(input)
+  })
+
+  it('does nothing without a workspaceId map', () => {
+    const input = '/workspace/SRC/w/wf-src'
+    expect(rewriteForkResourceUrls(input, { workflows: new Map([['wf-src', 'wf-dst']]) })).toBe(
+      input
+    )
+  })
+
+  it('rewrites a URL embedded mid-text', () => {
+    const input = 'See [chip](/workspace/SRC/knowledge/kb-src) here'
+    expect(rewriteForkResourceUrls(input, maps())).toBe(
+      'See [chip](/workspace/DST/knowledge/kb-dst) here'
+    )
+  })
+})