Fix bug

lib/utils/get-element-type.js

@@ -16,15 +16,16 @@ function getElementType(context, node, lazyElementCheck = false) {
 
  // check if the node contains a polymorphic prop
  const polymorphicPropName = settings?.github?.polymorphicPropName ?? 'as'
- const rawElement = getLiteralPropValue(getProp(node.attributes, polymorphicPropName)) ?? elementType(node)
 
- // if a component configuration does not exists, return the raw element
- if (!settings?.github?.components?.[rawElement]) return rawElement
+ const rawElement = elementType(node)
 
- const defaultComponent = settings.github.components[rawElement]
-
- // check if the default component is also defined in the configuration
- return defaultComponent ? defaultComponent : defaultComponent
+ if (getProp(node.attributes, polymorphicPropName)) {
+ return getLiteralPropValue(getProp(node.attributes, polymorphicPropName)) ?? rawElement
+ } else if (settings?.github?.components?.[rawElement]) {
+ return settings.github.components[rawElement]
+ } else {
+ return rawElement
+ }
 }
 
 module.exports = {getElementType}

tests/utils/get-element-type.mjs

@@ -64,14 +64,15 @@ describe('getElementType', function () {
  expect(getElementType({}, node)).to.equal('Box')
  })
 
- it('returns raw type when polymorphic prop is set to component, and not the mapped value', function () {
- // <Box as={Link} />
+ it('returns raw type when polymorphic prop is set to component even when there is a mapped value', function () {
+ // <Box as={isNavigationOpen ? 'generic' : 'navigation'} />
  const setting = mockSetting({
  Box: 'div',
  })
 
- // eslint-disable-next-line no-undef
- const node = mockJSXOpeningElement('Box', [mockJSXAttribute('as', Link)])
+ const node = mockJSXOpeningElement('Box', [
+ mockJSXConditionalAttribute('as', 'isNavigationOpen', 'generic', 'navigation'),
+ ])
  expect(getElementType(setting, node)).to.equal('Box')
  })
 })