mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 20:41:49 +03:00
2107c70e17
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/4996 GitOrigin-RevId: 26d367ab83df4b161f58133afab2788c6f0c9746
48 lines
2.4 KiB
Haskell
48 lines
2.4 KiB
Haskell
module Hasura.Backends.DataConnector.Adapter.ConfigTransform
|
|
( transformSourceConfig,
|
|
transformConnSourceConfig,
|
|
)
|
|
where
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
import Data.Aeson qualified as J
|
|
import Data.Environment qualified as Env
|
|
import Data.HashMap.Strict qualified as M
|
|
import Data.Text qualified as T
|
|
import Hasura.Backends.DataConnector.API qualified as API
|
|
import Hasura.Backends.DataConnector.Adapter.Types (ConnSourceConfig (ConnSourceConfig, template, value), SourceConfig)
|
|
import Hasura.Backends.DataConnector.Plan qualified as DC
|
|
import Hasura.Base.Error (Code (NotSupported), QErr, throw400)
|
|
import Hasura.Prelude
|
|
import Kriti qualified
|
|
import Kriti.CustomFunctions qualified as Kriti
|
|
import Kriti.Error qualified as Kriti
|
|
|
|
transformConfig :: (MonadError QErr m) => API.Config -> Maybe Text -> [(T.Text, J.Value)] -> Env.Environment -> m API.Config
|
|
transformConfig config maybeTemplate scope env = do
|
|
case maybeTemplate of
|
|
Nothing -> pure config
|
|
(Just t) ->
|
|
case Kriti.runKritiWith t (("$config", J.toJSON config) : scope) (additionalFunctions env) of
|
|
Left e -> throw400 NotSupported $ "transformConfig: Kriti template transform failed - " <> tshow (Kriti.serialize e)
|
|
Right (J.Object r) -> pure $ API.Config r
|
|
Right o -> throw400 NotSupported $ "transformConfig: Kriti did not decode into Object - " <> tshow o
|
|
|
|
transformSourceConfig :: (MonadError QErr m) => SourceConfig -> [(T.Text, J.Value)] -> Env.Environment -> m SourceConfig
|
|
transformSourceConfig sc@DC.SourceConfig {_scConfig, _scTemplate} scope env = do
|
|
transformedConfig <- transformConfig _scConfig _scTemplate scope env
|
|
pure sc {DC._scConfig = transformedConfig}
|
|
|
|
transformConnSourceConfig :: (MonadError QErr m) => ConnSourceConfig -> [(T.Text, J.Value)] -> Env.Environment -> m API.Config
|
|
transformConnSourceConfig ConnSourceConfig {value, template} scope env = transformConfig value template scope env
|
|
|
|
additionalFunctions :: Env.Environment -> M.HashMap T.Text (J.Value -> Either Kriti.CustomFunctionError J.Value)
|
|
additionalFunctions env = M.singleton "env" getEnv <> Kriti.basicFuncMap
|
|
where
|
|
getEnv :: J.Value -> Either Kriti.CustomFunctionError J.Value
|
|
getEnv x = case x of
|
|
J.Null -> Right $ J.Null
|
|
J.String k -> Right $ J.toJSON $ Env.lookupEnv env (T.unpack k)
|
|
_ -> Left $ Kriti.CustomFunctionError "Environment variable name should be a string"
|