freer-effects-0.3.0.0: examples/src/NonDet.hs

{-# LANGUAGE FlexibleContexts #-}
module NonDet where

import Control.Applicative
import Control.Monad
import Control.Monad.Freer

ifte :: Member NonDet r
     => Eff r a -> (a -> Eff r b) -> Eff r b -> Eff r b
ifte t th el = (t >>= th) <|> el

testIfte :: Member NonDet r => Eff r Int
testIfte = do
  n <- gen
  ifte (do d <- gen
           guard $ d < n && n `mod` d == 0)
       (const mzero)
       (return n)
  where gen = msum . fmap return $ [2..30]

testIfteRun :: [Int]
testIfteRun = run . makeChoiceA $ testIfte