module AWS.Secrets.Reader
  ( -- * Type
    SecretReader (..),
    bind,

    -- * Examples
    text,
    string,
    scientific,
    boundedInteger,
  )
where

import AWS.Secrets.Key (SecretKey, getSecretKeyText)
import Control.Applicative (Applicative, pure)
import qualified Data.Aeson as JSON
import qualified Data.Aeson.Key as JSON.Key
import Data.Function (($), (.))
import Data.Functor (Functor, fmap)
import Data.Functor.Compose (Compose (..))
import qualified Data.Aeson.KeyMap as KeyMap
import Data.Maybe (Maybe (..), maybe)
import Data.Scientific (Scientific)
import qualified Data.Scientific as Scientific
import Data.Semigroup ((<>))
import Data.Sequence (Seq)
import Data.String (String)
import Data.Text (Text)
import qualified Data.Text as Text
import Data.Validation (Validation (..), bindValidation)
import Prelude (Bounded, Integral)

newtype SecretReader a = SecretReader
  { forall a. SecretReader a -> Object -> Validation (Seq Text) a
apply :: JSON.Object -> Validation (Seq Text) a
  }
  deriving
    (forall a b. a -> SecretReader b -> SecretReader a
forall a b. (a -> b) -> SecretReader a -> SecretReader b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: forall a b. a -> SecretReader b -> SecretReader a
$c<$ :: forall a b. a -> SecretReader b -> SecretReader a
fmap :: forall a b. (a -> b) -> SecretReader a -> SecretReader b
$cfmap :: forall a b. (a -> b) -> SecretReader a -> SecretReader b
Functor, Functor SecretReader
forall a. a -> SecretReader a
forall a b. SecretReader a -> SecretReader b -> SecretReader a
forall a b. SecretReader a -> SecretReader b -> SecretReader b
forall a b.
SecretReader (a -> b) -> SecretReader a -> SecretReader b
forall a b c.
(a -> b -> c) -> SecretReader a -> SecretReader b -> SecretReader c
forall (f :: * -> *).
Functor f
-> (forall a. a -> f a)
-> (forall a b. f (a -> b) -> f a -> f b)
-> (forall a b c. (a -> b -> c) -> f a -> f b -> f c)
-> (forall a b. f a -> f b -> f b)
-> (forall a b. f a -> f b -> f a)
-> Applicative f
<* :: forall a b. SecretReader a -> SecretReader b -> SecretReader a
$c<* :: forall a b. SecretReader a -> SecretReader b -> SecretReader a
*> :: forall a b. SecretReader a -> SecretReader b -> SecretReader b
$c*> :: forall a b. SecretReader a -> SecretReader b -> SecretReader b
liftA2 :: forall a b c.
(a -> b -> c) -> SecretReader a -> SecretReader b -> SecretReader c
$cliftA2 :: forall a b c.
(a -> b -> c) -> SecretReader a -> SecretReader b -> SecretReader c
<*> :: forall a b.
SecretReader (a -> b) -> SecretReader a -> SecretReader b
$c<*> :: forall a b.
SecretReader (a -> b) -> SecretReader a -> SecretReader b
pure :: forall a. a -> SecretReader a
$cpure :: forall a. a -> SecretReader a
Applicative)
    via Compose ((->) JSON.Object) (Validation (Seq Text))

bind :: SecretReader a -> (a -> SecretReader b) -> SecretReader b
bind :: forall a b.
SecretReader a -> (a -> SecretReader b) -> SecretReader b
bind (SecretReader Object -> Validation (Seq Text) a
f) a -> SecretReader b
g = forall a. (Object -> Validation (Seq Text) a) -> SecretReader a
SecretReader \Object
o ->
  Object -> Validation (Seq Text) a
f Object
o forall e a b.
Validation e a -> (a -> Validation e b) -> Validation e b
`bindValidation` ((forall a b. (a -> b) -> a -> b
$ Object
o) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. SecretReader a -> Object -> Validation (Seq Text) a
apply forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> SecretReader b
g)

---

text :: SecretKey -> SecretReader Text
text :: SecretKey -> SecretReader Text
text SecretKey
k = forall a. (Object -> Validation (Seq Text) a) -> SecretReader a
SecretReader \Object
o ->
  forall a. Seq Text -> Maybe a -> Validation (Seq Text) a
