Bluefin demand analyser

bluefin.hs

{-# LANGUAGE GHC2021 #-}

import Bluefin.Eff
import Bluefin.Exception
import Bluefin.IO
import Control.Exception hiding (Exception, catch, throw)

{-# NOINLINE f #-}
f :: Int -> Int -> IO Int
f x y
  | x > 0 = throwIO (userError "What")
  | y > 0 = return 1
  | otherwise = return 2

{-# NOINLINE fEffIO #-}
fEffIO :: (e :> es) => IOE e -> Int -> Int -> Eff es Int
fEffIO io x y
  | x > 0 = effIO io (throwIO (userError "What"))
  | y > 0 = return 1
  | otherwise = return 2

{-# NOINLINE fEffException #-}
fEffException :: (e :> es) => Exception String e -> Int -> Int -> Eff es Int
fEffException ex x y
  | x > 0 = throw ex "What"
  | y > 0 = return 1
  | otherwise = return 2

-- ghci> mainIO
-- *** Exception: user error (What)
mainIO = f 2 undefined >>= print

-- ghci> mainEffIO
-- *** Exception: user error (What)
mainEffIO = runEff_ (\io -> fEffIO io 2 undefined) >>= print

-- ghci> mainEffException
-- *** Exception: user error (What)
mainEffException =
  runEff_
    ( \io ->
        catch
          (\ex -> fEffException ex 2 undefined)
          (effIO io . throwIO . userError)
    )
    >>= print