Emit comments for literal unions (fix #475)

Author: Boris Cherny

src/generator.ts

@@ -285,7 +285,9 @@ function generateRawType(ast: AST, options: Options): string {
  * Generate a Union or Intersection
  */
 function generateSetOperation(ast: TIntersection | TUnion, options: Options): string {
-  const members = (ast as TUnion).params.map(_ => generateType(_, options))
+  const members = (ast as TUnion).params.map(_ =>
+    [_.comment ? generateComment(_.comment) : null, generateType(_, options)].filter(Boolean).join('\n')
+  )
   const separator = ast.type === 'UNION' ? '|' : '&'
   return members.length === 1 ? members[0] : '(' + members.join(' ' + separator + ' ') + ')'
 }

test/__snapshots__/test/test.ts.md

@@ -0,0 +1,6 @@
+export const input = {
+  oneOf: [
+    {const: 'a', description: 'First comment (a).'},
+    {const: 'b', description: 'Second comment (b).'}
+  ]
+}