n8n/packages/cli/src/executionLifecycleHooks/restoreBinaryDataId.ts

import Container from 'typedi';
import { BinaryDataService } from 'n8n-core';
import type { IRun } from 'n8n-workflow';
import type { BinaryData } from 'n8n-core';

export function isMissingExecutionId(
	fileId: string,
	mode: BinaryData.NonDefaultMode,
	uuidV4CharLength = 36,
) {
	return mode === 'filesystem' ? uuidV4CharLength === fileId.length : fileId.includes('/temp/');
}

/**
 * Whenever the execution ID is not available to the binary data service at the
 * time of writing a binary data file, its name is missing the execution ID.
 *
 * This function restores the ID in the file name and run data reference.
 *
 * ```txt
 * filesystem:11869055-83c4-4493-876a-9092c4708b9b ->
 * filesystem:39011869055-83c4-4493-876a-9092c4708b9b
 *
 * s3:workflows/123/executions/temp/binary_data/69055-83c4-4493-876a-9092c4708b9b ->
 * s3:workflows/123/executions/390/binary_data/69055-83c4-4493-876a-9092c4708b9b
 * ```
 */
export async function restoreBinaryDataId(run: IRun, executionId: string) {
	const { runData } = run.data.resultData;

	const promises = Object.keys(runData).map(async (nodeName) => {
		const binaryDataId = runData[nodeName]?.[0]?.data?.main?.[0]?.[0]?.binary?.data.id;

		if (!binaryDataId) return;

		const [mode, fileId] = binaryDataId.split(':') as [BinaryData.NonDefaultMode, string];

		if (!isMissingExecutionId(fileId, mode)) return;

		const correctFileId =
			mode === 'filesystem' ? `${executionId}${fileId}` : fileId.replace('temp', executionId);

		await Container.get(BinaryDataService).rename(fileId, correctFileId);

		const correctBinaryDataId = `${mode}:${correctFileId}`;

		// @ts-expect-error Validated at the top
		run.data.resultData.runData[nodeName][0].data.main[0][0].binary.data.id = correctBinaryDataId;
	});

	await Promise.all(promises);
}