@protontech/drive-sdk 0.0.10

package/dist/internal/download/telemetry.js

@@ -0,0 +1,106 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DownloadTelemetry = void 0;
+const errors_1 = require("../../errors");
+const telemetry_1 = require("../../telemetry");
+const apiService_1 = require("../apiService");
+const uids_1 = require("../uids");
+class DownloadTelemetry {
+    telemetry;
+    sharesService;
+    logger;
+    constructor(telemetry, sharesService) {
+        this.telemetry = telemetry;
+        this.sharesService = sharesService;
+        this.telemetry = telemetry;
+        this.logger = this.telemetry.getLogger("download");
+        this.sharesService = sharesService;
+    }
+    getLoggerForRevision(revisionUid) {
+        return new telemetry_1.LoggerWithPrefix(this.logger, `revision ${revisionUid}`);
+    }
+    async downloadInitFailed(nodeUid, error) {
+        const { volumeId } = (0, uids_1.splitNodeUid)(nodeUid);
+        const errorCategory = getErrorCategory(error);
+        // No error category means ignored error from telemetry.
+        // For example, aborted request.
+        if (!errorCategory) {
+            return;
+        }
+        await this.sendTelemetry(volumeId, {
+            downloadedSize: 0,
+            error: errorCategory,
+        });
+    }
+    async downloadFailed(revisionUid, error, downloadedSize, claimedFileSize) {
+        const { volumeId } = (0, uids_1.splitNodeRevisionUid)(revisionUid);
+        const errorCategory = getErrorCategory(error);
+        // No error category means ignored error from telemetry.
+        // For example, aborted request.
+        if (!errorCategory) {
+            return;
+        }
+        await this.sendTelemetry(volumeId, {
+            downloadedSize,
+            claimedFileSize,
+            error: errorCategory,
+        });
+    }
+    async downloadFinished(revisionUid, downloadedSize) {
+        const { volumeId } = (0, uids_1.splitNodeRevisionUid)(revisionUid);
+        await this.sendTelemetry(volumeId, {
+            downloadedSize,
+            claimedFileSize: downloadedSize,
+        });
+    }
+    async sendTelemetry(volumeId, options) {
+        let context;
+        try {
+            context = await this.sharesService.getVolumeMetricContext(volumeId);
+        }
+        catch (error) {
+            this.logger.error('Failed to get metric context', error);
+        }
+        this.telemetry.logEvent({
+            eventName: 'download',
+            context,
+            ...options,
+        });
+    }
+}
+exports.DownloadTelemetry = DownloadTelemetry;
+function getErrorCategory(error) {
+    if (error instanceof errors_1.ValidationError) {
+        return undefined;
+    }
+    if (error instanceof errors_1.RateLimitedError) {
+        return 'rate_limited';
+    }
+    if (error instanceof errors_1.DecryptionError) {
+        return 'decryption_error';
+    }
+    if (error instanceof errors_1.IntegrityError) {
+        return 'integrity_error';
+    }
+    if (error instanceof apiService_1.APIHTTPError) {
+        if (error.statusCode >= 400 && error.statusCode < 500) {
+            return '4xx';
+        }
+        if (error.statusCode >= 500) {
+            return '5xx';
+        }
+    }
+    if (error instanceof Error) {
+        if (error.name === 'TimeoutError') {
+            return 'server_error';
+        }
+        if (error.name === 'OfflineError' || error.name === 'NetworkError' || error.message?.toLowerCase() === 'network error') {
+            return 'network_error';
+        }
+        if (error.name === 'AbortError') {
+            return undefined;
+        }
+    }
+    return 'unknown';
+}
+//# sourceMappingURL=telemetry.js.map

