cog_task/action/core/
template.rs

1use crate::action::{Action, StatefulAction};
2use crate::comm::QWriter;
3use crate::resource::{IoManager, ResourceManager};
4use crate::server::{AsyncSignal, Config, SyncSignal, ROOT_DIR};
5use eyre::{eyre, Context, Result};
6use serde::{Deserialize, Serialize};
7use std::collections::BTreeMap;
8use std::fs;
9use std::path::PathBuf;
10
11#[derive(Debug, Deserialize, Serialize)]
12#[serde(deny_unknown_fields)]
13pub struct Template {
14    src: PathBuf,
15    #[serde(default)]
16    params: BTreeMap<String, String>,
17}
18
19impl Action for Template {
20    fn init(self) -> Result<Box<dyn Action>> {
21        let path = ROOT_DIR.get().unwrap().join(&self.src);
22        let mut inner = fs::read_to_string(&path)
23            .wrap_err_with(|| format!("Failed to read `Template` source: {path:?}"))?;
24
25        for (k, v) in self.params.iter() {
26            let re = regex::Regex::new(&format!(r"\$\{{{k}\}}")).unwrap();
27            inner = re.replace_all(&inner, v).to_string();
28        }
29
30        ron::from_str::<Box<dyn Action>>(&inner).wrap_err("Failed to deserialize `Template`.")
31    }
32
33    #[inline]
34    fn stateful(
35        &self,
36        _io: &IoManager,
37        _res: &ResourceManager,
38        _config: &Config,
39        _sync_writer: &QWriter<SyncSignal>,
40        _async_writer: &QWriter<AsyncSignal>,
41    ) -> Result<Box<dyn StatefulAction>> {
42        Err(eyre!("Template can not be stateful."))
43    }
44}