n8n-enterprise-unlocked/packages/nodes-base/nodes/DataTable/actions/row/upsert.operation.ts

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

import { makeAddRow, getAddRow } from '../../common/addRow';
import { executeSelectMany, getSelectFields } from '../../common/selectMany';
import { getDataTableProxyExecute } from '../../common/utils';

export const FIELD: string = 'upsert';

const displayOptions: IDisplayOptions = {
	show: {
		resource: ['row'],
		operation: [FIELD],
	},
};

export const description: INodeProperties[] = [
	...getSelectFields(displayOptions),
	makeAddRow(FIELD, displayOptions),
];

export async function execute(
	this: IExecuteFunctions,
	index: number,
): Promise<INodeExecutionData[]> {
	const dataStoreProxy = await getDataTableProxyExecute(this, index);

	const row = getAddRow(this, index);

	const matches = await executeSelectMany(this, index, dataStoreProxy, true);

	// insert
	if (matches.length === 0) {
		const result = await dataStoreProxy.insertRows([row]);
		return result.map((json) => ({ json }));
	}

	// update
	const result = [];
	for (const match of matches) {
		const updatedRows = await dataStoreProxy.updateRows({
			data: row,
			filter: { id: match.json.id },
		});
		if (updatedRows.length !== 1) {
			throw new NodeOperationError(this.getNode(), 'invariant broken');
		}

		// The input object gets updated in the api call, somehow
		// And providing this column to the backend causes an unexpected column error
		// So let's just re-delete the field until we have a more aligned API
		delete row['updatedAt'];

		result.push(updatedRows[0]);
	}

	return result.map((json) => ({ json }));
}