@mui/codemod 6.0.0-alpha.5 → 6.0.0-alpha.7

package/node/deprecations/circular-progress-classes/circular-progress-classes.js

@@ -0,0 +1,77 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+var _postcssPlugin = require("./postcss-plugin");
+/**
+ * @param {import('jscodeshift').FileInfo} file
+ * @param {import('jscodeshift').API} api
+ */
+function transformer(file, api, options) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+  const printOptions = options.printOptions;
+  _postcssPlugin.classes.forEach(({
+    deprecatedClass,
+    replacementSelector
+  }) => {
+    const replacementSelectorPrefix = '&';
+    root.find(j.ImportDeclaration).filter(path => path.node.source.value.match(/^@mui\/material\/CircularProgress$/)).forEach(path => {
+      path.node.specifiers.forEach(specifier => {
+        if (specifier.type === 'ImportSpecifier' && specifier.imported.name === 'circularProgressClasses') {
+          const deprecatedAtomicClass = deprecatedClass.replace(`${deprecatedClass.split('-')[0]}-`, '');
+          root.find(j.MemberExpression, {
+            object: {
+              name: specifier.local.name
+            },
+            property: {
+              name: deprecatedAtomicClass
+            }
+          }).forEach(memberExpression => {
+            const parent = memberExpression.parentPath.parentPath.value;
+            if (parent.type === j.TemplateLiteral.name) {
+              const memberExpressionIndex = parent.expressions.findIndex(expression => expression === memberExpression.value);
+              const precedingTemplateElement = parent.quasis[memberExpressionIndex];
+              const atomicClasses = replacementSelector.replaceAll('MuiCircularProgress-', '').replaceAll(replacementSelectorPrefix, '').replaceAll(' > ', '').split('.').filter(Boolean);
+              if (precedingTemplateElement.value.raw.endsWith(deprecatedClass.startsWith(' ') ? `${replacementSelectorPrefix} .` : `${replacementSelectorPrefix}.`)) {
+                const atomicClassesArgs = [memberExpressionIndex, 1, ...atomicClasses.map(atomicClass => j.memberExpression(memberExpression.value.object, j.identifier(atomicClass)))];
+                parent.expressions.splice(...atomicClassesArgs);
+                if (replacementSelector.includes(' > ')) {
+                  const quasisArgs = [memberExpressionIndex, 1, j.templateElement({
+                    raw: precedingTemplateElement.value.raw.replace(' ', ''),
+                    cooked: precedingTemplateElement.value.cooked.replace(' ', '')
+                  }, false), j.templateElement({
+                    raw: ' > .',
+                    cooked: ' > .'
+                  }, false)];
+                  if (atomicClasses.length === 3) {
+                    quasisArgs.splice(3, 0, j.templateElement({
+                      raw: '.',
+                      cooked: '.'
+                    }, false));
+                  }
+                  parent.quasis.splice(...quasisArgs);
+                } else {
+                  parent.quasis.splice(memberExpressionIndex, 1, j.templateElement({
+                    raw: precedingTemplateElement.value.raw,
+                    cooked: precedingTemplateElement.value.cooked
+                  }, false), j.templateElement({
+                    raw: '.',
+                    cooked: '.'
+                  }, false));
+                }
+              }
+            }
+          });
+        }
+      });
+    });
+    const selectorRegex = new RegExp(`${replacementSelectorPrefix}${deprecatedClass}$`);
+    root.find(j.Literal, literal => typeof literal.value === 'string' && literal.value.match(selectorRegex)).forEach(path => {
+      path.replace(j.literal(path.value.value.replace(selectorRegex, `${replacementSelectorPrefix}${replacementSelector}`)));
+    });
+  });
+  return root.toSource(printOptions);
+}

package/node/deprecations/circular-progress-classes/index.js

@@ -0,0 +1,13 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+Object.defineProperty(exports, "default", {
+  enumerable: true,
+  get: function () {
+    return _circularProgressClasses.default;
+  }
+});
+var _circularProgressClasses = _interopRequireDefault(require("./circular-progress-classes"));

package/node/deprecations/circular-progress-classes/postcss-plugin.js

@@ -0,0 +1,33 @@
+"use strict";
+
+const classes = [{
+  deprecatedClass: ' .MuiCircularProgress-circleDeterminate',
+  replacementSelector: '.MuiCircularProgress-determinate > .MuiCircularProgress-circle'
+}, {
+  deprecatedClass: ' .MuiCircularProgress-circleIndeterminate',
+  replacementSelector: '.MuiCircularProgress-indeterminate > .MuiCircularProgress-circle'
+}];
+const plugin = () => {
+  return {
+    postcssPlugin: `Replace deprecated CircularProgress classes with new classes`,
+    Rule(rule) {
+      const {
+        selector
+      } = rule;
+      classes.forEach(({
+        deprecatedClass,
+        replacementSelector
+      }) => {
+        const selectorRegex = new RegExp(`${deprecatedClass}$`);
+        if (selector.match(selectorRegex)) {
+          rule.selector = selector.replace(selectorRegex, replacementSelector);
+        }
+      });
+    }
+  };
+};
+plugin.postcss = true;
+module.exports = {
+  plugin,
+  classes
+};

