n8n/packages/nodes-base/nodes/Amqp/AmqpTrigger.node.ts

import { ContainerOptions } from 'rhea';

import { ITriggerFunctions } from 'n8n-core';
import {
	IDataObject,
	INodeType,
	INodeTypeDescription,
	ITriggerResponse,
} from 'n8n-workflow';


export class AmqpTrigger implements INodeType {
	description: INodeTypeDescription = {
		displayName: 'AMQP Trigger',
		name: 'amqpTrigger',
		icon: 'file:amqp.png',
		group: ['trigger'],
		version: 1,
		description: 'Listens to AMQP 1.0 Messages',
		defaults: {
			name: 'AMQP Trigger',
			color: '#00FF00',
		},
		inputs: [],
		outputs: ['main'],
		credentials: [{
			name: 'amqp',
			required: true,
		}],
		properties: [
			// Node properties which the user gets displayed and
			// can change on the node.
			{
				displayName: 'Queue / Topic',
				name: 'sink',
				type: 'string',
				default: '',
				placeholder: 'topic://sourcename.something',
				description: 'name of the queue of topic to listen to',
			},
			{
				displayName: 'Clientname',
				name: 'clientname',
				type: 'string',
				default: '',
				placeholder: 'for durable/persistent topic subscriptions, example: "n8n"',
				description: 'Leave empty for non-durable topic subscriptions or queues. ',
			},
			{
				displayName: 'Subscription',
				name: 'subscription',
				type: 'string',
				default: '',
				placeholder: 'for durable/persistent topic subscriptions, example: "order-worker"',
				description: 'Leave empty for non-durable topic subscriptions or queues',
			},
			{
				displayName: 'Options',
				name: 'options',
				type: 'collection',
				placeholder: 'Add Option',
				default: {},
				options: [
					{
						displayName: 'Convert Body To String',
						name: 'jsonConvertByteArrayToString',
						type: 'boolean',
						default: false,
						description: 'Convert JSON Body content (["body"]["content"]) from Byte Array to string. Needed for Azure Service Bus.',
					},
					{
						displayName: 'JSON Parse Body',
						name: 'jsonParseBody',
						type: 'boolean',
						default: false,
						description: 'Parse the body to an object.',
					},
					{
						displayName: 'Only Body',
						name: 'onlyBody',
						type: 'boolean',
						default: false,
						description: 'Returns only the body property.',
					},
					{
						displayName: 'Messages per Cicle',
						name: 'pullMessagesNumber',
						type: 'number',
						default: 100,
						description: 'Number of messages to pull from the bus for every cicle',
					},
					{
						displayName: 'Sleep Time',
						name: 'sleepTime',
						type: 'number',
						default: 10,
						description: 'Milliseconds to sleep after every cicle.',
					},
					{
						displayName: 'Container ID',
						name: 'containerID',
						type: 'string',
						default: '',
						description: 'Will be used to pass to the RHEA Backend as container_id',
					},
					{
						displayName: 'Reconnect',
						name: 'reconnect',
						type: 'boolean',
						default: true,
						description: 'If on, the library will automatically attempt to reconnect if disconnected',
					},
					{
						displayName: 'Reconnect limit',
						name: 'reconnectLimit',
						type: 'number',
						default: 50,
						description: 'maximum number of reconnect attempts',
					},				
				],
			},
		],
	};


