Cater for mb shortcut

mui · Jul 22, 2024 · e0cc8c7 · e0cc8c7
1 parent 50bb4a7
commit e0cc8c7
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 2 deletions.
diff --git a/packages/mui-codemod/src/deprecations/typography-props/test-cases/actual.js b/packages/mui-codemod/src/deprecations/typography-props/test-cases/actual.js
@@ -15,5 +15,7 @@ import { Typography as MyTypography } from '@mui/material';
 <MyTypography paragraph={paragraph} className="my-class" />;
 <Typography paragraph sx={{ marginBottom: "32px" }} />;
 <MyTypography paragraph sx={{ marginBottom: "32px" }} />;
+<Typography paragraph sx={{ mb: "32px" }} />;
+<MyTypography paragraph sx={{ mb: "32px" }} />;
 <Typography paragraph sx={{ color: "black" }} />;
 <MyTypography paragraph sx={{ color: "black" }} />;
diff --git a/packages/mui-codemod/src/deprecations/typography-props/test-cases/expected.js b/packages/mui-codemod/src/deprecations/typography-props/test-cases/expected.js
@@ -27,6 +27,8 @@ import { Typography as MyTypography } from '@mui/material';
 }} />;
 <Typography sx={{ marginBottom: "32px" }} />;
 <MyTypography sx={{ marginBottom: "32px" }} />;
+<Typography sx={{ mb: "32px" }} />;
+<MyTypography sx={{ mb: "32px" }} />;
 <Typography
   sx={{
     color: "black",

diff --git a/packages/mui-codemod/src/deprecations/typography-props/test-cases/theme.actual.js b/packages/mui-codemod/src/deprecations/typography-props/test-cases/theme.actual.js
@@ -59,6 +59,17 @@ fn({
     },
   },
 });
+fn({
+  MuiTypography: {
+    defaultProps: {
+      paragraph,
+      className: "my-class",
+      sx: {
+        mb: "32px",
+      },
+    },
+  },
+});
 fn({
   MuiTypography: {
     defaultProps: {

diff --git a/packages/mui-codemod/src/deprecations/typography-props/test-cases/theme.expected.js b/packages/mui-codemod/src/deprecations/typography-props/test-cases/theme.expected.js
@@ -67,6 +67,17 @@ fn({
     defaultProps: {
       className: "my-class",
 
+      sx: {
+        mb: "32px",
+      }
+    },
+  },
+});
+fn({
+  MuiTypography: {
+    defaultProps: {
+      className: "my-class",
+
       sx: {
         color: "black",
         marginBottom: "16px"

diff --git a/packages/mui-codemod/src/deprecations/typography-props/typography-props.js b/packages/mui-codemod/src/deprecations/typography-props/typography-props.js
@@ -49,7 +49,9 @@ export default function transformer(file, api, options) {
     } else {
       const hasMarginBottom = elementPath.node.openingElement.attributes[
         sxIndex
-      ].value.expression.properties.some((key) => key.key.name === 'marginBottom');
+      ].value.expression.properties.some(
+        (key) => key.key.name === 'marginBottom' || key.key.name === 'mb',
+      );
 
       if (!hasMarginBottom) {
         assignObject(j, {
@@ -95,7 +97,7 @@ export default function transformer(file, api, options) {
       );
     } else {
       const hasMarginBottom = defaultPropsObject.value.properties[sxIndex].value.properties.some(
-        (key) => key.key.name === 'marginBottom',
+        (key) => key.key.name === 'marginBottom' || key.key.name === 'mb',
       );
 
       if (!hasMarginBottom) {