Encode unions

This commit is contained in:
Eugen Wissner 2023-01-08 17:33:25 +01:00
parent bb4375313e
commit ab4808c44d
Signed by: belka
GPG Key ID: A27FDC1E8EE902C0
2 changed files with 33 additions and 1 deletions

View File

@ -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')) =

View File

@ -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