More accurate import of escape

btd · Apr 11, 2016 · d66825f · d66825f
1 parent 8dc16fb
commit d66825f
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/src/index.js b/src/index.js
@@ -23,14 +23,13 @@ export default function( options = {} ) {
 
       const tpl = template( code, templateOptions );
 
-      const source = tpl.source;
-
-      const intro = /__e = _.escape/.test(source) ?
-        `
-        import escape from 'lodash-es/escape';
-        var _ = { escape: escape };
-        `:
-        '';
+      let hasEscape = false;
+      const source = tpl.source.replace(/__e = _.escape/, function() {
+        hasEscape = true;
+        return '__e = escape';
+      });
+
+      const intro = hasEscape ? "import escape from 'lodash-es/escape'" : "";
 
       return `
       ${intro}