diff --git a/src/Plugin.js b/src/Plugin.js index 8afe921..77282ab 100644 --- a/src/Plugin.js +++ b/src/Plugin.js @@ -2,8 +2,11 @@ import { join } from 'path'; import { addSideEffect, addDefault, addNamed } from '@babel/helper-module-imports'; function transCamel(_str, symbol) { - const str = _str[0].toLowerCase() + _str.substr(1); - return str.replace(/([A-Z])/g, $1 => `${symbol}${$1.toLowerCase()}`); + // e.g. QRCode + // First match: QR + // Second match: Code + const cells = _str.match(/([A-Z]+(?=[A-Z]|$))|([A-Z]?[^A-Z]+)/g) || []; + return cells.map(c => c.toLowerCase()).join(symbol); } function winPath(path) { diff --git a/test/fixtures/qr-code/actual.js b/test/fixtures/qr-code/actual.js new file mode 100644 index 0000000..62c1d3b --- /dev/null +++ b/test/fixtures/qr-code/actual.js @@ -0,0 +1,8 @@ +import { QRCode } from 'antd'; + +switch(QRCode){ + case QRCode: + console.log('foo'); + default: + console.log('bar') +} diff --git a/test/fixtures/qr-code/expected.js b/test/fixtures/qr-code/expected.js new file mode 100644 index 0000000..1f747ba --- /dev/null +++ b/test/fixtures/qr-code/expected.js @@ -0,0 +1,7 @@ +import _QRCode from "antd/lib/qr-code"; +switch (_QRCode) { + case _QRCode: + console.log('foo'); + default: + console.log('bar'); +} \ No newline at end of file diff --git a/test/fixtures/sample-ui/actual.js b/test/fixtures/sample-ui/actual.js new file mode 100644 index 0000000..a87f203 --- /dev/null +++ b/test/fixtures/sample-ui/actual.js @@ -0,0 +1,8 @@ +import { SampleUI } from 'antd'; + +switch(SampleUI){ + case SampleUI: + console.log('foo'); + default: + console.log('bar') +} diff --git a/test/fixtures/sample-ui/expected.js b/test/fixtures/sample-ui/expected.js new file mode 100644 index 0000000..67b0e3f --- /dev/null +++ b/test/fixtures/sample-ui/expected.js @@ -0,0 +1,7 @@ +import _SampleUI from "antd/lib/sample-ui"; +switch (_SampleUI) { + case _SampleUI: + console.log('foo'); + default: + console.log('bar'); +} \ No newline at end of file