package/dist/internal/download/telemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../../src/internal/download/telemetry.ts"],"names":[],"mappings":";;;AAAA,yCAAkG;AAElG,+CAAmD;AACnD,8CAA6C;AAC7C,kCAA6D;AAG7D,MAAa,iBAAiB;IAGN;IAAyC;IAFrD,MAAM,CAAS;IAEvB,YAAoB,SAA+B,EAAU,aAA4B;QAArE,cAAS,GAAT,SAAS,CAAsB;QAAU,kBAAa,GAAb,aAAa,CAAe;QACrF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACvC,CAAC;IAED,oBAAoB,CAAC,WAAmB;QACpC,OAAO,IAAI,4BAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,WAAW,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,KAAc;QACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,mBAAY,EAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE9C,wDAAwD;QACxD,gCAAgC;QAChC,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,OAAO;QACX,CAAC;QAED,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE;YAC/B,cAAc,EAAE,CAAC;YACjB,KAAK,EAAE,aAAa;SACvB,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,WAAmB,EAAE,KAAc,EAAE,cAAsB,EAAE,eAAwB;QACtG,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,2BAAoB,EAAC,WAAW,CAAC,CAAC;QACvD,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE9C,wDAAwD;QACxD,gCAAgC;QAChC,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,OAAO;QACX,CAAC;QAED,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE;YAC/B,cAAc;YACd,eAAe;YACf,KAAK,EAAE,aAAa;SACvB,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,WAAmB,EAAE,cAAsB;QAC9D,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,2BAAoB,EAAC,WAAW,CAAC,CAAC;QACvD,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE;YAC/B,cAAc;YACd,eAAe,EAAE,cAAc;SAClC,CAAC,CAAC;IACP,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,QAAgB,EAAE,OAI7C;QACG,IAAI,OAAO,CAAC;QACZ,IAAI,CAAC;YACD,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACxE,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;YACpB,SAAS,EAAE,UAAU;YACrB,OAAO;YACP,GAAG,OAAO;SACb,CAAC,CAAC;IACP,CAAC;CACJ;AAxED,8CAwEC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACpC,IAAI,KAAK,YAAY,wBAAe,EAAE,CAAC;QACnC,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,IAAI,KAAK,YAAY,yBAAgB,EAAE,CAAC;QACpC,OAAO,cAAc,CAAC;IAC1B,CAAC;IACD,IAAI,KAAK,YAAY,wBAAe,EAAE,CAAC;QACnC,OAAO,kBAAkB,CAAC;IAC9B,CAAC;IACD,IAAI,KAAK,YAAY,uBAAc,EAAE,CAAC;QAClC,OAAO,iBAAiB,CAAC;IAC7B,CAAC;IACD,IAAI,KAAK,YAAY,yBAAY,EAAE,CAAC;QAChC,IAAI,KAAK,CAAC,UAAU,IAAI,GAAG,IAAI,KAAK,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC;YACpD,OAAO,KAAK,CAAC;QACjB,CAAC;QACD,IAAI,KAAK,CAAC,UAAU,IAAI,GAAG,EAAE,CAAC;YAC1B,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IACD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QACzB,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YAChC,OAAO,cAAc,CAAC;QAC1B,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,IAAI,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,KAAK,eAAe,EAAE,CAAC;YACrH,OAAO,eAAe,CAAC;QAC3B,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC9B,OAAO,SAAS,CAAC;QACrB,CAAC;IACL,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC"}

package/dist/internal/download/telemetry.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/internal/download/telemetry.test.js

@@ -0,0 +1,111 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const errors_1 = require("../../errors");
+const apiService_1 = require("../apiService");
+const telemetry_1 = require("./telemetry");
+describe('DownloadTelemetry', () => {
+    let mockTelemetry;
+    let sharesService;
+    let downloadTelemetry;
+    const nodeUid = 'volumeId~nodeId';
+    const revisionUid = 'volumeId~nodeId~revisionId';
+    beforeEach(() => {
+        mockTelemetry = {
+            logEvent: jest.fn(),
+            getLogger: jest.fn().mockReturnValue({
+                info: jest.fn(),
+                warn: jest.fn(),
+                error: jest.fn(),
+                debug: jest.fn(),
+            }),
+        };
+        sharesService = {
+            getVolumeMetricContext: jest.fn().mockResolvedValue('own_volume'),
+        };
+        downloadTelemetry = new telemetry_1.DownloadTelemetry(mockTelemetry, sharesService);
+    });
+    it('should log failure during init (excludes file size)', async () => {
+        await downloadTelemetry.downloadInitFailed(nodeUid, new Error('Failed'));
+        expect(mockTelemetry.logEvent).toHaveBeenCalledWith({
+            eventName: "download",
+            context: "own_volume",
+            downloadedSize: 0,
+            error: "unknown",
+        });
+    });
+    it('should log failure download', async () => {
+        await downloadTelemetry.downloadFailed(revisionUid, new Error('Failed'), 123, 456);
+        expect(mockTelemetry.logEvent).toHaveBeenCalledWith({
+            eventName: "download",
+            context: "own_volume",
+            downloadedSize: 123,
+            claimedFileSize: 456,
+            error: "unknown",
+        });
+    });
+    it('should log successful download (excludes error)', async () => {
+        await downloadTelemetry.downloadFinished(revisionUid, 500);
+        expect(mockTelemetry.logEvent).toHaveBeenCalledWith({
+            eventName: "download",
+            context: "own_volume",
+            downloadedSize: 500,
+            claimedFileSize: 500,
+        });
+    });
+    describe('detect error category', () => {
+        const verifyErrorCategory = (error) => {
+            expect(mockTelemetry.logEvent).toHaveBeenCalledWith(expect.objectContaining({
+                error,
+            }));
+        };
+        it('should ignore ValidationError', async () => {
+            const error = new errors_1.ValidationError('Validation error');
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            expect(mockTelemetry.logEvent).not.toHaveBeenCalled();
+        });
+        it('should ignore AbortError', async () => {
+            const error = new Error('Aborted');
+            error.name = 'AbortError';
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            expect(mockTelemetry.logEvent).not.toHaveBeenCalled();
+        });
+        it('should detect "rate_limited" error for RateLimitedError', async () => {
+            const error = new errors_1.RateLimitedError('Rate limited');
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('rate_limited');
+        });
+        it('should detect "decryption_error" for DecryptionError', async () => {
+            const error = new errors_1.DecryptionError('Decryption failed');
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('decryption_error');
+        });
+        it('should detect "integrity_error" for IntegrityError', async () => {
+            const error = new errors_1.IntegrityError('Integrity check failed');
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('integrity_error');
+        });
+        it('should detect "4xx" error for APIHTTPError with 4xx status code', async () => {
+            const error = new apiService_1.APIHTTPError('Client error', 404);
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('4xx');
+        });
+        it('should detect "5xx" error for APIHTTPError with 5xx status code', async () => {
+            const error = new apiService_1.APIHTTPError('Server error', 500);
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('5xx');
+        });
+        it('should detect "server_error" for TimeoutError', async () => {
+            const error = new Error('Timeout');
+            error.name = 'TimeoutError';
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('server_error');
+        });
+        it('should detect "network_error" for NetworkError', async () => {
+            const error = new Error('Network error');
+            error.name = 'NetworkError';
+            await downloadTelemetry.downloadFailed(revisionUid, error, 100, 200);
+            verifyErrorCategory('network_error');
+        });
+    });
+});
+//# sourceMappingURL=telemetry.test.js.map

