assignInFile can write to different file

Glogster · Feb 24, 2016 · 85de6a2 · 85de6a2
1 parent ee37662
commit 85de6a2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -37,9 +37,12 @@ function assign(template, assigns, keyMaker, valueMaker) {
 }
 
 function assignInFile(file, assigns, keyMaker, valueMaker) {
-  return fs.readFileAsync(file, encoding)
+  const input = Array.isArray(file) ? file[0] : file;
+  const output = Array.isArray(file) ? file[1] : file;
+
+  return fs.readFileAsync(input, encoding)
     .then((content) => assign(content, assigns, keyMaker, valueMaker))
-    .then((content) => fs.writeFileAsync(file, content, encoding));
+    .then((content) => fs.writeFileAsync(output, content, encoding));
 }
 
 function md5FileHash(file) {