orFail [Text
"Missing: " forall a. Semigroup a => a -> a -> a
<> SecretKey -> Text
getSecretKeyText SecretKey
k] (forall v. Key -> KeyMap v -> Maybe v
KeyMap.lookup (SecretKey -> Key
key SecretKey
k) Object
o)
    forall e a b.
Validation e a -> (a -> Validation e b) -> Validation e b
`bindValidation` \Value
txt -> forall a. Seq Text -> Maybe a -> Validation (Seq Text) a
orFail [Text
"Not a string: " forall a. Semigroup a => a -> a -> a
<> SecretKey -> Text
getSecretKeyText SecretKey
k] (Value -> Maybe Text
castJsonString Value
txt)

string :: SecretKey -> SecretReader String
string :: SecretKey -> SecretReader String
string = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Text -> String
Text.unpack forall b c a. (b -> c) -> (a -> b) -> a -> c
. SecretKey -> SecretReader Text
text

scientific :: SecretKey -> SecretReader Scientific
scientific :: SecretKey -> SecretReader Scientific
scientific SecretKey
k = forall a. (Object -> Validation (Seq Text) a) -> SecretReader a
SecretReader \Object
o ->
  forall a. Seq Text -> Maybe a -> Validation (Seq Text) a
orFail [Text
"Missing: " forall a. Semigroup a => a -> a -> a
<> SecretKey -> Text
getSecretKeyText SecretKey
k] (forall v. Key -> KeyMap v -> Maybe v
KeyMap.lookup (SecretKey -> Key
key SecretKey
k) Object
o)
    forall e a b.
Validation e a -> (a -> Validation e b) -> Validation e b
`bindValidation` \Value
txt -> forall a. Seq Text -> Maybe a -> Validation (Seq Text) a
orFail [Text
"Not a string: " forall a. Semigroup a => a -> a -> a
<> SecretKey -> Text
getSecretKeyText SecretKey
k] (Value -> Maybe Scientific
castJsonNumber Value
txt)

boundedInteger :: (Integral i, Bounded i) => SecretKey -> SecretReader i
boundedInteger :: forall i. (Integral i, Bounded i) => SecretKey -> SecretReader i
boundedInteger SecretKey
k =
  SecretKey -> SecretReader Scientific
scientific SecretKey
k forall a b.
SecretReader a -> (a -> SecretReader b) -> SecretReader b
`bind` \Scientific
s -> forall a. (Object -> Validation (Seq Text) a) -> SecretReader a
SecretReader \Object
_ -> case forall i. (Integral i, Bounded i) => Scientific -> Maybe i
Scientific.toBoundedInteger Scientific
s of
    Maybe i
Nothing -> forall err a. err -> Validation err a
Failure [Text
"Not an integer or not within bounds: " forall a. Semigroup a => a -> a -> a
<> SecretKey -> Text
getSecretKeyText SecretKey
k]
    Just i
i -> forall (f :: * -> *) a. Applicative f => a -> f a
pure i
i

---

key :: SecretKey -> JSON.Key
key :: SecretKey -> Key
key = Text -> Key
JSON.Key.fromText forall b c a. (b -> c) -> (a -> b) -> a -> c
. SecretKey -> Text
getSecretKeyText

castJsonString :: JSON.Value -> Maybe Text
castJsonString :: Value -> Maybe Text
castJsonString = \case JSON.String Text
x -> forall a. a -> Maybe a
Just Text
x; Value
_ -> forall a. Maybe a
Nothing

castJsonNumber :: JSON.Value -> Maybe Scientific
castJsonNumber :: Value -> Maybe Scientific
castJsonNumber = \case JSON.Number Scientific
x -> forall a. a -> Maybe a
Just Scientific
x; Value
_ -> forall a. Maybe a
Nothing

orFail :: Seq Text -> Maybe a -> Validation (Seq Text) a
orFail :: forall a. Seq Text -> Maybe a -> Validation (Seq Text) a
orFail Seq Text
err = forall b a. b -> (a -> b) -> Maybe a -> b
maybe (forall err a. err -> Validation err a
Failure Seq Text
err) forall err a. a -> Validation err a
Success