jscodeshift codemod to change <React.Fragment> and <Fragment> to <>

fragment-component-to-jsx-fragment.js

module.exports = function (file, api, options) {
	const j = api.jscodeshift; // alias the jscodeshift API
	const root = j(file.source); // parse JS code into an AST

	let mutation = false;

	const isJSXIdentifier = (node, name) => j.JSXIdentifier.check(node) && node.name === name;

	// Remove the opening and closing element names if they are Fragment or React.Fragment
	const update = node => {
		const {openingElement, closingElement} = node.value;
		const {name} = openingElement;
		const isFragment = isJSXIdentifier(name, 'Fragment');
		const isReactFragment = j.JSXMemberExpression.check(name) && isJSXIdentifier(name.object, 'React') && isJSXIdentifier(name.property, 'Fragment');
		if ((isFragment || isReactFragment) && closingElement && !node.value.openingElement.attributes.length) {
			openingElement.name = null;
			closingElement.name = null;
			mutation = true;
		}
	};

	// find and update all React.Fragment elements
	root.findJSXElements()
		.forEach(update);

	// print
	if (mutation) {
		return root.toSource(options.printOptions);
	}

	return null;
};

module.exports.parser = 'babylon';