Merge pull request #10 from tikotzky/fix-issue-9

Fix stripping of statements on the first line and last line
YahooArchive · Jan 9, 2016 · d5dc2a1 · d5dc2a1
2 parents 5020e42 + 00b0875
commit d5dc2a1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lib/index.js b/lib/index.js
@@ -14,7 +14,7 @@ function StripFnLoader(source) {
 
     var toStrip = query.strip.join('|');
 
-    var regexPattern = new RegExp('\\n[ \\t]*(' + toStrip + ')\\([^\\);]+\\)[ \\t]*[;\\n]', 'g');
+    var regexPattern = new RegExp('(?:^|\\n)[ \\t]*(' + toStrip + ')\\([^\\);]+\\)[ \\t]*(?:$|[;\\n])', 'g');
 
     var transformed = source.replace(regexPattern, '\n');
 

diff --git a/tests/fixtures/app/index.js b/tests/fixtures/app/index.js
@@ -1,3 +1,4 @@
+console.log('a console.log on the first line should get stripped');
 /**
  * Copyright 2015, Yahoo! Inc.
  * Copyrights licensed under the New BSD License. See the accompanying LICENSE file for terms.
@@ -10,3 +11,4 @@ var makeFoo = function (bar, baz) {
     // This code would remain
     return new Foo(bar, baz);
 };
+console.log('a console.log on the last line without a semicolon should get stripped')