package/dist/internal/download/telemetry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.test.js","sourceRoot":"","sources":["../../../src/internal/download/telemetry.test.ts"],"names":[],"mappings":";;AAAA,yCAAkG;AAElG,8CAA6C;AAE7C,2CAAgD;AAEhD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAC/B,IAAI,aAAgD,CAAC;IACrD,IAAI,aAAyC,CAAC;IAC9C,IAAI,iBAAoC,CAAC;IAEzC,MAAM,OAAO,GAAG,iBAAiB,CAAC;IAClC,MAAM,WAAW,GAAG,4BAA4B,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACZ,aAAa,GAAG;YACZ,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;aACnB,CAAC;SAC2C,CAAC;QAElD,aAAa,GAAG;YACZ,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,YAAY,CAAC;SACpE,CAAA;QAED,iBAAiB,GAAG,IAAI,6BAAiB,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,iBAAiB,CAAC,kBAAkB,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEzE,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;YAChD,SAAS,EAAE,UAAU;YACrB,OAAO,EAAE,YAAY;YACrB,cAAc,EAAE,CAAC;YACjB,KAAK,EAAE,SAAS;SACnB,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,IAAI,KAAK,CAAC,QAAQ,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAEnF,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;YAChD,SAAS,EAAE,UAAU;YACrB,OAAO,EAAE,YAAY;YACrB,cAAc,EAAE,GAAG;YACnB,eAAe,EAAE,GAAG;YACpB,KAAK,EAAE,SAAS;SACnB,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,iBAAiB,CAAC,gBAAgB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAE3D,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;YAChD,SAAS,EAAE,UAAU;YACrB,OAAO,EAAE,YAAY;YACrB,cAAc,EAAE,GAAG;YACnB,eAAe,EAAE,GAAG;SACvB,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACnC,MAAM,mBAAmB,GAAG,CAAC,KAAa,EAAE,EAAE;YAC1C,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;gBACpB,KAAK;aACR,CAAC,CACL,CAAC;QACN,CAAC,CAAA;QAED,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,KAAK,GAAG,IAAI,wBAAe,CAAC,kBAAkB,CAAC,CAAC;YACtD,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;YACnC,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;YAC1B,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAErE,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,KAAK,GAAG,IAAI,yBAAgB,CAAC,cAAc,CAAC,CAAC;YACnD,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,cAAc,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,KAAK,GAAG,IAAI,wBAAe,CAAC,mBAAmB,CAAC,CAAC;YACvD,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,kBAAkB,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,KAAK,GAAG,IAAI,uBAAc,CAAC,wBAAwB,CAAC,CAAC;YAC3D,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,KAAK,GAAG,IAAI,yBAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;YACpD,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,KAAK,GAAG,IAAI,yBAAY,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;YACpD,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;YACnC,KAAK,CAAC,IAAI,GAAG,cAAc,CAAC;YAC5B,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,cAAc,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACzC,KAAK,CAAC,IAAI,GAAG,cAAc,CAAC;YAC5B,MAAM,iBAAiB,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACrE,mBAAmB,CAAC,eAAe,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/dist/internal/download/thumbnailDownloader.d.ts