package/node/deprecations/circular-progress-classes/postcss.config.js

@@ -0,0 +1,8 @@
+"use strict";
+
+const {
+  plugin
+} = require('./postcss-plugin');
+module.exports = {
+  plugins: [plugin]
+};

package/node/deprecations/circular-progress-classes/test-cases/actual.js

@@ -0,0 +1,7 @@
+"use strict";
+
+var _CircularProgress = require("@mui/material/CircularProgress");
+'& .MuiCircularProgress-circleDeterminate';
+'& .MuiCircularProgress-circleIndeterminate';
+`& .${_CircularProgress.circularProgressClasses.circleDeterminate}`;
+`& .${_CircularProgress.circularProgressClasses.circleIndeterminate}`;

package/node/deprecations/circular-progress-classes/test-cases/expected.js

@@ -0,0 +1,7 @@
+"use strict";
+
+var _CircularProgress = require("@mui/material/CircularProgress");
+"&.MuiCircularProgress-determinate > .MuiCircularProgress-circle";
+"&.MuiCircularProgress-indeterminate > .MuiCircularProgress-circle";
+`&.${_CircularProgress.circularProgressClasses.determinate} > .${_CircularProgress.circularProgressClasses.circle}`;
+`&.${_CircularProgress.circularProgressClasses.indeterminate} > .${_CircularProgress.circularProgressClasses.circle}`;

package/node/deprecations/divider-props/divider-props.js

@@ -20,27 +20,30 @@ function transformer(file, api, options) {
     root,
     componentName: 'Divider'
   }, elementPath => {
-    elementPath.node.openingElement.attributes = elementPath.node.openingElement.attributes.filter(attr => {
-      if (attr.type === 'JSXAttribute' && attr.name.name === 'light') {
-        return false;
-      }
-      return true;
-    });
-    const sxIndex = elementPath.node.openingElement.attributes.findIndex(attr => attr.type === 'JSXAttribute' && attr.name.name === 'sx');
-    if (sxIndex === -1) {
-      (0, _appendAttribute.default)(j, {
-        target: elementPath.node,
-        attributeName: 'sx',
-        expression: j.objectExpression([j.objectProperty(j.identifier('opacity'), j.literal('0.6'))])
+    const hasLightProp = elementPath.node.openingElement.attributes.findIndex(attr => attr.type === 'JSXAttribute' && attr.name.name === 'light') !== -1;
+    if (hasLightProp) {
+      elementPath.node.openingElement.attributes = elementPath.node.openingElement.attributes.filter(attr => {
+        if (attr.type === 'JSXAttribute' && attr.name.name === 'light') {
+          return false;
+        }
+        return true;
       });
-    } else {
-      const opacityIndex = elementPath.node.openingElement.attributes[sxIndex].value.expression.properties.findIndex(key => key.key.name === 'opacity');
-      if (opacityIndex === -1) {
-        (0, _assignObject.default)(j, {
-          target: elementPath.node.openingElement.attributes[sxIndex],
-          key: 'opacity',
-          expression: j.literal('0.6')
+      const sxIndex = elementPath.node.openingElement.attributes.findIndex(attr => attr.type === 'JSXAttribute' && attr.name.name === 'sx');
+      if (sxIndex === -1) {
+        (0, _appendAttribute.default)(j, {
+          target: elementPath.node,
+          attributeName: 'sx',
+          expression: j.objectExpression([j.objectProperty(j.identifier('opacity'), j.literal('0.6'))])
         });
+      } else {
+        const opacityIndex = elementPath.node.openingElement.attributes[sxIndex].value.expression.properties.findIndex(key => key.key.name === 'opacity');
+        if (opacityIndex === -1) {
+          (0, _assignObject.default)(j, {
+            target: elementPath.node.openingElement.attributes[sxIndex],
+            key: 'opacity',
+            expression: j.literal('0.6')
+          });
+        }
       }
     }
   });
@@ -50,17 +53,20 @@ function transformer(file, api, options) {
     }
   }).forEach(path => {
     const defaultPropsObject = path.value.value.properties.find(key => key.key.name === 'defaultProps');
-    defaultPropsObject.value.properties = defaultPropsObject.value.properties.filter(prop => {
-      var _prop$key;
-      return !['light'].includes(prop == null || (_prop$key = prop.key) == null ? void 0 : _prop$key.name);
-    });
-    const sxIndex = defaultPropsObject.value.properties.findIndex(prop => prop.key.name === 'sx');
-    if (sxIndex === -1) {
-      defaultPropsObject.value.properties.push(j.objectProperty(j.identifier('sx'), j.objectExpression([j.objectProperty(j.identifier('opacity'), j.literal('0.6'))])));
-    } else {
-      const opacityIndex = defaultPropsObject.value.properties[sxIndex].value.properties.findIndex(key => key.key.name === 'opacity');
-      if (opacityIndex === -1) {
-        defaultPropsObject.value.properties[sxIndex].value.properties.push(j.objectProperty(j.identifier('opacity'), j.literal('0.6')));
+    const hasLightProp = defaultPropsObject.value.properties.findIndex(prop => prop.key.name === 'light') !== -1;
+    if (hasLightProp) {
+      defaultPropsObject.value.properties = defaultPropsObject.value.properties.filter(prop => {
+        var _prop$key;
+        return !['light'].includes(prop == null || (_prop$key = prop.key) == null ? void 0 : _prop$key.name);
+      });
+      const sxIndex = defaultPropsObject.value.properties.findIndex(prop => prop.key.name === 'sx');
+      if (sxIndex === -1) {
+        defaultPropsObject.value.properties.push(j.objectProperty(j.identifier('sx'), j.objectExpression([j.objectProperty(j.identifier('opacity'), j.literal('0.6'))])));
+      } else {
+        const opacityIndex = defaultPropsObject.value.properties[sxIndex].value.properties.findIndex(key => key.key.name === 'opacity');
+        if (opacityIndex === -1) {
+          defaultPropsObject.value.properties[sxIndex].value.properties.push(j.objectProperty(j.identifier('opacity'), j.literal('0.6')));
+        }
       }
     }
   });

package/node/deprecations/divider-props/test-cases/actual.js

@@ -39,4 +39,7 @@ var _jsxRuntime = require("react/jsx-runtime");
   sx: {
     bgcolor: 'black'
   }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_Divider.default, {
+  className: "test"
 });

package/node/deprecations/divider-props/test-cases/expected.js

@@ -50,4 +50,7 @@ var _jsxRuntime = require("react/jsx-runtime");
     bgcolor: 'black',
     opacity: "0.6"
   }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_Divider.default, {
+  className: "test"
 });

package/node/deprecations/divider-props/test-cases/theme.actual.js

@@ -44,4 +44,11 @@ fn({
       }
     }
   }
+});
+fn({
+  MuiDivider: {
+    defaultProps: {
+      className: 'my-class'
+    }
+  }
 });

