Merge pull request #325 from researchgate/master

Add support for Flow literal string/number types

Author: tmcw

lib/flow_doctrine.js

@@ -11,6 +11,11 @@ var oneToOne = {
   }
 };
 
+var literalTypes = {
+  'StringLiteralTypeAnnotation': 'StringLiteral',
+  'NumberLiteralTypeAnnotation': 'NumberLiteral'
+};
+
 function flowDoctrine(type) {
 
   if (type.type in namedTypes) {
@@ -56,6 +61,13 @@ function flowDoctrine(type) {
       name: type.id.name
     };
   }
+
+  if (type.type in literalTypes) {
+    return {
+      type: literalTypes[type.type],
+      name: type.value
+    };
+  }
 }
 
 module.exports = flowDoctrine;

test/lib/flow_doctrine.js

@@ -112,6 +112,22 @@ test('flowDoctrine', function (t) {
       name: 'undefined'
     }, 'undefined');
 
+  t.deepEqual(flowDoctrine(toComment(
+      "/** add */function add(a: \"value\") { }"
+    ).context.ast.value.params[0].typeAnnotation.typeAnnotation),
+    {
+      type: 'StringLiteral',
+      name: 'value'
+    }, 'StringLiteral');
+
+  t.deepEqual(flowDoctrine(toComment(
+      "/** add */function add(a: 1) { }"
+    ).context.ast.value.params[0].typeAnnotation.typeAnnotation),
+    {
+      type: 'NumberLiteral',
+      name: '1'
+    }, 'NumberLiteral');
+
   t.end();
 });
 /* eslint-enable */