pub use middle::typing::ast::*;
use back::compiler::ExprCompilerFn;
use back::context::Context;
use syn::parse_quote;
pub struct Continuation
{
success: syn::Expr,
failure: syn::Expr
}
impl Continuation
{
pub fn new(success: syn::Expr, failure: syn::Expr) -> Self {
Continuation {
success: success,
failure: failure
}
}
pub fn compile_success(self, context: &mut Context,
compiler: ExprCompilerFn, idx: usize) -> Self
{
self.map_success(|success, failure|
context.compile_success(compiler, idx, success, failure))
}
pub fn compile_and_wrap(&self, context: &mut Context,
compiler: ExprCompilerFn, idx: usize, before_success: syn::Stmt) -> syn::Expr
{
let success = self.success.clone();
let s: syn::Expr = parse_quote!({#before_success #success});
context.compile_success(compiler, idx,
s,
self.failure.clone())
}
pub fn map_success<F>(mut self, f: F) -> Self where
F: FnOnce(syn::Expr, syn::Expr) -> syn::Expr
{
self.success = f(self.success, self.failure.clone());
self
}
pub fn unwrap_success(self) -> syn::Expr {
self.success
}
pub fn unwrap(self) -> (syn::Expr, syn::Expr) {
(self.success, self.failure)
}
}