madlang-2.2.0.1: src/Text/Madlibs/Cata/Run.hs

-- | Module containing functions to get `Text` from `RandTok`
module Text.Madlibs.Cata.Run where

import Text.Madlibs.Internal.Types
import Text.Madlibs.Internal.Utils
import qualified Data.Text as T
import Control.Monad.Random.Class

-- | Generate randomized text from a `RandTok`
run :: (MonadRandom m) => RandTok -> m T.Text
run tok@(List rs) = do
    value <- getRandomR (0,1) --(withSystemRandom . asGenST $ \gen -> uniform gen)
    let ret = ((snd . head) . filter ((>=value) . fst)) $ mkCdf tok
    case ret of
        (Value txt) -> pure txt
        tok@(List rs) -> run tok
run (Value txt) = pure txt

-- | Helper function to compute the cdf when we have a pdf
mkCdf :: RandTok -> [(Double, RandTok)]
mkCdf (List rs) = zip (cdf . (map fst) $ rs) (map snd rs)