Compat.hs

{-# LANGUAGE CPP #-}

module Autodocodec.Aeson.Compat where

#if MIN_VERSION_aeson(2,0,0)
import Data.Aeson.Key (Key)
import qualified Data.Aeson.Key as K
import Data.Aeson.KeyMap (KeyMap)
import qualified Data.Aeson.KeyMap as KM
#else
import qualified Data.HashMap.Strict as HM
import qualified Data.HashSet as HS
import qualified Data.Map as M
#endif
import qualified Data.Aeson as JSON
import qualified Data.Aeson.Types as JSON
import Data.Set (Set)
import qualified Data.Set as S
import Data.Text (Text)

#if MIN_VERSION_aeson(2,0,0)
toKey :: Text -> Key
toKey = K.fromText
#else
toKey :: Text -> Text
toKey = id
#endif

#if MIN_VERSION_aeson(2,0,0)
fromKey :: Key -> Text
fromKey = K.toText
#else
fromKey :: Text -> Text
fromKey = id
#endif

#if MIN_VERSION_aeson(2,0,0)
lookupKey :: Key -> KeyMap v -> Maybe v
lookupKey = KM.lookup
#else
lookupKey :: Text -> HashMap Text v -> Maybe v
lookupKey = HM.lookup
#endif

#if MIN_VERSION_aeson(2,0,0)
insert :: Key -> v -> KeyMap v -> KeyMap v
insert = KM.insert
#else
insert :: Text -> v -> HashMap Text v -> HashMap Text v
insert = HM.insert
#endif

#if MIN_VERSION_aeson(2,0,0)
fromList :: [(Key, v)] -> KeyMap v
fromList = KM.fromList
#else
fromList :: [(Text, v)] -> HashMap Text v
fromList = HM.fromList
#endif

#if MIN_VERSION_aeson(2,0,0)
toList :: KeyMap v -> [(Key, v)]
toList = KM.toList
#else
toList :: HashMap Text v -> [(Text, v)]
toList = HM.toList
#endif

#if MIN_VERSION_aeson(2,0,0)
keysSet :: KeyMap v -> Set Key
keysSet = S.fromList .  KM.keys
#else
keysSet :: HashHashMap Text v -> Set Key
keysSet = S.fromList . HS.toList . HM.keysSet
#endif

#if MIN_VERSION_aeson(2,0,0)
map :: (v1 -> v2) -> KeyMap v1 -> KeyMap v2
map = KM.map
#else
map :: (v1 -> v2) -> HashMap Text v1 -> HashMap Text v2
map = HM.map
#endif

liftToJSON ::
  (JSON.ToJSON1 f) =>
  (a -> JSON.Value) ->
  ([a] -> JSON.Value) ->
  f a ->
  JSON.Value
#if MIN_VERSION_aeson(2,2,0)
liftToJSON = JSON.liftToJSON (const False)
#else
liftToJSON = JSON.liftToJSON
#endif

liftToEncoding ::
  (JSON.ToJSON1 f) =>
  (a -> JSON.Encoding) ->
  ([a] -> JSON.Encoding) ->
  f a ->
  JSON.Encoding
#if MIN_VERSION_aeson(2,2,0)
liftToEncoding = JSON.liftToEncoding (const False)
#else
liftToEncoding = JSON.liftToEncoding
#endif

liftParseJSON ::
  (JSON.FromJSON1 f) =>
  (JSON.Value -> JSON.Parser a) ->
  (JSON.Value -> JSON.Parser [a]) ->
  JSON.Value ->
  JSON.Parser (f a)
#if MIN_VERSION_aeson(2,2,0)
liftParseJSON = JSON.liftParseJSON Nothing
#else
liftParseJSON = JSON.liftParseJSON
#endif