calrissian/include/monads.lfe

(defmacro monad (name)
  (case name
    ;; Provide the state monad in terms of the state transformer
    (''state `(transformer 'state 'identity))
    (_
     `(list_to_atom (lists:flatten (list "calrissian-"
                                         (atom_to_list ,name)
                                         "-monad"))))))

(defmacro transformer (name inner-monad)
  `(tuple (list_to_atom (lists:flatten (list "calrissian-"
                                             (atom_to_list ,name)
                                             "-transformer")))
          (monad ,inner-monad)))

(defmacro do-m args
  (let ((monad (car args))
        (statements (cdr args)))
    (calrissian-monad:do-transform monad statements)))

(defmacro >>= (monad m f)
  `(call ,monad '>>= ,m ,f))

(defmacro >> (monad m1 m2)
  `(call ,monad '>>= ,m1 (lambda (_) , m2)))

(defmacro return (monad expr)
  `(call ,monad 'return ,expr))

(defmacro fail (monad expr)
  `(call ,monad 'fail ,expr))

(defmacro sequence (monad list)
  `(: lists foldr
     (lambda (m acc) (mcons ,monad m acc))
     (return ,monad [])
     ,list))

(defmacro mcons (monad m mlist)
  `(do-m ,monad
         (x <- ,m)
         (rest <- ,mlist)
         (return ,monad (cons x rest))))