n8n-enterprise-unlocked/packages/nodes-base/nodes/Airtop/actions/file/upload.operation.ts

import type { IExecuteFunctions, INodeExecutionData, INodeProperties } from 'n8n-workflow';
import { NodeOperationError } from 'n8n-workflow';

import {
	createAndUploadFile,
	pushFileToSession,
	triggerFileInput,
	createFileBuffer,
} from './helpers';
import { validateRequiredStringField } from '../../GenericFunctions';
import {
	sessionIdField,
	windowIdField,
	elementDescriptionField,
	includeHiddenElementsField,
} from '../common/fields';

const displayOptions = {
	show: {
		resource: ['file'],
		operation: ['upload'],
	},
};

export const description: INodeProperties[] = [
	{
		...sessionIdField,
		description: 'The session ID to load the file into',
		displayOptions,
	},
	{
		...windowIdField,
		description: 'The window ID to trigger the file input in',
		displayOptions,
	},
	{
		displayName: 'File Name',
		name: 'fileName',
		type: 'string',
		default: '',
		required: true,
		description:
			'Name for the file to upload. For a session, all files loaded should have <b>unique names</b>.',
		displayOptions,
	},
	{
		displayName: 'File Type',
		name: 'fileType',
		type: 'options',
		options: [
			{
				name: 'Browser Download',
				value: 'browser_download',
			},
			{
				name: 'Screenshot',
				value: 'screenshot',
			},
			{
				name: 'Video',
				value: 'video',
			},
			{
				name: 'Customer Upload',
				value: 'customer_upload',
			},
		],
		default: 'customer_upload',
		description: "Choose the type of file to upload. Defaults to 'Customer Upload'.",
		displayOptions,
	},
	{
		displayName: 'Source',
		name: 'source',
		type: 'options',
		options: [
			{
				name: 'URL',
				value: 'url',
			},
			{
				name: 'Binary',
				value: 'binary',
			},
		],
		default: 'url',
		description: 'Source of the file to upload',
		displayOptions,
	},
	{
		displayName: 'Binary Property',
		name: 'binaryPropertyName',
		type: 'string',
		default: 'data',
		required: true,
		displayOptions: {
			show: {
				source: ['binary'],
				...displayOptions.show,
			},
		},
		description: 'Name of the binary property containing the file data',
	},
	{
		displayName: 'URL',
		name: 'url',
		type: 'string',
		default: '',
		required: true,
		displayOptions: {
			show: {
				source: ['url'],
				...displayOptions.show,
			},
		},
		description: 'URL from where to fetch the file to upload',
	},
	{
		displayName: 'Trigger File Input',
		name: 'triggerFileInputParameter',
		type: 'boolean',
		default: true,
		description:
			'Whether to automatically trigger the file input dialog in the current window. If disabled, the file will only be uploaded to the session without opening the file input dialog.',
		displayOptions,
	},
	{
		...elementDescriptionField,
		description: 'Optional description of the file input to interact with',
		placeholder: 'e.g. the file upload selection box',
		displayOptions: {
			show: {
				triggerFileInputParameter: [true],
				...displayOptions.show,
			},
		},
	},
	{
		...includeHiddenElementsField,
		displayOptions: {
			show: {
				triggerFileInputParameter: [true],
				...displayOptions.show,
			},
		},
	},
];

export async function execute(
	this: IExecuteFunctions,
	index: number,
): Promise<INodeExecutionData[]> {
	const sessionId = validateRequiredStringField.call(this, index, 'sessionId', 'Session ID');
	const windowId = validateRequiredStringField.call(this, index, 'windowId', 'Window ID');
	const fileName = this.getNodeParameter('fileName', index, '') as string;
	const fileType = this.getNodeParameter('fileType', index, 'customer_upload') as string;
	const source = this.getNodeParameter('source', index, 'url') as string;
	const url = this.getNodeParameter('url', index, '') as string;
	const binaryPropertyName = this.getNodeParameter('binaryPropertyName', index, '');
	const triggerFileInputParameter = this.getNodeParameter(
		'triggerFileInputParameter',
		index,
		true,
	) as boolean;
	const elementDescription = this.getNodeParameter('elementDescription', index, '') as string;
	const includeHiddenElements = this.getNodeParameter(
		'includeHiddenElements',
		index,
		false,
	) as boolean;
	// Get the file content based on source type
	const fileValue = source === 'url' ? url : binaryPropertyName;

	try {
		const fileBuffer = await createFileBuffer.call(this, source, fileValue, index);
		const fileId = await createAndUploadFile.call(this, fileName, fileBuffer, fileType);
		// Push file to session
		await pushFileToSession.call(this, fileId, sessionId);

		if (triggerFileInputParameter) {
			await triggerFileInput.call(this, {
				fileId,
				windowId,
				sessionId,
				elementDescription,
				includeHiddenElements,
			});
		}

		return this.helpers.returnJsonArray({
			sessionId,
			windowId,
			data: {
				fileId,
				message: 'File uploaded successfully',
			},
		});
	} catch (error) {
		throw new NodeOperationError(this.getNode(), error as Error);
	}
}