feat: add switch and switchCase treatment (#520)

* feat: add switch and switchCase treatment

* feat: add processing to entry array

* test: adjustment unit test

src/Plugin.js

@@ -266,6 +266,16 @@ export default class Plugin {
     this.buildExpressionHandler(node, ['callee', 'arguments'], path, state);
   }
 
+  SwitchStatement(path, state) {
+    const { node } = path;
+    this.buildExpressionHandler(node, ['discriminant'], path, state);
+  }
+
+  SwitchCase(path, state) {
+    const { node } = path;
+    this.buildExpressionHandler(node, ['test'], path, state);
+  }
+
   ClassDeclaration(path, state) {
     const { node } = path;
     this.buildExpressionHandler(node, ['superClass'], path, state);

src/index.js

@@ -99,6 +99,8 @@ export default function ({ types }) {
     'BinaryExpression',
     'NewExpression',
     'ClassDeclaration',
+    'SwitchStatement',
+    'SwitchCase',
   ];
 
   const ret = {

test/fixtures/switch/actual.js

@@ -0,0 +1,8 @@
+import { Select } from 'antd';
+
+  switch(Select){
+    case Select:
+      console.log('foo');
+    default:
+      console.log('bar')
+  }

test/fixtures/switch/expected.js

@@ -0,0 +1,13 @@
+"use strict";
+
+var _select = _interopRequireDefault(require("antd/lib/select"));
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+switch (_select.default) {
+  case _select.default:
+    console.log('foo');
+
+  default:
+    console.log('bar');
+}