@@ -0,0 +1,19 @@
+import { ThumbnailType, ProtonDriveTelemetry, ThumbnailResult } from "../../interface";
+import { DownloadAPIService } from "./apiService";
+import { DownloadCryptoService } from "./cryptoService";
+import { NodesService } from "./interface";
+export declare class ThumbnailDownloader {
+    private nodesService;
+    private apiService;
+    private cryptoService;
+    private logger;
+    private batchThumbnailToNodeUids;
+    private ongoingDownloads;
+    private bufferedThumbnails;
+    constructor(telemetry: ProtonDriveTelemetry, nodesService: NodesService, apiService: DownloadAPIService, cryptoService: DownloadCryptoService);
+    iterateThumbnails(nodeUids: string[], thumbnailType?: ThumbnailType, signal?: AbortSignal): AsyncGenerator<ThumbnailResult>;
+    private iterateThumbnailUids;
+    private requestBatchedThumbnailDownloads;
+    private iterateThumbnailDownloads;
+    private downloadThumbnail;
+}

package/dist/internal/download/thumbnailDownloader.js

@@ -0,0 +1,198 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ThumbnailDownloader = void 0;
+const ttag_1 = require("ttag");
+const interface_1 = require("../../interface");
+const errors_1 = require("../../errors");
+const telemetry_1 = require("../../telemetry");
+const errors_2 = require("../errors");
+/**
+ * Maximum number of thumbnails that can be downloaded at the same time.
+ */
+const MAX_DOWNLOAD_THUMBNAILS = 10;
+/**
+ * Maximum number of retries for thumbnail download and decryption.
+ */
+const MAX_THUMBNAIL_DOWNLOAD_ATTEMPTS = 2;
+class ThumbnailDownloader {
+    nodesService;
+    apiService;
+    cryptoService;
+    logger;
+    batchThumbnailToNodeUids = new Map();
+    ongoingDownloads = new Map();
+    bufferedThumbnails = [];
+    constructor(telemetry, nodesService, apiService, cryptoService) {
+        this.nodesService = nodesService;
+        this.apiService = apiService;
+        this.cryptoService = cryptoService;
+        this.logger = telemetry.getLogger("download");
+        this.nodesService = nodesService;
+        this.apiService = apiService;
+        this.cryptoService = cryptoService;
+    }
+    async *iterateThumbnails(nodeUids, thumbnailType = interface_1.ThumbnailType.Type1, signal) {
+        if (nodeUids.length === 0) {
+            return;
+        }
+        for await (const result of this.iterateThumbnailUids(nodeUids, thumbnailType, signal)) {
+            if (!result.ok) {
+                yield result;
+                continue;
+            }
+            this.batchThumbnailToNodeUids.set(result.thumbnailUid, result.nodeUid);
+            if (this.batchThumbnailToNodeUids.size >= MAX_DOWNLOAD_THUMBNAILS) {
+                await this.requestBatchedThumbnailDownloads(signal);
+            }
+            while (this.ongoingDownloads.size >= MAX_DOWNLOAD_THUMBNAILS) {
+                await Promise.race(this.ongoingDownloads.values());
+                yield* this.bufferedThumbnails;
+                this.bufferedThumbnails = [];
+            }
+        }
+        await this.requestBatchedThumbnailDownloads(signal);
+        while (this.ongoingDownloads.size > 0) {
+            await Promise.race(this.ongoingDownloads.values());
+            yield* this.bufferedThumbnails;
+            this.bufferedThumbnails = [];
+        }
+        yield* this.bufferedThumbnails;
+        this.bufferedThumbnails = [];
+    }
+    async *iterateThumbnailUids(nodeUids, thumbnailType, signal) {
+        for await (const node of this.nodesService.iterateNodes(nodeUids, signal)) {
+            if ('missingUid' in node) {
+                yield {
+                    nodeUid: node.missingUid,
+                    ok: false,
+                    error: (0, ttag_1.c)("Error").t `Node not found`,
+                };
+                continue;
+            }
+            if (node.type !== interface_1.NodeType.File) {
+                yield {
+                    nodeUid: node.uid,
+                    ok: false,
+                    error: (0, ttag_1.c)("Error").t `Node is not a file`,
+                };
+                continue;
+            }
+            let thumbnail;
+            if (node.activeRevision?.ok) {
+                thumbnail = node.activeRevision.value.thumbnails?.find((t) => t.type === thumbnailType);
+            }
+            if (!thumbnail) {
+                yield {
+                    nodeUid: node.uid,
+                    ok: false,
+                    error: (0, ttag_1.c)("Error").t `Node has no thumbnail`,
+                };
+                continue;
+            }
+            yield {
+                nodeUid: node.uid,
+                ok: true,
+                thumbnailUid: thumbnail.uid,
+            };
+        }
+    }
+    async requestBatchedThumbnailDownloads(signal) {
+        if (this.batchThumbnailToNodeUids.size === 0) {
+            return;
+        }
+        this.logger.debug(`Downloading thumbnail batch of size ${this.batchThumbnailToNodeUids.size}`);
+        for await (const downloadResult of this.iterateThumbnailDownloads(signal)) {
+            if (!downloadResult.ok) {
+                this.bufferedThumbnails.push({
+                    nodeUid: downloadResult.nodeUid,
+                    ok: false,
+                    error: downloadResult.error,
+                });
+                continue;
+            }
+            this.ongoingDownloads.set(downloadResult.nodeUid, downloadResult.downloadPromise
+                .then((thumbnail) => {
+                this.bufferedThumbnails.push({
+                    nodeUid: downloadResult.nodeUid,
+                    ok: true,
+                    thumbnail,
+                });
+            })
+                .catch((error) => {
+                this.bufferedThumbnails.push({
+                    nodeUid: downloadResult.nodeUid,
+                    ok: false,
+                    error: (0, errors_2.getErrorMessage)(error),
+                });
+            })
+                .finally(() => {
+                this.ongoingDownloads.delete(downloadResult.nodeUid);
+            }));
+        }
+        this.batchThumbnailToNodeUids.clear();
+    }
+    async *iterateThumbnailDownloads(signal) {
+        const missingThumbnailUids = new Set(this.batchThumbnailToNodeUids.keys());
+        for await (const result of this.apiService.iterateThumbnails(Array.from(this.batchThumbnailToNodeUids.keys()), signal)) {
+            const nodeUid = this.batchThumbnailToNodeUids.get(result.uid);
+            if (!nodeUid) {
+                this.logger.warn(`Unexpected thumbnail UID ${result.uid} returned from API`);
+                continue;
+            }
+            missingThumbnailUids.delete(result.uid);
+            if (!result.ok) {
+                yield {
+                    nodeUid,
+                    ok: false,
+                    error: result.error,
+                };
+                continue;
+            }
+            yield {
+                nodeUid,
+                ok: true,
+                downloadPromise: this.downloadThumbnail(nodeUid, result.bareUrl, result.token, signal),
+            };
+        }
+        for (const uid of missingThumbnailUids) {
+            const nodeUid = this.batchThumbnailToNodeUids.get(uid);
+            this.logger.warn(`Thumbnail UID ${uid} not found in API response`);
+            yield {
+                nodeUid,
+                ok: false,
+                error: (0, ttag_1.c)("Error").t `Thumbnail not found`,
+            };
+        }
+    }
+    async downloadThumbnail(nodeUid, bareUrl, token, signal) {
+        const logger = new telemetry_1.LoggerWithPrefix(this.logger, `thumbnail ${token}`);
+        let decryptedBlock = null;
+        let attempt = 0;
+        while (!decryptedBlock) {
+            logger.debug(`Downloading`);
+            attempt++;
+            try {
+                const [nodeKeys, encryptedBlock] = await Promise.all([
+                    this.nodesService.getNodeKeys(nodeUid),
+                    this.apiService.downloadBlock(bareUrl, token, undefined, signal),
+                ]);
+                if (!nodeKeys.contentKeyPacketSessionKey) {
+                    throw new errors_1.ValidationError((0, ttag_1.c)("Error").t `File has no content key`);
+                }
+                logger.debug(`Decrypting`);
+                decryptedBlock = await this.cryptoService.decryptThumbnail(encryptedBlock, nodeKeys.contentKeyPacketSessionKey);
+            }
+            catch (error) {
+                if (attempt <= MAX_THUMBNAIL_DOWNLOAD_ATTEMPTS) {
+                    logger.warn(`Thumbnail download failed #${attempt}, retrying: ${(0, errors_2.getErrorMessage)(error)}`);
+                    continue;
+                }
+                logger.error(`Thumbnail download failed`, error);
+                throw error;
+            }
+        }
+        return decryptedBlock;
+    }
+}
+exports.ThumbnailDownloader = ThumbnailDownloader;
+//# sourceMappingURL=thumbnailDownloader.js.map

