mk_lib/schema/
mod.rs

1mod command;
2mod include;
3mod plan;
4mod precondition;
5mod shell;
6mod task;
7mod task_context;
8mod task_dependency;
9mod task_root;
10mod use_cargo;
11mod use_npm;
12mod validation;
13
14use std::collections::HashSet;
15use std::fmt;
16use std::process::Stdio;
17use std::sync::{
18  Arc,
19  Mutex,
20};
21
22use once_cell::sync::Lazy;
23use regex::Regex;
24
25pub type ActiveTasks = Arc<Mutex<HashSet<String>>>;
26pub type CompletedTasks = Arc<Mutex<HashSet<String>>>;
27
28#[derive(Debug)]
29pub struct ExecutionInterrupted;
30
31impl fmt::Display for ExecutionInterrupted {
32  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
33    write!(f, "Execution interrupted")
34  }
35}
36
37impl std::error::Error for ExecutionInterrupted {}
38
39pub use command::*;
40pub use include::*;
41pub use plan::*;
42pub use precondition::*;
43pub use shell::*;
44pub use task::*;
45pub use task_context::*;
46pub use task_dependency::*;
47pub use task_root::*;
48pub use use_cargo::*;
49pub use use_npm::*;
50pub use validation::*;
51
52use crate::secrets::load_secret_value;
53
54static TEMPLATE_COMMAND_RE: Lazy<Regex> =
55  Lazy::new(|| Regex::new(r"^\$\{\{.+\}\}$").expect("valid template regex"));
56static TEMPLATE_EXPR_RE: Lazy<Regex> =
57  Lazy::new(|| Regex::new(r"\$\{\{\s*(.+?)\s*\}\}").expect("valid template expression regex"));
58
59pub fn is_shell_command(value: &str) -> anyhow::Result<bool> {
60  let re = Regex::new(r"^\$\(.+\)$")?;
61  Ok(re.is_match(value))
62}
63
64pub fn is_template_command(value: &str) -> anyhow::Result<bool> {
65  Ok(TEMPLATE_COMMAND_RE.is_match(value))
66}
67
68pub fn resolve_template_command_value(value: &str, context: &TaskContext) -> anyhow::Result<String> {
69  let value = value.trim_start_matches("${{").trim_end_matches("}}").trim();
70  resolve_template_expression(value, context)
71}
72
73pub fn resolve_template_expression(value: &str, context: &TaskContext) -> anyhow::Result<String> {
74  if value.starts_with("env.") {
75    let value = value.trim_start_matches("env.");
76    let value = context
77      .env_vars
78      .get(value)
79      .ok_or_else(|| anyhow::anyhow!("Environment variable '{}' is not defined", value))?;
80    Ok(value.to_string())
81  } else if value.starts_with("secrets.") {
82    let path = value.trim_start_matches("secrets.");
83    load_secret_value(
84      path,
85      context
86        .secret_config
87        .as_ref()
88        .ok_or_else(|| anyhow::anyhow!("Secret config missing from task context"))?,
89    )
90  } else if value.starts_with("outputs.") {
91    let name = value.trim_start_matches("outputs.");
92    context.get_task_output(name)?.ok_or_else(|| {
93      anyhow::anyhow!(
94        "Task output '{}' is not available. Ensure the task that produces it runs before this one.",
95        name
96      )
97    })
98  } else {
99    Ok(value.to_string())
100  }
101}
102
103pub fn interpolate_template_string(value: &str, context: &TaskContext) -> anyhow::Result<String> {
104  let mut result = String::with_capacity(value.len());
105  let mut last_end = 0usize;
106  for captures in TEMPLATE_EXPR_RE.captures_iter(value) {
107    let Some(full_match) = captures.get(0) else {
108      continue;
109    };
110    let Some(expr) = captures.get(1) else {
111      continue;
112    };
113    result.push_str(&value[last_end..full_match.start()]);
114    result.push_str(&resolve_template_expression(expr.as_str().trim(), context)?);
115    last_end = full_match.end();
116  }
117  result.push_str(&value[last_end..]);
118  Ok(result)
119}
120
121pub fn extract_output_references(value: &str) -> Vec<String> {
122  TEMPLATE_EXPR_RE
123    .captures_iter(value)
124    .filter_map(|captures| captures.get(1))
125    .map(|expr| expr.as_str().trim())
126    .filter_map(|expr| expr.strip_prefix("outputs."))
127    .map(str::to_string)
128    .collect()
129}
130
131pub fn contains_output_reference(value: &str) -> bool {
132  !extract_output_references(value).is_empty()
133}
134
135pub fn get_output_handler(verbose: bool) -> Stdio {
136  if verbose {
137    Stdio::piped()
138  } else {
139    Stdio::null()
140  }
141}
142
143#[cfg(test)]
144mod test {
145  use std::sync::Arc;
146
147  use super::*;
148
149  #[test]
150  fn test_interpolate_template_string_resolves_outputs() -> anyhow::Result<()> {
151    let root = Arc::new(TaskRoot::default());
152    let context = TaskContext::empty_with_root(root);
153    context.insert_task_output("version", "v1.2.3")?;
154    assert_eq!(
155      interpolate_template_string("tag=${{ outputs.version }}", &context)?,
156      "tag=v1.2.3"
157    );
158    Ok(())
159  }
160
161  #[test]
162  fn test_extract_output_references_finds_all_output_templates() {
163    assert_eq!(
164      extract_output_references("${{ outputs.first }}-${{ outputs.second }}"),
165      vec!["first".to_string(), "second".to_string()]
166    );
167  }
168}
mk_lib/schema/mod.rs

mk_lib/schema/
mod.rs
