@muyichengshayu/promptx 0.2.16 → 0.2.17

package/apps/server/src/appPaths.test.js

@@ -0,0 +1,52 @@
+import assert from 'node:assert/strict'
+import fs from 'node:fs'
+import os from 'node:os'
+import path from 'node:path'
+import test from 'node:test'
+
+function withEnv(overrides, fn) {
+  const previous = new Map()
+
+  for (const [key, value] of Object.entries(overrides)) {
+    previous.set(key, process.env[key])
+    if (typeof value === 'undefined') {
+      delete process.env[key]
+    } else {
+      process.env[key] = value
+    }
+  }
+
+  return Promise.resolve()
+    .then(fn)
+    .finally(() => {
+      for (const [key, value] of previous.entries()) {
+        if (typeof value === 'undefined') {
+          delete process.env[key]
+        } else {
+          process.env[key] = value
+        }
+      }
+    })
+}
+
+test('appPaths resolves storage under PROMPTX_HOME by default', async () => {
+  const tempHome = fs.mkdtempSync(path.join(os.tmpdir(), 'promptx-home-'))
+
+  await withEnv(
+    {
+      PROMPTX_HOME: tempHome,
+      PROMPTX_DATA_DIR: undefined,
+      PROMPTX_UPLOADS_DIR: undefined,
+      PROMPTX_TMP_DIR: undefined,
+    },
+    async () => {
+      const appPaths = await import(`./appPaths.js?test=${Date.now()}`)
+      const resolved = appPaths.ensurePromptxStorageReady()
+
+      assert.equal(resolved.promptxHomeDir, tempHome)
+      assert.equal(resolved.dataDir, path.join(tempHome, 'data'))
+      assert.equal(resolved.uploadsDir, path.join(tempHome, 'uploads'))
+      assert.equal(resolved.tmpDir, path.join(tempHome, 'tmp'))
+    }
+  )
+})

package/apps/server/src/assetRoutes.test.js

