Merge pull request #8 from calebthebrewer/immediate-code

Catch code that immediately follows a comment, fixes #9

Author: jonschlinkert

lib/code.js

@@ -6,7 +6,7 @@ var utils = require('./utils');
 function Code(str, comment) {
   str = utils.restore(str);
   var start = comment.loc.end.pos;
-  var lineno = comment.loc.end.line + 1;
+  var lineno = comment.loc.end.line;
   var ctx = {};
 
   var lines = str.split('\n').slice(lineno);

test/test.js

@@ -151,6 +151,18 @@ describe('code', function() {
     assert.equal(actual[0].code.value, 'var foo = "bar";');
   });
 
+  it('should get a code line that immediately follows the comment', function() {
+    var str = '/**\n * this is\n *\n * a comment\n*/\nvar foo = "bar";\n// var one = two';
+    var actual = extract(str);
+    assert.equal(actual[0].code.value, 'var foo = "bar";');
+  });
+
+  it('should not get a comment following a comment', function() {
+    var str = '/**\n * this is\n *\n * a comment\n*/\n// var one = two';
+    var actual = extract(str);
+    assert.equal(actual[0].code.value, '');
+  });
+
   it('should get the code starting and ending indices', function() {
     var str = '/**\n * this is\n *\n * a comment\n*/\n\n\nvar foo = "bar";\n';
     var actual = extract(str);