@mui/codemod
Version:
Codemod scripts for Material UI.
83 lines (82 loc) • 3.25 kB
JavaScript
;
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = transformer;
var _appendAttribute = _interopRequireDefault(require("../../util/appendAttribute"));
var _assignObject = _interopRequireDefault(require("../../util/assignObject"));
var _findComponentJSX = _interopRequireDefault(require("../../util/findComponentJSX"));
var _findComponentDefaultProps = _interopRequireDefault(require("../../util/findComponentDefaultProps"));
/**
* @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, _findComponentJSX.default)(j, {
root,
componentName: 'Divider',
packageName: options.packageName
}, elementPath => {
const lightProp = elementPath.node.openingElement.attributes.find(attr => attr.type === 'JSXAttribute' && attr.name.name === 'light');
if (!lightProp) {
return;
}
elementPath.node.openingElement.attributes = elementPath.node.openingElement.attributes.filter(attr => {
if (attr.type === 'JSXAttribute' && attr.name.name === 'light') {
return false;
}
return true;
});
const isLightPropTruthy = lightProp.value?.expression.value !== false;
if (!isLightPropTruthy) {
return;
}
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')
});
}
}
});
const defaultPropsPathCollection = (0, _findComponentDefaultProps.default)(j, {
root,
packageName: options.packageName,
componentName: 'Divider'
});
defaultPropsPathCollection.find(j.ObjectProperty, {
key: {
name: 'light'
}
}).forEach(path => {
const {
properties: defaultPropsProperties
} = path.parent.value;
if (path.value?.value.value === false) {
path.prune();
return;
}
const existingSx = defaultPropsProperties.find(prop => prop.key.name === 'sx');
if (!existingSx) {
defaultPropsProperties.push(j.property('init', j.identifier('sx'), j.objectExpression([j.objectProperty(j.identifier('opacity'), j.literal('0.6'))])));
} else if (!existingSx.value.properties.find(prop => prop.key.name === 'opacity')) {
existingSx.value.properties.push(j.property('init', j.identifier('opacity'), j.literal('0.6')));
}
path.prune();
});
return root.toSource(printOptions);
}