prescientmoon/graphql
1
0
Fork 0
forked from OSS/graphql
Code Pull Requests Activity

Encode unions

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/Language/GraphQL/AST/Encoder.hs
View File

@ -128,6 +128,13 @@ typeDefinition formatter = \case
<> optempty (directives formatter) directives' <> optempty (directives formatter) directives'
<> eitherFormat formatter " " "" <> eitherFormat formatter " " ""
<> bracesList formatter (fieldDefinition nextFormatter) fields' <> 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. _typeDefinition' -> "" -- TODO: Types missing.
where where
nextFormatter = incrementIndent formatter nextFormatter = incrementIndent formatter
@ -140,6 +147,19 @@ implementsInterfaces (Full.ImplementsInterfaces interfaces)
$ Text.intercalate " & " $ Text.intercalate " & "
$ toList interfaces $ 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 -> Lazy.Text.Text
description _formatter (Full.Description Nothing) = "" description _formatter (Full.Description Nothing) = ""
description formatter (Full.Description (Just description')) = description formatter (Full.Description (Just description')) =

12
tests/Language/GraphQL/AST/EncoderSpec.hs
View File

@ -217,3 +217,15 @@ spec = do
|] |]
actual = typeSystemDefinition pretty definition' actual = typeSystemDefinition pretty definition'
in actual `shouldBe` expected 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