@walkhi/code-relax 0.1.0-beta.76 → 0.1.0-beta.79

package/dist/src/message-upload-files.mjs

@@ -1,109 +1,115 @@
-import fs from 'node:fs/promises';
-import path from 'node:path';
-import crypto from 'node:crypto';
-import sharp from 'sharp';
-import media from '../shared/code-browser.cjs';
-import { validateInlineImage } from './message-images.mjs';
-import { createFileUploadSession, FILE_CHUNK_BYTES, MAX_FILE_BYTES, MAX_FILE_COUNT, MAX_TOTAL_BYTES } from './file-upload-session.mjs';
-import { fileRevision } from './file-transfer-range.mjs';
-
-const mimeTypes = { pdf: 'application/pdf', txt: 'text/plain', md: 'text/markdown', csv: 'text/csv',
-  json: 'application/json', zip: 'application/zip', doc: 'application/msword',
-  docx: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
-  xls: 'application/vnd.ms-excel', xlsx: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
-  ppt: 'application/vnd.ms-powerpoint', pptx: 'application/vnd.openxmlformats-officedocument.presentationml.presentation' };
-
-// Reuse the incoming-file contract without sending workstation bytes to the phone.
-export async function stageWorkstationMessageFile(file, uploadsRoot) {
-  const handle = await fs.open(file.path, 'r');
-  const session = createFileUploadSession(uploadsRoot);
-  try {
-    const stat = await handle.stat();
-    if (!stat.isFile() || stat.size < 1 || stat.size > MAX_FILE_BYTES) throw new Error('附件为空或超过 20 MiB。');
-    const validate = async () => {
-      if (fileRevision(await handle.stat()) !== file.revision || fileRevision(await fs.stat(file.path)) !== file.revision) {
-        throw new Error('电脑文件已变化，请重新添加。');
-      }
-    };
-    await validate();
-    const bytes = Buffer.alloc(stat.size);
-    for (let offset = 0; offset < bytes.length;) {
-      const { bytesRead } = await handle.read(bytes, offset, bytes.length - offset, offset);
-      if (!bytesRead) throw new Error('电脑文件读取不完整，请重新添加。');
-      offset += bytesRead;
-    }
-    await validate();
-    const sha256 = crypto.createHash('sha256').update(bytes).digest('hex');
-    const started = await session.receive({ type: 'start', name: file.name, size: bytes.length, sha256 });
-    for (let offset = 0; offset < bytes.length; offset += FILE_CHUNK_BYTES) {
-      await session.receive({ type: 'chunk', id: started.id, offset, data: bytes.subarray(offset, offset + FILE_CHUNK_BYTES) });
-    }
-    await validate();
-    const result = await session.receive({ type: 'finish', id: started.id });
-    await session.receive({ type: 'commit' });
-    return { ...result, mediaKind: media.mediaKind(file),
-      mimeType: media.mediaMime(file) || mimeTypes[path.extname(file.name).slice(1).toLowerCase()] || 'application/octet-stream' };
-  } finally { await handle.close(); await session.close(); }
-}
-
-async function localFile(file, directory) {
-  if (typeof file?.path !== 'string' || typeof file.name !== 'string' || !file.name || /[\r\n\0]/.test(file.name)) {
-    throw new Error('附件信息无效，请重新选择。');
-  }
-  const root = await fs.realpath(directory), resolved = await fs.realpath(file.path);
-  if (path.relative(root, path.dirname(resolved)) !== '') throw new Error('附件必须来自本次工作站上传目录。');
-  const handle = await fs.open(resolved, 'r');
-  try {
-    const stat = await handle.stat();
-    if (!stat.isFile() || stat.size < 1 || stat.size > MAX_FILE_BYTES) throw new Error('附件为空或超过 20 MiB。');
-    const bytes = await handle.readFile();
-    if (bytes.length !== stat.size) throw new Error('附件在读取时发生变化，请重新上传。');
-    const mimeType = media.mediaMime(file) || mimeTypes[path.extname(file.name).slice(1).toLowerCase()] || 'application/octet-stream';
-    if (!/^[a-z0-9.+-]+\/[a-z0-9.+-]+$/i.test(mimeType)) throw new Error('附件类型无效。');
-    return { name: file.name, mimeType, bytes };
-  } finally { await handle.close(); }
-}
-
-export async function readMessageUploads(body, uploadsRoot) {
-  const images = body.images ?? [], originals = body.originalImages ?? [], files = body.files ?? [];
-  if (![images, originals, files].every(Array.isArray) || images.length + originals.length > 4 || files.length + originals.length > MAX_FILE_COUNT) {
-    throw new Error('每条消息最多 4 张图片、4 个文件。');
-  }
-  const uploads = [];
-  let imageBytes = 0, fileBytes = 0;
-  for (const image of images) {
-    let file;
-    if (typeof image === 'string' && image.startsWith('data:')) {
-      validateInlineImage(image);
-      const mimeType = image.slice(5, image.indexOf(';'));
-      file = { name: `image.${mimeType.split('/')[1]}`, mimeType, bytes: Buffer.from(image.slice(image.indexOf(',') + 1), 'base64') };
-    } else {
-      if (typeof image !== 'string' || !/^[a-f0-9]{64}$/.test(image)) throw new Error('图片标识无效。');
-      let extension = 'webp';
-      try { await fs.access(path.join(uploadsRoot, `${image}.webp`)); }
-      catch (error) { if (error.code !== 'ENOENT') throw error; extension = 'png'; }
-      file = await localFile({ path: path.join(uploadsRoot, `${image}.${extension}`), name: `image.${extension}` }, uploadsRoot);
-    }
-    imageBytes += file.bytes.length;
-    if (imageBytes > 8 * 1024 * 1024) throw new Error('压缩图片合计不能超过 8 MiB。');
-    uploads.push(file);
-  }
-  for (const original of originals) {
-    if (typeof original !== 'string' || !/\.(png|jpe?g|webp|gif)$/i.test(original)) throw new Error('原图路径无效。');
-  }
-  for (const file of [...originals.map(value => ({ path: value, name: path.basename(value) })), ...files]) {
-    const uploaded = await localFile(file, path.join(uploadsRoot, 'incoming'));
-    fileBytes += uploaded.bytes.length;
-    if (fileBytes > MAX_TOTAL_BYTES) throw new Error('文件合计不能超过 40 MiB。');
-    uploads.push(uploaded);
-  }
-  for (const file of uploads) {
-    if (/^image\/(?:png|jpeg|webp|gif)$/.test(file.mimeType)) {
-      const metadata = await sharp(file.bytes).metadata();
-      file.width = metadata.width; file.height = metadata.height;
-      if (!file.width || !file.height) throw new Error('无法读取 图片尺寸。');
-    }
-  }
-  if (uploads.filter(file => file.width).length > 4) throw new Error('每条消息最多 4 张图片。');
-  return uploads;
-}
+import fs from 'node:fs/promises';
+import { openAsBlob } from 'node:fs';
+import path from 'node:path';
+import crypto from 'node:crypto';
+import sharp from 'sharp';
+import media from '../shared/code-browser.cjs';
+import { validateInlineImage } from './message-images.mjs';
+import { createFileUploadSession, FILE_CHUNK_BYTES, MAX_FILE_COUNT } from './file-upload-session.mjs';
+import { fileRevision } from './file-transfer-range.mjs';
+
+const mimeTypes = { pdf: 'application/pdf', txt: 'text/plain', md: 'text/markdown', csv: 'text/csv',
+  json: 'application/json', zip: 'application/zip', doc: 'application/msword',
+  docx: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+  xls: 'application/vnd.ms-excel', xlsx: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+  ppt: 'application/vnd.ms-powerpoint', pptx: 'application/vnd.openxmlformats-officedocument.presentationml.presentation' };
+
+// Reuse the incoming-file contract without sending workstation bytes to the phone.
+export async function stageWorkstationMessageFile(file, uploadsRoot) {
+  const handle = await fs.open(file.path, 'r');
+  const session = createFileUploadSession(uploadsRoot);
+  try {
+    const stat = await handle.stat();
+    if (!stat.isFile() || stat.size < 1) throw new Error('附件为空。');
+    const validate = async () => {
+      if (fileRevision(await handle.stat()) !== file.revision || fileRevision(await fs.stat(file.path)) !== file.revision) {
+        throw new Error('电脑文件已变化，请重新添加。');
+      }
+    };
+    await validate();
+    const bytes = Buffer.alloc(FILE_CHUNK_BYTES);
+    const hash = crypto.createHash('sha256');
+    const readPart = async offset => {
+      const length = Math.min(bytes.length, stat.size - offset);
+      let count = 0;
+      while (count < length) {
+        const { bytesRead } = await handle.read(bytes, count, length - count, offset + count);
+        if (!bytesRead) throw new Error('电脑文件读取不完整，请重新添加。');
+        count += bytesRead;
+      }
+      return bytes.subarray(0, length);
+    };
+    for (let offset = 0; offset < stat.size; offset += FILE_CHUNK_BYTES) hash.update(await readPart(offset));
+    await validate();
+    const sha256 = hash.digest('hex');
+    const started = await session.receive({ type: 'start', fileKey: '0', name: file.name, size: stat.size, sha256 });
+    for (let offset = 0; offset < stat.size; offset += FILE_CHUNK_BYTES) {
+      await session.receive({ type: 'chunk', id: started.id, offset, data: await readPart(offset) });
+    }
+    await validate();
+    const result = await session.receive({ type: 'finish', id: started.id });
+    await session.receive({ type: 'commit' });
+    return { ...result, mediaKind: media.mediaKind(file),
+      mimeType: media.mediaMime(file) || mimeTypes[path.extname(file.name).slice(1).toLowerCase()] || 'application/octet-stream' };
+  } finally { await handle.close(); await session.close(); }
+}
+
+async function localFile(file, directory) {
+  if (typeof file?.path !== 'string' || typeof file.name !== 'string' || !file.name || /[\r\n\0]/.test(file.name)) {
+    throw new Error('附件信息无效，请重新选择。');
+  }
+  const root = await fs.realpath(directory), resolved = await fs.realpath(file.path);
+  if (path.relative(root, path.dirname(resolved)) !== '') throw new Error('附件必须来自本次工作站上传目录。');
+  const handle = await fs.open(resolved, 'r');
+  try {
+    const stat = await handle.stat();
+    if (!stat.isFile() || stat.size < 1) throw new Error('附件为空。');
+    const bytes = await openAsBlob(resolved);
+    if (bytes.size !== stat.size) throw new Error('附件在读取时发生变化，请重新上传。');
+    const mimeType = media.mediaMime(file) || mimeTypes[path.extname(file.name).slice(1).toLowerCase()] || 'application/octet-stream';
+    if (!/^[a-z0-9.+-]+\/[a-z0-9.+-]+$/i.test(mimeType)) throw new Error('附件类型无效。');
+    return { name: file.name, mimeType, bytes, path: resolved };
+  } finally { await handle.close(); }
+}
+
+export async function readMessageUploads(body, uploadsRoot) {
+  const images = body.images ?? [], originals = body.originalImages ?? [], files = body.files ?? [];
+  if (![images, originals, files].every(Array.isArray) || images.length + originals.length > 4 || files.length + originals.length > MAX_FILE_COUNT) {
+    throw new Error('每条消息最多 4 张图片、4 个文件。');
+  }
+  const uploads = [];
+  let imageBytes = 0;
+  for (const image of images) {
+    let file;
+    if (typeof image === 'string' && image.startsWith('data:')) {
+      validateInlineImage(image);
+      const mimeType = image.slice(5, image.indexOf(';'));
+      file = { name: `image.${mimeType.split('/')[1]}`, mimeType, bytes: Buffer.from(image.slice(image.indexOf(',') + 1), 'base64') };
+    } else {
+      if (typeof image !== 'string' || !/^[a-f0-9]{64}$/.test(image)) throw new Error('图片标识无效。');
+      let extension = 'webp';
+      try { await fs.access(path.join(uploadsRoot, `${image}.webp`)); }
+      catch (error) { if (error.code !== 'ENOENT') throw error; extension = 'png'; }
+      file = await localFile({ path: path.join(uploadsRoot, `${image}.${extension}`), name: `image.${extension}` }, uploadsRoot);
+    }
+    imageBytes += file.bytes.size ?? file.bytes.length;
+    if (imageBytes > 8 * 1024 * 1024) throw new Error('压缩图片合计不能超过 8 MiB。');
+    uploads.push(file);
+  }
+  for (const original of originals) {
+    if (typeof original !== 'string' || !/\.(png|jpe?g|webp|gif)$/i.test(original)) throw new Error('原图路径无效。');
+  }
+  for (const file of [...originals.map(value => ({ path: value, name: path.basename(value) })), ...files]) {
+    const uploaded = await localFile(file, path.join(uploadsRoot, 'incoming'));
+    uploads.push(uploaded);
+  }
+  for (const file of uploads) {
+    if (/^image\/(?:png|jpeg|webp|gif)$/.test(file.mimeType)) {
+      const metadata = await sharp(file.path || file.bytes).metadata();
+      file.width = metadata.width; file.height = metadata.height;
+      if (!file.width || !file.height) throw new Error('无法读取 图片尺寸。');
+    }
+  }
+  if (uploads.filter(file => file.width).length > 4) throw new Error('每条消息最多 4 张图片。');
+  return uploads;
+}