feat: add partial support for onOf propType (#218)

README.md

@@ -16,6 +16,7 @@ Create typescript definitions files (d.ts) from react components.
 * ES6 and ES7 class syntax
 * Most PropTypes
   * any, array, bool, func, number, object, string, node, element, oneOfType, arrayOf, symbol
+* Partial support for oneOf PropType
 * required PropTypes
 * instanceOf PropTypes (when using API and giving a resolve function)
 * jsdoc

src/types.ts

@@ -55,6 +55,10 @@ export function get(astq: astqts.ASTQ, propertyAst: any, propTypesName: string|u
     case 'arrayOf':
       const typeDecl = get(astq, typeAst.arguments[0], propTypesName);
       return getTypeDeclaration(dom.create.array(typeDecl.type), !required);
+    case 'oneOf':
+      // FIXME: This should better be a real enum
+      const enumEntries = getEnumValues(typeAst.arguments[0].elements);
+      return getTypeDeclaration(dom.create.union(enumEntries as dom.Type[]), !required);
   }
 
   return {
@@ -95,3 +99,13 @@ function getComplexTypeName(astq: astqts.ASTQ, propertyAst: any,
   }
   return [required, undefined, typeAst];
 }
+
+function getEnumValues(oneOfTypes: any[]): any[] {
+  return oneOfTypes.map((element: any) => {
+    // FIXME: This are not named references!
+    if (element.type === 'StringLiteral') {
+      return dom.create.namedTypeReference(`'${element.value}'`);
+    }
+    return dom.create.namedTypeReference(element.value);
+  });
+}

tests/es6-class.d.ts

@@ -15,6 +15,7 @@ declare module 'component' {
         optionalNode?: React.ReactNode;
         optionalElement?: React.ReactElement<any>;
         optionalMessage?: typeof Message;
+        optionalEnum?: 'News' | 'Photos' | 1 | 2;
         optionalUnion?: string | number;
         optionalArrayOf?: number[];
         requiredFunc: (...args: any[])=>any;

tests/es6-class.jsx

@@ -23,7 +23,7 @@ Component.propTypes = {
 	optionalNode: React.PropTypes.node,
 	optionalElement: React.PropTypes.element,
 	optionalMessage: React.PropTypes.instanceOf(Message),
-	//optionalEnum: React.PropTypes.oneOf(['News', 'Photos']),
+	optionalEnum: React.PropTypes.oneOf(['News', 'Photos', 1, 2]),
 	optionalUnion: React.PropTypes.oneOfType([
 		React.PropTypes.string,
 		React.PropTypes.number