@aws-sdk/lib-transfer-manager 0.1.1 → 0.2.0

package/dist-cjs/submodules/transfer-manager/index.js

@@ -1,6 +1,7 @@
 const { isReadableStream, sdkStreamMixin } = require("@smithy/core/serde");
 const { Readable } = require("node:stream");
 const { S3Client, GetObjectCommand, ListObjectsV2Command, HeadObjectCommand, PutObjectCommand, CreateMultipartUploadCommand, CompleteMultipartUploadCommand, AbortMultipartUploadCommand, UploadPartCommand } = require("@aws-sdk/client-s3");
+const { setFeature } = require("@aws-sdk/core/client");
 const { isThrottlingError, isTransientError } = require("@smithy/core/retry");
 const { ReadStream, lstatSync, existsSync, unlinkSync, createReadStream } = require("node:fs");
 const { stat, mkdir, realpath, open, rename, unlink, opendir } = require("node:fs/promises");
@@ -488,6 +489,7 @@ class OrderedPartQueue {
     }
 }
 
+const TRANSFER_MANAGER_DIRECTORY_FEATURE = Symbol("transferManagerDirectoryFeature");
 class S3TransferManagerBase {
     static MIN_PART_SIZE = 5 * 1024 * 1024;
     s3;
@@ -541,6 +543,21 @@ class S3TransferManagerBase {
         };
         this.validateConfig();
     }
