feat(createSshConnection): add environment variable support to exec m…

…ethod
deploysteps · Nov 17, 2023 · 73d7727 · 73d7727
1 parent 3bca9a3
commit 73d7727
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/lib/utils/createSshConnection.js b/lib/utils/createSshConnection.js
@@ -149,7 +149,23 @@ export async function createSshConnection({ host, username, port = 22, privateKe
     interactive,
     script: (scriptContent, onData) => script(scriptContent, onData, false),
     sudoScript: (scriptContent, onData) => script(scriptContent, onData, true),
-    exec: (command, onData) => execRemote(connObj, command, onData, config),
+    exec: (command, optionsOrOnData, onData) => {
+      if (typeof optionsOrOnData === 'function') {
+        onData = optionsOrOnData;
+        optionsOrOnData = {};
+      }
+
+      let envString = '';
+      if (optionsOrOnData?.env) {
+        for (const [key, value] of Object.entries(optionsOrOnData.env)) {
+          envString += `${key}='${value}' `;
+        }
+      }
+
+      const fullCommand = `${envString}${command}`;
+
+      return execRemote(connObj, fullCommand, onData, config);
+    },
     close: () => {
       return new Promise(resolve => {
         connObj.once('close', resolve);