Encode unions

2 files changed, 33 insertions, 1 deletions

diff --git a/src/Language/GraphQL/AST/Encoder.hs b/src/Language/GraphQL/AST/Encoder.hs
index bd6dab8..ab17929 100644
--- a/src/Language/GraphQL/AST/Encoder.hs
+++ b/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
@@ -136,10 +143,23 @@ implementsInterfaces :: Foldable t => Full.ImplementsInterfaces t -> Lazy.Text
 implementsInterfaces (Full.ImplementsInterfaces interfaces)
     | null interfaces = mempty
     | otherwise = Lazy.Text.fromStrict
-        $ Text.append "implements"
+        $ Text.append "implements "
         $ 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')) =
diff --git a/tests/Language/GraphQL/AST/EncoderSpec.hs b/tests/Language/GraphQL/AST/EncoderSpec.hs
index 862fa34..5116c0e 100644
--- a/tests/Language/GraphQL/AST/EncoderSpec.hs
+++ b/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