use crate::config::paths::Paths; use include_dir::{include_dir, Dir}; use minijinja::{Environment, Error as MiniJinjaError, Value as MJValue}; use serde::Serialize; use std::path::PathBuf; static CORE_PROMPTS_DIR: Dir = include_dir!("$CARGO_MANIFEST_DIR/src/prompts"); static TEMPLATE_REGISTRY: &[(&str, &str)] = &[ ( "system.md", "Main system prompt that defines goose's personality and behavior", ), ( "compaction.md", "Prompt for summarizing conversation history when context limits are reached", ), ( "subagent_system.md", "System prompt for subagents spawned to handle specific tasks", ), ( "recipe.md", "Prompt for generating recipe files from conversations", ), ( "apps_create.md", "Prompt for generating new Goose apps based on the user instructions", ), ( "apps_iterate.md", "Prompt for updating existing Goose apps based on feedback", ), ( "permission_judge.md", "Prompt for analyzing tool operations for read-only detection", ), ( "plan.md", "Prompt used when goose creates step-by-step plans. CLI only", ), ( "tiny_model_system.md", "System prompt for tiny local models using shell command emulation", ), ( "session_name.md", "System prompt for generating short session names from conversation history", ), ]; /// Information about a template including its content and customization status #[derive(Debug, Clone, serde::Serialize, serde::Deserialize, utoipa::ToSchema)] pub struct Template { pub name: String, pub description: String, pub default_content: String, #[serde(skip_serializing_if = "Option::is_none")] pub user_content: Option, pub is_customized: bool, } fn user_prompts_dir() -> PathBuf { Paths::config_dir().join("prompts") } fn is_registered(name: &str) -> bool { TEMPLATE_REGISTRY.iter().any(|(n, _)| *n == name) } pub fn render_string( template_str: &str, context: &T, ) -> Result { let mut env = Environment::new(); env.set_trim_blocks(true); env.set_lstrip_blocks(true); env.add_template("template", template_str)?; let tmpl = env.get_template("template")?; let ctx = MJValue::from_serialize(context); let rendered = tmpl.render(ctx)?; Ok(rendered.trim().to_string()) } pub fn render_template(name: &str, context: &T) -> Result { if !is_registered(name) { return Err(MiniJinjaError::new( minijinja::ErrorKind::TemplateNotFound, format!("Template '{}' is not registered", name), )); } let user_path = user_prompts_dir().join(name); let template_str = if user_path.exists() { std::fs::read_to_string(&user_path).map_err(|e| { MiniJinjaError::new( minijinja::ErrorKind::InvalidOperation, format!("Failed to read user template: {}", e), ) })? } else { let file = CORE_PROMPTS_DIR.get_file(name).ok_or_else(|| { MiniJinjaError::new( minijinja::ErrorKind::TemplateNotFound, format!("Built-in template '{}' not found", name), ) })?; String::from_utf8_lossy(file.contents()).to_string() }; render_string(&template_str, context) } pub fn get_template(name: &str) -> Option