@strapi/data-transfer 5.10.4 → 5.11.1

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/engine/index.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,WAAW,EAAiC,MAAM,QAAQ,CAAC;AAUpE,OAAO,KAAK,EAEV,oBAAoB,EAIpB,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EAGb,eAAe,EACf,oBAAoB,EAEpB,iBAAiB,EACjB,wBAAwB,EACxB,YAAY,EACZ,mBAAmB,EAEnB,SAAS,EACV,MAAM,aAAa,CAAC;AAMrB,OAAO,EAEL,mBAAmB,EACnB,uBAAuB,EACxB,MAAM,qBAAqB,CAAC;AAK7B,eAAO,MAAM,eAAe,EAAE,aAAa,CAAC,aAAa,CAMvD,CAAC;AAEH,MAAM,MAAM,mBAAmB,GAAG,MAAM,CAAC,oBAAoB,EAAE,eAAe,CAAC,CAAC;AAEhF;;KAEK;AACL,eAAO,MAAM,oBAAoB,EAAE,mBAoBlC,CAAC;AAEF,eAAO,MAAM,wBAAwB,WAAW,CAAC;AACjD,eAAO,MAAM,uBAAuB,WAAW,CAAC;AAIhD,cAAM,cAAc,CAClB,CAAC,SAAS,eAAe,GAAG,eAAe,EAC3C,CAAC,SAAS,oBAAoB,GAAG,oBAAoB,CACrD,YAAW,eAAe;;IAE1B,cAAc,EAAE,eAAe,CAAC;IAEhC,mBAAmB,EAAE,oBAAoB,CAAC;IAE1C,OAAO,EAAE,sBAAsB,CAAC;IAOhC,QAAQ,EAAE;QAER,IAAI,EAAE,gBAAgB,CAAC;QAEvB,MAAM,EAAE,WAAW,CAAC;KACrB,CAAC;IAEF,WAAW,EAAE,mBAAmB,CAAC;IAUjC,YAAY,CAAC,OAAO,EAAE,iBAAiB;IAIvC,eAAe,CAAC,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY;IAOvD,mBAAmB,CAAC,KAAK,EAAE,KAAK;gBAgB1B,cAAc,EAAE,CAAC,EAAE,mBAAmB,EAAE,CAAC,EAAE,OAAO,EAAE,sBAAsB;IAatF;;OAEG;IACH,KAAK,CAAC,KAAK,EAAE,KAAK;IAMlB;;OAEG;IACH,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,uBAAuB;IAa3D;;OAEG;IACH,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM;IAO9C;;OAEG;IACH,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO;IAqR5C,eAAe,CAAC,KAAK,EAAE,aAAa;IAwG9B,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ9B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAa3B;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC;IAahC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAyCtB,cAAc;IAyDd,QAAQ,IAAI,OAAO,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAmD3C,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAwB/B,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAiBhC,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IA6CjC,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAsC9B,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB/B,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;CAc7C;AAED,eAAO,MAAM,oBAAoB,8EACf,CAAC,uBACI,CAAC,WACb,sBAAsB,KAC9B,eAAe,CAAC,EAAE,CAAC,CAErB,CAAC;AAEF,YAAY,EACV,cAAc,EACd,eAAe,EACf,sBAAsB,EACtB,eAAe,EACf,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EACpB,mBAAmB,EACnB,wBAAwB,EACxB,gBAAgB,GACjB,CAAC;AAEF,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/engine/index.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,WAAW,EAAiC,MAAM,QAAQ,CAAC;AAWpE,OAAO,KAAK,EAEV,oBAAoB,EAIpB,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EAGb,eAAe,EACf,oBAAoB,EAEpB,iBAAiB,EACjB,wBAAwB,EACxB,YAAY,EACZ,mBAAmB,EAEnB,SAAS,EACV,MAAM,aAAa,CAAC;AAMrB,OAAO,EAEL,mBAAmB,EACnB,uBAAuB,EACxB,MAAM,qBAAqB,CAAC;AAK7B,eAAO,MAAM,eAAe,EAAE,aAAa,CAAC,aAAa,CAMvD,CAAC;AAEH,MAAM,MAAM,mBAAmB,GAAG,MAAM,CAAC,oBAAoB,EAAE,eAAe,CAAC,CAAC;AAEhF;;KAEK;AACL,eAAO,MAAM,oBAAoB,EAAE,mBAoBlC,CAAC;AAEF,eAAO,MAAM,wBAAwB,WAAW,CAAC;AACjD,eAAO,MAAM,uBAAuB,WAAW,CAAC;AAIhD,cAAM,cAAc,CAClB,CAAC,SAAS,eAAe,GAAG,eAAe,EAC3C,CAAC,SAAS,oBAAoB,GAAG,oBAAoB,CACrD,YAAW,eAAe;;IAE1B,cAAc,EAAE,eAAe,CAAC;IAEhC,mBAAmB,EAAE,oBAAoB,CAAC;IAE1C,OAAO,EAAE,sBAAsB,CAAC;IAOhC,QAAQ,EAAE;QAER,IAAI,EAAE,gBAAgB,CAAC;QAEvB,MAAM,EAAE,WAAW,CAAC;KACrB,CAAC;IAEF,WAAW,EAAE,mBAAmB,CAAC;IAcjC,YAAY,CAAC,OAAO,EAAE,iBAAiB;IAIvC,eAAe,CAAC,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY;IAOvD,mBAAmB,CAAC,KAAK,EAAE,KAAK;gBAa1B,cAAc,EAAE,CAAC,EAAE,mBAAmB,EAAE,CAAC,EAAE,OAAO,EAAE,sBAAsB;IAatF;;OAEG;IACH,KAAK,CAAC,KAAK,EAAE,KAAK;IAMlB;;OAEG;IACH,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,uBAAuB;IAa3D;;OAEG;IACH,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM;IAO9C;;OAEG;IACH,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO;IAqR5C,eAAe,CAAC,KAAK,EAAE,aAAa;IAuH9B,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAM9B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAa3B;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC;IAahC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAyCtB,cAAc;IAyDd,QAAQ,IAAI,OAAO,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAmD3C,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAwB/B,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAiBhC,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IA6CjC,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAsC9B,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB/B,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;CAc7C;AAED,eAAO,MAAM,oBAAoB,8EACf,CAAC,uBACI,CAAC,WACb,sBAAsB,KAC9B,eAAe,CAAC,EAAE,CAAC,CAErB,CAAC;AAEF,YAAY,EACV,cAAc,EACd,eAAe,EACf,sBAAsB,EACtB,eAAe,EACf,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EACpB,mBAAmB,EACnB,wBAAwB,EACxB,gBAAgB,GACjB,CAAC;AAEF,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC"}

