From 51f066cbcd7a7d24c9e3885b8aef06d20d9d173d Mon Sep 17 00:00:00 2001
From: s <59780143+stwonary@users.noreply.github.com>
Date: Sun, 4 Apr 2021 04:33:22 -0400
Subject: [PATCH] :sparkles: Add RabbitMQ Header parameter (#1622)

* Add RabbitMQ Exchange's Header parameter

* code refactoring

* remove log
---
 .../nodes/RabbitMQ/DefaultOptions.ts          | 31 ++++++++++++++
 .../nodes/RabbitMQ/RabbitMQ.node.ts           | 42 ++++++++++++++++++-
 2 files changed, 72 insertions(+), 1 deletion(-)

diff --git a/packages/nodes-base/nodes/RabbitMQ/DefaultOptions.ts b/packages/nodes-base/nodes/RabbitMQ/DefaultOptions.ts
index 10d295dc036..bab50fa7435 100644
--- a/packages/nodes-base/nodes/RabbitMQ/DefaultOptions.ts
+++ b/packages/nodes-base/nodes/RabbitMQ/DefaultOptions.ts
@@ -36,6 +36,37 @@ export const rabbitDefaultOptions: Array<INodePropertyOptions | INodeProperties
 			},
 		],
 	},
+	{
+		displayName: 'Headers',
+		name: 'headers',
+		placeholder: 'Add Header',
+		description: 'Headers to add.',
+		type: 'fixedCollection',
+		typeOptions: {
+			multipleValues: true,
+		},
+		default: {},
+		options: [
+			{
+				name: 'header',
+				displayName: 'Header',
+				values: [
+					{
+						displayName: 'Key',
+						name: 'key',
+						type: 'string',
+						default: '',
+					},
+					{
+						displayName: 'Value',
+						name: 'value',
+						type: 'string',
+						default: '',
+					},
+				],
+			},
+		],
+	},
 	{
 		displayName: 'Auto Delete',
 		name: 'autoDelete',
diff --git a/packages/nodes-base/nodes/RabbitMQ/RabbitMQ.node.ts b/packages/nodes-base/nodes/RabbitMQ/RabbitMQ.node.ts
index e5d84437389..3b9cd60c6fc 100644
--- a/packages/nodes-base/nodes/RabbitMQ/RabbitMQ.node.ts
+++ b/packages/nodes-base/nodes/RabbitMQ/RabbitMQ.node.ts
@@ -208,6 +208,37 @@ export class RabbitMQ implements INodeType {
 							},
 						],
 					},
+					{
+						displayName: 'Headers',
+						name: 'headers',
+						placeholder: 'Add Header',
+						description: 'Headers to add.',
+						type: 'fixedCollection',
+						typeOptions: {
+							multipleValues: true,
+						},
+						default: {},
+						options: [
+							{
+								name: 'header',
+								displayName: 'Header',
+								values: [
+									{
+										displayName: 'Key',
+										name: 'key',
+										type: 'string',
+										default: '',
+									},
+									{
+										displayName: 'Value',
+										name: 'value',
+										type: 'string',
+										default: '',
+									},
+								],
+							},
+						],
+					},
 					{
 						displayName: 'Auto Delete',
 						name: 'autoDelete',
@@ -322,6 +353,15 @@ export class RabbitMQ implements INodeType {
 
 				const options = this.getNodeParameter('options', 0, {}) as IDataObject;
 
+				let headers : IDataObject = {};
+				if (options.headers && ((options.headers as IDataObject).header! as IDataObject[]).length) {
+					const additionalHeaders: IDataObject = {};
+					((options.headers as IDataObject).header as IDataObject[]).forEach((header: IDataObject) => {
+						additionalHeaders[header.key as string] = header.value;
+					});
+					headers = additionalHeaders;
+				}
+
 				channel = await rabbitmqConnectExchange.call(this, exchange, type, options);
 
 				const sendInputData = this.getNodeParameter('sendInputData', 0) as boolean;
@@ -336,7 +376,7 @@ export class RabbitMQ implements INodeType {
 						message = this.getNodeParameter('message', i) as string;
 					}
 
-					exchangePromises.push(channel.publish(exchange, routingKey, Buffer.from(message)));
+					exchangePromises.push(channel.publish(exchange, routingKey, Buffer.from(message), {headers}));
 				}
 
 				// @ts-ignore