cog-task 1.2.0

A general-purpose low-latency application to run cognitive tasks
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

use crate::action::{Action, StatefulAction};
use crate::comm::QWriter;
use crate::resource::{IoManager, ResourceManager};
use crate::server::{AsyncSignal, Config, SyncSignal, ROOT_DIR};
use eyre::{eyre, Context, Result};
use serde::{Deserialize, Serialize};
use std::collections::BTreeMap;
use std::fs;
use std::path::PathBuf;

#[derive(Debug, Deserialize, Serialize)]
#[serde(deny_unknown_fields)]
pub struct Template {
    src: PathBuf,
    #[serde(default)]
    params: BTreeMap<String, String>,
}

impl Action for Template {
    fn init(self) -> Result<Box<dyn Action>> {
        let path = ROOT_DIR.get().unwrap().join(&self.src);
        let mut inner = fs::read_to_string(&path)
            .wrap_err_with(|| format!("Failed to read `Template` source: {path:?}"))?;

        for (k, v) in self.params.iter() {
            let re = regex::Regex::new(&format!(r"\$\{{{k}\}}")).unwrap();
            inner = re.replace_all(&inner, v).to_string();
        }

        ron::from_str::<Box<dyn Action>>(&inner).wrap_err("Failed to deserialize `Template`.")
    }

    #[inline]
    fn stateful(
        &self,
        _io: &IoManager,
        _res: &ResourceManager,
        _config: &Config,
        _sync_writer: &QWriter<SyncSignal>,
        _async_writer: &QWriter<AsyncSignal>,
    ) -> Result<Box<dyn StatefulAction>> {
        Err(eyre!("Template can not be stateful."))
    }
}