Encode unions

src/Language/GraphQL/AST/Encoder.hs

@@ -128,6 +128,13 @@ typeDefinition formatter = \case
         <> optempty (directives formatter) directives'
         <> eitherFormat formatter " " ""
         <> bracesList formatter (fieldDefinition nextFormatter) fields'
+    Full.UnionTypeDefinition description' name' directives' members'
+        -> optempty (description formatter) description'
+        <> "union "
+        <> Lazy.Text.fromStrict name'
+        <> optempty (directives formatter) directives'
+        <> eitherFormat formatter " " ""
+        <> unionMemberTypes formatter members'
     _typeDefinition' -> "" -- TODO: Types missing.
   where
     nextFormatter = incrementIndent formatter
@@ -140,6 +147,19 @@ implementsInterfaces (Full.ImplementsInterfaces interfaces)
         $ Text.intercalate " & "
         $ toList interfaces
 
+unionMemberTypes :: Foldable t => Formatter -> Full.UnionMemberTypes t -> Lazy.Text
+unionMemberTypes formatter (Full.UnionMemberTypes memberTypes)
+    | null memberTypes = mempty
+    | Minified <- formatter = Lazy.Text.fromStrict
+        $ Text.append "= "
+        $ Text.intercalate " | "
+        $ toList memberTypes
+    | Pretty _ <- formatter
+        = Lazy.Text.append "="
+        $ Lazy.Text.concat
+        $ (("\n" <> indentSymbol <> "| ") <>) . Lazy.Text.fromStrict
+        <$> toList memberTypes
+
 description :: Formatter -> Full.Description -> Lazy.Text.Text
 description _formatter (Full.Description Nothing) = ""
 description formatter (Full.Description (Just description')) =

tests/Language/GraphQL/AST/EncoderSpec.hs

@@ -217,3 +217,15 @@ spec = do
                 |]
                 actual = typeSystemDefinition pretty definition'
              in actual `shouldBe` expected
+
+        it "encodes an enum definition" $
+            let definition' = Full.TypeDefinition
+                    $ Full.UnionTypeDefinition mempty "SearchResult" mempty
+                    $ Full.UnionMemberTypes ["Photo", "Person"]
+                expected = [gql|
+                  union SearchResult =
+                    | Photo
+                    | Person
+                |]
+                actual = typeSystemDefinition pretty definition'
+             in actual `shouldBe` expected