Add comment option.

index.js

@@ -11,7 +11,7 @@ function createBaseAST() {
   return ast;
 }
 
-function createStubFunctionASTNode(functionName) {
+function createStubFunctionASTNode(functionName, leadingComments) {
   var node = {
     type: 'FunctionDeclaration',
     id: {
@@ -27,35 +27,45 @@ function createStubFunctionASTNode(functionName) {
     generator: false,
     expression: false
   };
+  if (leadingComments) {
+    node.leadingComments = leadingComments;
+  }
   return node;
 }
 
-function _generateStubs(data) {
-  var ast = esprima.parse(data);
+function _generateStubs(data, options) {
+  var ast = esprima.parseScript(data, { attachComment: options.comment });
   var stubs = [];
   estraverse.traverse(ast, {
     leave: function (node) {
-      if (node.type === 'AssignmentExpression'
-        && node.operator === '='
-        && node.left.type === 'MemberExpression'
-        && node.left.object.type === 'Identifier'
-        && node.left.object.name === 'global') {
-        var functionName = node.left.property.name;
-        stubs.push(createStubFunctionASTNode(functionName));
+      if (node.type === 'ExpressionStatement'
+        && isGlobalAssignmentExpression(node.expression)) {
+        var functionName = node.expression.left.property.name;
+        stubs.push(createStubFunctionASTNode(functionName, node.leadingComments));
       }
     }
   });
 
   return stubs;
 }
 
-function generateStubs(source) {
+function isGlobalAssignmentExpression(node) {
+  return node.type === 'AssignmentExpression'
+    && node.operator === '='
+    && node.left.type === 'MemberExpression'
+    && node.left.object.type === 'Identifier'
+    && node.left.object.name === 'global'
+}
+
+function generateStubs(source, options) {
+  options = options || {comment: false};
+  var comment = !!options.comment;
   var baseAST = createBaseAST();
-  var stubs = _generateStubs(source);
+  var stubs = _generateStubs(source, options);
   stubs.forEach(function (stub) {
     baseAST.body.push(stub);
   });
-  return escodegen.generate(baseAST);
+  return escodegen.generate(baseAST, { comment: comment });
 }
 
 module.exports = generateStubs;

test/fixtures/expected.js

@@ -1,2 +1,12 @@
+/**
+ * Leading Block Comment for foo.
+ */
+// leading line comment for foo
 function foo() {
+}
+/**
+ * Leading Block Comment for boo.
+ */
+// leading line comment for boo
+function boo() {
 }

test/fixtures/source.js

@@ -1,2 +1,12 @@
+/**
+ * Leading Block Comment for foo.
+ */
+// leading line comment for foo
 global.foo = function () {
 };
+/**
+ * Leading Block Comment for boo.
+ */
+// leading line comment for boo
+global.boo = function () {
+};

test/index.test.js

@@ -6,7 +6,7 @@ var gasEntryGenerator = require('../');
 test('gas-entry-generator', function(t) {
   var source = fs.readFileSync(__dirname + '/fixtures/source.js', {encoding: 'utf8'});
   var expected = fs.readFileSync(__dirname + '/fixtures/expected.js', {encoding: 'utf8'});
-  var entryFunctions = gasEntryGenerator(source);
+  var entryFunctions = gasEntryGenerator(source, {comment: true});
   t.equal(entryFunctions.toString(), expected.toString(), 'actual output will match expected');
   t.end();
 });