n8n-enterprise-unlocked/packages/nodes-base/nodes/Snowflake/GenericFunctions.ts

import {
	IDataObject,
	INodeExecutionData,
} from 'n8n-workflow';

import * as snowflake from 'snowflake-sdk';

export function connect(conn: snowflake.Connection) {
	return new Promise((resolve, reject) => {
		conn.connect((err, conn) => {
			if (!err) {
				// @ts-ignore
				resolve();
			} else {
				reject(err);
			}
		});
	});
}

export function destroy(conn: snowflake.Connection) {
	return new Promise((resolve, reject) => {
		conn.destroy((err, conn) => {
			if (!err) {
				// @ts-ignore
				resolve();
			} else {
				reject(err);
			}
		});
	});
}

export function execute(conn: snowflake.Connection, sqlText: string, binds: snowflake.InsertBinds) {
	return new Promise((resolve, reject) => {
		conn.execute({
			sqlText,
			binds,
			complete: (err, stmt, rows) => {
				if (!err) {
					resolve(rows);
				} else {
					reject(err);
				}
			},
		});
	});
}

export function copyInputItems(items: INodeExecutionData[], properties: string[]): IDataObject[] {
	// Prepare the data to insert and copy it to be returned
	let newItem: IDataObject;
	return items.map((item) => {
		newItem = {};
		for (const property of properties) {
			if (item.json[property] === undefined) {
				newItem[property] = null;
			} else {
				newItem[property] = JSON.parse(JSON.stringify(item.json[property]));
			}
		}
		return newItem;
	});
}