	async trigger(this: ITriggerFunctions): Promise<ITriggerResponse> {

		const credentials = this.getCredentials('amqp');
		if (!credentials) {
			throw new Error('Credentials are mandatory!');
		}

		const sink = this.getNodeParameter('sink', '') as string;
		const clientname = this.getNodeParameter('clientname', '') as string;
		const subscription = this.getNodeParameter('subscription', '') as string;
		const options = this.getNodeParameter('options', {}) as IDataObject;
		const pullMessagesNumber = options.pullMessagesNumber || 100;
		const container_id = options.containerID as string;
		const containerReconnect = options.reconnect || true as boolean;
		const containerReconnectLimit = options.reconnectLimit || 50 as number;

		if (sink === '') {
			throw new Error('Queue or Topic required!');
		}

		let durable = false;

		if (subscription && clientname) {
			durable = true;
		}

		const container = require('rhea');

		/*
			Values are documentet here: https://github.com/amqp/rhea#container
		 */
		const connectOptions: ContainerOptions = {
			host: credentials.hostname,
			hostname: credentials.hostname,
			port: credentials.port,
			reconnect: containerReconnect,			    
			reconnect_limit: containerReconnectLimit,	
		};
		if (credentials.username || credentials.password) {
			connectOptions.username = credentials.username;
			connectOptions.password = credentials.password;
		}
		if (credentials.transportType) {
			connectOptions.transport = credentials.transportType;
		}
		if(container_id) {
			connectOptions.id = container_id;
			connectOptions.container_id = container_id;
		}

		let lastMsgId: number | undefined = undefined;
		const self = this;

		container.on('receiver_open', (context: any) => { // tslint:disable-line:no-any
			context.receiver.add_credit(pullMessagesNumber);
		});

		container.on('message', (context: any) => { // tslint:disable-line:no-any
			// ignore duplicate message check, don't think it's necessary, but it was in the rhea-lib example code
			if (context.message.message_id && context.message.message_id === lastMsgId) {
				return;
			}
			lastMsgId = context.message.message_id;

			let data = context.message;

			if (options.jsonConvertByteArrayToString === true && data.body.content !== undefined) {
				// The buffer is not ready... Stringify and parse back to load it.
				const cont = JSON.stringify(data.body.content);
				data.body = String.fromCharCode.apply(null, JSON.parse(cont).data);
			}

			if (options.jsonConvertByteArrayToString === true && data.body.content !== undefined) {
				// The buffer is not ready... Stringify and parse back to load it.
				const content = JSON.stringify(data.body.content);
				data.body = String.fromCharCode.apply(null, JSON.parse(content).data);
			}

			if (options.jsonParseBody === true) {
				data.body = JSON.parse(data.body);
			}
			if (options.onlyBody === true) {
				data = data.body;
			}


			self.emit([self.helpers.returnJsonArray([data])]);

			if (context.receiver.credit === 0) {
				setTimeout(() => {
					context.receiver.add_credit(pullMessagesNumber);
				}, options.sleepTime as number || 10);
			}
		});

		const connection = container.connect(connectOptions);
		let clientOptions = undefined;
		if (durable) {
			clientOptions = {
				name: subscription,
				source: {
					address: sink,
					durable: 2,
					expiry_policy: 'never',
				},
				credit_window: 0,	// prefetch 1
			};
		} else {
			clientOptions = {
				source: {
					address: sink,
				},
				credit_window: 0,	// prefetch 1
			};
		}
		connection.open_receiver(clientOptions);


		// The "closeFunction" function gets called by n8n whenever
		// the workflow gets deactivated and can so clean up.
		async function closeFunction() {
			container.removeAllListeners('receiver_open');
			container.removeAllListeners('message');
			connection.close();
		}

		// The "manualTriggerFunction" function gets called by n8n
		// when a user is in the workflow editor and starts the
		// workflow manually.
		// for AMQP it doesn't make much sense to wait here but
		// for a new user who doesn't know how this works, it's better to wait and show a respective info message
		async function manualTriggerFunction() {
			await new Promise((resolve, reject) => {
				const timeoutHandler = setTimeout(() => {
					reject(new Error('Aborted, no message received within 30secs. This 30sec timeout is only set for "manually triggered execution". Active Workflows will listen indefinitely.'));
				}, 30000);
				container.on('message', (context: any) => { // tslint:disable-line:no-any
					// Check if the only property present in the message is body
					// in which case we only emit the content of the body property
					// otherwise we emit all properties and their content
					if (Object.keys(context.message)[0] === 'body' && Object.keys(context.message).length === 1) {
						self.emit([self.helpers.returnJsonArray([context.message.body])]);
					} else {
						self.emit([self.helpers.returnJsonArray([context.message])]);
					}
					clearTimeout(timeoutHandler);
					resolve(true);
				});
			});
		}

		return {
			closeFunction,
			manualTriggerFunction,
		};

	}
}