[Snyk] Upgrade pseudoexec from 0.1.4 to 0.2.0

bin/cml/runner.js

@@ -118,6 +118,7 @@ const runCloud = async (opts) => {
       cloud,
       cloudRegion: region,
       cloudType: type,
+      cloudMetadata: metadata,
       cloudGpu: gpu,
       cloudHddSize: hddSize,
       cloudSshPrivate: sshPrivate,
@@ -151,6 +152,7 @@ const runCloud = async (opts) => {
         cloud,
         region,
         type,
+        metadata,
         gpu: gpu === 'tesla' ? 'v100' : gpu,
         hddSize,
         sshPrivate,
@@ -190,6 +192,7 @@ const runCloud = async (opts) => {
           instanceLaunchTime: attributes.instance_launch_time,
           instanceType: attributes.instance_type,
           labels: attributes.labels,
+          metadata: attributes.metadata,
           name: attributes.name,
           region: attributes.region,
           repo: attributes.repo,
@@ -467,6 +470,19 @@ exports.builder = (yargs) =>
         description:
           'Instance type. Choices: [m, l, xl]. Also supports native types like i.e. t2.micro'
       },
+      cloudMetadata: {
+        type: 'array',
+        string: true,
+        coerce: (items) => {
+          const keyValuePairs = items.map((item) => [
+            ...item.split(/=(.+)/),
+            null
+          ]);
+          return Object.fromEntries(keyValuePairs);
+        },
+        description:
+          'Key Value pairs to associate cml-runner instance on the provider i.e. tags/labels "key=value"'
+      },
       cloudGpu: {
         type: 'string',
         choices: ['nogpu', 'k80', 'v100', 'tesla'],

bin/cml/runner.test.js

@@ -42,6 +42,9 @@ Options:
                                                    [string] [default: \\"us-west\\"]
   --cloud-type                Instance type. Choices: [m, l, xl]. Also supports
                               native types like i.e. t2.micro           [string]
+  --cloud-metadata            Key Value pairs to associate cml-runner instance
+                              on the provider i.e. tags/labels \\"key=value\\"
+                                                                         [array]
   --cloud-gpu                 GPU type.
                              [string] [choices: \\"nogpu\\", \\"k80\\", \\"v100\\", \\"tesla\\"]
   --cloud-hdd-size            HDD size in GB                            [number]

package.json

@@ -77,7 +77,7 @@
     "node-forge": "^0.10.0",
     "node-ssh": "^12.0.0",
     "proxy-agent": "^5.0.0",
-    "pseudoexec": "^0.1.4",
+    "pseudoexec": "^0.2.0",
     "semver": "^7.3.5",
     "simple-git": "^2.45.1",
     "strip-url-auth": "^1.0.1",

src/terraform.js

@@ -48,6 +48,9 @@ const destroy = async (opts = {}) => {
   );
 };
 
+const mapCloudMetadata = (metadata) =>
+  Object.entries(metadata).map(([key, value]) => `${key} = "${value || ''}"`);
+
 const iterativeProviderTpl = () => {
   return `
 terraform {
@@ -74,6 +77,7 @@ const iterativeCmlRunnerTpl = (opts = {}) => {
     name,
     single,
     type,
+    metadata,
     gpu,
     hddSize,
     sshPrivate,
@@ -83,7 +87,7 @@ const iterativeCmlRunnerTpl = (opts = {}) => {
     awsSecurityGroup
   } = opts;
 
-  return `
+  const template = `
 ${iterativeProviderTpl()}
 
 resource "iterative_cml_runner" "runner" {
@@ -108,8 +112,14 @@ resource "iterative_cml_runner" "runner" {
   ${spotPrice ? `spot_price = ${spotPrice}` : ''}
   ${startupScript ? `startup_script = "${startupScript}"` : ''}
   ${awsSecurityGroup ? `aws_security_group = "${awsSecurityGroup}"` : ''}
+  ${
+    metadata
+      ? `metadata = {\n    ${mapCloudMetadata(metadata).join('\n    ')}\n  }`
+      : ''
+  }
 }
 `;
+  return template;
 };
 
 const checkMinVersion = async () => {