fiddler 4.9.1

Data Stream processor 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

use crate::config::register_plugin;
use crate::config::ItemType;
use crate::config::{ConfigSpec, ExecutionType};
use crate::CallbackChan;
use crate::Message;
use crate::{Closer, Error, Input};
use async_std::io;
use async_trait::async_trait;
use fiddler_macros::fiddler_registration_func;
use serde_yaml::Value;

/// Command to exit stdin input mode
const STDIN_EXIT_COMMAND: &str = "exit()";

pub struct StdIn {}

#[async_trait]
impl Input for StdIn {
    async fn read(&mut self) -> Result<(Message, Option<CallbackChan>), Error> {
        let mut buffer = String::new();
        let stdin = io::stdin();
        let _ = stdin
            .read_line(&mut buffer)
            .await
            .map_err(|_| Error::EndOfInput)?;

        // remove new line character
        let _ = buffer.pop();

        if buffer == STDIN_EXIT_COMMAND {
            return Err(Error::EndOfInput);
        };

        Ok((
            Message {
                bytes: buffer.into_bytes(),
                ..Default::default()
            },
            None,
        ))
    }
}

impl Closer for StdIn {}

#[fiddler_registration_func]
fn create_stdin(_conf: Value) -> Result<ExecutionType, Error> {
    Ok(ExecutionType::Input(Box::new(StdIn {})))
}

pub(super) fn register_stdin() -> Result<(), Error> {
    let config = "type: object";
    let conf_spec = ConfigSpec::from_schema(config)?;

    register_plugin("stdin".into(), ItemType::Input, conf_spec, create_stdin)
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn register_plugin() {
        register_stdin().unwrap()
    }
}