fix usage of codeframe

babel · Jun 1, 2017 · f1cfbe0 · f1cfbe0
1 parent 523fae4
commit f1cfbe0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/packages/babel-core/src/transformation/file/index.js b/packages/babel-core/src/transformation/file/index.js
@@ -432,7 +432,13 @@ export default class File extends Store {
 
       const loc = err.loc;
       if (loc) {
-        err.codeFrame = codeFrame(code, loc.line, loc.column + 1, this.opts);
+        const location = {
+          start: {
+            line: loc.line,
+            column: loc.column + 1
+          },
+        };
+        err.codeFrame = codeFrame(code, location, this.opts);
         message += "\n" + err.codeFrame;
       }
 

diff --git a/packages/babel-traverse/src/path/replacement.js b/packages/babel-traverse/src/path/replacement.js
@@ -74,8 +74,14 @@ export function replaceWithSourceString(replacement) {
   } catch (err) {
     const loc = err.loc;
     if (loc) {
+      const location = {
+        start: {
+          line: loc.line,
+          column: loc.column + 1
+        },
+      };
       err.message += " - make sure this is an expression.";
-      err.message += "\n" + codeFrame(replacement, loc.line, loc.column + 1);
+      err.message += "\n" + codeFrame(replacement, location);
     }
     throw err;
   }