mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 12:31:52 +03:00
65 lines
1.8 KiB
Haskell
65 lines
1.8 KiB
Haskell
module Hasura.GraphQL.Utils
|
|
( showName
|
|
, groupTuples
|
|
, groupListWith
|
|
, mkMapWith
|
|
, showNames
|
|
, simpleGraphQLQuery
|
|
, getBaseTyWithNestedLevelsCount
|
|
) where
|
|
|
|
import Hasura.Prelude
|
|
|
|
import qualified Data.HashMap.Strict as Map
|
|
import qualified Data.List.NonEmpty as NE
|
|
import qualified Data.Text as T
|
|
import qualified Language.GraphQL.Draft.Syntax as G
|
|
|
|
showName :: G.Name -> Text
|
|
showName name = "\"" <> G.unName name <> "\""
|
|
|
|
getBaseTyWithNestedLevelsCount :: G.GType -> (G.Name, Int)
|
|
getBaseTyWithNestedLevelsCount ty = go ty 0
|
|
where
|
|
go :: G.GType -> Int -> (G.Name, Int)
|
|
go gType ctr =
|
|
case gType of
|
|
G.TypeNamed _ n -> (n, ctr)
|
|
G.TypeList _ gType' -> flip go (ctr + 1) gType'
|
|
|
|
groupListWith
|
|
:: (Eq k, Hashable k, Foldable t, Functor t)
|
|
=> (v -> k) -> t v -> Map.HashMap k (NE.NonEmpty v)
|
|
groupListWith f l =
|
|
groupTuples $ fmap (\v -> (f v, v)) l
|
|
|
|
groupTuples
|
|
:: (Eq k, Hashable k, Foldable t)
|
|
=> t (k, v) -> Map.HashMap k (NE.NonEmpty v)
|
|
groupTuples =
|
|
foldr groupFlds Map.empty
|
|
where
|
|
groupFlds (k, v) m = case Map.lookup k m of
|
|
Nothing -> Map.insert k (v NE.:| []) m
|
|
Just s -> Map.insert k (v NE.<| s) m
|
|
|
|
-- either duplicate keys or the map
|
|
mkMapWith
|
|
:: (Eq k, Hashable k, Foldable t, Functor t)
|
|
=> (v -> k) -> t v -> Either (NE.NonEmpty k) (Map.HashMap k v)
|
|
mkMapWith f l =
|
|
case NE.nonEmpty dups of
|
|
Just dupsNE -> Left dupsNE
|
|
Nothing -> Right $ Map.map NE.head mapG
|
|
where
|
|
mapG = groupListWith f l
|
|
dups = Map.keys $ Map.filter ((> 1) . length) mapG
|
|
|
|
showNames :: (Foldable t) => t G.Name -> Text
|
|
showNames names =
|
|
T.intercalate ", " $ map G.unName $ toList names
|
|
|
|
-- A simple graphql query to be used in generators
|
|
simpleGraphQLQuery :: Text
|
|
simpleGraphQLQuery = "query {author {id name}}"
|