@mui/codemod 5.14.4 → 5.14.5

package/README.md

@@ -266,12 +266,20 @@ The associated breaking change was done in [#34997](https://github.com/mui/mater
 
 #### `date-pickers-moved-to-x`
 
-Rename the imports of date and time pickers from `@mui/lab` to `@mui/x-date-pickers` and `@mui/x-date-pickers-pro`.
+Rename the imports of Date and Time Pickers from `@mui/lab` to `@mui/x-date-pickers` and `@mui/x-date-pickers-pro`.
 
 ```bash
 npx @mui/codemod v5.0.0/date-pickers-moved-to-x <path>
 ```
 
+#### `tree-view-moved-to-x`
+
+Rename the imports of Tree View from `@mui/lab` to `@mui/x-tree-view`.
+
+```bash
+npx @mui/codemod v5.0.0/tree-view-moved-to-x <path>
+```
+
 #### 🚀 `preset-safe`
 
 A combination of all important transformers for migrating v4 to v5. ⚠️ This codemod should be run only once.

package/node/v5.0.0/tree-view-moved-to-x.js

@@ -0,0 +1,86 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+const _exports = {
+  TreeView: {
+    default: 'TreeView',
+    named: ['treeViewClasses', 'TreeViewClasses', 'TreeViewClassKey', 'getTreeViewUtilityClass', 'TreeViewPropsBase', 'TreeViewProps', 'SingleSelectTreeViewProps', 'MultiSelectTreeViewProps']
+  },
+  TreeItem: {
+    default: 'TreeItem',
+    named: ['useTreeItem', 'treeItemClasses', 'TreeItemClasses', 'TreeItemClassKey', 'getTreeItemUtilityClass', 'TreeItemProps', 'TreeItemContentProps']
+  }
+};
+const buildLookup = () => {
+  return Object.fromEntries(Object.entries(_exports).flatMap(([entryPoint, entryPointData]) => [entryPointData.default, ...entryPointData.named].map(exportName => [exportName, {
+    entryPoint
+  }])));
+};
+
+/**
+ * @param {import('jscodeshift').FileInfo} file
+ * @param {import('jscodeshift').API} api
+ */
+function transformer(fileInfo, api, options) {
+  const j = api.jscodeshift;
+  const printOptions = options.printOptions || {
+    quote: 'single'
+  };
+  const lookup = buildLookup();
+  const root = j(fileInfo.source);
+  root.find(j.ImportDeclaration).forEach(path => {
+    const importSource = path.node.source.value;
+    const subPackageImportMatch = importSource.match(/@mui\/lab\/(.*)/);
+    if (subPackageImportMatch !== null) {
+      const subModule = subPackageImportMatch[1];
+      if (subModule.startsWith('internal')) {
+        console.warn('Imports from `@mui/lab/internal` are not supported');
+        return;
+      }
+      if (_exports[subModule]) {
+        /**
+         * @type {import('jscodeshift').ASTPath}
+         */
+        const sourcePath = path.get('source');
+        const targetPackage = '@mui/x-tree-view';
+        const targetImportPath = `${targetPackage}/${subModule}`;
+        sourcePath.replace(j.stringLiteral(targetImportPath));
+        const importDeclaration = path.value;
+        importDeclaration.specifiers = importDeclaration.specifiers.map(specifier => {
+          if (specifier.type === 'ImportDefaultSpecifier') {
+            const localName = specifier.local.name;
+            return j.importSpecifier(j.identifier(subModule), j.identifier(localName));
+          }
+          return specifier;
+        });
+      }
+    } else if (importSource === '@mui/lab') {
+      // Sieve import specifiers into /core and /lab
+      const xImportSpecifiers = [];
+      const labImportSpecifiers = [];
+      path.node.specifiers.forEach(specifier => {
+        if (specifier.type === 'ImportSpecifier') {
+          const lookupValue = lookup[specifier.imported.name];
+          if (lookupValue) {
+            xImportSpecifiers.push(specifier);
+          } else {
+            labImportSpecifiers.push(specifier);
+          }
+        } else {
+          // `import Lab from '@material-ui/lab'`
+          // `import * as Lab from '@material-ui/lab'`
+          // These imports would require scope analysis.
+          console.warn(`Can't handle ${specifier.type}`);
+        }
+      });
+      if (xImportSpecifiers.length > 0) {
+        const targetPackage = '@mui/x-tree-view';
+        path.replace(j.importDeclaration(xImportSpecifiers, j.stringLiteral(targetPackage)), j.importDeclaration(labImportSpecifiers, j.stringLiteral('@mui/lab')));
+      }
+    }
+  }).toSource(printOptions);
+  return root.toSource(printOptions);
+}

package/node/v5.0.0/tree-view-moved-to-x.test/actual-root.js

@@ -0,0 +1,3 @@
+"use strict";
+
+var _lab = require("@mui/lab");

package/node/v5.0.0/tree-view-moved-to-x.test/actual-sub-module.js

@@ -0,0 +1,8 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+var _TreeView = _interopRequireWildcard(require("@mui/lab/TreeView"));
+var _TreeItem = _interopRequireWildcard(require("@mui/lab/TreeItem"));
+var _Timeline = _interopRequireDefault(require("@mui/lab/Timeline"));
+function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }

package/node/v5.0.0/tree-view-moved-to-x.test/expected-root.js

@@ -0,0 +1,4 @@
+"use strict";
+
+var _xTreeView = require("@mui/x-tree-view");
+var _lab = require("@mui/lab");

package/node/v5.0.0/tree-view-moved-to-x.test/expected-sub-module.js

@@ -0,0 +1,6 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+var _TreeView = require("@mui/x-tree-view/TreeView");
+var _TreeItem = require("@mui/x-tree-view/TreeItem");
+var _Timeline = _interopRequireDefault(require("@mui/lab/Timeline"));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mui/codemod",
-  "version": "5.14.4",
+  "version": "5.14.5",
   "bin": "./codemod.js",
   "private": false,
   "author": "MUI Team",