secure-upload-fastify-sdk 1.0.4 → 1.0.6

package/dist/index.mjs

@@ -300,7 +300,12 @@ function toFileRecord(row) {
     meta: row.meta && typeof row.meta === "object" ? row.meta : null,
     createdAt: row.createdAt,
     updatedAt: row.updatedAt,
-    deletedAt: row.deletedAt
+    deletedAt: row.deletedAt,
+    // v1.5.1+:水印溯源字段(可选;旧 schema/老库里 row 可能没有,需要兜底)
+    watermarked: row.watermarked ?? false,
+    isOriginal: row.isOriginal ?? true,
+    parentId: row.parentId ?? null,
+    retentionExpiresAt: row.retentionExpiresAt ?? null
   };
 }
 var PrismaMultipartRepository = class {
@@ -466,6 +471,27 @@ var PrismaFileRepository = class {
     });
     return toFileRecord(row);
   }
+  /**
+   * v1.5.1: 钩子 onComplete 返回 patch 后 SDK 调这个方法
+   *   - 只更新传入的字段;其他字段保留原值
+   *   - 文件不存在 → 抛错(由调用方决定是否捕获)
+   *   - 业务字段 watermarked/isOriginal/parentId/retentionExpiresAt 需要 schema v1.4.0+
+   */
+  async update(id, patch) {
+    const data = {};
+    if (patch.storageKey !== void 0) data.storageKey = patch.storageKey;
+    if (patch.size !== void 0) data.size = BigInt(patch.size);
+    if (patch.contentType !== void 0) data.contentType = patch.contentType;
+    if (patch.meta !== void 0) data.meta = patch.meta;
+    if (patch.watermarked !== void 0) data.watermarked = patch.watermarked;
+    if (patch.isOriginal !== void 0) data.isOriginal = patch.isOriginal;
+    if (patch.parentId !== void 0) data.parentId = patch.parentId;
+    if (patch.retentionExpiresAt !== void 0) {
+      data.retentionExpiresAt = patch.retentionExpiresAt;
+    }
+    const row = await this.prisma.file.update({ where: { id }, data });
+    return toFileRecord(row);
+  }
   async softDelete(id, metaPatch) {
     if (metaPatch) {
       const cur = await this.prisma.file.findUnique({ where: { id } });
@@ -692,6 +718,30 @@ var InMemoryFileRepository = class {
     };
     this.store.set(id, next);
   }
+  /**
+   * v1.5.1: 钩子 onComplete 返回 patch 后 SDK 调这个方法
+   *   - 只更新传入的字段;其他字段保留原值
+   *   - 文件不存在 → 抛错(由调用方决定是否捕获)
+   */
+  async update(id, patch) {
+    const rec = this.store.get(id);
+    if (!rec) throw new Error("NOT_FOUND");
+    const now = /* @__PURE__ */ new Date();
+    const next = {
+      ...rec,
+      storageKey: patch.storageKey ?? rec.storageKey,
+      size: patch.size !== void 0 ? BigInt(patch.size) : rec.size,
+      contentType: patch.contentType ?? rec.contentType,
+      meta: patch.meta === void 0 ? rec.meta : patch.meta,
+      watermarked: patch.watermarked ?? rec.watermarked,
+      isOriginal: patch.isOriginal ?? rec.isOriginal,
+      parentId: patch.parentId === void 0 ? rec.parentId : patch.parentId,
+      retentionExpiresAt: patch.retentionExpiresAt === void 0 ? rec.retentionExpiresAt : patch.retentionExpiresAt,
+      updatedAt: now
+    };
+    this.store.set(id, next);
+    return next;
+  }
   async listOrphans(opts) {
     const sinceMs = opts.sinceDays * 24 * 3600 * 1e3;
     const list = [];
@@ -1139,6 +1189,47 @@ async function lock(rt, key, ttlMs, fn, retries = 0) {
   }
   return await memoryWithLock(key, fn, retries);
 }
+async function streamToBuffer(stream) {
+  const chunks = [];
+  for await (const chunk of stream) {
+    chunks.push(typeof chunk === "string" ? Buffer.from(chunk) : chunk);
+  }
+  return Buffer.concat(chunks);
+}
+async function runOnComplete(rt, file, log) {
+  if (!rt.onComplete) return file;
+  try {
+    const patch = await rt.onComplete({
+      file,
+      storage: rt.storage,
+      log,
+      getOriginalBuffer: async () => {
+        const stream = await Promise.resolve(rt.storage.get(file.storageKey));
+        return streamToBuffer(stream);
+      }
+    });
+    if (!patch || Object.keys(patch).length === 0) {
+      return file;
+    }
+    const updated = await rt.fileRepo.update(file.id, patch);
+    log.info(
+      {
+        fileId: file.id,
+        patchKeys: Object.keys(patch),
+        newStorageKey: patch.storageKey,
+        watermarked: patch.watermarked
+      },
+      "\u2705 onComplete \u94A9\u5B50\u5DF2\u5E94\u7528 patch"
+    );
+    return updated;
+  } catch (err) {
+    log.warn(
+      { err, fileId: file.id, uploadId: file.uploadId },
+      "\u26A0\uFE0F onComplete \u94A9\u5B50\u6267\u884C\u5931\u8D25,\u5DF2\u8DF3\u8FC7\u4E1A\u52A1\u540E\u5904\u7406(file record \u4FDD\u6301\u539F\u72B6)"
+    );
+    return file;
+  }
+}
 async function registerMultipartRoutes(app, rt) {
   if (!rt.config.enabled) {
     app.log.warn("\u26A0\uFE0F  multipart \u4E0A\u4F20\u5DF2\u7981\u7528 (enabled=false)");
@@ -1444,8 +1535,14 @@ async function registerMultipartRoutes(app, rt) {
           contentHash: finalHash,
           meta: rec.meta ?? void 0
         });
+        const finalFile = await runOnComplete(rt, file2, req.log);
         await rt.multipartRepo.markCompleted(rec.id, fileId);
-        return { idempotent: false, fileId: file2.id, size: merged.size, finalHash };
+        return {
+          idempotent: false,
+          fileId: finalFile.id,
+          size: Number(finalFile.size),
+          finalHash
+        };
       }
     );
     if (!result.idempotent) {
@@ -1458,7 +1555,8 @@ async function registerMultipartRoutes(app, rt) {
         uploadId: body.uploadId,
         fileId: result.fileId,
         size: result.size,
-        userId: rec.userId
+        userId: rec.userId,
+        watermarkHook: rt.onComplete ? "on" : "off"
       },
       "\u2705 multipart: complete"
     );
@@ -1934,7 +2032,8 @@ async function secureUploadPluginImpl(app, options) {
         ...options.strategy ?? {}
       }
     },
-    validateSession: options.sessionValidator ?? defaultSessionValidator
+    validateSession: options.sessionValidator ?? defaultSessionValidator,
+    onComplete: options.onComplete ?? null
   };
   app.decorate("secureUpload", runtime);
   await registerMultipartRoutes(app, runtime);