mamba 0.3.6

A transpiler which converts Mamba files to Python 3 files
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::{ASTTy, Context};
use crate::check::ast::NodeTy;
use crate::generate::ast::node::Core;
use crate::generate::convert::common::convert_vec;
use crate::generate::convert::convert_node;
use crate::generate::convert::state::{Imports, State};
use crate::generate::result::{GenResult, UnimplementedErr};

pub fn convert_builder(ast: &ASTTy, imp: &mut Imports, state: &State, ctx: &Context) -> GenResult {
    match &ast.node {
        NodeTy::DictBuilder { from, to, conditions } => {
            let from = Box::from(convert_node(from, imp, state, ctx)?);
            let to = Box::from(convert_node(to, imp, state, ctx)?);

            if let Some(col) = conditions.first() {
                let conds = conditions.strip_prefix(&[col.clone()]).expect("Unreachable");
                let conds = convert_vec(conds, imp, state, ctx)?;
                let col = Box::from(convert_node(col, imp, state, ctx)?);
                Ok(Core::DictComprehension { from, to, col, conds })
            } else {
                Err(Box::from(UnimplementedErr::new(ast, "Cannot be empty")))
            }
        }
        NodeTy::ListBuilder { item, conditions } => {
            let expr = Box::from(convert_node(item, imp, state, ctx)?);

            if let Some(col) = conditions.first() {
                let conds = conditions.strip_prefix(&[col.clone()]).expect("Unreachable");
                let conds = convert_vec(conds, imp, state, ctx)?;
                let col = Box::from(convert_node(col, imp, state, ctx)?);
                Ok(Core::List { elements: vec![Core::Comprehension { expr, col, conds }] })
            } else {
                Err(Box::from(UnimplementedErr::new(ast, "Cannot be empty")))
            }
        }
        NodeTy::SetBuilder { item, conditions } => {
            let expr = Box::from(convert_node(item, imp, state, ctx)?);

            if let Some(col) = conditions.first() {
                let conds = conditions.strip_prefix(&[col.clone()]).expect("Unreachable");
                let conds = convert_vec(conds, imp, state, ctx)?;
                let col = Box::from(convert_node(col, imp, state, ctx)?);
                Ok(Core::Set { elements: vec![Core::Comprehension { expr, col, conds }] })
            } else {
                Err(Box::from(UnimplementedErr::new(ast, "Cannot be empty")))
            }
        }
        other => {
            let msg = format!("Expected call flow but was: {other:?}.");
            Err(Box::from(UnimplementedErr::new(ast, &msg)))
        }
    }
}