Merge pull request activepieces#5344 from alinperghel/feat/rabbitmq-p…

…iece feat(rabbitmq): add support for vhosts
youngsecurity · Aug 13, 2024 · 0f2cd8d · 0f2cd8d
2 parents 0ae02ba + f2cc7fc
commit 0f2cd8d
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/packages/pieces/community/rabbitmq/package.json b/packages/pieces/community/rabbitmq/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@activepieces/piece-rabbitmq",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "dependencies": {
     "amqplib": "^0.10.4"
   },

diff --git a/packages/pieces/community/rabbitmq/src/index.ts b/packages/pieces/community/rabbitmq/src/index.ts
@@ -27,6 +27,11 @@ export const rabbitmqAuth = PieceAuth.CustomAuth({
       description: "Port",
       required: true,
     }),
+    vhost: Property.ShortText({
+      displayName: "Virtual Host",
+      description: "Virtual Host",
+      required: false,
+    }),
   },
 });
 

diff --git a/packages/pieces/community/rabbitmq/src/lib/common/index.ts b/packages/pieces/community/rabbitmq/src/lib/common/index.ts
@@ -15,5 +15,10 @@ export async function rabbitmqConnect(
 }
 
 function createAmqpURI(auth: PiecePropValueSchema<typeof rabbitmqAuth>): string {
-  return `amqp://${auth.username}:${auth.password}@${auth.host}:${auth.port}`;
+  const uri = `amqp://${auth.username}:${auth.password}@${auth.host}:${auth.port}`;
+
+  if (!auth.vhost) {
+    return uri;
+  }
+  return `${uri}/${auth.vhost}`;
 }