babel-plugin-typescript-to-proptypes/src/addToClass.ts

Generate React PropTypes from TypeScript prop interfaces.

import { types as t } from '@babel/core';
import { convertToPropTypes } from './convertBabelToPropTypes';
import { extractGenericTypeNames } from './extractGenericTypeNames';
import { createPropTypesObject, mergePropTypes } from './propTypes';
import { ConvertState } from './types';

function findStaticProperty(
	node: t.ClassDeclaration,
	name: string,
): t.ClassMethod | t.ClassProperty | undefined {
	return node.body.body.find(
		(property) =>
			t.isClassProperty(property, { static: true }) &&
			t.isIdentifier(property.key, { name }) &&
			(t.isObjectExpression(property.value) || t.isCallExpression(property.value)),
	) as t.ClassMethod | t.ClassProperty;
}

export function addToClass(node: t.ClassDeclaration, state: ConvertState) {
	if (!node.superTypeParameters || node.superTypeParameters.params.length <= 0) {
		return;
	}

	const defaultProps = findStaticProperty(node, 'defaultProps');
	const defaultPropsKeyList: string[] = [];

	if (defaultProps && t.isClassProperty(defaultProps) && t.isObjectExpression(defaultProps.value)) {
		defaultProps.value.properties.forEach((prop) => {
			if (t.isProperty(prop) && t.isIdentifier(prop.key)) {
				defaultPropsKeyList.push(prop.key.name);
			}
		});
	}

	const typeNames = extractGenericTypeNames(node.superTypeParameters.params[0]);
	const propTypesList = convertToPropTypes(
		state.componentTypes,
		typeNames,
		state,
		defaultPropsKeyList,
	);

	if (typeNames.length === 0 || propTypesList.length === 0) {
		return;
	}

	const propTypes = findStaticProperty(node, 'propTypes');

	if (propTypes) {
		if (t.isClassProperty(propTypes)) {
			propTypes.value = mergePropTypes(propTypes.value, propTypesList, state);
		}
	} else {
		const staticProperty = t.classProperty(
			t.identifier('propTypes'),
			createPropTypesObject(propTypesList, state),
		);

		staticProperty.static = true;

		node.body.body.unshift(staticProperty);
	}
}