@mui/x-codemod 8.0.0-beta.0 → 8.0.0-beta.2

package/README.md

@@ -29,7 +29,7 @@ Options:
   --jscodeshift Pass options directly to jscodeshift                  [array]
 
 Examples:
-  npx @mui/x-codemod@next v7.0.0/preset-safe src
+  npx @mui/x-codemod@next v8.0.0/preset-safe src
   npx @mui/x-codemod@next v6.0.0/component-rename-prop src --
   --component=DataGrid --from=prop --to=newProp
 ```
@@ -326,6 +326,10 @@ The list includes these transformers
 - [`remove-stabilized-v8-experimentalFeatures`](#remove-stabilized-v8-experimentalFeatures)
 - [`remove-props`](#remove-props)
 - [`rename-props`](#rename-props)
+- [`rename-imports`](#rename-imports)
+- [`reform-row-selection-model`](#reform-row-selection-model)
+- [`rename-package`](#rename-package)
+- [`add-showToolbar-prop`](#add-showToolbar-prop)
 
 #### `remove-stabilized-v8-experimentalFeatures`
 
@@ -353,11 +357,13 @@ The list includes these props:
 
 - `indeterminateCheckboxAction`
 - `rowPositionsDebounceMs`
+- `resetPageOnSortFilter`
 
 ```diff
  <DataGrid
 -  indeterminateCheckboxAction="deselect"
 -  rowPositionsDebounceMs={100}
+-  resetPageOnSortFilter
  />
 ```
 
@@ -374,11 +380,26 @@ Rename props to the new ones.
 The list includes these props:
 
 - `unstable_rowSpanning` to `rowSpanning`
+- `unstable_dataSource` to `dataSource`
+- `unstable_dataSourceCache` to `dataSourceCache`
+- `unstable_lazyLoading` to `lazyLoading`
+- `unstable_lazyLoadingRequestThrottleMs` to `lazyLoadingRequestThrottleMs`
+- `unstable_onDataSourceError` to `onDataSourceError`
 
 ```diff
  <DataGrid
 -  unstable_rowSpanning
+-  unstable_dataSource={dataSource}
+-  unstable_dataSourceCache={dataSourceCache}
+-  unstable_lazyLoading
+-  unstable_lazyLoadingRequestThrottleMs={100}
+-  unstable_onDataSourceError={() => {}}
 +  rowSpanning
++  dataSource={dataSource}
++  dataSourceCache={dataSourceCache}
++  lazyLoading
++  lazyLoadingRequestThrottleMs={100}
++  onDataSourceError={() => {}}
  />
 ```
 
@@ -388,6 +409,81 @@ The list includes these props:
 npx @mui/x-codemod@next v8.0.0/data-grid/rename-props <path>
 ```
 
+#### `rename-imports`
+
+This codemod renames the imports of the Data Grid components. The list includes these imports:
+
+- `selectedGridRowsSelector` to `gridRowSelectionIdsSelector`
+- `selectedGridRowsCountSelector` to `gridRowSelectionCountSelector`
+
+```diff
+-import { selectedGridRowsSelector, selectedGridRowsCountSelector } from '@mui/x-data-grid';
++import { gridRowSelectionIdsSelector, gridRowSelectionCountSelector } from '@mui/x-data-grid';
+```
+
+<!-- #default-branch-switch -->
+
+```bash
+npx @mui/x-codemod@next v8.0.0/data-grid/rename-imports <path>
+```
+
+#### `reform-row-selection-model`
+
+Reforms the controlled `rowSelectionModel` prop value to the new one.
+
+```diff
+-const [rowSelectionModel, setRowSelectionModel] = React.useState([1, 2]);
++const [rowSelectionModel, setRowSelectionModel] = React.useState({
++  type: 'include',
++  ids: new Set([1, 2]),
++});
+
+ <DataGrid
+  rowSelectionModel={rowSelectionModel}
+  onRowSelectionModelChange={setRowSelectionModel}
+ />
+```
+
+<!-- #default-branch-switch -->
+
+```bash
+npx @mui/x-codemod@next v8.0.0/data-grid/reform-row-selection-model <path>
+```
+
+#### `rename-package`
+
+Reorganizes the imports moved from `@mui/x-data-grid-pro` and `@mui/x-data-grid-premium`.
+
+```diff
+-import { LicenseInfo } from '@mui/x-data-grid-pro';
++import { LicenseInfo } from '@mui/x-license';
+```
+
+<!-- #default-branch-switch -->
+
+```bash
+npx @mui/x-codemod@next v8.0.0/data-grid/rename-package <path>
+```
+
+#### `add-showToolbar-prop`
+
+Adds the `showToolbar` prop to the Data Grid components that are using `slots.toolbar` prop.
+
+```diff
+ <DataGridPremium
+  slots={{
+    toolbar: GridToolbar,
+  }}
++ showToolbar
+ />
+```
+
+<!-- #default-branch-switch -->
+
+```bash
+npx @mui/x-codemod@next v8.0.0/data-grid/add-showToolbar-prop <path>
+```
+
 ### Pickers codemods
 
 #### `preset-safe` for Pickers v8.0.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mui/x-codemod",
