Make fields that are enum types required

awakesecurity · Aug 11, 2020 · 52a8b88 · 52a8b88
1 parent 1eb688b
commit 52a8b88
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/Proto3/Suite/DotProto/Generate.hs b/src/Proto3/Suite/DotProto/Generate.hs
@@ -620,6 +620,9 @@ dotProtoMessageD ctxt parentIdent messageIdent messageParts = do
           DotProtoMessageOneOf ident _ -> do
             name <- dpIdentUnqualName ident
             pure [(name, False)]
+          DotProtoMessageDefinition (DotProtoEnum _ ident _) -> do
+            name <- dpIdentUnqualName ident
+            pure [(name, True)]
           _                            -> do pure []
 
     fieldss <- traverse getEither messageParts