Generate files first, then write them out

Author: Tyler Waters

lib/generator.js

@@ -126,9 +126,10 @@ const generateOperationFile = (config, operation, operation_name) => new Promise
  * @param   {Object}  config Configuration options
  * @returns {Promise}
  */
-const generateOperationFiles = config => new Promise((resolve, reject) => {
+
+const generateOperationFiles = config => {
   const files = {};
-  _.each(config.data.openapi.paths, (path, path_name) => {
+  _.forEach(config.data.openapi.paths, (path, path_name) => {
     const operation_name = path.endpointName;
     if (files[operation_name] === undefined) {
       files[operation_name] = [];
@@ -142,11 +143,12 @@ const generateOperationFiles = config => new Promise((resolve, reject) => {
       subresource: (path_name.substring(operation_name.length+1) || '/').replace(/}/g, '').replace(/{/g, ':')
     });
 
-    Promise.all(
-      _.map(files, (operation, operation_name) => generateOperationFile(config, operation, operation_name))
-    ).then(resolve).catch(reject);
   });
-});
+
+  return Promise.all(
+    _.map(files, (operation, operation_name) => generateOperationFile(config, operation, operation_name))
+  );
+};
 
 /**
  * Generates the directory structure.