- dsh web gains --provider/--model: a non-deepseek provider mounts the matching pi-ai catalog route (ambient credentials), making image input reachable from the shipped Web assembly; requires an explicit --model - attachment-local syncs the publication directories after the hard-link publish so a reported durable reference survives a crash (POSIX; Windows relies on filesystem metadata journaling) - the attachment seam gains storage-free validateImage; the host validates a complete multi-image prompt before persisting any member, so one malformed image cannot strand valid members as unreferenced objects - startSession sends before navigating: a rejected first send keeps the empty state, its error strip, and the complete draft mounted - the webserver rejects an undeclared-length body the moment it crosses the configured limit instead of draining a potentially endless stream to EOF
59 lines
2.4 KiB
TypeScript
59 lines
2.4 KiB
TypeScript
import { Context } from 'cordis'
|
|
import { existsSync } from 'node:fs'
|
|
import { mkdtemp, rm } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { describe, expect, it } from 'vitest'
|
|
import LocalAttachmentStore, {
|
|
DEFAULT_MAX_IMAGE_BYTES,
|
|
DEFAULT_MAX_IMAGE_PIXELS,
|
|
DEFAULT_MAX_IMAGES_PER_MESSAGE,
|
|
DEFAULT_MAX_MESSAGE_IMAGE_BYTES,
|
|
} from '../src/index.ts'
|
|
|
|
describe('local attachment service', () => {
|
|
it('resolves every omitted admission limit explicitly', () => {
|
|
const service = new LocalAttachmentStore(new Context(), {})
|
|
expect(service.imageLimits).toEqual({
|
|
maxImageBytes: DEFAULT_MAX_IMAGE_BYTES,
|
|
maxImagesPerMessage: DEFAULT_MAX_IMAGES_PER_MESSAGE,
|
|
maxMessageImageBytes: DEFAULT_MAX_MESSAGE_IMAGE_BYTES,
|
|
maxImagePixels: DEFAULT_MAX_IMAGE_PIXELS,
|
|
mediaTypes: ['image/png', 'image/jpeg', 'image/webp', 'image/gif'],
|
|
})
|
|
})
|
|
|
|
it('saves and reads through the service boundary', async () => {
|
|
const dshHome = await mkdtemp(join(tmpdir(), 'dsh-attachment-service-'))
|
|
try {
|
|
const service = new LocalAttachmentStore(new Context(), { dshHome })
|
|
const data = Uint8Array.from(Buffer.from(
|
|
'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII=',
|
|
'base64',
|
|
))
|
|
const ref = await service.saveImage({ data, mediaType: 'image/png' })
|
|
await expect(service.readImage(ref)).resolves.toEqual({ ref, data })
|
|
} finally {
|
|
await rm(dshHome, { recursive: true, force: true })
|
|
}
|
|
})
|
|
|
|
it('validates without persisting: a rejected image leaves no storage root behind', async () => {
|
|
const dshHome = await mkdtemp(join(tmpdir(), 'dsh-attachment-validate-'))
|
|
try {
|
|
const service = new LocalAttachmentStore(new Context(), { dshHome })
|
|
expect(() => { service.validateImage({ data: Uint8Array.of(1, 2, 3), mediaType: 'image/png' }) })
|
|
.toThrow(/Unsupported or malformed image data/)
|
|
const valid = Uint8Array.from(Buffer.from(
|
|
'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII=',
|
|
'base64',
|
|
))
|
|
expect(() => { service.validateImage({ data: valid, mediaType: 'image/png' }) }).not.toThrow()
|
|
// Validation is storage-free: nothing below the root may exist yet.
|
|
expect(existsSync(service.root)).toBe(false)
|
|
} finally {
|
|
await rm(dshHome, { recursive: true, force: true })
|
|
}
|
|
})
|
|
})
|