diff --git a/codemods/qs/index.js b/codemods/qs/index.js index 6cf8ebc..a9d3556 100644 --- a/codemods/qs/index.js +++ b/codemods/qs/index.js @@ -165,8 +165,20 @@ export default function (options) { const nameMatch = imp.getMatch('NAME'); if (nameMatch) { - importName = nameMatch.text(); - edits.push(nameMatch.replace('pq')); + const namespaceImport = nameMatch.find({ + rule: { + kind: 'identifier', + inside: { + kind: 'namespace_import', + }, + }, + }); + if (namespaceImport) { + importName = namespaceImport.text(); + } else { + importName = nameMatch.text(); + } + edits.push(nameMatch.replace('* as pq')); } edits.push(source.replace(`${quoteType}picoquery${quoteType}`)); diff --git a/test/fixtures/qs/basic/after.js b/test/fixtures/qs/basic/after.js index 8832aff..e2b88d4 100644 --- a/test/fixtures/qs/basic/after.js +++ b/test/fixtures/qs/basic/after.js @@ -1,4 +1,4 @@ -import pq from 'picoquery'; +import * as pq from 'picoquery'; const obj = {foo: 'bar'}; const query = 'foo=bar'; diff --git a/test/fixtures/qs/basic/result.js b/test/fixtures/qs/basic/result.js index 8832aff..e2b88d4 100644 --- a/test/fixtures/qs/basic/result.js +++ b/test/fixtures/qs/basic/result.js @@ -1,4 +1,4 @@ -import pq from 'picoquery'; +import * as pq from 'picoquery'; const obj = {foo: 'bar'}; const query = 'foo=bar';