secure-upload-fastify-sdk 1.0.5 → 1.0.6

package/dist/server/fastify-plugin.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fastify-plugin.d.ts","sourceRoot":"","sources":["../../src/server/fastify-plugin.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AAGH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAC5F,OAAO,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AAQhG,OAAO,KAAK,EACV,mBAAmB,EAIpB,MAAM,SAAS,CAAC;AAmDjB,iBAAe,sBAAsB,CACnC,GAAG,EAAE,eAAe,EACpB,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,IAAI,CAAC,CAqHf;;AAED,wBAGG;AAKH,OAAO,EACL,YAAY,EACZ,eAAe,EACf,yBAAyB,EACzB,oBAAoB,EACpB,2BAA2B,EAC3B,sBAAsB,GACvB,CAAC"}
+{"version":3,"file":"fastify-plugin.d.ts","sourceRoot":"","sources":["../../src/server/fastify-plugin.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AAGH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAC5F,OAAO,EAAE,2BAA2B,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AAQhG,OAAO,KAAK,EACV,mBAAmB,EAIpB,MAAM,SAAS,CAAC;AAmDjB,iBAAe,sBAAsB,CACnC,GAAG,EAAE,eAAe,EACpB,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,IAAI,CAAC,CAsHf;;AAED,wBAGG;AAKH,OAAO,EACL,YAAY,EACZ,eAAe,EACf,yBAAyB,EACzB,oBAAoB,EACpB,2BAA2B,EAC3B,sBAAsB,GACvB,CAAC"}

package/dist/server/index.d.ts

@@ -12,6 +12,6 @@
  *   import { chooseStrategy, suggestConcurrency, DEFAULT_STRATEGY_CONFIG } from 'secure-upload-fastify-sdk/strategy';
  */
 export { default, default as secureUploadPlugin, LocalStorage, InMemoryStorage, PrismaMultipartRepository, PrismaFileRepository, InMemoryMultipartRepository, InMemoryFileRepository, } from './fastify-plugin';
-export type { SecureUploadOptions, SecureUploadRuntime, SecureMode, RepositoryConfig, StorageConfig, PathPrefixConfig, SessionValidator, InitResponse, PartResponse, CompleteResponse, AbortResponse, StatusResponse, MultipartRecord, FileRecord, } from './types';
+export type { SecureUploadOptions, SecureUploadRuntime, SecureMode, RepositoryConfig, StorageConfig, PathPrefixConfig, SessionValidator, InitResponse, PartResponse, CompleteResponse, AbortResponse, StatusResponse, MultipartRecord, FileRecord, OnCompleteHook, OnCompleteContext, OnCompletePatch, } from './types';
 export { ErrorCode, SecureError, getHttpStatus, } from './errors';
 //# sourceMappingURL=index.d.ts.map

package/dist/server/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EACL,OAAO,EACP,OAAO,IAAI,kBAAkB,EAC7B,YAAY,EACZ,eAAe,EACf,yBAAyB,EACzB,oBAAoB,EACpB,2BAA2B,EAC3B,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EACV,mBAAmB,EACnB,mBAAmB,EACnB,UAAU,EACV,gBAAgB,EAChB,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,YAAY,EACZ,YAAY,EACZ,gBAAgB,EAChB,aAAa,EACb,cAAc,EACd,eAAe,EACf,UAAU,GACX,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,GACd,MAAM,UAAU,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EACL,OAAO,EACP,OAAO,IAAI,kBAAkB,EAC7B,YAAY,EACZ,eAAe,EACf,yBAAyB,EACzB,oBAAoB,EACpB,2BAA2B,EAC3B,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EACV,mBAAmB,EACnB,mBAAmB,EACnB,UAAU,EACV,gBAAgB,EAChB,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,YAAY,EACZ,YAAY,EACZ,gBAAgB,EAChB,aAAa,EACb,cAAc,EACd,eAAe,EACf,UAAU,EACV,cAAc,EACd,iBAAiB,EACjB,eAAe,GAChB,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,GACd,MAAM,UAAU,CAAC"}

package/dist/server/index.js

@@ -346,7 +346,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 {
@@ -512,6 +517,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 } });
@@ -738,6 +764,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 = [];
@@ -1185,6 +1235,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)");
@@ -1490,8 +1581,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) {
@@ -1504,7 +1601,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"
     );
@@ -1980,7 +2078,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);