package/dist/index.js

@@ -1,6 +1,7 @@
 'use strict';
 
 var stream$1 = require('stream');
+var promises = require('stream/promises');
 var path = require('path');
 var os = require('os');
 var streamChain = require('stream-chain');
@@ -654,8 +655,7 @@ const TRANSFER_STAGES = Object.freeze([
 };
 const DEFAULT_VERSION_STRATEGY = 'ignore';
 const DEFAULT_SCHEMA_STRATEGY = 'strict';
-var _metadata$1 = /*#__PURE__*/ _class_private_field_loose_key$6("_metadata"), _schema = /*#__PURE__*/ _class_private_field_loose_key$6("_schema"), _handlers = /*#__PURE__*/ _class_private_field_loose_key$6("_handlers"), // Save the currently open stream so that we can access it at any time
-_currentStream = /*#__PURE__*/ _class_private_field_loose_key$6("_currentStream"), /**
+var _metadata$1 = /*#__PURE__*/ _class_private_field_loose_key$6("_metadata"), _schema = /*#__PURE__*/ _class_private_field_loose_key$6("_schema"), _handlers = /*#__PURE__*/ _class_private_field_loose_key$6("_handlers"), _currentStreamController = /*#__PURE__*/ _class_private_field_loose_key$6("_currentStreamController"), _aborted = /*#__PURE__*/ _class_private_field_loose_key$6("_aborted"), /**
    * Create and return a transform stream based on the given stage and options.
    *
    * Allowed transformations includes 'filter' and 'map'.
@@ -770,11 +770,9 @@ class TransferEngine {
     }
     // Cause an ongoing transfer to abort gracefully
     async abortTransfer() {
-        const err = new TransferEngineError('fatal', 'Transfer aborted.');
-        if (!_class_private_field_loose_base$6(this, _currentStream)[_currentStream]) {
-            throw err;
-        }
-        _class_private_field_loose_base$6(this, _currentStream)[_currentStream].destroy(err);
+        _class_private_field_loose_base$6(this, _aborted)[_aborted] = true;
+        _class_private_field_loose_base$6(this, _currentStreamController)[_currentStreamController]?.abort();
+        throw new TransferEngineError('fatal', 'Transfer aborted.');
     }
     async init() {
         // Resolve providers' resource and store
@@ -1084,7 +1082,11 @@ class TransferEngine {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, _currentStream, {
+        Object.defineProperty(this, _currentStreamController, {
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, _aborted, {
             writable: true,
             value: void 0
         });
@@ -1094,6 +1096,7 @@ class TransferEngine {
             schemaDiff: [],
             errors: {}
         };
+        _class_private_field_loose_base$6(this, _aborted)[_aborted] = false;
         this.diagnostics = createDiagnosticReporter();
         validateProvider('source', sourceProvider);
         validateProvider('destination', destinationProvider);
@@ -1292,6 +1295,9 @@ function assertSchemasMatching(sourceSchemas, destinationSchemas) {
     }
 }
 async function transferStage(options) {
+    if (_class_private_field_loose_base$6(this, _aborted)[_aborted]) {
+        throw new TransferEngineError('fatal', 'Transfer aborted.');
+    }
     const { stage, source, destination, transform, tracker } = options;
     const updateEndTime = ()=>{
         const stageData = this.progress.data[stage];
@@ -1323,27 +1329,42 @@ async function transferStage(options) {
         return;
     }
     _class_private_field_loose_base$6(this, _emitStageUpdate)[_emitStageUpdate]('start', stage);
-    await new Promise((resolve, reject)=>{
-        let stream = source;
+    try {
+        const streams = [
+            source
+        ];
         if (transform) {
-            stream = stream.pipe(transform);
+            streams.push(transform);
         }
         if (tracker) {
-            stream = stream.pipe(tracker);
-        }
-        _class_private_field_loose_base$6(this, _currentStream)[_currentStream] = stream.pipe(destination).on('error', (e)=>{
-            updateEndTime();
-            _class_private_field_loose_base$6(this, _emitStageUpdate)[_emitStageUpdate]('error', stage);
-            this.reportError(e, 'error');
+            streams.push(tracker);
+        }
+        streams.push(destination);
+        // NOTE: to debug/confirm backpressure issues from misbehaving stream, uncomment the following lines
+        // source.on('pause', () => console.log(`[${stage}] Source paused due to backpressure`));
+        // source.on('resume', () => console.log(`[${stage}] Source resumed`));
+        // destination.on('drain', () =>
+        //   console.log(`[${stage}] Destination drained, resuming data flow`)
+        // );
+        // destination.on('error', (err) => console.error(`[${stage}] Destination error:`, err));
+        const controller = new AbortController();
+        const { signal } = controller;
+        // Store the controller so you can cancel later
+        _class_private_field_loose_base$6(this, _currentStreamController)[_currentStreamController] = controller;
+        await promises.pipeline(streams, {
+            signal
+        });
+        _class_private_field_loose_base$6(this, _emitStageUpdate)[_emitStageUpdate]('finish', stage);
+    } catch (e) {
+        updateEndTime();
+        _class_private_field_loose_base$6(this, _emitStageUpdate)[_emitStageUpdate]('error', stage);
+        this.reportError(e, 'error');
+        if (!destination.destroyed) {
             destination.destroy(e);
-            reject(e);
-        }).on('close', ()=>{
-            _class_private_field_loose_base$6(this, _currentStream)[_currentStream] = undefined;
-            updateEndTime();
-            resolve();
-        });
-    });
-    _class_private_field_loose_base$6(this, _emitStageUpdate)[_emitStageUpdate]('finish', stage);
+        }
+    } finally{
+        updateEndTime();
+    }
 }
 async function resolveProviderResource() {
     const sourceMetadata = await this.sourceProvider.getMetadata();
@@ -1915,9 +1936,7 @@ const createLinkQuery = (strapi, trx)=>{
                 assignOrderColumns();
                 const qb = connection.insert(payload).into(addSchema(joinTable.name));
                 if (trx) {
-                    await trx.transaction(async (nestedTrx)=>{
-                        await qb.transacting(nestedTrx);
-                    });
+                    await qb.transacting(trx);
                 }
             }
             if ('morphColumn' in attribute && attribute.morphColumn) {
@@ -2370,7 +2389,7 @@ class LocalStrapiDestinationProvider {
                     const provider = strapi.config.get('plugin::upload').provider;
                     const fileId = fileEntitiesMapper?.[uploadData.id];
                     if (!fileId) {
-                        callback(new Error(`File ID not found for ID: ${uploadData.id}`));
+                        return callback(new Error(`File ID not found for ID: ${uploadData.id}`));
                     }
                     try {
                         await strapi.plugin('upload').provider.uploadStream(uploadData);
@@ -2421,9 +2440,9 @@ class LocalStrapiDestinationProvider {
                                 provider
                             }
                         });
-                        callback();
+                        return callback();
                     } catch (error) {
-                        callback(new Error(`Error while uploading asset ${chunk.filename} ${error}`));
+                        return callback(new Error(`Error while uploading asset ${chunk.filename} ${error}`));
                     }
                 });
             }
@@ -2960,7 +2979,7 @@ function reportError(message, error) {
 }
 function handleStreamError(streamType, err) {
     const { message, stack } = err;
-    const errorMessage = `Error in ${streamType} read stream: ${message}`;
+    const errorMessage = `[Data transfer] Error in ${streamType} read stream: ${message}`;
     const formattedError = {
         message: errorMessage,
         stack,
@@ -3132,17 +3151,6 @@ const connectToWebsocket = (address, options, diagnostics)=>{
 const trimTrailingSlash = (input)=>{
     return input.replace(/\/$/, '');
 };
-const wait = (ms)=>{
-    return new Promise((resolve)=>{
-        setTimeout(resolve, ms);
-    });
-};
-const waitUntil = async (test, interval)=>{
-    while(!test()){
-        await wait(interval);
-    }
-    return Promise.resolve();
-};
 
 const TRANSFER_PATH = '/transfer/runner';
 const TRANSFER_METHODS = [
@@ -3563,6 +3571,17 @@ class RemoteStrapiSourceProvider {
         });
         // Init the asset map
         const assets = {};
+        // Watch for stalled assets; if we don't receive a chunk within timeout, abort transfer
+        const resetTimeout = (assetID)=>{
+            if (assets[assetID].timeout) {
+                clearTimeout(assets[assetID].timeout);
+            }
+            assets[assetID].timeout = setTimeout(()=>{
+                _class_private_field_loose_base$2(this, _reportInfo$2)[_reportInfo$2](`Asset ${assetID} transfer stalled, aborting.`);
+                assets[assetID].status = 'errored';
+                assets[assetID].stream.destroy(new Error(`Asset ${assetID} transfer timed out`));
+            }, this.options.streamTimeout);
+        };
         stream/**
        * Process a payload of many transfer assets and performs the following tasks:
        * - Start: creates a stream for new assets.
@@ -3573,56 +3592,46 @@ class RemoteStrapiSourceProvider {
                 const { action, assetID } = item;
                 // Creates the stream to send the incoming asset through
                 if (action === 'start') {
-                    // Ignore the item if a transfer has already been started for the same asset ID
+                    // if a transfer has already been started for the same asset ID, something is wrong
                     if (assets[assetID]) {
-                        continue;
+                        throw new Error(`Asset ${assetID} already started`);
                     }
+                    _class_private_field_loose_base$2(this, _reportInfo$2)[_reportInfo$2](`Asset ${assetID} starting`);
                     // Register the asset
                     assets[assetID] = {
                         ...item.data,
                         stream: new stream$1.PassThrough(),
-                        status: 'idle',
+                        status: 'ok',
                         queue: []
                     };
+                    resetTimeout(assetID);
                     // Connect the individual asset stream to the main asset stage stream
                     // Note: nothing is transferred until data chunks are fed to the asset stream
                     await this.writeAsync(pass, assets[assetID]);
-                } else if (action === 'stream') {
-                    // If the asset hasn't been registered, or if it's been closed already, then ignore the message
+                } else if (action === 'stream' || action === 'end') {
+                    // If the asset hasn't been registered, or if it's been closed already, something is wrong
                     if (!assets[assetID]) {
-                        continue;
+                        throw new Error(`No id matching ${assetID} for stream action`);
                     }
-                    switch(assets[assetID].status){
-                        // The asset is ready to accept a new chunk, write it now
-                        case 'idle':
-                            await writeAssetChunk(assetID, item.data);
-                            break;
-                        // The resource is busy, queue the current chunk so that it gets transferred as soon as possible
-                        case 'busy':
-                            assets[assetID].queue.push(item);
-                            break;
+                    // On every action, reset the timeout timer
+                    if (action === 'stream') {
+                        resetTimeout(assetID);
+                    } else {
+                        clearTimeout(assets[assetID].timeout);
                     }
-                } else if (action === 'end') {
-                    // If the asset has already been closed, or if it was never registered, ignore the command
-                    if (!assets[assetID]) {
-                        continue;
+                    if (assets[assetID].status === 'closed') {
+                        throw new Error(`Asset ${assetID} is closed`);
                     }
-                    switch(assets[assetID].status){
-                        // There's no ongoing activity, the asset is ready to be closed
-                        case 'idle':
-                        case 'errored':
-                            await closeAssetStream(assetID);
-                            break;
-                        // The resource is busy, wait for a different state and close the stream.
-                        case 'busy':
-                            await Promise.race([
-                                // Either: wait for the asset to be ready to be closed
-                                waitUntil(()=>assets[assetID].status !== 'busy', 100),
-                                // Or: if the last chunks are still not processed after ten seconds
-                                wait(10000)
-                            ]);
-                            await closeAssetStream(assetID);
-                            break;
+                    assets[assetID].queue.push(item);
+                }
+            }
+            // each new payload will start new processQueue calls, which may cause some extra calls
+            // it's essentially saying "start processing this asset again, I added more data to the queue"
+            for(const assetID in assets){
+                if (Object.prototype.hasOwnProperty.call(assets, assetID)) {
+                    const asset = assets[assetID];
+                    if (asset.queue?.length > 0) {
+                        await processQueue(assetID);
                     }
                 }
             }
@@ -3630,38 +3639,48 @@ class RemoteStrapiSourceProvider {
             pass.end();
         });
         /**
-     * Writes a chunk of data for the specified asset with the given id.
-     */ const writeAssetChunk = async (id, data)=>{
+     * Start processing the queue for a given assetID
+     *
+     * Even though this is a loop that attempts to process the entire queue, it is safe to call this more than once
+     * for the same asset id because the queue is shared globally, the items are shifted off, and immediately written
+     */ const processQueue = async (id)=>{
             if (!assets[id]) {
                 throw new Error(`Failed to write asset chunk for "${id}". Asset not found.`);
             }
-            const { status: currentStatus } = assets[id];
-            if (currentStatus !== 'idle') {
+            const asset = assets[id];
+            const { status: currentStatus } = asset;
+            if ([
+                'closed',
+                'errored'
+            ].includes(currentStatus)) {
                 throw new Error(`Failed to write asset chunk for "${id}". The asset is currently "${currentStatus}"`);
             }
-            const nextItemInQueue = ()=>assets[id].queue.shift();
-            try {
-                // Lock the asset
-                assets[id].status = 'busy';
-                // Save the current chunk
-                await unsafe_writeAssetChunk(id, data);
-                // Empty the queue if needed
-                let item = nextItemInQueue();
-                while(item){
-                    await unsafe_writeAssetChunk(id, item.data);
-                    item = nextItemInQueue();
+            while(asset.queue.length > 0){
+                const data = asset.queue.shift();
+                if (!data) {
+                    throw new Error(`Invalid chunk found for ${id}`);
+                }
+                try {
+                    // if this is an end chunk, close the asset stream
+                    if (data.action === 'end') {
+                        _class_private_field_loose_base$2(this, _reportInfo$2)[_reportInfo$2](`Ending asset stream for ${id}`);
+                        await closeAssetStream(id);
+                        break; // Exit the loop after closing the stream
+                    }
+                    // Save the current chunk
+                    await writeChunkToStream(id, data);
+                } catch  {
+                    if (!assets[id]) {
+                        throw new Error(`No id matching ${id} for writeAssetChunk`);
+                    }
                 }
-                // Unlock the asset
-                assets[id].status = 'idle';
-            } catch  {
-                assets[id].status = 'errored';
             }
         };
         /**
      * Writes a chunk of data to the asset's stream.
      *
      * Only check if the targeted asset exists, no other validation is done.
-     */ const unsafe_writeAssetChunk = async (id, data)=>{
+     */ const writeChunkToStream = async (id, data)=>{
             const asset = assets[id];
             if (!asset) {
                 throw new Error(`Failed to write asset chunk for "${id}". Asset not found.`);
@@ -3682,9 +3701,11 @@ class RemoteStrapiSourceProvider {
             await new Promise((resolve, reject)=>{
                 const { stream } = assets[id];
                 stream.on('close', ()=>{
-                    delete assets[id];
                     resolve();
-                }).on('error', reject).end();
+                }).on('error', (e)=>{
+                    assets[id].status = 'errored';
+                    reject(new Error(`Failed to close asset "${id}". Asset stream error: ${e.toString()}`));
+                }).end();
             });
         };
         return pass;
@@ -3799,6 +3820,9 @@ class RemoteStrapiSourceProvider {
         });
         this.name = 'source::remote-strapi';
         this.type = 'source';
+        this.defaultOptions = {
+            streamTimeout: 15000
+        };
         this.writeAsync = (stream, data)=>{
             return new Promise((resolve, reject)=>{
                 stream.write(data, (error)=>{
@@ -3809,7 +3833,10 @@ class RemoteStrapiSourceProvider {
                 });
             });
         };
-        this.options = options;
+        this.options = {
+            ...this.defaultOptions,
+            ...options
+        };
         this.ws = null;
         this.dispatcher = null;
     }
@@ -4742,6 +4769,18 @@ const createPullController = handlerControllerFactory((proto)=>({
                 kind: 'warning'
             });
         },
+        onError (error) {
+            this.diagnostics?.report({
+                details: {
+                    message: error.message,
+                    error,
+                    createdAt: new Date(),
+                    name: error.name,
+                    severity: 'fatal'
+                },
+                kind: 'error'
+            });
+        },
         assertValidTransferAction (action) {
             // Abstract the constant to string[] to allow looser check on the given action
             const validActions = VALID_TRANSFER_ACTIONS;
@@ -4815,6 +4854,15 @@ const createPullController = handlerControllerFactory((proto)=>({
             let batch = [];
             const stream = this.streams?.[stage];
             const batchLength = ()=>Buffer.byteLength(JSON.stringify(batch));
+            const maybeConfirm = async (data)=>{
+                try {
+                    await this.confirm(data);
+                } catch (error) {
+                    // Handle the error, log it, or take other appropriate actions
+                    strapi?.log.error(`[Data transfer] Message confirmation failed: ${error?.message}`);
+                    this.onError(error);
+                }
+            };
             const sendBatch = async ()=>{
                 await this.confirm({
                     type: 'transfer',
@@ -4823,6 +4871,7 @@ const createPullController = handlerControllerFactory((proto)=>({
                     error: null,
                     id
                 });
+                batch = [];
             };
             if (!stream) {
                 throw new ProviderTransferError(`No available stream found for ${stage}`);
@@ -4833,7 +4882,6 @@ const createPullController = handlerControllerFactory((proto)=>({
                         batch.push(chunk);
                         if (batchLength() >= batchSize) {
                             await sendBatch();
-                            batch = [];
                         }
                     } else {
                         await this.confirm({
@@ -4849,7 +4897,6 @@ const createPullController = handlerControllerFactory((proto)=>({
                 }
                 if (batch.length > 0 && stage !== 'assets') {
                     await sendBatch();
-                    batch = [];
                 }
                 await this.confirm({
                     type: 'transfer',
@@ -4859,7 +4906,8 @@ const createPullController = handlerControllerFactory((proto)=>({
                     id
                 });
             } catch (e) {
-                await this.confirm({
+                // TODO: if this confirm fails, can we abort the whole transfer?
+                await maybeConfirm({
                     type: 'transfer',
                     data: null,
                     ended: true,
@@ -4908,7 +4956,7 @@ const createPullController = handlerControllerFactory((proto)=>({
                     };
                     const BATCH_MAX_SIZE = 1024 * 1024; // 1MB
                     if (!assets) {
-                        throw new Error('bad');
+                        throw new Error('Assets read stream could not be created');
                     }
                     /**
          * Generates batches of 1MB of data from the assets stream to avoid