package/dist/internal/download/thumbnailDownloader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thumbnailDownloader.js","sourceRoot":"","sources":["../../../src/internal/download/thumbnailDownloader.ts"],"names":[],"mappings":";;;AAAA,+BAAyB;AAEzB,+CAAyG;AACzG,yCAA+C;AAC/C,+CAAmD;AAInD,sCAA4C;AAE5C;;GAEG;AACH,MAAM,uBAAuB,GAAG,EAAE,CAAC;AAEnC;;GAEG;AACH,MAAM,+BAA+B,GAAG,CAAC,CAAC;AAE1C,MAAa,mBAAmB;IAYhB;IACA;IACA;IAbJ,MAAM,CAAS;IAEf,wBAAwB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrD,gBAAgB,GAAG,IAAI,GAAG,EAAyB,CAAC;IACpD,kBAAkB,GAGpB,EAAE,CAAC;IAET,YACI,SAA+B,EACvB,YAA0B,EAC1B,UAA8B,EAC9B,aAAoC;QAFpC,iBAAY,GAAZ,YAAY,CAAc;QAC1B,eAAU,GAAV,UAAU,CAAoB;QAC9B,kBAAa,GAAb,aAAa,CAAuB;QAE5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC9C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,CAAC,iBAAiB,CACpB,QAAkB,EAClB,aAAa,GAAG,yBAAa,CAAC,KAAK,EACnC,MAAoB;QAEpB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO;QACX,CAAC;QAED,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,aAAa,EAAE,MAAM,CAAC,EAAE,CAAC;YACpF,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,MAAM,CAAC;gBACb,SAAS;YACb,CAAC;YAED,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YACvE,IAAI,IAAI,CAAC,wBAAwB,CAAC,IAAI,IAAI,uBAAuB,EAAE,CAAC;gBAChE,MAAM,IAAI,CAAC,gCAAgC,CAAC,MAAM,CAAC,CAAC;YACxD,CAAC;YAED,OAAO,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,uBAAuB,EAAE,CAAC;gBAC3D,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnD,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC;gBAC/B,IAAI,CAAC,kBAAkB,GAAG,EAAE,CAAC;YACjC,CAAC;QACL,CAAC;QAED,MAAM,IAAI,CAAC,gCAAgC,CAAC,MAAM,CAAC,CAAC;QAEpD,OAAO,IAAI,CAAC,gBAAgB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACpC,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;YACnD,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC;YAC/B,IAAI,CAAC,kBAAkB,GAAG,EAAE,CAAC;QACjC,CAAC;QAED,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC;QAC/B,IAAI,CAAC,kBAAkB,GAAG,EAAE,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB,CAAC,QAAkB,EAAE,aAA4B,EAAE,MAAoB;QAItG,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC;YACxE,IAAI,YAAY,IAAI,IAAI,EAAE,CAAC;gBACvB,MAAM;oBACF,OAAO,EAAE,IAAI,CAAC,UAAU;oBACxB,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,gBAAgB;iBACtC,CAAA;gBACD,SAAS;YACb,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAQ,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM;oBACF,OAAO,EAAE,IAAI,CAAC,GAAG;oBACjB,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,oBAAoB;iBAC1C,CAAA;gBACD,SAAS;YACb,CAAC;YAED,IAAI,SAAS,CAAC;YACd,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE,EAAE,CAAC;gBAC1B,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,EAAE,IAAI,CAClD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAClC,CAAC;YACN,CAAC;YACD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,MAAM;oBACF,OAAO,EAAE,IAAI,CAAC,GAAG;oBACjB,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,uBAAuB;iBAC7C,CAAA;gBACD,SAAS;YACb,CAAC;YAED,MAAM;gBACF,OAAO,EAAE,IAAI,CAAC,GAAG;gBACjB,EAAE,EAAE,IAAI;gBACR,YAAY,EAAE,SAAS,CAAC,GAAG;aAC9B,CAAA;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,gCAAgC,CAAC,MAAoB;QAC/D,IAAI,IAAI,CAAC,wBAAwB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC3C,OAAO;QACX,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC,CAAC;QAE/F,IAAI,KAAK,EAAE,MAAM,cAAc,IAAI,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,EAAE,CAAC;YACxE,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;gBACrB,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;oBACzB,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,cAAc,CAAC,KAAK;iBAC9B,CAAC,CAAC;gBACH,SAAS;YACb,CAAC;YAED,IAAI,CAAC,gBAAgB,CAAC,GAAG,CACrB,cAAc,CAAC,OAAO,EACtB,cAAc,CAAC,eAAe;iBACzB,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE;gBAChB,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;oBACzB,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,EAAE,EAAE,IAAI;oBACR,SAAS;iBACZ,CAAC,CAAC;YACP,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;oBACzB,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAA,wBAAe,EAAC,KAAK,CAAC;iBAChC,CAAC,CAAC;YACP,CAAC,CAAC;iBACD,OAAO,CAAC,GAAG,EAAE;gBACV,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACzD,CAAC,CAAC,CACT,CAAC;QACN,CAAC;QAED,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,MAAoB;QAIzD,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC,CAAC;QAE3E,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,IAAI,CAAC,UAAU,CAAC,iBAAiB,CACxD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAChD,MAAM,CACT,EAAE,CAAC;YACA,MAAM,OAAO,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC9D,IAAI,CAAC,OAAO,EAAE,CAAC;gBACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,MAAM,CAAC,GAAG,oBAAoB,CAAC,CAAC;gBAC7E,SAAS;YACb,CAAC;YAED,oBAAoB,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAExC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM;oBACF,OAAO;oBACP,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,MAAM,CAAC,KAAK;iBACtB,CAAA;gBACD,SAAS;YACb,CAAC;YAED,MAAM;gBACF,OAAO;gBACP,EAAE,EAAE,IAAI;gBACR,eAAe,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;aACzF,CAAA;QACL,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,oBAAoB,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;YACxD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,GAAG,4BAA4B,CAAC,CAAC;YACnE,MAAM;gBACF,OAAO;gBACP,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,qBAAqB;aAC3C,CAAA;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,OAAe,EAAE,OAAe,EAAE,KAAa,EAAE,MAAoB;QACjG,MAAM,MAAM,GAAG,IAAI,4BAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,KAAK,EAAE,CAAC,CAAC;QAEvE,IAAI,cAAc,GAAsB,IAAI,CAAC;QAC7C,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,OAAO,CAAC,cAAc,EAAE,CAAC;YACrB,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC5B,OAAO,EAAE,CAAC;YAEV,IAAI,CAAC;gBACD,MAAM,CAAC,QAAQ,EAAE,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;oBACjD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,OAAO,CAAC;oBACtC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC;iBACnE,CAAC,CAAC;gBAEH,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE,CAAC;oBACvC,MAAM,IAAI,wBAAe,CAAC,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,yBAAyB,CAAC,CAAC;gBACrE,CAAC;gBAED,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC3B,cAAc,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,cAAc,EAAE,QAAQ,CAAC,0BAA0B,CAAC,CAAC;YACpH,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACtB,IAAI,OAAO,IAAI,+BAA+B,EAAE,CAAC;oBAC7C,MAAM,CAAC,IAAI,CAAC,8BAA8B,OAAO,eAAe,IAAA,wBAAe,EAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBAC1F,SAAS;gBACb,CAAC;gBAED,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;gBACjD,MAAM,KAAK,CAAC;YAChB,CAAC;QACL,CAAC;QAED,OAAO,cAAc,CAAC;IAC1B,CAAC;CACJ;AArOD,kDAqOC"}