-  "version": "8.0.0-beta.0",
+  "version": "8.0.0-beta.2",
   "bin": "./codemod.js",
   "private": false,
   "author": "MUI Team",
@@ -30,7 +30,7 @@
     "@babel/traverse": "^7.26.10",
     "jscodeshift": "17.1.2",
     "yargs": "^17.7.2",
-    "@mui/x-internals": "8.0.0-beta.0"
+    "@mui/x-internals": "8.0.0-beta.2"
   },
   "devDependencies": {
     "@types/jscodeshift": "^0.12.0",

package/v8.0.0/data-grid/add-showToolbar-prop/index.js

@@ -0,0 +1,60 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+function transformer(file, api, options) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+
+  // List of DataGrid components
+  const dataGridComponents = new Set(['DataGrid', 'DataGridPro', 'DataGridPremium']);
+  // List of allowed import sources
+  const dataGridSources = new Set(['@mui/x-data-grid', '@mui/x-data-grid-pro', '@mui/x-data-grid-premium']);
+
+  // Find relevant DataGrid imports
+  const importedDataGrids = new Set();
+  root.find(j.ImportDeclaration).forEach(path => {
+    if (dataGridSources.has(path.node.source.value)) {
+      path.node.specifiers?.forEach(specifier => {
+        if (specifier.imported && dataGridComponents.has(specifier.imported.name)) {
+          const localName = specifier.local?.name;
+          if (localName) {
+            importedDataGrids.add(localName);
+          }
+        }
+      });
+    }
+  });
+  if (importedDataGrids.size === 0) {
+    return file.source;
+  }
+  root.find(j.JSXOpeningElement).forEach(path => {
+    if (!importedDataGrids.has(path.node.name.name)) {
+      return;
+    }
+    let hasSlotsToolbar = false;
+    let hasShowToolbar = false;
+    path.node.attributes?.forEach(attr => {
+      if (attr.type === 'JSXAttribute') {
+        if (attr.name.name === 'slots') {
+          if (attr.value && attr.value.type === 'JSXExpressionContainer' && attr.value.expression.type === 'ObjectExpression') {
+            hasSlotsToolbar = attr.value.expression.properties.some(prop => prop.key.name === 'toolbar');
+          }
+        }
+        if (attr.name.name === 'showToolbar') {
+          hasShowToolbar = true;
+        }
+      }
+    });
+    if (hasSlotsToolbar && !hasShowToolbar) {
+      path.node.attributes?.push(j.jsxAttribute(j.jsxIdentifier('showToolbar')));
+    }
+  });
+  const printOptions = options.printOptions || {
+    quote: 'single',
+    trailingComma: true
+  };
+  return root.toSource(printOptions);
+}

package/v8.0.0/data-grid/preset-safe/index.js

@@ -8,9 +8,17 @@ exports.default = transformer;
 var _removeStabilizedExperimentalFeatures = _interopRequireDefault(require("../remove-stabilized-experimentalFeatures"));
 var _removeProps = _interopRequireDefault(require("../remove-props"));
 var _renameProps = _interopRequireDefault(require("../rename-props"));
