Handle array of strings too

jen-huang · jen-huang · commit b804d8a64460 · 2021-02-05T12:16:31.000-08:00
diff --git a/x-pack/plugins/fleet/server/services/epm/agent/agent.test.ts b/x-pack/plugins/fleet/server/services/epm/agent/agent.test.ts
@@ -27,13 +27,18 @@ optional_field: {{optional_field}}
 {{/if}}
 foo: {{bar}}
 some_text_field: {{should_be_text}}
+multi_text_field:
+{{#each multi_text}}
+  - {{this}}
+{{/each}}
       `;
     const vars = {
       paths: { value: ['/usr/local/var/log/nginx/access.log'] },
       password: { type: 'password', value: '' },
       optional_field: { type: 'text', value: undefined },
       bar: { type: 'text', value: 'bar' },
       should_be_text: { type: 'text', value: '1234' },
+      multi_text: { type: 'text', value: ['1234', 'foo', 'bar'] },
     };
 
     const output = compileTemplate(vars, streamTemplate);
@@ -45,6 +50,7 @@ some_text_field: {{should_be_text}}
       password: '',
       foo: 'bar',
       some_text_field: '1234',
+      multi_text_field: ['1234', 'foo', 'bar'],
     });
   });
 
diff --git a/x-pack/plugins/fleet/server/services/epm/agent/agent.ts b/x-pack/plugins/fleet/server/services/epm/agent/agent.ts
@@ -58,6 +58,10 @@ function replaceVariablesInYaml(yamlVariables: { [k: string]: any }, yaml: any)
   return yaml;
 }
 
+const maybeEscapeNumericString = (value: string) => {
+  return value.length && !isNaN(+value) ? `"${value}"` : value;
+};
+
 function buildTemplateVariables(variables: PackagePolicyConfigRecord, templateStr: string) {
   const yamlValues: { [k: string]: any } = {};
   const vars = Object.entries(variables).reduce((acc, [key, recordEntry]) => {
@@ -84,13 +88,14 @@ function buildTemplateVariables(variables: PackagePolicyConfigRecord, templateSt
       const yamlKeyPlaceholder = `##${key}##`;
       varPart[lastKeyPart] = `"${yamlKeyPlaceholder}"`;
       yamlValues[yamlKeyPlaceholder] = recordEntry.value ? safeLoad(recordEntry.value) : null;
-    } else if (
-      recordEntry.type &&
-      recordEntry.type === 'text' &&
-      recordEntry.value?.length &&
-      !isNaN(+recordEntry.value)
-    ) {
-      varPart[lastKeyPart] = `"${recordEntry.value}"`;
+    } else if (recordEntry.type && recordEntry.type === 'text' && recordEntry.value?.length) {
+      if (Array.isArray(recordEntry.value)) {
+        varPart[lastKeyPart] = recordEntry.value.map((value: string) =>
+          maybeEscapeNumericString(value)
+        );
+      } else {
+        varPart[lastKeyPart] = maybeEscapeNumericString(recordEntry.value);
+      }
     } else {
       varPart[lastKeyPart] = recordEntry.value;
     }
