Organize Weigh measures about weigh HOT 5 CLOSED

data ToWeight = Simple String Action | Group String [ToWeight]

-- | Weigh specification monad.
newtype Weigh a =
  Weigh {runWeigh :: State (Config, [ToWeight]) a}
  deriving (Monad,Functor,Applicative)

[...]

-- | Write out an action.
tellAction :: String -> Action -> Weigh ()
tellAction name action = Weigh $ modify $ second $ \x -> x ++ [Simple name action]

wgroup :: String -> Weigh () -> Weigh ()
wgroup str wei = do
  start <- Weigh get
  let startL = length $ snd start
  wei
  Weigh $ modify $ second $ \x -> take startL x ++ [Group str $ drop startL x]

mainWith $ do
  wgroup "fst" $ do
    func "1" f 1
    func "2" f 2
  wgroup "snd" $ do
    func "3" f 3