+var _reformRowSelectionModel = _interopRequireDefault(require("../reform-row-selection-model"));
+var _renameImports = _interopRequireDefault(require("../rename-imports"));
+var _renamePackage = _interopRequireDefault(require("../rename-package"));
+var _addShowToolbarProp = _interopRequireDefault(require("../add-showToolbar-prop"));
 function transformer(file, api, options) {
   file.source = (0, _removeStabilizedExperimentalFeatures.default)(file, api, options);
   file.source = (0, _renameProps.default)(file, api, options);
   file.source = (0, _removeProps.default)(file, api, options);
+  file.source = (0, _reformRowSelectionModel.default)(file, api, options);
+  file.source = (0, _renameImports.default)(file, api, options);
+  file.source = (0, _renamePackage.default)(file, api, options);
+  file.source = (0, _addShowToolbarProp.default)(file, api, options);
   return file.source;
 }

package/v8.0.0/data-grid/reform-row-selection-model/index.js

@@ -0,0 +1,47 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+// @ts-nocheck - TODO: fix this
+
+const componentNames = ['DataGrid', 'DataGridPro', 'DataGridPremium'];
+const attrName = 'rowSelectionModel';
+function transformer(file, api, options) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+
+  // Step 1: Collect variable names used in <DataGrid rowSelectionModel={...} />
+  const usedInDataGrid = new Set();
+  root.find(j.JSXOpeningElement).filter(path => j.JSXIdentifier.check(path.node.name) && componentNames.some(name => name === path.node.name.name)).forEach(path => {
+    path.node.attributes?.forEach(attr => {
+      if (j.JSXAttribute.check(attr) && j.JSXIdentifier.check(attr.name) && attr.name.name === attrName && attr.value && j.JSXExpressionContainer.check(attr.value) && j.Identifier.check(attr.value.expression)) {
+        usedInDataGrid.add(attr.value.expression.name);
+      }
+    });
+  });
+  if (usedInDataGrid.size === 0) {
+    return file.source; // No relevant transformations needed
+  }
+
+  // Step 2: Convert only relevant `useState` or `useMemo` variables
+  root.find(j.VariableDeclarator).filter(path => j.ArrayPattern.check(path.node.id) && path.node.id.elements.length === 2 && j.Identifier.check(path.node.id.elements[0]) && usedInDataGrid.has(path.node.id.elements[0].name) &&
+  // Only modify if used in `componentNames`
+  path.node.init && j.CallExpression.check(path.node.init) && (j.MemberExpression.check(path.node.init.callee) && j.Identifier.check(path.node.init.callee.object) && ['React', 'useState', 'useMemo'].includes(path.node.init.callee.object.name) || ['useState', 'useMemo'].includes(path.node.init.callee.name)) &&
+  // Handle direct calls
+  path.node.init.arguments.length > 0 && (j.ArrayExpression.check(path.node.init.arguments[0]) || j.ArrowFunctionExpression.check(path.node.init.arguments[0]) && j.BlockStatement.check(path.node.init.arguments[0].body) === false && j.ArrayExpression.check(path.node.init.arguments[0].body))).forEach(path => {
+    const arrayExpression = j.ArrayExpression.check(path.node.init?.arguments[0]) ? path.node.init?.arguments[0] : path.node.init?.arguments[0]?.body;
+    const newObject = j.objectExpression([j.property('init', j.identifier('type'), j.literal('include')), j.property('init', j.identifier('ids'), j.newExpression(j.identifier('Set'), [arrayExpression]))]);
+    if (j.ArrayExpression.check(path.node.init.arguments[0])) {
+      path.node.init.arguments[0] = newObject;
+    } else {
+      path.node.init.arguments[0].body = newObject;
+    }
+  });
+  const printOptions = options?.printOptions || {
+    quote: 'single',
+    trailingComma: true
+  };
+  return root.toSource(printOptions);
+}

package/v8.0.0/data-grid/remove-props/index.js

