dsh-mobile-plugin 0.1.74 → 0.1.76

package/lib/mobile/uploads.js

@@ -0,0 +1,220 @@
+'use strict';
+
+/**
+ * 移动端文件分片上传：把手机分享进来的文件落到宿主机固定收件目录。
+ *
+ * 四步：begin（登记文件名，拿 uploadId 与片大小）→ chunk（按 index 顺序追加）→
+ * finish（落盘成 `<收件根>/<日期>/<文件名>`，重名自动加后缀）→ abort（放弃）。
+ *
+ * 为什么分片，而不是一次 POST 传完：
+ *   - 一次传完会把「能传多大的文件」变成一条运维配置 —— 中转的
+ *     `MAX_BODY_BYTES`（dsh-mobile-server/lib/config.js）默认 50MB，插件自己的
+ *     readJsonBody 默认只有 256KB。分片后单请求只有 1MB，与这些上限无关。
+ *   - 手机侧网络不稳，分片能只重传坏掉的那一片，而不是整个文件。
+ *
+ * 为什么片内容走 base64 而不是原始字节：隧道协议本身支持原始字节
+ * （lib/relay/protocol.js 的 httpBody 帧），但 App 侧没有任何「发二进制 body」的
+ * 先例，RN 的 fetch 对 ArrayBuffer body 的支持也随版本变；图片通道本来就以 base64
+ * 走 JSON，这里沿用同一条确定能用的路。代价是 33% 膨胀（单片 1MB → 约 1.37MB），
+ * 仍远低于任何一层上限。
+ *
+ * 顺序与幂等：chunk 只接受「下一片」，重复送达同一片按重传处理（回当前 received，
+ * 不再追加），跳号直接 409 —— 客户端因此不需要在重试前先问进度。
+ *
+ * 落盘根目录固定 `~/.dsh/mobile-inbox`（收件专目录，与工作区解耦，工作区被删也不
+ * 影响已收到的文件）。`DSH_MOBILE_INBOX` 可覆盖，只给测试用。
+ */
+
+const fsp = require('node:fs/promises');
+const os = require('node:os');
+const path = require('node:path');
+const crypto = require('node:crypto');
+
+const { sendJson, readJsonBody } = require('../shared/http-util');
+
+// 片大小 256KB（base64 后约 341KB）：客户端一次请求的硬预算是 8s（`app/src/api/core.ts`
+// 的 REQUEST_TIMEOUT_MS），1MB 片在慢速上行网络上会撞超时；片越小越稳，代价只是多几次往返。
+const CHUNK_BYTES = 256 * 1024;
+/** 单片请求体上限：1MB base64 后约 1.37MB，留出信封与转义余量。 */
+const CHUNK_MAX_BODY = 4 * 1024 * 1024;
+/** 文件名上限（按字符算，避开各文件系统的字节上限差异）。 */
+const NAME_MAX_CHARS = 120;
+/** 暂存残片的最长寿命：手机上传中断后没人来 abort 的清理依据。 */
+const STAGING_TTL_MS = 24 * 60 * 60 * 1000;
+/** 分片必须是标准 base64：Node 的 Buffer.from 会静默丢弃非法字符，光靠解码判不出坏片。 */
+const BASE64_RE = /^[A-Za-z0-9+/]*={0,2}$/;
+
+/** 进行中的上传：uploadId → { name, stagingPath, received, bytes }。 */
+const uploads = new Map();
+
+/** 收件根目录。每次取用都重算，好让测试用环境变量分派。 */
+function inboxRoot() {
+  const override = process.env.DSH_MOBILE_INBOX;
+  return override ? path.resolve(override) : path.join(os.homedir(), '.dsh', 'mobile-inbox');
+}
+
+/**
+ * 清成一个能安全当作文件名的名字：去掉路径成分、文件系统保留字符与控制字符，
+ * 压掉连续空白，掐掉前导点（避免生成隐藏文件），并限长（保住扩展名）。
+ */
+function sanitizeName(raw) {
+  const base = path.basename(String(raw == null ? '' : raw).replace(/\\/g, '/'));
+  const cleaned = base
+    .replace(/[/\\:*?"<>|\u0000-\u001f\u007f]/g, '_')
+    .replace(/\s+/g, ' ')
+    .trim()
+    .replace(/^\.+/, '')
+    .trim();
+  if (!cleaned) return '未命名文件';
+  if (cleaned.length <= NAME_MAX_CHARS) return cleaned;
+  const ext = path.extname(cleaned).slice(0, 16);
+  return cleaned.slice(0, NAME_MAX_CHARS - ext.length) + ext;
+}
+
+/** 本地日期 yyyy-mm-dd（收件目录按天分堆，`os` 本地时区更贴近「今天收到的」）。 */
+function today() {
+  const now = new Date();
+  const pad = (n) => String(n).padStart(2, '0');
+  return `${now.getFullYear()}-${pad(now.getMonth() + 1)}-${pad(now.getDate())}`;
+}
+
+/** 在 dir 下给 name 找一个不冲突的落点（`名字-1.ext`、`名字-2.ext`…）。 */
+async function uniquePath(dir, name) {
+  const ext = path.extname(name);
+  const stem = name.slice(0, name.length - ext.length) || '未命名文件';
+  for (let i = 0; i < 200; i += 1) {
+    const candidate = path.join(dir, i === 0 ? name : `${stem}-${i}${ext}`);
+    try {
+      await fsp.access(candidate);
+    } catch {
+      return candidate;
+    }
+  }
+  return path.join(dir, `${stem}-${Date.now()}${ext}`);
+}
+
+/** 清掉过期残片；尽力而为，失败不影响本次上传。 */
+async function sweepStaging(stagingDir) {
+  let entries;
+  try {
+    entries = await fsp.readdir(stagingDir, { withFileTypes: true });
+  } catch {
+    return;
+  }
+  const cutoff = Date.now() - STAGING_TTL_MS;
+  await Promise.all(entries.map(async (entry) => {
+    if (!entry.isFile() || !entry.name.endsWith('.part')) return;
+    const target = path.join(stagingDir, entry.name);
+    try {
+      const stat = await fsp.stat(target);
+      if (stat.mtimeMs < cutoff) await fsp.unlink(target);
+    } catch { /* 已被别人清掉 */ }
+  }));
+}
+
+function fail(res, status, code, message) {
+  return sendJson(res, status, { ok: false, error: { code, message } });
+}
+
+/** POST /api/mobile/v1/upload/begin → { ok, uploadId, chunkSize }。 */
+async function handleUploadBegin(_ctx, req, res) {
+  const body = await readJsonBody(req);
+  if (!body || typeof body.name !== 'string' || !body.name.trim()) {
+    return fail(res, 400, 'bad-request', '需要 { name }');
+  }
+  const stagingDir = path.join(inboxRoot(), '.staging');
+  try {
+    await fsp.mkdir(stagingDir, { recursive: true });
+  } catch (error) {
+    return fail(res, 500, 'inbox-unavailable', `收件目录不可用：${error.message}`);
+  }
+  void sweepStaging(stagingDir);
+
+  const uploadId = crypto.randomUUID();
+  const stagingPath = path.join(stagingDir, `${uploadId}.part`);
+  try {
+    await fsp.writeFile(stagingPath, Buffer.alloc(0));
+  } catch (error) {
+    return fail(res, 500, 'inbox-unavailable', `无法建立暂存文件：${error.message}`);
+  }
+  uploads.set(uploadId, {
+    name: sanitizeName(body.name),
+    stagingPath,
+    received: 0,
+    bytes: 0,
+  });
+  sendJson(res, 200, { ok: true, uploadId, chunkSize: CHUNK_BYTES });
+}
+
+/** POST /api/mobile/v1/upload/chunk { uploadId, index, data } → { ok, received }。 */
+async function handleUploadChunk(_ctx, req, res) {
+  const body = await readJsonBody(req, CHUNK_MAX_BODY);
+  if (!body) return fail(res, 413, 'too-large', '分片过大或不是合法 JSON');
+
+  const job = uploads.get(String(body.uploadId == null ? '' : body.uploadId));
+  if (!job) return fail(res, 404, 'unknown-upload', '上传会话不存在或已结束');
+
+  const index = Number(body.index);
+  if (!Number.isInteger(index) || index < 0) return fail(res, 400, 'bad-request', 'index 必须是非负整数');
+  if (typeof body.data !== 'string') return fail(res, 400, 'bad-request', 'data 必须是 base64 字符串');
+  if (index < job.received) return sendJson(res, 200, { ok: true, received: job.received });
+  if (index > job.received) {
+    return fail(res, 409, 'out-of-order', `期望分片 ${job.received}，收到 ${index}`);
+  }
+
+  const chunk = Buffer.from(body.data, 'base64');
+  if (!BASE64_RE.test(body.data) || body.data.length % 4 !== 0) {
+    return fail(res, 400, 'bad-request', 'data 不是合法 base64');
+  }
+  try {
+    await fsp.appendFile(job.stagingPath, chunk);
+  } catch (error) {
+    uploads.delete(String(body.uploadId));
+    return fail(res, 500, 'inbox-unavailable', `写入暂存文件失败：${error.message}`);
+  }
+  job.received += 1;
+  job.bytes += chunk.length;
+  sendJson(res, 200, { ok: true, received: job.received });
+}
+
+/** POST /api/mobile/v1/upload/finish { uploadId } → { ok, path, name, bytes }。 */
+async function handleUploadFinish(_ctx, req, res) {
+  const body = await readJsonBody(req);
+  const uploadId = String((body && body.uploadId) || '');
+  const job = uploads.get(uploadId);
+  if (!job) return fail(res, 404, 'unknown-upload', '上传会话不存在或已结束');
+  uploads.delete(uploadId);
+
+  const dir = path.join(inboxRoot(), today());
+  try {
+    await fsp.mkdir(dir, { recursive: true });
+    const target = await uniquePath(dir, job.name);
+    await fsp.rename(job.stagingPath, target);
+    sendJson(res, 200, { ok: true, path: target, name: path.basename(target), bytes: job.bytes });
+  } catch (error) {
+    try { await fsp.unlink(job.stagingPath); } catch { /* 已经不在了 */ }
+    return fail(res, 500, 'inbox-unavailable', `落盘失败：${error.message}`);
+  }
+}
+
+/** POST /api/mobile/v1/upload/abort { uploadId } → { ok }。 */
+async function handleUploadAbort(_ctx, req, res) {
+  const body = await readJsonBody(req);
+  const uploadId = String((body && body.uploadId) || '');
+  const job = uploads.get(uploadId);
+  if (job) {
+    uploads.delete(uploadId);
+    try { await fsp.unlink(job.stagingPath); } catch { /* 已经不在了 */ }
+  }
+  sendJson(res, 200, { ok: true });
+}
+
+module.exports = {
+  handleUploadBegin,
+  handleUploadChunk,
+  handleUploadFinish,
+  handleUploadAbort,
+  sanitizeName,
+  inboxRoot,
+  CHUNK_BYTES,
+};

package/lib/mobile/workspaces.js

@@ -15,7 +15,7 @@
  */
 
 const { sendJson, readJsonBody } = require('../shared/http-util');
-const { listRevision } = require('../shared/list-revision.js');
+const { stableRevision } = require('../shared/list-revision.js');
 // 会话列表缓存归 sessions.js 管。归档/移动会话会改变列表行本身，必须让那边立刻失效，
 // 否则手机端按完「归档会话」，那一行最长 10 秒还留在侧栏里，看着像没归档。
 const { invalidateSessionList } = require('./sessions.js');
@@ -74,7 +74,8 @@ async function buildWorkspacesBody(ctx) {
     // App 把 generatedAt + revision 存进快照用于展示「数据时间」，emptyConfirmed
     // 让「用户真的删光了工作区」不必再靠连续两次空去猜。
     generatedAt: Date.now(),
-    revision: listRevision(items),
+    // 归档集合也是工作区快照语义的一部分；仅它变化时也必须让镜像推送判为新版本。
+    revision: stableRevision({ items, archivedSessionIds }),
     emptyConfirmed: items.length === 0,
   };
 }