support binary expression

mui · Jun 4, 2024 · 7533f08 · 7533f08
1 parent eec27cc
commit 7533f08
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 1 deletion.
diff --git a/packages/mui-codemod/src/v6.0.0/sx-prop/sx-v6.js b/packages/mui-codemod/src/v6.0.0/sx-prop/sx-v6.js
@@ -222,7 +222,10 @@ export default function sxV6(file, api, options) {
               (returnExpression.type === 'CallExpression' &&
                 getObjectKey(returnExpression.callee)?.name === 'theme') ||
               (returnExpression.type === 'MemberExpression' &&
-                getObjectKey(returnExpression)?.name === 'theme')
+                getObjectKey(returnExpression)?.name === 'theme') ||
+              (returnExpression.type === 'BinaryExpression' &&
+                (getObjectKey(returnExpression.left)?.name === 'theme' ||
+                  getObjectKey(returnExpression.right)?.name === 'theme'))
             ) {
               data.replaceValue?.(returnExpression);
               rootThemeCallback(data);

diff --git a/packages/mui-codemod/src/v6.0.0/sx-prop/test-cases/sx-value-callback.actual.js b/packages/mui-codemod/src/v6.0.0/sx-prop/test-cases/sx-value-callback.actual.js
@@ -87,3 +87,11 @@ function FacebookCircularProgress(props) {
     p: 4,
   }}
 ></Box>;
+
+<Backdrop
+  sx={{ color: '#fff', zIndex: (theme) => theme.zIndex.drawer + 1 }}
+  open={open}
+  onClick={handleClose}
+>
+  <CircularProgress color="inherit" />
+</Backdrop>;
diff --git a/packages/mui-codemod/src/v6.0.0/sx-prop/test-cases/sx-value-callback.expected.js b/packages/mui-codemod/src/v6.0.0/sx-prop/test-cases/sx-value-callback.expected.js
@@ -98,3 +98,14 @@ function FacebookCircularProgress(props) {
     p: 4
   })}
 ></Box>;
+
+<Backdrop
+  sx={theme => ({
+    color: '#fff',
+    zIndex: theme.zIndex.drawer + 1
+  })}
+  open={open}
+  onClick={handleClose}
+>
+  <CircularProgress color="inherit" />
+</Backdrop>;