fix(parser): treat ComponentType as elementType

merceyz · Jul 23, 2019 · 53f1e21 · 53f1e21
1 parent d0b0a82
commit 53f1e21
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/src/parser.ts b/src/parser.ts
@@ -256,16 +256,18 @@ export function parseFromProgram(
       declaration &&
       ts.isPropertySignature(declaration) &&
       declaration.type &&
-      ts.isTypeReferenceNode(declaration.type) &&
-      declaration.type.typeName.getText() === 'React.ElementType'
+      ts.isTypeReferenceNode(declaration.type)
     ) {
-      return t.propTypeNode(
-        symbol.getName(),
-        getDocumentation(symbol),
-        declaration.questionToken
-          ? t.unionNode([t.undefinedNode(), t.elementNode('elementType')])
-          : t.elementNode('elementType'),
-      );
+      const name = declaration.type.typeName.getText();
+      if (name === 'React.ElementType' || name === 'React.ComponentType') {
+        return t.propTypeNode(
+          symbol.getName(),
+          getDocumentation(symbol),
+          declaration.questionToken
+            ? t.unionNode([t.undefinedNode(), t.elementNode('elementType')])
+            : t.elementNode('elementType'),
+        );
+      }
     }
 
     const type = declaration