@mui/codemod/util/propsToObject.js

Codemod scripts for Material UI.

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = propsToObject;
function propsToObject({
  j,
  root,
  componentName,
  aliasName,
  propName,
  props
}) {
  function buildObject(node, value) {
    const shorthand = node.value.expression && node.value.expression.name === node.name.name;
    const property = j.objectProperty(j.identifier(node.name.name), node.value.expression ? node.value.expression : node.value);
    property.shorthand = shorthand;
    value.push(property);
    return value;
  }
  const result = aliasName ? root.find(j.JSXElement, {
    openingElement: {
      name: {
        property: {
          name: componentName
        }
      }
    }
  }) : root.findJSXElements(componentName);
  return result.forEach(path => {
    if (!aliasName || aliasName && path.node.openingElement.name.object.name === aliasName) {
      let propValue = [];
      const attributes = path.node.openingElement.attributes;
      attributes.forEach((node, index) => {
        // Only transform whitelisted props
        if (node.type === 'JSXAttribute' && props.includes(node.name.name)) {
          propValue = buildObject(node, propValue);
          delete attributes[index];
        }
      });
      if (propValue.length > 0) {
        const propNameAttr = attributes.find(attr => attr?.name?.name === propName);
        if (propNameAttr) {
          (propNameAttr.value.expression?.properties || []).push(...j.objectExpression(propValue).properties);
        } else {
          attributes.push(j.jsxAttribute(j.jsxIdentifier(propName), j.jsxExpressionContainer(j.objectExpression(propValue))));
        }
      }
    }
  });
}