Emit comments for literal unions (fix #475)

bcherny · Aug 7, 2022 · 91cd642 · 91cd642
1 parent 34de194
commit 91cd642
Show file tree

Hide file tree

Showing 4 changed files with 177,073 additions and 31,690 deletions.
diff --git a/src/generator.ts b/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 + ' ') + ')'
 }

diff --git a/test/__snapshots__/test/test.ts.md b/test/__snapshots__/test/test.ts.md
diff --git a/test/__snapshots__/test/test.ts.snap b/test/__snapshots__/test/test.ts.snap
diff --git a/test/e2e/enum.4.ts b/test/e2e/enum.4.ts
@@ -0,0 +1,6 @@
+export const input = {
+  oneOf: [
+    {const: 'a', description: 'First comment (a).'},
+    {const: 'b', description: 'Second comment (b).'}
+  ]
+}