Merge pull request #3 from Kitsumon/a

Update ssm-helper.js
Kitsumon · Apr 17, 2023 · 4a853e6 · 4a853e6
2 parents 2d04d53 + c2eb5a8
commit 4a853e6
Showing 1 changed file with 31 additions and 17 deletions.
diff --git a/ssm-helper.js b/ssm-helper.js
@@ -4,27 +4,41 @@ const getParameters = async (ssmPath, getChildren, decryption, region) => {
     AWS.config.update({region: region});
     const ssm = new AWS.SSM();
     let params;
+    const ssmParams = [];
+
     let promise;
-    if (getChildren)
-    {
-        params =
+    let nextToken;
+    while(!promise) {
+        if (getChildren)
         {
-            Path: ssmPath,
-            WithDecryption: decryption
-        };
-        promise = ssm.getParametersByPath(params).promise();
-    }
-    else
-    {
-        params =
+            params =
+            {
+                Path: ssmPath,
+                WithDecryption: decryption,
+                NextToken: nextToken
+            };
+            promise = ssm.getParametersByPath(params).promise();
+        }
+        else
+        {
+            params =
+            {
+                Names: [ssmPath],
+                WithDecryption: decryption,                
+                NextToken: nextToken
+            };
+            promise = ssm.getParameters(params).promise();
+        }
+
+        const result = await promise;
+        ssmParams.push(...result.Parameters);
+        if (result.NextToken)
         {
-            Names: [ssmPath],
-            WithDecryption: decryption
-        };
-        promise = ssm.getParameters(params).promise();;
+            nextToken = result.NextToken;
+            promise = null;
+        }
     }
-    const result = await promise;
-    return result.Parameters;
+    return ssmParams;
 }
 
 module.exports = {getParameters};