rust_to_dtr 0.16.0

Convert Rust Soroban contracts to DTR format.
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use crate::common::compilation_state;
use crate::instruction::Instruction;
use crate::translate::expression::parse_expression;
use crate::{
    errors::not_translatable_error::NotTranslatableError, //, translate::expression::parse_expression,
};
use syn::ExprUnary;

pub fn handle_unary_expression(
    expr: &ExprUnary,
    compilation_state: &mut compilation_state::CompilationState,
) -> Result<Vec<Instruction>, NotTranslatableError> {
    let global_uuid = compilation_state.get_global_uuid();

    let unary_arg_name = format!("UNARY_ARGUMENT_{}", global_uuid);

    let original_assignment = compilation_state.next_assignment.clone();
    let mut preceding_instructions = parse_expression(
        &expr.expr,
        &mut compilation_state.with_assignment(Some(unary_arg_name.to_string())),
    )?;
    compilation_state.with_assignment(original_assignment);

    preceding_instructions.push(Instruction::from_compilation_state(
        "evaluate".to_string(),
        vec![determine_unary_operation(&expr.op), unary_arg_name],
        compilation_state,
    ));

    Ok(preceding_instructions)
}

fn determine_unary_operation(op: &syn::UnOp) -> String {
    match op {
        syn::UnOp::Not(_) => "!".to_string(),
        syn::UnOp::Neg(_) => "-".to_string(),
        _ => panic!("Unsupported unary operation"),
    }
}

#[cfg(test)]
mod tests {
    use crate::common::compilation_state::CompilationState;
    use crate::instruction::Instruction;
    use crate::translate::expression::unary_expression::handle_unary_expression;
    use syn::parse_quote;

    #[test]
    fn test_handle_unary_negation_expression() {
        let mut compilation_state = CompilationState::new();
        let expr = parse_quote! { !a };
        let instructions = handle_unary_expression(&expr, &mut compilation_state).unwrap();
        assert_eq!(
            instructions,
            vec![
                Instruction::new(
                    1,
                    "assign".to_string(),
                    vec!["a".to_string()],
                    "UNARY_ARGUMENT_0".to_string(),
                    0
                ),
                Instruction::new(
                    2,
                    "evaluate".to_string(),
                    vec!["!".to_string(), "UNARY_ARGUMENT_0".to_string()],
                    "".to_string(),
                    0
                ),
            ]
        );
    }

    #[test]
    fn test_handle_unary_minus_expression() {
        let mut compilation_state = CompilationState::new();
        let expr = parse_quote! { -a };
        let instructions = handle_unary_expression(&expr, &mut compilation_state).unwrap();
        assert_eq!(
            instructions,
            vec![
                Instruction::new(
                    1,
                    "assign".to_string(),
                    vec!["a".to_string()],
                    "UNARY_ARGUMENT_0".to_string(),
                    0
                ),
                Instruction::new(
                    2,
                    "evaluate".to_string(),
                    vec!["-".to_string(), "UNARY_ARGUMENT_0".to_string()],
                    "".to_string(),
                    0
                ),
            ]
        );
    }
}