support both v4 and v5-alpha, beta

mui · Oct 5, 2021 · 927f135 · 927f135
1 parent eb72833
commit 927f135
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/packages/mui-codemod/src/v5.0.0/optimal-imports.js b/packages/mui-codemod/src/v5.0.0/optimal-imports.js
@@ -6,7 +6,11 @@ import getJSExports from '../util/getJSExports';
 if (process.env.NODE_ENV === 'test') {
   const resolve = require.resolve;
   require.resolve = (source) =>
-    resolve(source.replace(/^@material-ui\/core\/es/, '../../../mui-material/src'));
+    resolve(
+      source
+        .replace(/^@material-ui\/core\/es/, '../../../mui-material/src')
+        .replace(/^@material-ui\/core\/modern/, '../../../mui-material/src'),
+    );
 }
 
 export default function transformer(fileInfo, api, options) {
@@ -47,11 +51,18 @@ export default function transformer(fileInfo, api, options) {
     }
     const targetImportPath = `${targetModule}/${subpath}`;
 
-    const whitelist = getJSExports(
-      require.resolve(`${importModule}/es/${subpath}`, {
+    let loader;
+    try {
+      loader = require.resolve(`${importModule}/modern/${subpath}`, {
         paths: [dirname(fileInfo.path)],
-      }),
-    );
+      });
+    } catch (error) {
+      loader = require.resolve(`${importModule}/es/${subpath}`, {
+        paths: [dirname(fileInfo.path)],
+      });
+    }
+
+    const whitelist = getJSExports(loader);
 
     path.node.specifiers.forEach((specifier, index) => {
       if (!path.node.specifiers.length) {