feature: convert-import-to-call: add loc

coderaiser · Jun 20, 2023 · 04f7587 · 04f7587
1 parent 4761331
commit 04f7587
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 3 deletions.
diff --git a/lib/convert-import-to-call.js b/lib/convert-import-to-call.js
@@ -3,12 +3,24 @@
 const {CallExpression, Import} = require('@babel/types');
 const setLiteral = require('./set-literal');
 
+const IMPORT_LENGTH = 'import'.length;
+
 module.exports = (path) => {
-    const {source} = path.node;
+    const {loc, source} = path.node;
+    const calleeNode = Import();
+
+    calleeNode.loc = {
+        start: loc.start,
+        end: {
+            line: loc.start.line,
+            column: loc.start.column + IMPORT_LENGTH,
+        },
+    };
 
     setLiteral(source);
 
-    const callNode = CallExpression(Import(), [source]);
+    const callNode = CallExpression(calleeNode, [source]);
+    callNode.loc = loc;
 
     path.replaceWith(callNode);
 };
diff --git a/test/fixture/import-expression.json b/test/fixture/import-expression.json
@@ -32,7 +32,17 @@
                 "expression": {
                     "type": "CallExpression",
                     "callee": {
-                        "type": "Import"
+                        "type": "Import",
+                        "loc": {
+                            "start": {
+                                "line": 1,
+                                "column": 0
+                            },
+                            "end": {
+                                "line": 1,
+                                "column": 6
+                            }
+                        }
                     },
                     "arguments": [
                         {
@@ -56,6 +66,16 @@
                             }
                         }
                     ],
+                    "loc": {
+                        "start": {
+                            "line": 1,
+                            "column": 0
+                        },
+                        "end": {
+                            "line": 1,
+                            "column": 11
+                        }
+                    },
                     "trailingComments": [],
                     "leadingComments": [],
                     "innerComments": []