blob: b34f63585e76ea30d670318952de44911ddd6556 [file] [log] [blame]
import propName from './propName';
const DEFAULT_OPTIONS = {
ignoreCase: true,
};
/**
* Returns the JSXAttribute itself or undefined, indicating the prop
* is not present on the JSXOpeningElement.
*
*/
export default function getProp(props = [], prop = '', options = DEFAULT_OPTIONS) {
function getName(name) { return options.ignoreCase ? name.toUpperCase() : name; }
const propToFind = getName(prop);
function isPropToFind(property) {
return property.type === 'Property'
&& property.key.type === 'Identifier'
&& propToFind === getName(property.key.name);
}
const foundAttribute = props.find((attribute) => {
// If the props contain a spread prop, try to find the property in the object expression.
if (attribute.type === 'JSXSpreadAttribute') {
return attribute.argument.type === 'ObjectExpression'
&& propToFind !== getName('key') // https://github.com/reactjs/rfcs/pull/107
&& attribute.argument.properties.some(isPropToFind);
}
return propToFind === getName(propName(attribute));
});
if (foundAttribute && foundAttribute.type === 'JSXSpreadAttribute') {
return propertyToJSXAttribute(foundAttribute.argument.properties.find(isPropToFind));
}
return foundAttribute;
}
function propertyToJSXAttribute(node) {
const { key, value } = node;
return {
type: 'JSXAttribute',
name: { type: 'JSXIdentifier', name: key.name, ...getBaseProps(key) },
value: value.type === 'Literal'
? value
: { type: 'JSXExpressionContainer', expression: value, ...getBaseProps(value) },
...getBaseProps(node),
};
}
function getBaseProps({
start,
end,
loc,
range,
}) {
return {
loc: getBaseLocation(loc),
...(start !== undefined ? { start } : {}),
...(end !== undefined ? { end } : {}),
...(range !== undefined ? { range } : {}),
};
}
function getBaseLocation({
start,
end,
source,
filename,
}) {
return {
start,
end,
...(source !== undefined ? { source } : {}),
...(filename !== undefined ? { filename } : {}),
};
}