package/dist/internal/download/thumbnailDownloader.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/internal/download/thumbnailDownloader.test.js

@@ -0,0 +1,179 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const telemetry_1 = require("../../tests/telemetry");
+const thumbnailDownloader_1 = require("./thumbnailDownloader");
+describe('ThumbnailDownloader', () => {
+    let telemetry;
+    let nodesService;
+    let apiService;
+    let cryptoService;
+    let downloader;
+    beforeEach(() => {
+        telemetry = (0, telemetry_1.getMockTelemetry)();
+        // @ts-expect-error No need to implement all methods for mocking
+        nodesService = {
+            iterateNodes: jest.fn().mockImplementation(async function* (nodeUids) {
+                for (const nodeUid of nodeUids) {
+                    yield {
+                        uid: nodeUid,
+                        type: 'file',
+                        activeRevision: {
+                            ok: true,
+                            value: {
+                                thumbnails: [{ type: 1, uid: `thumb-${nodeUid}` }],
+                            },
+                        },
+                    };
+                }
+            }),
+            getNodeKeys: jest.fn().mockReturnValue({
+                contentKeyPacketSessionKey: 'contentKeyPacketSessionKey',
+            }),
+        };
+        // @ts-expect-error No need to implement all methods for mocking
+        apiService = {
+            iterateThumbnails: jest.fn().mockImplementation(async function* (thumbnailUids) {
+                for (const thumbnailUid of thumbnailUids) {
+                    yield {
+                        uid: thumbnailUid,
+                        ok: true,
+                        bareUrl: `url-${thumbnailUid}`,
+                        token: `token-${thumbnailUid}`,
+                    };
+                }
+            }),
+            downloadBlock: jest.fn().mockResolvedValue(new Uint8Array([1, 2, 3])),
+        };
+        // @ts-expect-error No need to implement all methods for mocking
+        cryptoService = {
+            decryptThumbnail: jest.fn().mockImplementation(async (thumbnail) => thumbnail),
+        };
+        downloader = new thumbnailDownloader_1.ThumbnailDownloader(telemetry, nodesService, apiService, cryptoService);
+    });
+    it('should handle all success cases', async () => {
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1', 'node2', 'node3']));
+        expect(results).toEqual([
+            { nodeUid: 'node1', ok: true, thumbnail: new Uint8Array([1, 2, 3]) },
+            { nodeUid: 'node2', ok: true, thumbnail: new Uint8Array([1, 2, 3]) },
+            { nodeUid: 'node3', ok: true, thumbnail: new Uint8Array([1, 2, 3]) },
+        ]);
+        expect(nodesService.iterateNodes).toHaveBeenCalledWith(['node1', 'node2', 'node3'], undefined);
+        expect(apiService.iterateThumbnails).toHaveBeenCalledWith(['thumb-node1', 'thumb-node2', 'thumb-node3'], undefined);
+        expect(nodesService.getNodeKeys).toHaveBeenCalledTimes(3);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(3);
+        expect(cryptoService.decryptThumbnail).toHaveBeenCalledTimes(3);
+        expect(cryptoService.decryptThumbnail).toHaveBeenCalledWith(new Uint8Array([1, 2, 3]), 'contentKeyPacketSessionKey');
+    });
+    it('should handle no requested node', async () => {
+        const results = await Array.fromAsync(downloader.iterateThumbnails([]));
+        expect(results).toEqual([]);
+        expect(nodesService.iterateNodes).not.toHaveBeenCalled();
+        expect(apiService.iterateThumbnails).not.toHaveBeenCalled();
+    });
+    it('should handle failure when requesting nodes', async () => {
+        nodesService.iterateNodes = jest.fn().mockImplementation(() => {
+            throw new Error('Failed to fetch nodes');
+        });
+        const results = Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        await expect(results).rejects.toThrow('Failed to fetch nodes');
+        expect(apiService.iterateThumbnails).not.toHaveBeenCalled();
+    });
+    it('should handle missing node', async () => {
+        nodesService.iterateNodes = jest.fn().mockImplementation(async function* () {
+            yield { missingUid: 'node1' };
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Node not found' }]);
+        expect(apiService.iterateThumbnails).not.toHaveBeenCalled();
+    });
+    it('should handle node that is not a file', async () => {
+        nodesService.iterateNodes = jest.fn().mockImplementation(async function* () {
+            yield { uid: 'node1', type: 'folder' };
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Node is not a file' }]);
+        expect(apiService.iterateThumbnails).not.toHaveBeenCalled();
+    });
+    it('should handle node without requested thumbnail', async () => {
+        nodesService.iterateNodes = jest.fn().mockImplementation(async function* () {
+            yield { uid: 'node1', type: 'file', activeRevision: { ok: true, value: { thumbnails: [] } } };
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Node has no thumbnail' }]);
+        expect(apiService.iterateThumbnails).not.toHaveBeenCalled();
+    });
+    it('should handle API failure to provide token for thumbnail', async () => {
+        apiService.iterateThumbnails = jest.fn().mockImplementation(async function* () {
+            yield { uid: 'thumb-node1', ok: false, error: 'Failed to fetch token' };
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Failed to fetch token' }]);
+        expect(apiService.downloadBlock).not.toHaveBeenCalled();
+    });
+    it('should handle API providing unexpected thumbnail', async () => {
+        apiService.iterateThumbnails = jest.fn().mockImplementation(async function* () {
+            yield { uid: 'thumb-unexpected', ok: true, thumbnail: new Uint8Array([1, 2, 3]) };
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Thumbnail not found' }]);
+        expect(apiService.downloadBlock).not.toHaveBeenCalled();
+    });
+    it('should handle failure when downloading block', async () => {
+        apiService.downloadBlock = jest.fn().mockRejectedValue(new Error('Failed to download thumbnail'));
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Failed to download thumbnail' }]);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(3);
+    });
+    it('should handle one-off failure when downloading block', async () => {
+        let callCount = 0;
+        apiService.downloadBlock = jest.fn().mockImplementation(() => {
+            callCount++;
+            if (callCount === 1) {
+                return Promise.reject(new Error('Failed to download block'));
+            }
+            return Promise.resolve(new Uint8Array([1, 2, 3]));
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: true, thumbnail: new Uint8Array([1, 2, 3]) }]);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(2);
+    });
+    it('should handle failure when getting node keys', async () => {
+        nodesService.getNodeKeys = jest.fn().mockRejectedValue(new Error('Failed to get node keys'));
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Failed to get node keys' }]);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(3);
+    });
+    it('should handle one-off failure when getting node keys', async () => {
+        let callCount = 0;
+        nodesService.getNodeKeys = jest.fn().mockImplementation(() => {
+            callCount++;
+            if (callCount === 1) {
+                return Promise.reject(new Error('Failed to get node keys'));
+            }
+            return Promise.resolve({ contentKeyPacketSessionKey: 'contentKeyPacketSessionKey' });
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: true, thumbnail: new Uint8Array([1, 2, 3]) }]);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(2);
+    });
+    it('should handle failure when decrypting block', async () => {
+        cryptoService.decryptThumbnail = jest.fn().mockRejectedValue(new Error('Failed to decrypt thumbnail'));
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: false, error: 'Failed to decrypt thumbnail' }]);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(3);
+    });
+    it('should handle one-off failure when decrypting block', async () => {
+        let callCount = 0;
+        cryptoService.decryptThumbnail = jest.fn().mockImplementation(() => {
+            callCount++;
+            if (callCount === 1) {
+                return Promise.reject(new Error('Failed to decrypt thumbnail'));
+            }
+            return Promise.resolve(new Uint8Array([1, 2, 3]));
+        });
+        const results = await Array.fromAsync(downloader.iterateThumbnails(['node1']));
+        expect(results).toEqual([{ nodeUid: 'node1', ok: true, thumbnail: new Uint8Array([1, 2, 3]) }]);
+        expect(apiService.downloadBlock).toHaveBeenCalledTimes(2);
+    });
+});
+//# sourceMappingURL=thumbnailDownloader.test.js.map