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};
#[block]
#[derive(BlockProps, Debug)]
#[category = "collections"]
pub struct Keys {
#[input(name = "input", kind = "Dict")]
pub input: InputImpl,
#[output(kind = "List")]
pub out: OutputImpl,
}
impl Block for Keys {
async fn execute(&mut self) {
self.read_inputs_until_ready().await;
match self.input.get_value() {
Some(Value::Dict(dict)) => self.out.set(
dict.keys()
.map(|k| k.as_str().into())
.collect::<Vec<_>>()
.into(),
),
_ => self.out.set(Vec::default().into()),
}
}
}
#[cfg(test)]
mod test {
use libhaystack::dict;
use libhaystack::val::Dict;
use libhaystack::val::Value;
use crate::{
base::block::Block, base::block::test_utils::write_block_inputs, blocks::collections::Keys,
};
#[tokio::test]
async fn test_keys_block() {
let mut block = Keys::new();
write_block_inputs([(&mut block.input, Dict::default())]).await;
block.execute().await;
assert_eq!(block.out.value, vec![].into());
write_block_inputs([(&mut block.input, dict! {"a" => 1})]).await;
block.execute().await;
assert_eq!(block.out.value, vec!["a".into()].into());
}
}