package/node/deprecations/divider-props/test-cases/theme.expected.js

@@ -49,4 +49,11 @@ fn({
       }
     }
   }
+});
+fn({
+  MuiDivider: {
+    defaultProps: {
+      className: 'my-class'
+    }
+  }
 });

package/node/deprecations/speed-dial-props/index.js

@@ -0,0 +1,13 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+Object.defineProperty(exports, "default", {
+  enumerable: true,
+  get: function () {
+    return _speedDialProps.default;
+  }
+});
+var _speedDialProps = _interopRequireDefault(require("./speed-dial-props"));

package/node/deprecations/speed-dial-props/speed-dial-props.js

@@ -0,0 +1,31 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = transformer;
+var _movePropIntoSlots = _interopRequireDefault(require("../utils/movePropIntoSlots"));
+var _movePropIntoSlotProps = _interopRequireDefault(require("../utils/movePropIntoSlotProps"));
+/**
+ * @param {import('jscodeshift').FileInfo} file
+ * @param {import('jscodeshift').API} api
+ */
+function transformer(file, api, options) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+  const printOptions = options.printOptions;
+  (0, _movePropIntoSlots.default)(j, {
+    root,
+    componentName: 'SpeedDial',
+    propName: 'TransitionComponent',
+    slotName: 'transition'
+  });
+  (0, _movePropIntoSlotProps.default)(j, {
+    root,
+    componentName: 'SpeedDial',
+    propName: 'TransitionProps',
+    slotName: 'transition'
+  });
+  return root.toSource(printOptions);
+}

package/node/deprecations/speed-dial-props/test-cases/actual.js

