Merge 51e3ff9 into 7e4afb3

kayomarz · Jun 6, 2021 · 3e4b924 · 3e4b924
2 parents 7e4afb3 + 51e3ff9
commit 3e4b924
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 33 deletions.
diff --git a/bin/unwrap.js b/bin/unwrap.js
@@ -3,24 +3,37 @@
 'use strict';
 
 const attr = require('../lib/dynamodb-data-types').AttributeValue;
-const content = process.argv[2];
 
-if (!content) {
-  throw new Error('Input data required.')
+if (process.stdin.isTTY) {
+  unwrap(process.argv[2]);
+} else {
+  let data = "";
+  process.stdin.on('data', function(chunk) {
+    data += chunk;
+  });
+  process.stdin.on('end', function() {
+    unwrap(data);
+  });
 }
 
-let payload = '';
-try {
-  payload = JSON.parse(content);
-}
-catch (err) {
-  payload = content
-}
-const output = attr.unwrap(payload);
+function unwrap(content) {
+  if (!content) {
+    throw new Error('Input data required.')
+  }
 
-if (typeof output === 'object') {
-  process.stdout.write(JSON.stringify(output) + '\n');
-}
-else {
-  process.stdout.write(output + '\n');
-}
+  let payload = '';
+  try {
+    payload = JSON.parse(content);
+  }
+  catch (err) {
+    payload = content
+  }
+  const output = attr.unwrap(payload);
+
+  if (typeof output === 'object') {
+    process.stdout.write(JSON.stringify(output) + '\n');
+  }
+  else {
+    process.stdout.write(output + '\n');
+  }
+}
diff --git a/bin/wrap.js b/bin/wrap.js
@@ -3,24 +3,37 @@
 'use strict';
 
 const attr = require('../lib/dynamodb-data-types').AttributeValue;
-const content = process.argv[2];
 
-if (!content) {
-  throw new Error('Input data required.')
+if (process.stdin.isTTY) {
+  wrap(process.argv[2]);
+} else {
+  let data = "";
+  process.stdin.on('data', function(chunk) {
+    data += chunk;
+  });
+  process.stdin.on('end', function() {
+    wrap(data);
+  });
 }
 
-let payload = '';
-try {
-  payload = JSON.parse(content);
-}
-catch (err) {
-  payload = content
-}
-const output = attr.wrap(payload);
+function wrap(content) {
+  if (!content) {
+    throw new Error('Input data required.')
+  }
 
-if (typeof output === 'object') {
-  process.stdout.write(JSON.stringify(output) + '\n');
-}
-else {
-  process.stdout.write(output + '\n');
+  let payload = '';
+  try {
+    payload = JSON.parse(content);
+  }
+  catch (err) {
+    payload = content
+  }
+  const output = attr.wrap(payload);
+
+  if (typeof output === 'object') {
+    process.stdout.write(JSON.stringify(output) + '\n');
+  }
+  else {
+    process.stdout.write(output + '\n');
+  }
 }