module Hasura.RQL.Types.ResultCustomization
( AliasMapping,
singletonAliasMapping,
ResultCustomizer,
applyResultCustomizer,
applyAliasMapping,
modifyFieldByName,
customizeTypeNameString,
)
where
import Data.Aeson.Ordered qualified as JO
import Data.HashMap.Strict as Map
import Data.Monoid (Endo (..))
import Hasura.Prelude
import Language.GraphQL.Draft.Syntax qualified as G
newtype AliasMapping = AliasMapping {AliasMapping -> Name -> Name
unAliasMapping :: G.Name -> G.Name}
deriving (b -> AliasMapping -> AliasMapping
NonEmpty AliasMapping -> AliasMapping
AliasMapping -> AliasMapping -> AliasMapping
(AliasMapping -> AliasMapping -> AliasMapping)
-> (NonEmpty AliasMapping -> AliasMapping)
-> (forall b. Integral b => b -> AliasMapping -> AliasMapping)
-> Semigroup AliasMapping
forall b. Integral b => b -> AliasMapping -> AliasMapping
forall a.
(a -> a -> a)
-> (NonEmpty a -> a)
-> (forall b. Integral b => b -> a -> a)
-> Semigroup a
stimes :: b -> AliasMapping -> AliasMapping
$cstimes :: forall b. Integral b => b -> AliasMapping -> AliasMapping
sconcat :: NonEmpty AliasMapping -> AliasMapping
$csconcat :: NonEmpty AliasMapping -> AliasMapping
<> :: AliasMapping -> AliasMapping -> AliasMapping
$c<> :: AliasMapping -> AliasMapping -> AliasMapping
Semigroup, Semigroup AliasMapping
AliasMapping
Semigroup AliasMapping
-> AliasMapping
-> (AliasMapping -> AliasMapping -> AliasMapping)
-> ([AliasMapping] -> AliasMapping)
-> Monoid AliasMapping
[AliasMapping] -> AliasMapping
AliasMapping -> AliasMapping -> AliasMapping
forall a.
Semigroup a -> a -> (a -> a -> a) -> ([a] -> a) -> Monoid a
mconcat :: [AliasMapping] -> AliasMapping
$cmconcat :: [AliasMapping] -> AliasMapping
mappend :: AliasMapping -> AliasMapping -> AliasMapping
$cmappend :: AliasMapping -> AliasMapping -> AliasMapping
mempty :: AliasMapping
$cmempty :: AliasMapping
$cp1Monoid :: Semigroup AliasMapping
Monoid) via (Endo G.Name)
singletonAliasMapping :: G.Name -> G.Name -> AliasMapping
singletonAliasMapping :: Name -> Name -> AliasMapping
singletonAliasMapping Name
fieldName Name
alias = (Name -> Name) -> AliasMapping
AliasMapping ((Name -> Name) -> AliasMapping) -> (Name -> Name) -> AliasMapping
forall a b. (a -> b) -> a -> b
$ \Name
fieldName' ->
if Name
fieldName Name -> Name -> Bool
forall a. Eq a => a -> a -> Bool
== Name
fieldName'
then Name
alias
else Name
fieldName'
newtype ResultCustomizer = ResultCustomizer {ResultCustomizer -> AliasMapping -> Value -> Value
unResultCustomizer :: AliasMapping -> JO.Value -> JO.Value}
deriving (b -> ResultCustomizer -> ResultCustomizer
NonEmpty ResultCustomizer -> ResultCustomizer
ResultCustomizer -> ResultCustomizer -> ResultCustomizer
(ResultCustomizer -> ResultCustomizer -> ResultCustomizer)
-> (NonEmpty ResultCustomizer -> ResultCustomizer)
-> (forall b.
Integral b =>
b -> ResultCustomizer -> ResultCustomizer)
-> Semigroup ResultCustomizer
forall b. Integral b => b -> ResultCustomizer -> ResultCustomizer
forall a.
(a -> a -> a)
-> (NonEmpty a -> a)
-> (forall b. Integral b => b -> a -> a)
-> Semigroup a
stimes :: b -> ResultCustomizer -> ResultCustomizer
$cstimes :: forall b. Integral b => b -> ResultCustomizer -> ResultCustomizer
sconcat :: NonEmpty ResultCustomizer -> ResultCustomizer
$csconcat :: NonEmpty ResultCustomizer -> ResultCustomizer
<> :: ResultCustomizer -> ResultCustomizer -> ResultCustomizer
$c<> :: ResultCustomizer -> ResultCustomizer -> ResultCustomizer
Semigroup, Semigroup ResultCustomizer
ResultCustomizer
Semigroup ResultCustomizer
-> ResultCustomizer
-> (ResultCustomizer -> ResultCustomizer -> ResultCustomizer)
-> ([ResultCustomizer] -> ResultCustomizer)
-> Monoid ResultCustomizer
[ResultCustomizer] -> ResultCustomizer
ResultCustomizer -> ResultCustomizer -> ResultCustomizer
forall a.
Semigroup a -> a -> (a -> a -> a) -> ([a] -> a) -> Monoid a
mconcat :: [ResultCustomizer] -> ResultCustomizer
$cmconcat :: [ResultCustomizer] -> ResultCustomizer
mappend :: ResultCustomizer -> ResultCustomizer -> ResultCustomizer
$cmappend :: ResultCustomizer -> ResultCustomizer -> ResultCustomizer
mempty :: ResultCustomizer
$cmempty :: ResultCustomizer
$cp1Monoid :: Semigroup ResultCustomizer
Monoid) via (AliasMapping -> Endo JO.Value)
applyResultCustomizer :: ResultCustomizer -> JO.Value -> JO.Value
applyResultCustomizer :: ResultCustomizer -> Value -> Value
applyResultCustomizer = ((AliasMapping -> Value -> Value) -> AliasMapping -> Value -> Value
forall a b. (a -> b) -> a -> b
$ AliasMapping
forall a. Monoid a => a
mempty) ((AliasMapping -> Value -> Value) -> Value -> Value)
-> (ResultCustomizer -> AliasMapping -> Value -> Value)
-> ResultCustomizer
-> Value
-> Value
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ResultCustomizer -> AliasMapping -> Value -> Value
unResultCustomizer
applyAliasMapping :: AliasMapping -> ResultCustomizer -> ResultCustomizer
applyAliasMapping :: AliasMapping -> ResultCustomizer -> ResultCustomizer
applyAliasMapping AliasMapping
aliasMapping (ResultCustomizer AliasMapping -> Value -> Value
m) =
(AliasMapping -> Value -> Value) -> ResultCustomizer
ResultCustomizer ((AliasMapping -> Value -> Value) -> ResultCustomizer)
-> (AliasMapping -> Value -> Value) -> ResultCustomizer
forall a b. (a -> b) -> a -> b
$ AliasMapping -> Value -> Value
m (AliasMapping -> Value -> Value)
-> (AliasMapping -> AliasMapping) -> AliasMapping -> Value -> Value
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (AliasMapping -> AliasMapping -> AliasMapping
forall a. Semigroup a => a -> a -> a
<> AliasMapping
aliasMapping)
modifyFieldByName :: G.Name -> ResultCustomizer -> ResultCustomizer
modifyFieldByName :: Name -> ResultCustomizer -> ResultCustomizer
modifyFieldByName Name
fieldName ResultCustomizer {AliasMapping -> Value -> Value
unResultCustomizer :: AliasMapping -> Value -> Value
unResultCustomizer :: ResultCustomizer -> AliasMapping -> Value -> Value
..} =
(AliasMapping -> Value -> Value) -> ResultCustomizer
ResultCustomizer ((AliasMapping -> Value -> Value) -> ResultCustomizer)
-> (AliasMapping -> Value -> Value) -> ResultCustomizer
forall a b. (a -> b) -> a -> b
$ \AliasMapping {Name -> Name
unAliasMapping :: Name -> Name
unAliasMapping :: AliasMapping -> Name -> Name
..} ->
let applyCustomizer :: Value -> Value
applyCustomizer = AliasMapping -> Value -> Value
unResultCustomizer AliasMapping
forall a. Monoid a => a
mempty
modifyFieldByName' :: Value -> Value
modifyFieldByName' = \case
JO.Object Object
o -> Object -> Value
JO.Object (Object -> Value) -> Object -> Value
forall a b. (a -> b) -> a -> b
$ (Value -> Value) -> Text -> Object -> Object
JO.adjust Value -> Value
applyCustomizer (Name -> Text
G.unName (Name -> Text) -> Name -> Text
forall a b. (a -> b) -> a -> b
$ Name -> Name
unAliasMapping Name
fieldName) Object
o
JO.Array Array
a -> Array -> Value
JO.Array (Array -> Value) -> Array -> Value
forall a b. (a -> b) -> a -> b
$ Value -> Value
modifyFieldByName' (Value -> Value) -> Array -> Array
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Array
a
Value
v -> Value
v
in Value -> Value
modifyFieldByName'
customizeTypeNameString :: HashMap G.Name G.Name -> ResultCustomizer
customizeTypeNameString :: HashMap Name Name -> ResultCustomizer
customizeTypeNameString HashMap Name Name
typeNameMap | HashMap Name Name -> Bool
forall k v. HashMap k v -> Bool
Map.null HashMap Name Name
typeNameMap = ResultCustomizer
forall a. Monoid a => a
mempty
customizeTypeNameString HashMap Name Name
typeNameMap =
(AliasMapping -> Value -> Value) -> ResultCustomizer
ResultCustomizer ((AliasMapping -> Value -> Value) -> ResultCustomizer)
-> (AliasMapping -> Value -> Value) -> ResultCustomizer
forall a b. (a -> b) -> a -> b
$ \AliasMapping
_aliasMapping -> \case
JO.String Text
t -> Text -> Value
JO.String (Text -> Value) -> Text -> Value
forall a b. (a -> b) -> a -> b
$
Text -> Maybe Text -> Text
forall a. a -> Maybe a -> a
fromMaybe Text
t (Maybe Text -> Text) -> Maybe Text -> Text
forall a b. (a -> b) -> a -> b
$ do
Name
typeName <- Text -> Maybe Name
G.mkName Text
t
Name -> Text
G.unName (Name -> Text) -> Maybe Name -> Maybe Text
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Name -> HashMap Name Name -> Maybe Name
forall k v. (Eq k, Hashable k) => k -> HashMap k v -> Maybe v
Map.lookup Name
typeName HashMap Name Name
typeNameMap
Value
v -> Value
v