Files
simstudioai--sim/apps/sim/lib/uploads/utils/embedded-image-ref.test.ts
T
wehub-resource-sync d25d482dc2
Publish CLI Package / publish-npm (push) Waiting to run
Publish Python SDK / publish-pypi (push) Waiting to run
Publish TypeScript SDK / publish-npm (push) Waiting to run
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:20:55 +08:00

61 lines
2.4 KiB
TypeScript

import { describe, expect, it } from 'vitest'
import {
extractEmbeddedFileRef,
extractEmbeddedFileRefs,
} from '@/lib/uploads/utils/embedded-image-ref'
const KEY = 'workspace/W1/1700000000000-deadbeefdeadbeef-photo.png'
const ENCODED = encodeURIComponent(KEY)
describe('extractEmbeddedFileRef', () => {
it('parses serve-url embeds (encoded, raw, and s3/blob prefixed) to the workspace key', () => {
expect(extractEmbeddedFileRef(`/api/files/serve/${ENCODED}?context=workspace`)).toEqual({
key: KEY,
})
expect(extractEmbeddedFileRef(`/api/files/serve/s3/${ENCODED}`)).toEqual({ key: KEY })
expect(extractEmbeddedFileRef(`/api/files/serve/blob/${ENCODED}`)).toEqual({ key: KEY })
})
it('parses view-url and in-app-path embeds to the file id', () => {
expect(extractEmbeddedFileRef('/api/files/view/wf_YwDXi8eWOkTxn0sbgChlB')).toEqual({
fileId: 'wf_YwDXi8eWOkTxn0sbgChlB',
})
expect(extractEmbeddedFileRef('/workspace/W1/files/wf_abc')).toEqual({ fileId: 'wf_abc' })
})
it('returns null for external, data, and non-workspace serve urls', () => {
expect(extractEmbeddedFileRef('https://cdn.example.com/a.png')).toBeNull()
expect(extractEmbeddedFileRef('data:image/png;base64,AAAA')).toBeNull()
expect(extractEmbeddedFileRef('/api/files/serve/profile-pictures%2Fu1%2Favatar.png')).toBeNull()
})
})
describe('extractEmbeddedFileRefs', () => {
it('collects de-duplicated keys and ids from a document via the shared parser', () => {
const content = `
![a](/api/files/serve/${ENCODED}?context=workspace)
![b](/api/files/view/wf_abc)
![c](/workspace/W1/files/4bdaf6c4-072e-464e-891d-b6af3b5fe2cc)
![dup](/api/files/serve/s3/${ENCODED})
![ext](https://cdn.example.com/x.png)
![pub](/api/files/serve/profile-pictures%2Fu1%2Favatar.png)
`
const { keys, ids } = extractEmbeddedFileRefs(content)
expect(keys).toEqual([KEY])
expect(ids.sort()).toEqual(['4bdaf6c4-072e-464e-891d-b6af3b5fe2cc', 'wf_abc'].sort())
})
it('caps total references (keys + ids) at 50 combined', () => {
const ids = Array.from(
{ length: 40 },
(_, i) => `/api/files/view/wf_${String(i).padStart(6, '0')}`
)
const keys = Array.from(
{ length: 40 },
(_, i) => `/api/files/serve/${encodeURIComponent(`workspace/W1/k${i}.png`)}`
)
const { keys: k, ids: d } = extractEmbeddedFileRefs([...ids, ...keys].join(' '))
expect(k.length + d.length).toBe(50)
})
})