aboutsummaryrefslogtreecommitdiffstats
path: root/src/Data/JLD.hs
blob: d60e5a1a37ba5082023b35488c46c9fd1d7a6356 (plain) (blame)
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
module Data.JLD (
    module Data.JLD.Mime,
    module Data.JLD.Error,
    module Data.JLD.Options,
    JLDExpandParams (..),
    expand,
) where

import Data.JLD.Prelude

import Data.JLD.Context (buildActiveContext)
import Data.JLD.Control.Monad.RES (evalREST, runREST)
import Data.JLD.Error
import Data.JLD.Expansion (JLDEParams (..))
import Data.JLD.Expansion qualified as E (expand)
import Data.JLD.Mime
import Data.JLD.Model.ActiveContext (ActiveContext (..), newActiveContext)
import Data.JLD.Model.Keyword (Keyword (..))
import Data.JLD.Model.URI (uriToIri)
import Data.JLD.Monad (JLDEnv, JLDState, newEnv, newState)
import Data.JLD.Options
import Data.JLD.Util (flattenSingletonArray, valueToArray)

import Data.Aeson (Value (..))
import Data.Aeson.KeyMap qualified as KM
import Data.Vector qualified as V (singleton)
import Text.URI (URI)

data JLDExpandParams e m = JLDExpandParams
    { jldExpandParamsExpandContext :: Maybe Value
    , jldExpandParamsFrameExpansion :: Bool
    , jldExpandParamsEnv :: JLDEnv e m
    , jldExpandParamsState :: JLDState
    }
    deriving (Show)

expand :: Monad m => Value -> URI -> (JLDExpandParams () m -> JLDExpandParams e m) -> m (Either (JLDError e) Value, JLDState)
expand document baseUrl paramsFn = do
    let JLDExpandParams{..} =
            paramsFn
                JLDExpandParams
                    { jldExpandParamsExpandContext = Nothing
                    , jldExpandParamsFrameExpansion = False
                    , jldExpandParamsEnv = newEnv id
                    , jldExpandParamsState = newState id
                    }

        activeContext = newActiveContext \ac -> ac{activeContextBaseUrl = Just baseUrl, activeContextBaseIri = uriToIri baseUrl}
        expansionParams p = p{jldeParamsFrameExpansion = jldExpandParamsFrameExpansion}

    -- 6.
    let maybeExpandContext =
            jldExpandParamsExpandContext <&> flattenSingletonArray .> \case
                Array expandedContext -> Array expandedContext
                (Object expandedContext) | Just ctx <- KM.lookup (show KeywordContext) expandedContext -> ctx
                expandedContext -> Array <| V.singleton expandedContext

    activeContext' <- case maybeExpandContext of
        Just expandContext ->
            buildActiveContext activeContext expandContext (Just baseUrl) id
                |> evalREST jldExpandParamsEnv jldExpandParamsState
                |> fmap (fromRight activeContext)
        Nothing -> pure activeContext

    -- 8.
    (result, state') <-
        E.expand activeContext' document baseUrl expansionParams
            |> runREST jldExpandParamsEnv jldExpandParamsState

    let result' = case result of
            -- 8.1.
            Right (Object expanded)
                | KM.size expanded == 1
                , Just expanded' <- KM.lookup (show KeywordGraph) expanded ->
                    Right <. Array <| valueToArray expanded'
            -- 8.2.
            Right Null -> Right <| Array mempty
            -- 8.3.
            Right expanded -> Right <. Array <| valueToArray expanded
            --
            Left err -> Left err

    pure (result', state')