Lifei Zhou
126 lines
4.0 KiB
Rust
126 lines
4.0 KiB
Rust
use anyhow::Result;
|
|
use console::style;
|
|
|
|
use goose::recipe::Recipe;
|
|
use minijinja::UndefinedBehavior;
|
|
use serde_json::Value as JsonValue;
|
|
use serde_yaml::Value as YamlValue;
|
|
use std::collections::HashMap;
|
|
|
|
use crate::recipes::search_recipe::retrieve_recipe_file;
|
|
|
|
/// Loads and validates a recipe from a YAML or JSON file
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `path` - Path to the recipe file (YAML or JSON)
|
|
/// * `log` - whether to log information about the recipe or not
|
|
/// * `params` - optional parameters to render the recipe with
|
|
///
|
|
/// # Returns
|
|
///
|
|
/// The parsed recipe struct if successful
|
|
///
|
|
/// # Errors
|
|
///
|
|
/// Returns an error if:
|
|
/// - The file doesn't exist
|
|
/// - The file can't be read
|
|
/// - The YAML/JSON is invalid
|
|
/// - The required fields are missing
|
|
pub fn load_recipe(
|
|
recipe_name: &str,
|
|
log: bool,
|
|
params: Option<Vec<(String, String)>>,
|
|
) -> Result<Recipe> {
|
|
let content = retrieve_recipe_file(recipe_name)?;
|
|
|
|
// Check if any parameters were provided
|
|
let rendered_content = match params {
|
|
None => content,
|
|
Some(params) => render_content_with_params(&content, ¶ms)?,
|
|
};
|
|
|
|
let recipe: Recipe;
|
|
if serde_json::from_str::<JsonValue>(&rendered_content).is_ok() {
|
|
recipe = serde_json::from_str(&rendered_content)?
|
|
} else if serde_yaml::from_str::<YamlValue>(&rendered_content).is_ok() {
|
|
recipe = serde_yaml::from_str(&rendered_content)?
|
|
} else {
|
|
return Err(anyhow::anyhow!(
|
|
"Unsupported file format for recipe file. Expected .yaml or .json"
|
|
));
|
|
}
|
|
|
|
if log {
|
|
// Display information about the loaded recipe
|
|
println!(
|
|
"{} {}",
|
|
style("Loading recipe:").green().bold(),
|
|
style(&recipe.title).green()
|
|
);
|
|
println!("{} {}", style("Description:").dim(), &recipe.description);
|
|
|
|
println!(); // Add a blank line for spacing
|
|
}
|
|
|
|
Ok(recipe)
|
|
}
|
|
|
|
fn render_content_with_params(content: &str, params: &[(String, String)]) -> Result<String> {
|
|
// Turn params into HashMap
|
|
let param_map: HashMap<String, String> = params.iter().cloned().collect();
|
|
|
|
// Create a minijinja environment and context
|
|
let mut env = minijinja::Environment::new();
|
|
env.set_undefined_behavior(UndefinedBehavior::Strict);
|
|
let template = env.template_from_str(content)
|
|
.map_err(|_| anyhow::anyhow!("Failed to render recipe, please check if the recipe has proper syntax for variables: eg: {{ variable_name }}"))?;
|
|
|
|
// Render the template with the parameters
|
|
template.render(param_map).map_err(|_| {
|
|
anyhow::anyhow!(
|
|
"Failed to render the recipe - please check if all required parameters are provided"
|
|
)
|
|
})
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_render_content_with_params() {
|
|
// Test basic parameter substitution
|
|
let content = "Hello {{ name }}!";
|
|
let params = vec![("name".to_string(), "World".to_string())];
|
|
let result = render_content_with_params(content, ¶ms).unwrap();
|
|
assert_eq!(result, "Hello World!");
|
|
|
|
// Test multiple parameters
|
|
let content = "{{ greeting }} {{ name }}!";
|
|
let params = vec![
|
|
("greeting".to_string(), "Hi".to_string()),
|
|
("name".to_string(), "Alice".to_string()),
|
|
];
|
|
let result = render_content_with_params(content, ¶ms).unwrap();
|
|
assert_eq!(result, "Hi Alice!");
|
|
|
|
// Test missing parameter results in error
|
|
let content = "Hello {{ missing }}!";
|
|
let params = vec![];
|
|
let err = render_content_with_params(content, ¶ms).unwrap_err();
|
|
assert!(err
|
|
.to_string()
|
|
.contains("please check if all required parameters"));
|
|
|
|
// Test invalid template syntax results in error
|
|
let content = "Hello {{ unclosed";
|
|
let params = vec![];
|
|
let err = render_content_with_params(content, ¶ms).unwrap_err();
|
|
assert!(err
|
|
.to_string()
|
|
.contains("please check if the recipe has proper syntax"));
|
|
}
|
|
}
|