@@ -0,0 +1,40 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+var _SpeedDial = _interopRequireDefault(require("@mui/material/SpeedDial"));
+var _material = require("@mui/material");
+var _jsxRuntime = require("react/jsx-runtime");
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_SpeedDial.default, {
+  TransitionComponent: CustomTransition,
+  TransitionProps: CustomTransitionProps
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_material.SpeedDial, {
+  TransitionComponent: CustomTransition,
+  TransitionProps: CustomTransitionProps
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_SpeedDial.default, {
+  TransitionComponent: CustomTransition,
+  TransitionProps: CustomTransitionProps,
+  slots: {
+    root: 'div'
+  }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_material.SpeedDial, {
+  TransitionComponent: CustomTransition,
+  TransitionProps: CustomTransitionProps,
+  slots: (0, _extends2.default)({}, outerSlots)
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_SpeedDial.default, {
+  TransitionComponent: ComponentTransition,
+  TransitionProps: CustomTransitionProps,
+  slots: {
+    root: 'div',
+    transition: SlotTransition
+  }
+});
+// should skip non MUI components
+/*#__PURE__*/(0, _jsxRuntime.jsx)(NonMuiSpeedDial, {
+  TransitionComponent: CustomTransition,
+  TransitionProps: CustomTransitionProps
+});

package/node/deprecations/speed-dial-props/test-cases/expected.js

@@ -0,0 +1,54 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+var _SpeedDial = _interopRequireDefault(require("@mui/material/SpeedDial"));
+var _material = require("@mui/material");
+var _jsxRuntime = require("react/jsx-runtime");
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_SpeedDial.default, {
+  slots: {
+    transition: CustomTransition
+  },
+  slotProps: {
+    transition: CustomTransitionProps
+  }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_material.SpeedDial, {
+  slots: {
+    transition: CustomTransition
+  },
+  slotProps: {
+    transition: CustomTransitionProps
+  }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_SpeedDial.default, {
+  slots: {
+    root: 'div',
+    transition: CustomTransition
+  },
+  slotProps: {
+    transition: CustomTransitionProps
+  }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_material.SpeedDial, {
+  slots: (0, _extends2.default)({}, outerSlots, {
+    transition: CustomTransition
+  }),
+  slotProps: {
+    transition: CustomTransitionProps
+  }
+});
+/*#__PURE__*/(0, _jsxRuntime.jsx)(_SpeedDial.default, {
+  slots: {
+    root: 'div',
+    transition: SlotTransition
+  },
+  slotProps: {
+    transition: CustomTransitionProps
+  }
+});
+// should skip non MUI components
+/*#__PURE__*/(0, _jsxRuntime.jsx)(NonMuiSpeedDial, {
+  TransitionComponent: CustomTransition,
+  TransitionProps: CustomTransitionProps
+});

package/node/deprecations/speed-dial-props/test-cases/theme.actual.js

@@ -0,0 +1,33 @@
+"use strict";
+
+fn({
+  MuiSpeedDial: {
+    defaultProps: {
+      TransitionComponent: CustomTransition,
+      TransitionProps: CustomTransitionProps
+    }
+  }
+});
+fn({
+  MuiSpeedDial: {
+    defaultProps: {
+      TransitionComponent: CustomTransition,
+      TransitionProps: CustomTransitionProps,
+      slots: {
+        root: 'div'
+      }
+    }
+  }
+});
+fn({
+  MuiSpeedDial: {
+    defaultProps: {
+      TransitionComponent: ComponentTransition,
+      TransitionProps: CustomTransitionProps,
+      slots: {
+        root: 'div',
+        transition: SlotTransition
+      }
+    }
+  }
+});

package/node/deprecations/speed-dial-props/test-cases/theme.expected.js

@@ -0,0 +1,40 @@
+"use strict";
+
+fn({
+  MuiSpeedDial: {
+    defaultProps: {
+      slots: {
+        transition: CustomTransition
+      },
+      slotProps: {
+        transition: CustomTransitionProps
+      }
+    }
+  }
+});
+fn({
+  MuiSpeedDial: {
+    defaultProps: {
+      slots: {
+        root: 'div',
+        transition: CustomTransition
+      },
+      slotProps: {
+        transition: CustomTransitionProps
+      }
+    }
+  }
+});
+fn({
+  MuiSpeedDial: {
+    defaultProps: {
+      slots: {
+        root: 'div',
+        transition: SlotTransition
+      },
+      slotProps: {
+        transition: CustomTransitionProps
+      }
+    }
+  }
+});

package/node/util/getReturnExpression.js

@@ -0,0 +1,20 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = getReturnExpression;
+/**
+ * @param {import('ast-types').namedTypes.ArrowFunctionExpression | import('ast-types').namedTypes.FunctionExpression} node
+ */
+function getReturnExpression(node) {
+  let body = node.body;
+  if (body === 'BlockStatement') {
+    body = body.body;
+  }
+  if (Array.isArray(body)) {
+    var _body$find;
+    return (_body$find = body.find(statement => statement.type === 'ReturnStatement')) == null ? void 0 : _body$find.argument;
+  }
+  return body;
+}