mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-23 23:43:44 +03:00
7aef17863a
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/9001 GitOrigin-RevId: 5053e4c881f0d037b9170b434bcf533284a827a6
71 lines
2.4 KiB
Haskell
71 lines
2.4 KiB
Haskell
module Hasura.LogicalModel.Common
|
|
( toFieldInfo,
|
|
columnsFromFields,
|
|
logicalModelFieldsToFieldInfo,
|
|
)
|
|
where
|
|
|
|
import Data.Bifunctor (bimap)
|
|
import Data.HashMap.Strict qualified as HashMap
|
|
import Data.HashMap.Strict.InsOrd qualified as InsOrdHashMap
|
|
import Data.Text.Extended (ToTxt (toTxt))
|
|
import Hasura.LogicalModel.NullableScalarType (NullableScalarType (..))
|
|
import Hasura.LogicalModel.Types (LogicalModelField (..))
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types.Backend (Backend (..))
|
|
import Hasura.RQL.Types.Column (ColumnInfo (..), ColumnMutability (..), ColumnType (..), fromCol)
|
|
import Hasura.RQL.Types.Table (FieldInfo (..), FieldInfoMap)
|
|
import Language.GraphQL.Draft.Syntax qualified as G
|
|
|
|
columnsFromFields ::
|
|
InsOrdHashMap.InsOrdHashMap k (LogicalModelField b) ->
|
|
InsOrdHashMap.InsOrdHashMap k (NullableScalarType b)
|
|
columnsFromFields =
|
|
InsOrdHashMap.mapMaybe
|
|
( \case
|
|
LogicalModelScalarField
|
|
{ lmfType = nstType,
|
|
lmfNullable = nstNullable,
|
|
lmfDescription = nstDescription
|
|
} ->
|
|
Just (NullableScalarType {..})
|
|
_ -> Nothing
|
|
)
|
|
|
|
toFieldInfo :: forall b. (Backend b) => InsOrdHashMap.InsOrdHashMap (Column b) (NullableScalarType b) -> Maybe [FieldInfo b]
|
|
toFieldInfo fields =
|
|
traverseWithIndex
|
|
(\i -> fmap FIColumn . logicalModelToColumnInfo i)
|
|
(InsOrdHashMap.toList fields)
|
|
|
|
traverseWithIndex :: (Applicative m) => (Int -> aa -> m bb) -> [aa] -> m [bb]
|
|
traverseWithIndex f = zipWithM f [0 ..]
|
|
|
|
logicalModelToColumnInfo :: forall b. (Backend b) => Int -> (Column b, NullableScalarType b) -> Maybe (ColumnInfo b)
|
|
logicalModelToColumnInfo i (column, NullableScalarType {..}) = do
|
|
name <- G.mkName (toTxt column)
|
|
pure $
|
|
ColumnInfo
|
|
{ ciColumn = column,
|
|
ciName = name,
|
|
ciPosition = i,
|
|
ciType = ColumnScalar nstType,
|
|
ciIsNullable = nstNullable,
|
|
ciDescription = G.Description <$> nstDescription,
|
|
ciMutability = ColumnMutability {_cmIsInsertable = False, _cmIsUpdatable = False}
|
|
}
|
|
|
|
logicalModelFieldsToFieldInfo ::
|
|
forall b.
|
|
(Backend b) =>
|
|
InsOrdHashMap.InsOrdHashMap (Column b) (LogicalModelField b) ->
|
|
FieldInfoMap (FieldInfo b)
|
|
logicalModelFieldsToFieldInfo =
|
|
HashMap.fromList
|
|
. fmap (bimap (fromCol @b) FIColumn)
|
|
. fromMaybe mempty
|
|
. traverseWithIndex
|
|
(\i (column, lmf) -> (,) column <$> logicalModelToColumnInfo i (column, lmf))
|
|
. InsOrdHashMap.toList
|
|
. columnsFromFields
|