@lacrous/omnicloud 0.2.25 → 0.2.27

package/dist/index.cjs

@@ -1282,6 +1282,15 @@ var StorageEngine = class {
       throw mapProviderError("Storage listing failed", error);
     }
   }
+  /** Read-only search for stored objects with this content hash and size. */
+  async findByCaption(sha256, size) {
+    if (!this.provider.findByCaption) return null;
+    try {
+      return await this.provider.findByCaption(sha256, size);
+    } catch (error) {
+      throw mapProviderError("Storage lookup failed", error);
+    }
+  }
   async stat(ref) {
     try {
       return await this.provider.stat(ref);
@@ -1500,6 +1509,12 @@ function uploadRequestFingerprint(input) {
 }
 
 // ../core/src/services/file-service.ts
+function contentSha256(input) {
+  return input.spooled?.sha256 ?? sha256Hex(input.data ?? Buffer.alloc(0));
+}
+function contentSize(input) {
+  return input.spooled?.size ?? input.data?.byteLength ?? 0;
+}
 function connectionCausedFailure(error) {
   if (error instanceof TelegramConnectionError) return true;
   if (error instanceof UploadFailedError) {
@@ -1557,6 +1572,9 @@ var FileService = class {
     if (op?.status === "UPLOADING" && op.telegramMessageId !== null) {
       return this.commitStoredObject(userId, input, operations, op);
     }
+    if (op?.status === "UNKNOWN") {
+      return this.resolveUnknown(userId, input, operations, op);
+    }
     if (!op) {
       try {
         op = await operations.create({ userId, operationId, requestFingerprint: fingerprint });
@@ -1586,11 +1604,54 @@ var FileService = class {
       const status = connectionCausedFailure(error) ? "UNKNOWN" : "FAILED";
       await operations.update(op.id, {
         status,
+        sha256: contentSha256(input),
+        size: contentSize(input),
         error: error instanceof Error ? error.message.slice(0, 200) : "upload failed"
       });
       throw error;
     }
   }
+  /**
+   * Resolves an operation whose Telegram outcome is unknown. The channel is searched
+   * for the object by its content hash and size. Exactly one match is adopted, zero
+   * matches means nothing was stored so the upload is sent again, and anything
+   * ambiguous stays UNKNOWN so that no duplicate is written.
+   */
+  async resolveUnknown(userId, input, operations, op) {
+    if (op.sha256 === null || op.size === null) {
+      throw new ConflictError(
+        "This upload's outcome is unknown and cannot be checked automatically. Ask an administrator to reconcile it."
+      );
+    }
+    const engine = await this.engineFor(userId);
+    const matches = await engine.findByCaption(op.sha256, op.size);
+    if (matches === null) {
+      throw new ConflictError(
+        "This upload's outcome is unknown and the storage backend cannot search for it."
+      );
+    }
+    if (matches.length > 1) {
+      throw new ConflictError(
+        "This upload's outcome is unknown and more than one matching object exists. Nothing was written."
+      );
+    }
+    if (matches.length === 1) {
+      const stored = matches[0];
+      await operations.update(op.id, { telegramMessageId: Number(stored.messageId) });
+      await operations.claim(op.id, "UNKNOWN", "COMMITTING");
+      const record = await this.commitRecord(userId, input, {
+        name: sanitizeFileName(input.name),
+        mimeType: mimeFromFilename(sanitizeFileName(input.name)),
+        size: op.size,
+        sha256: op.sha256,
+        telegramMessageId: Number(stored.messageId)
+      });
+      await operations.update(op.id, { status: "COMPLETED", fileId: record.id });
+      return record;
+    }
+    await operations.claim(op.id, "UNKNOWN", "FAILED");
+    return this.uploadOnce(userId, input, op.operationId, operations);
+  }
   /**
    * Follows an operation another request is driving, for a bounded time. Returns
    * the finished file if the winner completes, or the stored object's commit if
@@ -3037,6 +3098,7 @@ var TelegramStorageProvider = class {
         file,
         forceDocument: true,
         workers: 1,
+        caption: input.sha256,
         // Surface transport-level progress to the caller. GramJS reports a
         // 0..1 fraction. `isCanceled` lets us abort an in-flight upload.
         progressCallback: Object.assign(
@@ -3194,6 +3256,41 @@ var TelegramStorageProvider = class {
     }
     return objects;
   }
+  /**
+   * Finds stored documents whose caption equals the content hash and whose size
+   * matches. Used to recover an upload whose outcome was unknown.
+   */
+  async findByCaption(sha256, size) {
+    const matches = [];
+    try {
+      for await (const message of this.client.iterMessages(this.peer, {
+        search: sha256,
+        limit: 20
+      })) {
+        if (!(message instanceof import_telegram2.Api.Message)) continue;
+        if (message.message !== sha256) continue;
+        const document = message.document;
+        if (!(document instanceof import_telegram2.Api.Document)) continue;
+        if (Number(document.size.toString()) !== size) continue;
+        let fileName;
+        for (const attr of document.attributes) {
+          if (attr instanceof import_telegram2.Api.DocumentAttributeFilename) {
+            fileName = attr.fileName;
+            break;
+          }
+        }
+        matches.push({
+          messageId: message.id.toString(),
+          name: fileName ?? "file",
+          size,
+          mimeType: document.mimeType
+        });
+      }
+    } catch (error) {
+      throw mapTelegramError(error, "Telegram lookup failed");
+    }
+    return matches;
+  }
   async stat(ref) {
     const message = await this.getMessage(ref.messageId);
     if (!message?.media) return null;