module LaunchDarkly.Server.Integrations.TestData.FlagBuilder
( FlagBuilder(..)
, UserKey
, VariationIndex
, newFlagBuilder
, booleanFlag
, on
, fallthroughVariation
, offVariation
, variationForAllUsers
, valueForAllUsers
, variationForUser
, variations
, buildFlag
, UserAttribute
, ifMatch
, ifNotMatch
, FlagRuleBuilder
, andMatch
, andNotMatch
, thenReturn
, Variation
)
where
import qualified Data.Aeson as Aeson
import Data.Map.Strict (Map)
import qualified Data.Map.Strict as Map
import Data.Text (Text)
import qualified Data.Text as T
import GHC.Natural (Natural)
import qualified LaunchDarkly.Server.Features as F
import qualified LaunchDarkly.Server.Operators as Op
import Data.Function ((&))
type UserKey = Text
type VariationIndex = Integer
trueVariationForBoolean, falseVariationForBoolean :: VariationIndex
trueVariationForBoolean :: Integer
trueVariationForBoolean = Integer
0
falseVariationForBoolean :: Integer
falseVariationForBoolean = Integer
1
variationForBoolean :: Bool -> VariationIndex
variationForBoolean :: Bool -> Integer
variationForBoolean Bool
True = Integer
trueVariationForBoolean
variationForBoolean Bool
False = Integer
falseVariationForBoolean
data FlagBuilder = FlagBuilder
{ FlagBuilder -> UserKey
fbKey :: Text
, FlagBuilder -> Maybe Integer
fbOffVariation :: Maybe VariationIndex
, FlagBuilder -> Bool
fbOn :: Bool
, FlagBuilder -> Maybe Integer
fbFallthroughVariation :: Maybe VariationIndex
, FlagBuilder -> [Value]
fbVariations :: [Aeson.Value]
, FlagBuilder -> Map UserKey Integer
fbTargetMap :: Map UserKey VariationIndex
, FlagBuilder -> [FlagRule]
fbRules :: [FlagRule]
} deriving (Int -> FlagBuilder -> ShowS
[FlagBuilder] -> ShowS
FlagBuilder -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [FlagBuilder] -> ShowS
$cshowList :: [FlagBuilder] -> ShowS
show :: FlagBuilder -> String
$cshow :: FlagBuilder -> String
showsPrec :: Int -> FlagBuilder -> ShowS
$cshowsPrec :: Int -> FlagBuilder -> ShowS
Show)
fbTargets :: FlagBuilder -> [F.Target]
fbTargets :: FlagBuilder -> [Target]
fbTargets FlagBuilder
flagBuilder =
forall k a. Map k a -> [a]
Map.elems forall a b. (a -> b) -> a -> b
$
forall k a b. (k -> a -> b) -> Map k a -> Map k b
Map.mapWithKey (forall a b c. (a -> b -> c) -> b -> a -> c
flip [UserKey] -> Integer -> Target
F.Target) forall a b. (a -> b) -> a -> b
$
forall k a b. (k -> a -> b -> b) -> b -> Map k a -> b
Map.foldrWithKey forall {k} {a}. Ord k => a -> k -> Map k [a] -> Map k [a]
go forall a. Monoid a => a
mempty (FlagBuilder -> Map UserKey Integer
fbTargetMap FlagBuilder
flagBuilder)
where
go :: a -> k -> Map k [a] -> Map k [a]
go a
userKey k
variation =
forall k a. Ord k => (a -> a -> a) -> k -> a -> Map k a -> Map k a
Map.insertWith forall a. Semigroup a => a -> a -> a
(<>) k
variation [a
userKey]
buildFlag :: Natural -> FlagBuilder -> F.Flag
buildFlag :: Natural -> FlagBuilder -> Flag
buildFlag Natural
version FlagBuilder
flagBuilder =
F.Flag
{ $sel:key:Flag :: UserKey
F.key = FlagBuilder -> UserKey
fbKey FlagBuilder
flagBuilder
, $sel:version:Flag :: Natural
F.version = Natural
version
, $sel:on:Flag :: Bool
F.on = FlagBuilder -> Bool
fbOn FlagBuilder
flagBuilder
, $sel:trackEvents:Flag :: Bool
F.trackEvents = Bool
False
, $sel:trackEventsFallthrough:Flag :: Bool
F.trackEventsFallthrough = Bool
False
, $sel:deleted:Flag :: Bool
F.deleted = Bool
False
, $sel:prerequisites:Flag :: [Prerequisite]
F.prerequisites = []
, $sel:salt:Flag :: UserKey
F.salt = UserKey
"salt"
, $sel:targets:Flag :: [Target]
F.targets = FlagBuilder -> [Target]
fbTargets FlagBuilder
flagBuilder
, $sel:rules:Flag :: [Rule]
F.rules = forall num a b. Integral num => (num -> a -> b) -> [a] -> [b]
mapWithIndex Integer -> FlagRule -> Rule
convertFlagRule (FlagBuilder -> [FlagRule]
fbRules FlagBuilder
flagBuilder)
, $sel:fallthrough:Flag :: VariationOrRollout
F.fallthrough = Maybe Integer -> Maybe Rollout -> VariationOrRollout
F.VariationOrRollout (FlagBuilder -> Maybe Integer
fbFallthroughVariation FlagBuilder
flagBuilder) forall a. Maybe a
Nothing
, $sel:offVariation:Flag :: Maybe Integer
F.offVariation = FlagBuilder -> Maybe Integer
fbOffVariation FlagBuilder
flagBuilder
, $sel:variations:Flag :: [Value]
F.variations = FlagBuilder -> [Value]
fbVariations FlagBuilder
flagBuilder
, $sel:debugEventsUntilDate:Flag :: Maybe Natural
F.debugEventsUntilDate = forall a. Maybe a
Nothing
, $sel:clientSideAvailability:Flag :: ClientSideAvailability
F.clientSideAvailability = Bool -> Bool -> Bool -> ClientSideAvailability
F.ClientSideAvailability Bool
False Bool
False Bool
False
}
mapWithIndex :: Integral num => (num -> a -> b) -> [a] -> [b]
mapWithIndex :: forall num a b. Integral num => (num -> a -> b) -> [a] -> [b]
mapWithIndex num -> a -> b
f [a]
l =
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry num -> a -> b
f) (forall a b. [a] -> [b] -> [(a, b)]
zip [num
0..] [a]
l)
newFlagBuilder :: Text -> FlagBuilder
newFlagBuilder :: UserKey -> FlagBuilder
newFlagBuilder UserKey
key =
FlagBuilder
{ $sel:fbKey:FlagBuilder :: UserKey
fbKey = UserKey
key
, $sel:fbOffVariation:FlagBuilder :: Maybe Integer
fbOffVariation = forall a. Maybe a
Nothing
, $sel:fbOn:FlagBuilder :: Bool
fbOn = Bool
True
, $sel:fbFallthroughVariation:FlagBuilder :: Maybe Integer
fbFallthroughVariation = forall a. Maybe a
Nothing
, $sel:fbVariations:FlagBuilder :: [Value]
fbVariations = forall a. Monoid a => a
mempty
, $sel:fbTargetMap:FlagBuilder :: Map UserKey Integer
fbTargetMap = forall a. Monoid a => a
mempty
, $sel:fbRules:FlagBuilder :: [FlagRule]
fbRules = forall a. Monoid a => a
mempty
}
booleanFlagVariations :: [Aeson.Value]
booleanFlagVariations :: [Value]
booleanFlagVariations = [Bool -> Value
Aeson.Bool Bool
True, Bool -> Value
Aeson.Bool Bool
False]
isBooleanFlag :: FlagBuilder -> Bool
isBooleanFlag :: FlagBuilder -> Bool
isBooleanFlag FlagBuilder
flagBuilder
| [Value]
booleanFlagVariations forall a. Eq a => a -> a -> Bool
== FlagBuilder -> [Value]
fbVariations FlagBuilder
flagBuilder = Bool
True
| Bool
otherwise = Bool
False
booleanFlag :: FlagBuilder -> FlagBuilder
booleanFlag :: FlagBuilder -> FlagBuilder
booleanFlag FlagBuilder
flagBuilder
| FlagBuilder -> Bool
isBooleanFlag FlagBuilder
flagBuilder =
FlagBuilder
flagBuilder
| Bool
otherwise =
FlagBuilder
flagBuilder
forall a b. a -> (a -> b) -> b
& [Value] -> FlagBuilder -> FlagBuilder
variations [Value]
booleanFlagVariations
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
fallthroughVariation Integer
trueVariationForBoolean
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
offVariation Integer
falseVariationForBoolean
on :: Bool
-> FlagBuilder
-> FlagBuilder
on :: Bool -> FlagBuilder -> FlagBuilder
on Bool
isOn FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbOn:FlagBuilder :: Bool
fbOn = Bool
isOn }
clearRules :: FlagBuilder -> FlagBuilder
clearRules :: FlagBuilder -> FlagBuilder
clearRules FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbRules:FlagBuilder :: [FlagRule]
fbRules = forall a. Monoid a => a
mempty }
clearUserTargets :: FlagBuilder -> FlagBuilder
clearUserTargets :: FlagBuilder -> FlagBuilder
clearUserTargets FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbTargetMap:FlagBuilder :: Map UserKey Integer
fbTargetMap = forall a. Monoid a => a
mempty }
valueForAllUsers :: Aeson.ToJSON value
=> value
-> FlagBuilder
-> FlagBuilder
valueForAllUsers :: forall value. ToJSON value => value -> FlagBuilder -> FlagBuilder
valueForAllUsers value
val FlagBuilder
fb =
FlagBuilder
fb forall a b. a -> (a -> b) -> b
& [Value] -> FlagBuilder -> FlagBuilder
variations [forall a. ToJSON a => a -> Value
Aeson.toJSON value
val]
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
variationForAllUsers (Integer
0 :: VariationIndex)
variations :: [Aeson.Value]
-> FlagBuilder
-> FlagBuilder
variations :: [Value] -> FlagBuilder -> FlagBuilder
variations [Value]
values FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbVariations:FlagBuilder :: [Value]
fbVariations = [Value]
values }
class Variation val where
fallthroughVariation :: val
-> FlagBuilder
-> FlagBuilder
offVariation :: val
-> FlagBuilder
-> FlagBuilder
variationForAllUsers :: val
-> FlagBuilder
-> FlagBuilder
variationForUser :: UserKey
-> val
-> FlagBuilder
-> FlagBuilder
thenReturn :: val
-> FlagRuleBuilder
-> FlagBuilder
instance Variation Integer where
fallthroughVariation :: Integer -> FlagBuilder -> FlagBuilder
fallthroughVariation Integer
variationIndex FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbFallthroughVariation:FlagBuilder :: Maybe Integer
fbFallthroughVariation = forall a. a -> Maybe a
Just Integer
variationIndex }
offVariation :: Integer -> FlagBuilder -> FlagBuilder
offVariation Integer
variationIndex FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbOffVariation:FlagBuilder :: Maybe Integer
fbOffVariation = forall a. a -> Maybe a
Just Integer
variationIndex }
variationForAllUsers :: Integer -> FlagBuilder -> FlagBuilder
variationForAllUsers Integer
variationIndex FlagBuilder
fb =
FlagBuilder
fb forall a b. a -> (a -> b) -> b
& Bool -> FlagBuilder -> FlagBuilder
on Bool
True
forall a b. a -> (a -> b) -> b
& FlagBuilder -> FlagBuilder
clearRules
forall a b. a -> (a -> b) -> b
& FlagBuilder -> FlagBuilder
clearUserTargets
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
fallthroughVariation Integer
variationIndex
variationForUser :: UserKey -> Integer -> FlagBuilder -> FlagBuilder
variationForUser UserKey
userKey Integer
variationIndex FlagBuilder
fb =
FlagBuilder
fb{ $sel:fbTargetMap:FlagBuilder :: Map UserKey Integer
fbTargetMap = forall k a. Ord k => k -> a -> Map k a -> Map k a
Map.insert UserKey
userKey Integer
variationIndex (FlagBuilder -> Map UserKey Integer
fbTargetMap FlagBuilder
fb) }
thenReturn :: Integer -> FlagRuleBuilder -> FlagBuilder
thenReturn Integer
variationIndex FlagRuleBuilder
ruleBuilder =
let fb :: FlagBuilder
fb = FlagRuleBuilder -> FlagBuilder
frbBaseBuilder FlagRuleBuilder
ruleBuilder
in FlagBuilder
fb{ $sel:fbRules:FlagBuilder :: [FlagRule]
fbRules = [Clause] -> Integer -> FlagRule
FlagRule (FlagRuleBuilder -> [Clause]
frbClauses FlagRuleBuilder
ruleBuilder) Integer
variationIndex forall a. a -> [a] -> [a]
: FlagBuilder -> [FlagRule]
fbRules FlagBuilder
fb }
instance Variation Bool where
fallthroughVariation :: Bool -> FlagBuilder -> FlagBuilder
fallthroughVariation Bool
value FlagBuilder
fb =
FlagBuilder
fb forall a b. a -> (a -> b) -> b
& FlagBuilder -> FlagBuilder
booleanFlag
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
fallthroughVariation (Bool -> Integer
variationForBoolean Bool
value)
offVariation :: Bool -> FlagBuilder -> FlagBuilder
offVariation Bool
value FlagBuilder
fb =
FlagBuilder
fb forall a b. a -> (a -> b) -> b
& FlagBuilder -> FlagBuilder
booleanFlag
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
offVariation (Bool -> Integer
variationForBoolean Bool
value)
variationForAllUsers :: Bool -> FlagBuilder -> FlagBuilder
variationForAllUsers Bool
value FlagBuilder
fb =
FlagBuilder
fb forall a b. a -> (a -> b) -> b
& FlagBuilder -> FlagBuilder
booleanFlag
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagBuilder -> FlagBuilder
variationForAllUsers (Bool -> Integer
variationForBoolean Bool
value)
variationForUser :: UserKey -> Bool -> FlagBuilder -> FlagBuilder
variationForUser UserKey
userKey Bool
value FlagBuilder
fb =
FlagBuilder
fb forall a b. a -> (a -> b) -> b
& FlagBuilder -> FlagBuilder
booleanFlag
forall a b. a -> (a -> b) -> b
& forall val.
Variation val =>
UserKey -> val -> FlagBuilder -> FlagBuilder
variationForUser UserKey
userKey (Bool -> Integer
variationForBoolean Bool
value)
thenReturn :: Bool -> FlagRuleBuilder -> FlagBuilder
thenReturn Bool
value FlagRuleBuilder
ruleBuilder =
FlagRuleBuilder
ruleBuilder { $sel:frbBaseBuilder:FlagRuleBuilder :: FlagBuilder
frbBaseBuilder = FlagBuilder -> FlagBuilder
booleanFlag forall a b. (a -> b) -> a -> b
$ FlagRuleBuilder -> FlagBuilder
frbBaseBuilder FlagRuleBuilder
ruleBuilder }
forall a b. a -> (a -> b) -> b
& forall val. Variation val => val -> FlagRuleBuilder -> FlagBuilder
thenReturn (Bool -> Integer
variationForBoolean Bool
value)
type UserAttribute = Text
ifMatch :: UserAttribute
-> [Aeson.Value]
-> FlagBuilder
-> FlagRuleBuilder
ifMatch :: UserKey -> [Value] -> FlagBuilder -> FlagRuleBuilder
ifMatch UserKey
userAttribute [Value]
values FlagBuilder
fb =
FlagBuilder -> FlagRuleBuilder
newFlagRuleBuilder FlagBuilder
fb
forall a b. a -> (a -> b) -> b
& UserKey -> [Value] -> FlagRuleBuilder -> FlagRuleBuilder
andMatch UserKey
userAttribute [Value]
values
ifNotMatch :: UserAttribute
-> [Aeson.Value]
-> FlagBuilder
-> FlagRuleBuilder
ifNotMatch :: UserKey -> [Value] -> FlagBuilder -> FlagRuleBuilder
ifNotMatch UserKey
userAttribute [Value]
values FlagBuilder
fb =
FlagBuilder -> FlagRuleBuilder
newFlagRuleBuilder FlagBuilder
fb
forall a b. a -> (a -> b) -> b
& UserKey -> [Value] -> FlagRuleBuilder -> FlagRuleBuilder
andNotMatch UserKey
userAttribute [Value]
values
data Clause = Clause
{ Clause -> UserKey
clauseAttribute :: UserAttribute
, Clause -> [Value]
clauseValues :: [Aeson.Value]
, Clause -> Bool
clauseNegate :: Bool
} deriving (Int -> Clause -> ShowS
[Clause] -> ShowS
Clause -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [Clause] -> ShowS
$cshowList :: [Clause] -> ShowS
show :: Clause -> String
$cshow :: Clause -> String
showsPrec :: Int -> Clause -> ShowS
$cshowsPrec :: Int -> Clause -> ShowS
Show)
data FlagRule = FlagRule
{ FlagRule -> [Clause]
frClauses :: [Clause]
, FlagRule -> Integer
frVariation :: VariationIndex
} deriving (Int -> FlagRule -> ShowS
[FlagRule] -> ShowS
FlagRule -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [FlagRule] -> ShowS
$cshowList :: [FlagRule] -> ShowS
show :: FlagRule -> String
$cshow :: FlagRule -> String
showsPrec :: Int -> FlagRule -> ShowS
$cshowsPrec :: Int -> FlagRule -> ShowS
Show)
convertFlagRule :: Integer -> FlagRule -> F.Rule
convertFlagRule :: Integer -> FlagRule -> Rule
convertFlagRule Integer
idx FlagRule
flagRule =
F.Rule
{ $sel:id:Rule :: UserKey
F.id = String -> UserKey
T.pack forall a b. (a -> b) -> a -> b
$ String
"rule" forall a. Semigroup a => a -> a -> a
<> forall a. Show a => a -> String
show Integer
idx
, $sel:variationOrRollout:Rule :: VariationOrRollout
F.variationOrRollout = Maybe Integer -> Maybe Rollout -> VariationOrRollout
F.VariationOrRollout (forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ FlagRule -> Integer
frVariation FlagRule
flagRule) forall a. Maybe a
Nothing
, $sel:clauses:Rule :: [Clause]
F.clauses = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Clause -> Clause
convertClause (FlagRule -> [Clause]
frClauses FlagRule
flagRule)
, $sel:trackEvents:Rule :: Bool
F.trackEvents = Bool
False
}
convertClause :: Clause -> F.Clause
convertClause :: Clause -> Clause
convertClause Clause
clause =
F.Clause
{ $sel:attribute:Clause :: UserKey
F.attribute = Clause -> UserKey
clauseAttribute Clause
clause
, $sel:negate:Clause :: Bool
F.negate = Clause -> Bool
clauseNegate Clause
clause
, $sel:values:Clause :: [Value]
F.values = Clause -> [Value]
clauseValues Clause
clause
, $sel:op:Clause :: Op
F.op = Op
Op.OpIn
}
data FlagRuleBuilder = FlagRuleBuilder
{ FlagRuleBuilder -> [Clause]
frbClauses :: [Clause]
, FlagRuleBuilder -> FlagBuilder
frbBaseBuilder :: FlagBuilder
} deriving (Int -> FlagRuleBuilder -> ShowS
[FlagRuleBuilder] -> ShowS
FlagRuleBuilder -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [FlagRuleBuilder] -> ShowS
$cshowList :: [FlagRuleBuilder] -> ShowS
show :: FlagRuleBuilder -> String
$cshow :: FlagRuleBuilder -> String
showsPrec :: Int -> FlagRuleBuilder -> ShowS
$cshowsPrec :: Int -> FlagRuleBuilder -> ShowS
Show)
newFlagRuleBuilder :: FlagBuilder -> FlagRuleBuilder
newFlagRuleBuilder :: FlagBuilder -> FlagRuleBuilder
newFlagRuleBuilder FlagBuilder
baseBuilder =
FlagRuleBuilder
{ $sel:frbClauses:FlagRuleBuilder :: [Clause]
frbClauses = forall a. Monoid a => a
mempty
, $sel:frbBaseBuilder:FlagRuleBuilder :: FlagBuilder
frbBaseBuilder = FlagBuilder
baseBuilder
}
andMatch :: UserAttribute
-> [Aeson.Value]
-> FlagRuleBuilder
-> FlagRuleBuilder
andMatch :: UserKey -> [Value] -> FlagRuleBuilder -> FlagRuleBuilder
andMatch UserKey
userAttribute [Value]
values FlagRuleBuilder
ruleBuilder =
FlagRuleBuilder
ruleBuilder{ $sel:frbClauses:FlagRuleBuilder :: [Clause]
frbClauses = UserKey -> [Value] -> Bool -> Clause
Clause UserKey
userAttribute [Value]
values Bool
False forall a. a -> [a] -> [a]
: FlagRuleBuilder -> [Clause]
frbClauses FlagRuleBuilder
ruleBuilder }
andNotMatch :: UserAttribute
-> [Aeson.Value]
-> FlagRuleBuilder
-> FlagRuleBuilder
andNotMatch :: UserKey -> [Value] -> FlagRuleBuilder -> FlagRuleBuilder
andNotMatch UserKey
userAttribute [Value]
values FlagRuleBuilder
ruleBuilder =
FlagRuleBuilder
ruleBuilder{ $sel:frbClauses:FlagRuleBuilder :: [Clause]
frbClauses = UserKey -> [Value] -> Bool -> Clause
Clause UserKey
userAttribute [Value]
values Bool
True forall a. a -> [a] -> [a]
: FlagRuleBuilder -> [Clause]
frbClauses FlagRuleBuilder
ruleBuilder }