/
Schema.purs
470 lines (416 loc) · 17 KB
/
Schema.purs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
-- | Codegen functions to get purs schema code from graphQL schemas
module GraphQL.Client.CodeGen.Schema
( schemaFromGqlToPurs
, schemasFromGqlToPurs
) where
import Prelude
import Data.Argonaut.Decode (decodeJson)
import Data.Argonaut.Encode (encodeJson)
import Data.Array (filter, notElem, nub, nubBy)
import Data.Array as Array
import Data.Either (Either(..), hush)
import Data.Foldable (fold, foldMap, intercalate)
import Data.Function (on)
import Data.FunctorWithIndex (mapWithIndex)
import Data.GraphQL.AST as AST
import Data.GraphQL.Parser (document)
import Data.List (List, mapMaybe)
import Data.List as List
import Data.Map (Map, lookup, unions)
import Data.Map as Map
import Data.Maybe (Maybe(..), fromMaybe, fromMaybe', maybe)
import Data.Monoid (guard)
import Data.Newtype (unwrap)
import Data.String (Pattern(..), codePointFromChar, contains, take)
import Data.String as String
import Data.String.CodePoints (takeWhile)
import Data.String.Extra (pascalCase)
import Data.Traversable (sequence, traverse)
import Data.Tuple (Tuple(..))
import Effect.Aff (Aff)
import GraphQL.Client.CodeGen.GetSymbols (getSymbols, symbolsToCode)
import GraphQL.Client.CodeGen.Lines (commentPrefix, docComment, fromLines, indent, toLines)
import GraphQL.Client.CodeGen.Template.Enum as Enum
import GraphQL.Client.CodeGen.Template.Schema as Schema
import GraphQL.Client.CodeGen.Types (FilesToWrite, GqlEnum, GqlInput, InputOptions, PursGql)
import Text.Parsing.Parser (ParseError, runParser)
schemasFromGqlToPurs :: InputOptions -> Array GqlInput -> Aff (Either ParseError FilesToWrite)
schemasFromGqlToPurs opts_ = traverse (schemaFromGqlToPursWithCache opts) >>> map sequence >>> map (map collectSchemas)
where
modulePrefix = foldMap (_ <> ".") opts.modulePath
opts = opts_ { fieldTypeOverrides = fieldTypeOverrides }
fieldTypeOverrides =
Map.unions
$ opts_.fieldTypeOverrides
# mapWithIndex \gqlObjectName obj ->
Map.fromFoldable
[ Tuple gqlObjectName obj
, Tuple (gqlObjectName <> "InsertInput") obj
, Tuple (gqlObjectName <> "MinFields") obj
, Tuple (gqlObjectName <> "MaxFields") obj
, Tuple (gqlObjectName <> "SetInput") obj
, Tuple (gqlObjectName <> "BoolExp") $ map (\o -> o { typeName = o.typeName <> "ComparisonExp" }) obj
]
collectSchemas :: Array PursGql -> FilesToWrite
collectSchemas pursGqls =
{ schemas:
pursGqls
<#> \pg ->
{ code:
Schema.template
{ name: pg.moduleName
, mainSchemaCode: pg.mainSchemaCode
, idImport: opts.idImport
, enums: map _.name pg.enums
, modulePrefix
}
, path: opts.dir <> "/Schema/" <> pg.moduleName <> ".purs"
}
, enums:
nubBy (compare `on` _.path)
$ pursGqls
>>= \pg ->
pg.enums
<#> \{ name, values, description } ->
{ code:
Enum.template modulePrefix
{ name
, values
, description
, imports: opts.enumImports
, customCode: opts.customEnumCode
}
, path: opts.dir <> "/Enum/" <> name <> ".purs"
}
, symbols:
pursGqls >>= _.symbols
# \syms ->
{ path: opts.dir <> "/Symbols.purs", code: symbolsToCode modulePrefix syms }
}
-- | Given a gql doc this will create the equivalent purs gql schema
schemaFromGqlToPursWithCache :: InputOptions -> GqlInput -> Aff (Either ParseError PursGql)
schemaFromGqlToPursWithCache opts { schema, moduleName } = go opts.cache
where
go Nothing = pure $ schemaFromGqlToPurs opts { schema, moduleName }
go (Just { get, set }) = do
jsonMay <- get schema
eVal <- case jsonMay >>= decodeJson >>> hush of
Nothing -> go Nothing
Just res -> pure $ Right res
case eVal of
Right val -> set { key: schema, val: encodeJson val }
_ -> pure unit
pure $ eVal
schemaFromGqlToPurs :: InputOptions -> GqlInput -> Either ParseError PursGql
schemaFromGqlToPurs opts { schema, moduleName } =
runParser schema document
<#> \ast ->
let
symbols = Array.fromFoldable $ getSymbols ast
in
{ mainSchemaCode: gqlToPursMainSchemaCode opts ast
, enums: gqlToPursEnums opts.gqlScalarsToPursTypes ast
, symbols
, moduleName
}
toImport ::
forall r.
String ->
Array
{ moduleName :: String
| r
} ->
Array String
toImport mainCode =
map
( \t ->
guard (contains (Pattern t.moduleName) mainCode)
$ "\nimport "
<> t.moduleName
<> " as "
<> t.moduleName
)
gqlToPursMainSchemaCode :: InputOptions -> AST.Document -> String
gqlToPursMainSchemaCode { gqlScalarsToPursTypes, externalTypes, fieldTypeOverrides, useNewtypesForRecords } doc =
imports
<> guard (imports /= "") "\n"
<> "\n"
<> mainCode
where
imports =
fold $ nub
$ toImport mainCode (Array.fromFoldable externalTypes)
<> toImport mainCode (Array.fromFoldable $ unions fieldTypeOverrides)
<> toImport mainCode [ { moduleName: "Data.Argonaut.Core" } ]
mainCode = unwrap doc # mapMaybe definitionToPurs # removeDuplicateDefinitions # intercalate "\n\n"
removeDuplicateDefinitions = Array.fromFoldable >>> nubBy (compare `on` getDefinitionTypeName) >>> List.fromFoldable
getDefinitionTypeName :: String -> String
getDefinitionTypeName =
takeWhile (notEq (codePointFromChar '='))
>>> toLines
>>> filter (\l -> take (String.length commentPrefix) l /= commentPrefix)
>>> fromLines
definitionToPurs :: AST.Definition -> Maybe String
definitionToPurs = case _ of
AST.Definition_ExecutableDefinition _ -> Nothing
AST.Definition_TypeSystemDefinition def -> typeSystemDefinitionToPurs def
AST.Definition_TypeSystemExtension _ -> Nothing
typeSystemDefinitionToPurs :: AST.TypeSystemDefinition -> Maybe String
typeSystemDefinitionToPurs = case _ of
AST.TypeSystemDefinition_SchemaDefinition schemaDefinition -> Just $ schemaDefinitionToPurs schemaDefinition
AST.TypeSystemDefinition_TypeDefinition typeDefinition -> typeDefinitionToPurs typeDefinition
AST.TypeSystemDefinition_DirectiveDefinition directiveDefinition -> directiveDefinitionToPurs directiveDefinition
schemaDefinitionToPurs :: AST.SchemaDefinition -> String
schemaDefinitionToPurs (AST.SchemaDefinition { rootOperationTypeDefinition }) = map rootOperationTypeDefinitionToPurs rootOperationTypeDefinition # intercalate "\n\n"
rootOperationTypeDefinitionToPurs :: AST.RootOperationTypeDefinition -> String
rootOperationTypeDefinitionToPurs (AST.RootOperationTypeDefinition { operationType, namedType }) =
"type "
<> opStr
<> " = "
<> (namedTypeToPurs_ namedType)
where
opStr = case operationType of
AST.Query -> "Query"
AST.Mutation -> "Mutation"
AST.Subscription -> "Subscription"
typeDefinitionToPurs :: AST.TypeDefinition -> Maybe String
typeDefinitionToPurs = case _ of
AST.TypeDefinition_ScalarTypeDefinition scalarTypeDefinition -> Just $ scalarTypeDefinitionToPurs scalarTypeDefinition
AST.TypeDefinition_ObjectTypeDefinition objectTypeDefinition -> Just $ objectTypeDefinitionToPurs objectTypeDefinition
AST.TypeDefinition_InterfaceTypeDefinition interfaceTypeDefinition -> interfaceTypeDefinitionToPurs interfaceTypeDefinition
AST.TypeDefinition_UnionTypeDefinition unionTypeDefinition -> unionTypeDefinitionToPurs unionTypeDefinition
AST.TypeDefinition_EnumTypeDefinition enumTypeDefinition -> enumTypeDefinitionToPurs enumTypeDefinition
AST.TypeDefinition_InputObjectTypeDefinition inputObjectTypeDefinition -> Just $ inputObjectTypeDefinitionToPurs inputObjectTypeDefinition
scalarTypeDefinitionToPurs :: AST.ScalarTypeDefinition -> String
scalarTypeDefinitionToPurs (AST.ScalarTypeDefinition { description, name }) =
guard (notElem tName builtInTypes)
$ docComment description
<> "type "
<> tName
<> " = "
<> typeAndModule.moduleName
<> "."
<> typeAndModule.typeName
where
tName = typeName_ name
typeAndModule =
lookup tName externalTypes
# fromMaybe
{ moduleName: "Data.Argonaut.Core"
, typeName: "Json -- Unknown scalar type. Add " <> tName <> " to externalTypes in codegen options to override this behaviour"
}
builtInTypes = [ "Int", "Number", "String", "Boolean" ]
objectTypeDefinitionToPurs :: AST.ObjectTypeDefinition -> String
objectTypeDefinitionToPurs ( AST.ObjectTypeDefinition
{ description
, fieldsDefinition
, name
}
) =
let
tName = typeName_ name
in
docComment description
<> if useNewtypesForRecords then
"newtype "
<> typeName_ name
<> (fieldsDefinition # foldMap \fd -> " = " <> typeName_ name <> " " <> fieldsDefinitionToPurs tName fd)
<> "\nderive instance newtype"
<> tName
<> " :: Newtype "
<> tName
<> " _"
<> "\ninstance argToGql"
<> tName
<> " :: (Newtype "
<> tName
<> " {| p}, RecordArg p a u) => ArgGql "
<> tName
<> " { | a }"
else
"type "
<> typeName_ name
<> (fieldsDefinition # foldMap \fd -> " = " <> fieldsDefinitionToPurs tName fd)
fieldsDefinitionToPurs :: String -> AST.FieldsDefinition -> String
fieldsDefinitionToPurs objectName (AST.FieldsDefinition fieldsDefinition) =
indent
$ "\n{ "
<> intercalate "\n, " (map (fieldDefinitionToPurs objectName) fieldsDefinition)
<> "\n}"
fieldDefinitionToPurs :: String -> AST.FieldDefinition -> String
fieldDefinitionToPurs objectName ( AST.FieldDefinition
{ description
, name
, argumentsDefinition
, type: tipe
}
) =
inlineComment description
<> name
<> " :: "
<> foldMap argumentsDefinitionToPurs argumentsDefinition
<> case lookup objectName fieldTypeOverrides >>= lookup name of
Nothing -> typeToPurs tipe
Just out -> case tipe of
AST.Type_NonNullType _ -> out.moduleName <> "." <> out.typeName
AST.Type_ListType _ -> wrapArray $ out.moduleName <> "." <> out.typeName
_ -> wrapMaybe $ out.moduleName <> "." <> out.typeName
argumentsDefinitionToPurs :: AST.ArgumentsDefinition -> String
argumentsDefinitionToPurs (AST.ArgumentsDefinition inputValueDefinitions) =
indent
$ "\n{ "
<> intercalate "\n, " (map inputValueDefinitionsToPurs inputValueDefinitions)
<> "\n}\n==> "
inputValueDefinitionsToPurs :: AST.InputValueDefinition -> String
inputValueDefinitionsToPurs ( AST.InputValueDefinition
{ description
, name
, type: tipe
}
) =
inlineComment description
<> name
<> " :: "
<> argTypeToPurs tipe
interfaceTypeDefinitionToPurs :: AST.InterfaceTypeDefinition -> Maybe String
interfaceTypeDefinitionToPurs (AST.InterfaceTypeDefinition _) = Nothing
unionTypeDefinitionToPurs :: AST.UnionTypeDefinition -> Maybe String
unionTypeDefinitionToPurs (AST.UnionTypeDefinition _) = Nothing
enumTypeDefinitionToPurs :: AST.EnumTypeDefinition -> Maybe String
enumTypeDefinitionToPurs (AST.EnumTypeDefinition _) = Nothing
inputObjectTypeDefinitionToPurs :: AST.InputObjectTypeDefinition -> String
inputObjectTypeDefinitionToPurs ( AST.InputObjectTypeDefinition
{ description
, inputFieldsDefinition
, name
}
) =
let
tName = typeName_ name
in
docComment description
<> "newtype "
<> tName
<> ( inputFieldsDefinition
# foldMap \(AST.InputFieldsDefinition fd) ->
" = "
<> tName
<> inputValueToFieldsDefinitionToPurs tName fd
)
<> "\nderive instance newtype"
<> tName
<> " :: Newtype "
<> tName
<> " _"
<> "\ninstance argToGql"
<> tName
<> " :: (Newtype "
<> tName
<> " {| p}, RecordArg p a u) => ArgGql "
<> tName
<> " { | a }"
inputValueToFieldsDefinitionToPurs :: String -> List AST.InputValueDefinition -> String
inputValueToFieldsDefinitionToPurs objectName definitions =
indent
$ "\n{ "
<> intercalate "\n, " (map (inputValueDefinitionToPurs objectName) definitions)
<> "\n}"
inputValueDefinitionToPurs :: String -> AST.InputValueDefinition -> String
inputValueDefinitionToPurs objectName ( AST.InputValueDefinition
{ description
, name
, type: tipe
}
) =
inlineComment description
<> name
<> " :: "
<> case lookup objectName fieldTypeOverrides >>= lookup name of
Nothing -> argTypeToPurs tipe
Just out -> case tipe of
AST.Type_NonNullType _ -> wrapNotNull $ out.moduleName <> "." <> out.typeName
AST.Type_ListType _ -> wrapArray $ out.moduleName <> "." <> out.typeName
_ -> out.moduleName <> "." <> out.typeName
directiveDefinitionToPurs :: AST.DirectiveDefinition -> Maybe String
directiveDefinitionToPurs _ = Nothing
argTypeToPurs :: AST.Type -> String
argTypeToPurs = case _ of
(AST.Type_NamedType namedType) -> namedTypeToPurs_ namedType
(AST.Type_ListType listType) -> argListTypeToPurs listType
(AST.Type_NonNullType notNullType) -> wrapNotNull $ argNotNullTypeToPurs notNullType
argNotNullTypeToPurs :: AST.NonNullType -> String
argNotNullTypeToPurs = case _ of
AST.NonNullType_NamedType t -> namedTypeToPurs_ t
AST.NonNullType_ListType t -> argListTypeToPurs t
argListTypeToPurs :: AST.ListType -> String
argListTypeToPurs (AST.ListType t) = "(Array " <> argTypeToPurs t <> ")"
wrapNotNull s = "(NotNull " <> s <> ")"
typeToPurs :: AST.Type -> String
typeToPurs = case _ of
(AST.Type_NamedType namedType) -> namedTypeToPursNullable namedType
(AST.Type_ListType listType) -> listTypeToPursNullable listType
(AST.Type_NonNullType notNullType) -> notNullTypeToPurs notNullType
namedTypeToPursNullable :: AST.NamedType -> String
namedTypeToPursNullable = wrapMaybe <<< namedTypeToPurs_
listTypeToPursNullable :: AST.ListType -> String
listTypeToPursNullable t = wrapMaybe $ listTypeToPurs t
wrapMaybe s = "(Maybe " <> s <> ")"
notNullTypeToPurs :: AST.NonNullType -> String
notNullTypeToPurs = case _ of
AST.NonNullType_NamedType t -> namedTypeToPurs_ t
AST.NonNullType_ListType t -> listTypeToPurs t
listTypeToPurs :: AST.ListType -> String
listTypeToPurs (AST.ListType t) = wrapArray $ typeToPurs t
wrapArray s = "(Array " <> s <> ")"
typeName_ = typeName gqlScalarsToPursTypes
namedTypeToPurs_ = namedTypeToPurs gqlScalarsToPursTypes
gqlToPursEnums :: Map String String -> AST.Document -> Array GqlEnum
gqlToPursEnums gqlScalarsToPursTypes = unwrap >>> mapMaybe definitionToEnum >>> Array.fromFoldable
where
definitionToEnum :: AST.Definition -> Maybe GqlEnum
definitionToEnum = case _ of
AST.Definition_TypeSystemDefinition def -> typeSystemDefinitionToPurs def
_ -> Nothing
typeSystemDefinitionToPurs :: AST.TypeSystemDefinition -> Maybe GqlEnum
typeSystemDefinitionToPurs = case _ of
AST.TypeSystemDefinition_TypeDefinition typeDefinition -> typeDefinitionToPurs typeDefinition
_ -> Nothing
typeDefinitionToPurs :: AST.TypeDefinition -> Maybe GqlEnum
typeDefinitionToPurs = case _ of
AST.TypeDefinition_EnumTypeDefinition (AST.EnumTypeDefinition enumTypeDefinition) ->
Just
{ name: typeName_ enumTypeDefinition.name
, description: enumTypeDefinition.description
, values: maybe [] enumValuesDefinitionToPurs enumTypeDefinition.enumValuesDefinition
}
_ -> Nothing
enumValuesDefinitionToPurs :: AST.EnumValuesDefinition -> Array String
enumValuesDefinitionToPurs def =
Array.fromFoldable $ unwrap def
<#> \(AST.EnumValueDefinition { enumValue }) ->
unwrap enumValue
typeName_ = typeName gqlScalarsToPursTypes
namedTypeToPurs :: Map String String -> AST.NamedType -> String
namedTypeToPurs gqlScalarsToPursTypes (AST.NamedType str) = typeName gqlScalarsToPursTypes str
inlineComment :: Maybe String -> String
inlineComment = foldMap (\str -> "\n{- " <> str <> " -}\n")
typeName :: Map String String -> String -> String
typeName gqlScalarsToPursTypes str =
lookup str gqlScalarsToPursTypes
# fromMaybe' \_ -> case pascalCase str of
"Id" -> "ID"
"Float" -> "Number"
"Numeric" -> "Number"
"Bigint" -> "Number"
"Smallint" -> "Int"
"Integer" -> "Int"
"Int" -> "Int"
"Int2" -> "Int"
"Int4" -> "Int"
"Int8" -> "Int"
"Text" -> "String"
"Citext" -> "String"
"Jsonb" -> "Json"
"Timestamp" -> "DateTime"
"Timestamptz" -> "DateTime"
s -> s