getElementType.js

/**
 * @flow
 */

import type { JSXOpeningElement } from 'ast-types-flow';
import hasOwn from 'hasown';
import includes from 'array-includes';
import { elementType, getProp, getLiteralPropValue } from 'jsx-ast-utils';

import type { ESLintContext } from '../../flow/eslint';

const getElementType = (context: ESLintContext): ((node: JSXOpeningElement) => string) => {
  const { settings } = context;
  const polymorphicPropName = settings['jsx-a11y']?.polymorphicPropName;
  const polymorphicAllowList = settings['jsx-a11y']?.polymorphicAllowList;

  const componentMap = settings['jsx-a11y']?.components;

  return (node: JSXOpeningElement): string => {
    const polymorphicProp = polymorphicPropName ? getLiteralPropValue(getProp(node.attributes, polymorphicPropName)) : undefined;

    let rawType = elementType(node);
    if (
      polymorphicProp
      && (!polymorphicAllowList || includes(polymorphicAllowList, rawType))
    ) {
      rawType = polymorphicProp;
    }

    if (!componentMap) {
      return rawType;
    }

    return hasOwn(componentMap, rawType) ? componentMap[rawType] : rawType;
  };
};

export default getElementType;