+    send(command, options) {
+        const directoryFeature = options?.[TRANSFER_MANAGER_DIRECTORY_FEATURE];
+        command.middlewareStack.add((next, context) => async (args) => {
+            setFeature(context, "S3_TRANSFER", "G");
+            if (directoryFeature) {
+                setFeature(context, directoryFeature, directoryFeature === "S3_TRANSFER_UPLOAD_DIRECTORY" ? "9" : "+");
+            }
+            return next(args);
+        }, {
+            name: "s3TransferManagerFeature",
+            step: "initialize",
+            override: true,
+        });
+        return this.s3.send(command, (options ?? {}));
+    }
     addEventListener(type, callback, options) {
         const eventType = type;
         const listeners = this.eventListeners[eventType];
@@ -807,7 +824,7 @@ class S3TransferManagerBase {
         };
         let initialResponse;
         try {
-            initialResponse = await this.s3.send(new GetObjectCommand(initialPartRequest), transferOptions);
+            initialResponse = await this.send(new GetObjectCommand(initialPartRequest), transferOptions);
         }
         catch (error) {
             removeLocalEventListeners();
@@ -903,7 +920,7 @@ class S3TransferManagerBase {
                                 expectedLength,
                                 resultToken,
                             };
-                            await this.s3.send(new GetObjectCommand(partGetRequest), {
+                            await this.send(new GetObjectCommand(partGetRequest), {
                                 ...transferOptions,
                                 downloadDataToFile: downloadOptions,
                             });
@@ -917,7 +934,7 @@ class S3TransferManagerBase {
                             });
                         }
                         else {
-                            const partResponse = await this.s3.send(new GetObjectCommand(partGetRequest), transferOptions);
+                            const partResponse = await this.send(new GetObjectCommand(partGetRequest), transferOptions);
                             if (partResponse.Body && partResponse.Body instanceof Readable) {
                                 partResponse.Body.on("error", () => { });
                             }
@@ -1006,7 +1023,7 @@ class S3TransferManagerBase {
                 Range: `bytes=0-${rangeSize - 1}`,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const initialResponse = await this.s3.send(new GetObjectCommand(initialRangeRequest), transferOptions);
+            const initialResponse = await this.send(new GetObjectCommand(initialRangeRequest), transferOptions);
             if (initialResponse.Body && initialResponse.Body instanceof Readable) {
                 initialResponse.Body.on("error", () => { });
             }
@@ -1094,7 +1111,7 @@ class S3TransferManagerBase {
                                 expectedLength,
                                 resultToken,
                             };
-                            const rangeResponse = await this.s3.send(new GetObjectCommand(rangeGetRequest), {
+                            const rangeResponse = await this.send(new GetObjectCommand(rangeGetRequest), {
                                 ...transferOptions,
                                 downloadDataToFile: downloadOptions,
                             });
@@ -1109,7 +1126,7 @@ class S3TransferManagerBase {
                             });
                         }
                         else {
-                            const rangeResponse = await this.s3.send(new GetObjectCommand(rangeGetRequest), transferOptions);
+                            const rangeResponse = await this.send(new GetObjectCommand(rangeGetRequest), transferOptions);
                             this.validateRangeDownload(`bytes=${start}-${end}`, rangeResponse.ContentRange);
                             if (rangeResponse.Body && rangeResponse.Body instanceof Readable) {
                                 rangeResponse.Body.on("error", () => { });
@@ -1284,6 +1301,7 @@ class S3TransferManagerBase {
                     }
                     await this.uploadInternal(putRequest, {
                         abortSignal: abortController.signal,
+                        [TRANSFER_MANAGER_DIRECTORY_FEATURE]: "S3_TRANSFER_UPLOAD_DIRECTORY",
                     }, false);
                     objectsUploaded++;
                     transferredFiles++;
@@ -1418,8 +1436,9 @@ class S3TransferManagerBase {
                     Delimiter: undefined,
                     ContinuationToken: continuationToken,
                 };
-                const listResponse = await this.s3.send(new ListObjectsV2Command(listRequest), {
+                const listResponse = await this.send(new ListObjectsV2Command(listRequest), {
                     abortSignal: abortController.signal,
+                    [TRANSFER_MANAGER_DIRECTORY_FEATURE]: "S3_TRANSFER_DOWNLOAD_DIRECTORY",
                 });
                 continuationToken = listResponse.IsTruncated ? listResponse.NextContinuationToken : undefined;
                 for (const object of listResponse.Contents ?? []) {
@@ -1463,6 +1482,7 @@ class S3TransferManagerBase {
                                 destination: localPath,
                             }, {
                                 abortSignal: abortController.signal,
+                                [TRANSFER_MANAGER_DIRECTORY_FEATURE]: "S3_TRANSFER_DOWNLOAD_DIRECTORY",
                             }, false);
                             objectsDownloaded++;
                             transferredFiles++;
@@ -1555,7 +1575,7 @@ class S3TransferManagerBase {
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
             try {
-                const initialPart = await this.s3.send(new GetObjectCommand(initialPartRequest), transferOptions);
+                const initialPart = await this.send(new GetObjectCommand(initialPartRequest), transferOptions);
                 await internalEventHandler.afterInitialGetObject();
                 const partSize = initialPart.ContentLength;
                 const initialETag = initialPart.ETag ?? undefined;
@@ -1588,8 +1608,7 @@ class S3TransferManagerBase {
                         partRequests.push(getObjectRequest);
                         partTasks.push(() => {
                             this.checkAborted(transferOptions);
-                            return this.s3
-                                .send(new GetObjectCommand(getObjectRequest), transferOptions)
+                            return this.send(new GetObjectCommand(getObjectRequest), transferOptions)
                                 .then((response) => {
                                 this.validatePartDownload(response.ContentRange, part, partSize ?? 0);
                                 if (response.Body && typeof response.Body.getReader === "function") {
@@ -1630,7 +1649,7 @@ class S3TransferManagerBase {
                         ChecksumMode: "ENABLED",
                     }),
                 };
-                const getObject = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+                const getObject = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
                 totalSize = getObject.ContentRange ? Number.parseInt(getObject.ContentRange.split("/")[1]) : 0;
                 this.dispatchTransferInitiatedEvent(request, totalSize);
                 if (getObject.Body) {
@@ -1652,13 +1671,13 @@ class S3TransferManagerBase {
     async downloadByRange(request, transferOptions, streams, requests, metadata, checksumValidationEnabled) {
         let streamConsumedCallback;
         this.checkAborted(transferOptions);
-        const headResponse = await this.s3.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
+        const headResponse = await this.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
         if (headResponse.ContentLength === 0) {
             const getObjectRequest = {
                 ...request,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const response = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+            const response = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
             this.dispatchTransferInitiatedEvent(request, 0);
             if (response.Body)
                 streams.push(Promise.resolve(response.Body));
@@ -1685,7 +1704,7 @@ class S3TransferManagerBase {
                 Range: `bytes=${left}-${right}`,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const initialRangeGet = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+            const initialRangeGet = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
             await internalEventHandler.afterInitialGetObject();
             this.validateRangeDownload(`bytes=${left}-${right}`, initialRangeGet.ContentRange);
             initialETag = initialRangeGet.ETag ?? undefined;
@@ -1732,8 +1751,7 @@ class S3TransferManagerBase {
             rangeRequests.push(getObjectRequest);
             rangeTasks.push(() => {
                 this.checkAborted(transferOptions);
-                return this.s3
-                    .send(new GetObjectCommand(getObjectRequest), transferOptions)
+                return this.send(new GetObjectCommand(getObjectRequest), transferOptions)
                     .then((response) => {
                     this.validateRangeDownload(range, response.ContentRange);
                     if (response.Body && typeof response.Body.getReader === "function") {
@@ -1776,7 +1794,7 @@ class S3TransferManagerBase {
             PartNumber: 1,
             ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
         };
-        const initialPart = await this.s3.send(new GetObjectCommand(initialPartRequest), transferOptions);
+        const initialPart = await this.send(new GetObjectCommand(initialPartRequest), transferOptions);
         await internalEventHandler.afterInitialGetObject();
         const partSize = initialPart.ContentLength;
         const initialETag = initialPart.ETag ?? undefined;
@@ -1856,8 +1874,7 @@ class S3TransferManagerBase {
                     checksumAlgorithm: checksumValidationEnabled ? "CRC32" : undefined,
                     resultToken,
                 };
-                this.s3
-                    .send(new GetObjectCommand(partGetRequest), {
+                this.send(new GetObjectCommand(partGetRequest), {
                     ...transferOptions,
                     downloadStream: transferOptions_,
                 })
@@ -1923,14 +1940,14 @@ class S3TransferManagerBase {
     }
     async downloadByRangeWithWorkers(request, transferOptions, streams, requests, metadata, checksumValidationEnabled) {
         this.checkAborted(transferOptions);
-        const headResponse = await this.s3.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
+        const headResponse = await this.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
         await internalEventHandler.afterInitialGetObject();
         if (headResponse.ContentLength === 0) {
             const getObjectRequest = {
                 ...request,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const response = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+            const response = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
             this.dispatchTransferInitiatedEvent(request, 0);
             if (response.Body)
                 streams.push(Promise.resolve(response.Body));
@@ -2008,8 +2025,7 @@ class S3TransferManagerBase {
                     checksumAlgorithm: checksumValidationEnabled ? "CRC32" : undefined,
                     resultToken,
                 };
-                this.s3
-                    .send(new GetObjectCommand(rangeGetRequest), {
+                this.send(new GetObjectCommand(rangeGetRequest), {
                     ...transferOptions,
                     downloadStream: transferOpts,
                 })
@@ -2280,7 +2296,7 @@ class S3TransferManagerBase {
     async uploadSinglePart(request, contentLength, transferOptions, emitEvents) {
         const putObjectRequest = { ...request };
         this.checkAborted(transferOptions);
-        const response = await this.s3.send(new PutObjectCommand(putObjectRequest), transferOptions);
+        const response = await this.send(new PutObjectCommand(putObjectRequest), transferOptions);
         if (emitEvents) {
             this.dispatchEvent(Object.assign(new Event("bytesTransferred"), {
                 request,
@@ -2370,7 +2386,7 @@ class S3TransferManagerBase {
         if (!createMpuRequest.ChecksumAlgorithm && this.requestChecksumCalculation === "WHEN_SUPPORTED") {
             createMpuRequest.ChecksumAlgorithm = request.ChecksumAlgorithm ?? "CRC32";
         }
-        const createMpuResponse = await this.s3.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
+        const createMpuResponse = await this.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
         const uploadId = createMpuResponse.UploadId;
         const checksumAlgorithm = createMpuRequest.ChecksumAlgorithm;
         const checksumHeader = checksumAlgorithm ? `x-amz-checksum-${checksumAlgorithm.toLowerCase()}` : undefined;
@@ -2395,7 +2411,7 @@ class S3TransferManagerBase {
                     }),
                 };
                 try {
-                    return await this.s3.send(new UploadPartCommand(partRequest), {
+                    return await this.send(new UploadPartCommand(partRequest), {
                         ...transferOptions,
                         dataSource,
                     });
@@ -2484,7 +2500,7 @@ class S3TransferManagerBase {
                     completeRequest.ChecksumCRC64NVME = request.ChecksumCRC64NVME;
             }
             try {
-                return await this.s3.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
+                return await this.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
             }
             catch (completeMpuError) {
                 this.logger.warn(`CompleteMultipartUpload failed for upload ID ${uploadId}: ${completeMpuError.message}`);
@@ -2498,7 +2514,7 @@ class S3TransferManagerBase {
                 UploadId: uploadId,
             };
             try {
-                await this.s3.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
+                await this.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
             }
             catch (abortError) {
                 this.logger.warn(`Failed to abort multipart upload ${uploadId}:`, abortError);
@@ -2522,7 +2538,7 @@ class S3TransferManagerBase {
         if (!createMpuRequest.ChecksumAlgorithm && this.requestChecksumCalculation === "WHEN_SUPPORTED") {
             createMpuRequest.ChecksumAlgorithm = request.ChecksumAlgorithm ?? "CRC32";
         }
-        const createMpuResponse = await this.s3.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
+        const createMpuResponse = await this.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
         const uploadId = createMpuResponse.UploadId;
         const abortController = new AbortController();
         const uploadAbortSignal = abortController.signal;
@@ -2547,7 +2563,7 @@ class S3TransferManagerBase {
                             ChecksumAlgorithm: request.ChecksumAlgorithm,
                         }),
                     };
-                    const partResponse = await this.s3.send(new UploadPartCommand(partRequest), transferOptions);
+                    const partResponse = await this.send(new UploadPartCommand(partRequest), transferOptions);
                     const checksumAlgorithm = partRequest.ChecksumAlgorithm;
                     if (checksumAlgorithm &&
                         !partResponse.ChecksumCRC32 &&
@@ -2616,7 +2632,7 @@ class S3TransferManagerBase {
                     completeRequest.ChecksumCRC64NVME = request.ChecksumCRC64NVME;
             }
             try {
-                return await this.s3.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
+                return await this.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
             }
             catch (completeMpuError) {
                 this.logger.warn(`CompleteMultipartUpload failed for upload ID ${uploadId}: ${completeMpuError.message}`);
@@ -2630,7 +2646,7 @@ class S3TransferManagerBase {
                 UploadId: uploadId,
             };
             try {
-                await this.s3.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
+                await this.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
             }
             catch (abortError) {
                 this.logger.warn(`Failed to abort multipart upload ${uploadId}:`, abortError);

package/dist-es/submodules/transfer-manager/S3TransferManager.js

@@ -1,4 +1,5 @@
 import { AbortMultipartUploadCommand, CompleteMultipartUploadCommand, CreateMultipartUploadCommand, GetObjectCommand, HeadObjectCommand, ListObjectsV2Command, PutObjectCommand, S3Client, UploadPartCommand, } from "@aws-sdk/client-s3";
+import { setFeature } from "@aws-sdk/core/client";
 import { isThrottlingError, isTransientError } from "@smithy/core/retry";
 import { createReadStream, existsSync } from "node:fs";
 import { open, opendir, realpath, stat } from "node:fs/promises";
@@ -9,6 +10,7 @@ import { createDestinationDirectory, createLocalParentDirectories, deriveLocalPa
 import { FileManager } from "./file-manager";
 import { LogLevel } from "./log-level";
 import { OrderedPartQueue } from "./ordered-part-queue";
+const TRANSFER_MANAGER_DIRECTORY_FEATURE = Symbol("transferManagerDirectoryFeature");
 class S3TransferManagerBase {
     static MIN_PART_SIZE = 5 * 1024 * 1024;
     s3;
@@ -62,6 +64,21 @@ class S3TransferManagerBase {
         };
         this.validateConfig();
     }
+    send(command, options) {
+        const directoryFeature = options?.[TRANSFER_MANAGER_DIRECTORY_FEATURE];
+        command.middlewareStack.add((next, context) => async (args) => {
+            setFeature(context, "S3_TRANSFER", "G");
+            if (directoryFeature) {
+                setFeature(context, directoryFeature, directoryFeature === "S3_TRANSFER_UPLOAD_DIRECTORY" ? "9" : "+");
+            }
+            return next(args);
+        }, {
+            name: "s3TransferManagerFeature",
+            step: "initialize",
+            override: true,
+        });
+        return this.s3.send(command, (options ?? {}));
+    }
     addEventListener(type, callback, options) {
         const eventType = type;
         const listeners = this.eventListeners[eventType];
@@ -328,7 +345,7 @@ class S3TransferManagerBase {
         };
         let initialResponse;
         try {
-            initialResponse = await this.s3.send(new GetObjectCommand(initialPartRequest), transferOptions);
+            initialResponse = await this.send(new GetObjectCommand(initialPartRequest), transferOptions);
         }
         catch (error) {
             removeLocalEventListeners();
@@ -424,7 +441,7 @@ class S3TransferManagerBase {
                                 expectedLength,
                                 resultToken,
                             };
-                            await this.s3.send(new GetObjectCommand(partGetRequest), {
+                            await this.send(new GetObjectCommand(partGetRequest), {
                                 ...transferOptions,
                                 downloadDataToFile: downloadOptions,
                             });
@@ -438,7 +455,7 @@ class S3TransferManagerBase {
                             });
                         }
                         else {
-                            const partResponse = await this.s3.send(new GetObjectCommand(partGetRequest), transferOptions);
+                            const partResponse = await this.send(new GetObjectCommand(partGetRequest), transferOptions);
                             if (partResponse.Body && partResponse.Body instanceof Readable) {
                                 partResponse.Body.on("error", () => { });
                             }
@@ -527,7 +544,7 @@ class S3TransferManagerBase {
                 Range: `bytes=0-${rangeSize - 1}`,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const initialResponse = await this.s3.send(new GetObjectCommand(initialRangeRequest), transferOptions);
+            const initialResponse = await this.send(new GetObjectCommand(initialRangeRequest), transferOptions);
             if (initialResponse.Body && initialResponse.Body instanceof Readable) {
                 initialResponse.Body.on("error", () => { });
             }
@@ -615,7 +632,7 @@ class S3TransferManagerBase {
                                 expectedLength,
                                 resultToken,
                             };
-                            const rangeResponse = await this.s3.send(new GetObjectCommand(rangeGetRequest), {
+                            const rangeResponse = await this.send(new GetObjectCommand(rangeGetRequest), {
                                 ...transferOptions,
                                 downloadDataToFile: downloadOptions,
                             });
@@ -630,7 +647,7 @@ class S3TransferManagerBase {
                             });
                         }
                         else {
-                            const rangeResponse = await this.s3.send(new GetObjectCommand(rangeGetRequest), transferOptions);
+                            const rangeResponse = await this.send(new GetObjectCommand(rangeGetRequest), transferOptions);
                             this.validateRangeDownload(`bytes=${start}-${end}`, rangeResponse.ContentRange);
                             if (rangeResponse.Body && rangeResponse.Body instanceof Readable) {
                                 rangeResponse.Body.on("error", () => { });
@@ -805,6 +822,7 @@ class S3TransferManagerBase {
                     }
                     await this.uploadInternal(putRequest, {
                         abortSignal: abortController.signal,
+                        [TRANSFER_MANAGER_DIRECTORY_FEATURE]: "S3_TRANSFER_UPLOAD_DIRECTORY",
                     }, false);
                     objectsUploaded++;
                     transferredFiles++;
@@ -939,8 +957,9 @@ class S3TransferManagerBase {
                     Delimiter: undefined,
                     ContinuationToken: continuationToken,
                 };
-                const listResponse = await this.s3.send(new ListObjectsV2Command(listRequest), {
+                const listResponse = await this.send(new ListObjectsV2Command(listRequest), {
                     abortSignal: abortController.signal,
+                    [TRANSFER_MANAGER_DIRECTORY_FEATURE]: "S3_TRANSFER_DOWNLOAD_DIRECTORY",
                 });
                 continuationToken = listResponse.IsTruncated ? listResponse.NextContinuationToken : undefined;
                 for (const object of listResponse.Contents ?? []) {
@@ -984,6 +1003,7 @@ class S3TransferManagerBase {
                                 destination: localPath,
                             }, {
                                 abortSignal: abortController.signal,
+                                [TRANSFER_MANAGER_DIRECTORY_FEATURE]: "S3_TRANSFER_DOWNLOAD_DIRECTORY",
                             }, false);
                             objectsDownloaded++;
                             transferredFiles++;
@@ -1076,7 +1096,7 @@ class S3TransferManagerBase {
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
             try {
-                const initialPart = await this.s3.send(new GetObjectCommand(initialPartRequest), transferOptions);
+                const initialPart = await this.send(new GetObjectCommand(initialPartRequest), transferOptions);
                 await internalEventHandler.afterInitialGetObject();
                 const partSize = initialPart.ContentLength;
                 const initialETag = initialPart.ETag ?? undefined;
@@ -1109,8 +1129,7 @@ class S3TransferManagerBase {
                         partRequests.push(getObjectRequest);
                         partTasks.push(() => {
                             this.checkAborted(transferOptions);
-                            return this.s3
-                                .send(new GetObjectCommand(getObjectRequest), transferOptions)
+                            return this.send(new GetObjectCommand(getObjectRequest), transferOptions)
                                 .then((response) => {
                                 this.validatePartDownload(response.ContentRange, part, partSize ?? 0);
                                 if (response.Body && typeof response.Body.getReader === "function") {
@@ -1151,7 +1170,7 @@ class S3TransferManagerBase {
                         ChecksumMode: "ENABLED",
                     }),
                 };
-                const getObject = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+                const getObject = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
                 totalSize = getObject.ContentRange ? Number.parseInt(getObject.ContentRange.split("/")[1]) : 0;
                 this.dispatchTransferInitiatedEvent(request, totalSize);
                 if (getObject.Body) {
@@ -1173,13 +1192,13 @@ class S3TransferManagerBase {
     async downloadByRange(request, transferOptions, streams, requests, metadata, checksumValidationEnabled) {
         let streamConsumedCallback;
         this.checkAborted(transferOptions);
-        const headResponse = await this.s3.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
+        const headResponse = await this.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
         if (headResponse.ContentLength === 0) {
             const getObjectRequest = {
                 ...request,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const response = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+            const response = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
             this.dispatchTransferInitiatedEvent(request, 0);
             if (response.Body)
                 streams.push(Promise.resolve(response.Body));
@@ -1206,7 +1225,7 @@ class S3TransferManagerBase {
                 Range: `bytes=${left}-${right}`,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const initialRangeGet = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+            const initialRangeGet = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
             await internalEventHandler.afterInitialGetObject();
             this.validateRangeDownload(`bytes=${left}-${right}`, initialRangeGet.ContentRange);
             initialETag = initialRangeGet.ETag ?? undefined;
@@ -1253,8 +1272,7 @@ class S3TransferManagerBase {
             rangeRequests.push(getObjectRequest);
             rangeTasks.push(() => {
                 this.checkAborted(transferOptions);
-                return this.s3
-                    .send(new GetObjectCommand(getObjectRequest), transferOptions)
+                return this.send(new GetObjectCommand(getObjectRequest), transferOptions)
                     .then((response) => {
                     this.validateRangeDownload(range, response.ContentRange);
                     if (response.Body && typeof response.Body.getReader === "function") {
@@ -1297,7 +1315,7 @@ class S3TransferManagerBase {
             PartNumber: 1,
             ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
         };
-        const initialPart = await this.s3.send(new GetObjectCommand(initialPartRequest), transferOptions);
+        const initialPart = await this.send(new GetObjectCommand(initialPartRequest), transferOptions);
         await internalEventHandler.afterInitialGetObject();
         const partSize = initialPart.ContentLength;
         const initialETag = initialPart.ETag ?? undefined;
@@ -1377,8 +1395,7 @@ class S3TransferManagerBase {
                     checksumAlgorithm: checksumValidationEnabled ? "CRC32" : undefined,
                     resultToken,
                 };
-                this.s3
-                    .send(new GetObjectCommand(partGetRequest), {
+                this.send(new GetObjectCommand(partGetRequest), {
                     ...transferOptions,
                     downloadStream: transferOptions_,
                 })
@@ -1444,14 +1461,14 @@ class S3TransferManagerBase {
     }
     async downloadByRangeWithWorkers(request, transferOptions, streams, requests, metadata, checksumValidationEnabled) {
         this.checkAborted(transferOptions);
-        const headResponse = await this.s3.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
+        const headResponse = await this.send(new HeadObjectCommand({ Bucket: request.Bucket, Key: request.Key }), transferOptions);
         await internalEventHandler.afterInitialGetObject();
         if (headResponse.ContentLength === 0) {
             const getObjectRequest = {
                 ...request,
                 ...(checksumValidationEnabled && { ChecksumMode: "ENABLED" }),
             };
-            const response = await this.s3.send(new GetObjectCommand(getObjectRequest), transferOptions);
+            const response = await this.send(new GetObjectCommand(getObjectRequest), transferOptions);
             this.dispatchTransferInitiatedEvent(request, 0);
             if (response.Body)
                 streams.push(Promise.resolve(response.Body));
@@ -1529,8 +1546,7 @@ class S3TransferManagerBase {
                     checksumAlgorithm: checksumValidationEnabled ? "CRC32" : undefined,
                     resultToken,
                 };
-                this.s3
-                    .send(new GetObjectCommand(rangeGetRequest), {
+                this.send(new GetObjectCommand(rangeGetRequest), {
                     ...transferOptions,
                     downloadStream: transferOpts,
                 })
@@ -1801,7 +1817,7 @@ class S3TransferManagerBase {
     async uploadSinglePart(request, contentLength, transferOptions, emitEvents) {
         const putObjectRequest = { ...request };
         this.checkAborted(transferOptions);
-        const response = await this.s3.send(new PutObjectCommand(putObjectRequest), transferOptions);
+        const response = await this.send(new PutObjectCommand(putObjectRequest), transferOptions);
         if (emitEvents) {
             this.dispatchEvent(Object.assign(new Event("bytesTransferred"), {
                 request,
@@ -1891,7 +1907,7 @@ class S3TransferManagerBase {
         if (!createMpuRequest.ChecksumAlgorithm && this.requestChecksumCalculation === "WHEN_SUPPORTED") {
             createMpuRequest.ChecksumAlgorithm = request.ChecksumAlgorithm ?? "CRC32";
         }
-        const createMpuResponse = await this.s3.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
+        const createMpuResponse = await this.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
         const uploadId = createMpuResponse.UploadId;
         const checksumAlgorithm = createMpuRequest.ChecksumAlgorithm;
         const checksumHeader = checksumAlgorithm ? `x-amz-checksum-${checksumAlgorithm.toLowerCase()}` : undefined;
@@ -1916,7 +1932,7 @@ class S3TransferManagerBase {
                     }),
                 };
                 try {
-                    return await this.s3.send(new UploadPartCommand(partRequest), {
+                    return await this.send(new UploadPartCommand(partRequest), {
                         ...transferOptions,
                         dataSource,
                     });
@@ -2005,7 +2021,7 @@ class S3TransferManagerBase {
                     completeRequest.ChecksumCRC64NVME = request.ChecksumCRC64NVME;
             }
             try {
-                return await this.s3.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
+                return await this.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
             }
             catch (completeMpuError) {
                 this.logger.warn(`CompleteMultipartUpload failed for upload ID ${uploadId}: ${completeMpuError.message}`);
@@ -2019,7 +2035,7 @@ class S3TransferManagerBase {
                 UploadId: uploadId,
             };
             try {
-                await this.s3.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
+                await this.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
             }
             catch (abortError) {
                 this.logger.warn(`Failed to abort multipart upload ${uploadId}:`, abortError);
@@ -2043,7 +2059,7 @@ class S3TransferManagerBase {
         if (!createMpuRequest.ChecksumAlgorithm && this.requestChecksumCalculation === "WHEN_SUPPORTED") {
             createMpuRequest.ChecksumAlgorithm = request.ChecksumAlgorithm ?? "CRC32";
         }
-        const createMpuResponse = await this.s3.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
+        const createMpuResponse = await this.send(new CreateMultipartUploadCommand(createMpuRequest), transferOptions);
         const uploadId = createMpuResponse.UploadId;
         const abortController = new AbortController();
         const uploadAbortSignal = abortController.signal;
@@ -2068,7 +2084,7 @@ class S3TransferManagerBase {
                             ChecksumAlgorithm: request.ChecksumAlgorithm,
                         }),
                     };
-                    const partResponse = await this.s3.send(new UploadPartCommand(partRequest), transferOptions);
+                    const partResponse = await this.send(new UploadPartCommand(partRequest), transferOptions);
                     const checksumAlgorithm = partRequest.ChecksumAlgorithm;
                     if (checksumAlgorithm &&
                         !partResponse.ChecksumCRC32 &&
@@ -2137,7 +2153,7 @@ class S3TransferManagerBase {
                     completeRequest.ChecksumCRC64NVME = request.ChecksumCRC64NVME;
             }
             try {
-                return await this.s3.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
+                return await this.send(new CompleteMultipartUploadCommand(completeRequest), transferOptions);
             }
             catch (completeMpuError) {
                 this.logger.warn(`CompleteMultipartUpload failed for upload ID ${uploadId}: ${completeMpuError.message}`);
@@ -2151,7 +2167,7 @@ class S3TransferManagerBase {
                 UploadId: uploadId,
             };
             try {
-                await this.s3.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
+                await this.send(new AbortMultipartUploadCommand(abortRequest), transferOptions);
             }
             catch (abortError) {
                 this.logger.warn(`Failed to abort multipart upload ${uploadId}:`, abortError);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-sdk/lib-transfer-manager",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "description": "S3 Transfer Manager for AWS SDK JS v3",
   "homepage": "https://github.com/aws/aws-sdk-js-v3/tree/main/preview/lib-transfer-manager",
   "license": "Apache-2.0",
@@ -85,15 +85,16 @@
     "test:browser:watch": "yarn g:vitest watch -c vitest.config.browser.mts"
   },
   "dependencies": {
-    "@aws-sdk/checksums": "^3.1001.4",
-    "@smithy/core": "^3.36.0",
+    "@aws-sdk/checksums": "^3.1001.5",
+    "@aws-sdk/core": "^3.978.5",
+    "@smithy/core": "^3.36.1",
     "@smithy/node-http-handler": "^4.12.1",
     "@smithy/types": "^4.19.0",
     "tslib": "^2.6.2"
   },
   "devDependencies": {
-    "@aws-sdk/client-s3": "3.1153.0",
-    "@aws-sdk/xhr-http-handler": "3.1153.0",
+    "@aws-sdk/client-s3": "3.1155.0",
+    "@aws-sdk/xhr-http-handler": "3.1155.0",
     "@tsconfig/recommended": "1.0.1",
     "@types/node": "^20.14.8",
     "concurrently": "7.0.0",
@@ -102,7 +103,7 @@
     "typescript": "~7.0.2"
   },
   "peerDependencies": {
-    "@aws-sdk/client-s3": "^3.1153.0"
+    "@aws-sdk/client-s3": "^3.1155.0"
   },
   "engines": {
     "node": ">=20.0.0"