logic-mesh 1.0.0

Control logic engine using event based and reactive blocks written in Rust.
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

// Copyright (c) 2022-2023, Radu Racariu.

use uuid::Uuid;

use crate::base::{
    block::{Block, BlockDesc, BlockProps, BlockState},
    input::{InputProps, input_reader::InputReader},
    output::Output,
};

use libhaystack::val::{Value, kind::HaystackKind};

use crate::{blocks::InputImpl, blocks::OutputImpl};

/// Gets the element specified at key from the input and outputs the
/// element's value.
#[block]
#[derive(BlockProps, Debug)]
#[category = "collections"]
pub struct GetElement {
    #[input(name = "input", kind = "Null")]
    pub input: InputImpl,
    #[input(name = "key", kind = "Null")]
    pub key: InputImpl,
    #[output(kind = "Null")]
    pub out: OutputImpl,
}

impl Block for GetElement {
    async fn execute(&mut self) {
        self.read_inputs_until_ready().await;

        if let Some(value) = match (self.input.get_value(), self.key.get_value()) {
            (Some(Value::Dict(dict)), Some(Value::Str(key))) => dict.get(key.as_str()),
            (Some(Value::List(list)), Some(Value::Number(index))) => list.get(index.value as usize),
            _ => None,
        } {
            self.out.set(value.clone());
        } else {
            self.out.set(Value::Null);
        }
    }
}

#[cfg(test)]
mod test {
    use libhaystack::val::Dict;
    use libhaystack::{dict, val::Value};

    use crate::{
        base::block::Block, base::block::test_utils::write_block_inputs,
        blocks::collections::GetElement,
    };

    #[tokio::test]
    async fn test_get_element_block() {
        let mut block = GetElement::new();

        write_block_inputs([
            (&mut block.input, Value::Dict(dict! {"a" => 1})),
            (&mut block.key, "a".into()),
        ])
        .await;

        block.execute().await;
        assert_eq!(block.out.value, 1.into());

        write_block_inputs([(&mut block.key, "x")]).await;
        block.execute().await;
        assert_eq!(block.out.value, Value::Null);

        write_block_inputs([(&mut block.input, ""), (&mut block.key, "a")]).await;

        block.execute().await;
        assert_eq!(block.out.value, Value::Null);

        write_block_inputs([
            (&mut block.input, Value::List(vec![1.into(), 2.into()])),
            (&mut block.key, 1.into()),
        ])
        .await;

        block.execute().await;
        assert_eq!(block.out.value, 2.into());

        write_block_inputs([(&mut block.key, 100)]).await;
        block.execute().await;
        assert_eq!(block.out.value, Value::Null);
    }
}