@@ -0,0 +1,168 @@
+import assert from 'node:assert/strict'
+import fs from 'node:fs'
+import os from 'node:os'
+import path from 'node:path'
+import Fastify from 'fastify'
+import multipart from '@fastify/multipart'
+import test from 'node:test'
+import { createCanvas } from '@napi-rs/canvas'
+
+import { registerAssetRoutes } from './assetRoutes.js'
+
+function createMultipartBody(fileName, mimeType, buffer) {
+  const boundary = `----promptx-test-${Date.now()}`
+  const head = Buffer.from(
+    `--${boundary}\r\n`
+    + `Content-Disposition: form-data; name="file"; filename="${fileName}"\r\n`
+    + `Content-Type: ${mimeType}\r\n\r\n`
+  )
+  const tail = Buffer.from(`\r\n--${boundary}--\r\n`)
+
+  return {
+    boundary,
+    body: Buffer.concat([head, buffer, tail]),
+  }
+}
+
+function createTestPngBuffer(width = 2000, height = 1000) {
+  const canvas = createCanvas(width, height)
+  const context = canvas.getContext('2d')
+  context.fillStyle = '#0ea5e9'
+  context.fillRect(0, 0, width, height)
+  return canvas.toBuffer('image/png')
+}
+
+function createTestSvgBuffer(width = 120, height = 80) {
+  return Buffer.from(
+    `<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${height}" viewBox="0 0 ${width} ${height}">`
+    + '<rect width="100%" height="100%" fill="#0ea5e9"/>'
+    + '</svg>',
+    'utf8'
+  )
+}
+
+test('asset routes reject missing upload file', async () => {
+  const app = Fastify()
+  await app.register(multipart)
+  registerAssetRoutes(app, {
+    createTempFilePath: () => '',
+    importPdfBlocks: async () => ({ blocks: [] }),
+    normalizeUploadFileName: (name) => name,
+    removeAssetFiles: () => {},
+    tmpDir: 'tmp',
+    uploadsDir: 'uploads',
+  })
+  await app.ready()
+
+  try {
+    const uploadResponse = await app.inject({
+      method: 'POST',
+      url: '/api/uploads',
+    })
+    assert.equal(uploadResponse.statusCode, 406)
+
+    const pdfResponse = await app.inject({
+      method: 'POST',
+      url: '/api/imports/pdf',
+    })
+    assert.equal(pdfResponse.statusCode, 406)
+  } finally {
+    await app.close()
+  }
+})
+
+test('asset routes keep uploaded png format and original bytes', async () => {
+  const rootDir = fs.mkdtempSync(path.join(os.tmpdir(), 'promptx-asset-routes-'))
+  const uploadsDir = path.join(rootDir, 'uploads')
+  const tmpDir = path.join(rootDir, 'tmp')
+  fs.mkdirSync(uploadsDir, { recursive: true })
+  fs.mkdirSync(tmpDir, { recursive: true })
+
+  const app = Fastify()
+  await app.register(multipart)
+  registerAssetRoutes(app, {
+    createTempFilePath: (_dir, fileName) => path.join(tmpDir, fileName),
+    importPdfBlocks: async () => ({ blocks: [] }),
+    normalizeUploadFileName: (name) => name,
+    removeAssetFiles: () => {},
+    tmpDir,
+    uploadsDir,
+  })
+  await app.ready()
+
+  try {
+    const sourceBuffer = createTestPngBuffer()
+    const payload = createMultipartBody('demo.png', 'image/png', sourceBuffer)
+    const response = await app.inject({
+      method: 'POST',
+      url: '/api/uploads',
+      headers: {
+        'content-type': `multipart/form-data; boundary=${payload.boundary}`,
+      },
+      payload: payload.body,
+    })
+
+    assert.equal(response.statusCode, 201)
+    const body = response.json()
+    assert.match(body.url || '', /^\/uploads\/.+\.png$/)
+    assert.equal(body.width, 2000)
+    assert.equal(body.height, 1000)
+    assert.equal(body.mimeType, 'image/png')
+    assert.equal(typeof body.size, 'number')
+    assert.ok(body.size > 0)
+
+    const outputPath = path.join(uploadsDir, path.basename(body.url))
+    assert.equal(fs.existsSync(outputPath), true)
+    assert.deepEqual(fs.readFileSync(outputPath), sourceBuffer)
+  } finally {
+    await app.close()
+    fs.rmSync(rootDir, { recursive: true, force: true })
+  }
+})
+
+test('asset routes keep uploaded svg format and original bytes', async () => {
+  const rootDir = fs.mkdtempSync(path.join(os.tmpdir(), 'promptx-asset-routes-svg-'))
+  const uploadsDir = path.join(rootDir, 'uploads')
+  const tmpDir = path.join(rootDir, 'tmp')
+  fs.mkdirSync(uploadsDir, { recursive: true })
+  fs.mkdirSync(tmpDir, { recursive: true })
+
+  const app = Fastify()
+  await app.register(multipart)
+  registerAssetRoutes(app, {
+    createTempFilePath: (_dir, fileName) => path.join(tmpDir, fileName),
+    importPdfBlocks: async () => ({ blocks: [] }),
+    normalizeUploadFileName: (name) => name,
+    removeAssetFiles: () => {},
+    tmpDir,
+    uploadsDir,
+  })
+  await app.ready()
+
+  try {
+    const sourceBuffer = createTestSvgBuffer()
+    const payload = createMultipartBody('logo.svg', 'image/svg+xml', sourceBuffer)
+    const response = await app.inject({
+      method: 'POST',
+      url: '/api/uploads',
+      headers: {
+        'content-type': `multipart/form-data; boundary=${payload.boundary}`,
+      },
+      payload: payload.body,
+    })
+
+    assert.equal(response.statusCode, 201)
+    const body = response.json()
+    assert.match(body.url || '', /^\/uploads\/.+\.svg$/)
+    assert.equal(body.mimeType, 'image/svg+xml')
+    assert.equal(typeof body.size, 'number')
+    assert.ok(body.size > 0)
+
+    const outputPath = path.join(uploadsDir, path.basename(body.url))
+    assert.equal(fs.existsSync(outputPath), true)
+    assert.deepEqual(fs.readFileSync(outputPath), sourceBuffer)
+  } finally {
+    await app.close()
+    fs.rmSync(rootDir, { recursive: true, force: true })
+  }
+})