@@ -7,7 +7,7 @@ Object.defineProperty(exports, "__esModule", {
 exports.default = transformer;
 var _removeProps = _interopRequireDefault(require("../../../util/removeProps"));
 const componentNames = ['DataGrid', 'DataGridPro', 'DataGridPremium'];
-const props = ['indeterminateCheckboxAction', 'rowPositionsDebounceMs'];
+const props = ['indeterminateCheckboxAction', 'rowPositionsDebounceMs', 'resetPageOnSortFilter'];
 function transformer(file, api, options) {
   const j = api.jscodeshift;
   const root = j(file.source);

package/v8.0.0/data-grid/rename-imports/index.js

@@ -0,0 +1,28 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+var _renameImports = require("../../../util/renameImports");
+function transformer(file, api, options) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+  const printOptions = options.printOptions || {
+    quote: 'single',
+    trailingComma: true
+  };
+  (0, _renameImports.renameImports)({
+    j,
+    root,
+    packageNames: ['@mui/x-data-grid', '@mui/x-data-grid-pro', '@mui/x-data-grid-premium'],
+    imports: [{
+      oldEndpoint: 'DataGrid',
+      importsMapping: {
+        selectedGridRowsSelector: 'gridRowSelectionIdsSelector',
+        selectedGridRowsCountSelector: 'gridRowSelectionCountSelector'
+      }
+    }]
+  });
+  return root.toSource(printOptions);
+}

package/v8.0.0/data-grid/rename-package/index.js

@@ -0,0 +1,34 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+// TODO: Make it generic and move to utils
+function transformer(file, api, options) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+  root.find(j.ImportDeclaration).filter(path => ['@mui/x-data-grid-pro', '@mui/x-data-grid-premium'].includes(path.node.source.value)).forEach(path => {
+    const specifiers = path.node.specifiers;
+    const filteredSpecifiers = specifiers.filter(spec => !(j.ImportSpecifier.check(spec) && spec.imported.name === 'LicenseInfo'));
+
+    // If `LicenseInfo` was found and removed
+    if (filteredSpecifiers.length !== specifiers.length) {
+      const licenseImport = j.importDeclaration([j.importSpecifier(j.identifier('LicenseInfo'))], j.stringLiteral('@mui/x-license'));
+      if (filteredSpecifiers.length > 0) {
+        // Keep other imports but remove `LicenseInfo`
+        path.node.specifiers = filteredSpecifiers;
+        // Insert new import right after the modified import
+        j(path).insertAfter(licenseImport);
+      } else {
+        // Remove import entirely and insert new one at the same position
+        j(path).replaceWith(licenseImport);
+      }
+    }
+  });
+  const printOptions = options?.printOptions || {
+    quote: 'single',
+    trailingComma: true
+  };
+  return root.toSource(printOptions);
+}

package/v8.0.0/data-grid/rename-package/rename-package.js

@@ -0,0 +1,33 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+var _path = _interopRequireDefault(require("path"));
+var _chai = require("chai");
+var _jscodeshift = _interopRequireDefault(require("jscodeshift"));
+var _ = _interopRequireDefault(require("."));
+var _readFile = _interopRequireDefault(require("../../../util/readFile"));
+function read(fileName) {
+  return (0, _readFile.default)(_path.default.join(__dirname, fileName));
+}
+describe('v8.0.0/data-grid', () => {
+  describe('rename package', () => {
+    it('transforms props as needed', () => {
+      const actual = (0, _.default)({
+        source: read('./actual.spec.js')
+      }, {
+        jscodeshift: _jscodeshift.default
+      });
+      const expected = read('./expected.spec.js');
+      (0, _chai.expect)(actual).to.equal(expected, 'The transformed version should be correct');
+    });
+    it('should be idempotent', () => {
+      const actual = (0, _.default)({
+        source: read('./expected.spec.js')
+      }, {
+        jscodeshift: _jscodeshift.default
+      });
+      const expected = read('./expected.spec.js');
+      (0, _chai.expect)(actual).to.equal(expected, 'The transformed version should be correct');
+    });
+  });
+});

package/v8.0.0/data-grid/rename-props/index.js

@@ -8,7 +8,12 @@ exports.default = transformer;
 var _renameProps = _interopRequireDefault(require("../../../util/renameProps"));
 const componentNames = ['DataGrid', 'DataGridPro', 'DataGridPremium'];
 const props = {
-  unstable_rowSpanning: 'rowSpanning'
+  unstable_rowSpanning: 'rowSpanning',
+  unstable_dataSource: 'dataSource',
+  unstable_dataSourceCache: 'dataSourceCache',
+  unstable_lazyLoading: 'lazyLoading',
+  unstable_lazyLoadingRequestThrottleMs: 'lazyLoadingRequestThrottleMs',
+  unstable_onDataSourceError: 'onDataSourceError'
 };
 function transformer(file, api, options) {
   const j = api.jscodeshift;