use anyhow::Result;
pub fn init(output_path: Option<&str>) -> Result<()> {
let path = output_path.unwrap_or("tasks.toml");
if std::path::Path::new(path).exists() {
anyhow::bail!("{path} already exists — use -o <file> to choose a different name");
}
let (dir, team, verify, language) = detect_project_defaults();
let template = render_template(&dir, &team, &verify, &language);
std::fs::write(path, &template)?;
println!("[aid] Created {path}");
println!("[aid] Edit the [[task]] entries, then run: aid batch {path} --parallel");
Ok(())
}
fn detect_project_defaults() -> (String, String, String, String) {
let project = crate::project::detect_project();
match project {
Some(config) => {
let dir = ".".to_string();
let team = config.team.unwrap_or_default();
let verify = config.verify.unwrap_or_default();
let language = config.language.unwrap_or_default();
(dir, team, verify, language)
}
None => (String::new(), String::new(), String::new(), String::new()),
}
}
fn render_template(dir: &str, team: &str, verify: &str, language: &str) -> String {
let mut lines = vec![
"# Batch task file for aid".to_string(),
"# Docs: aid batch --help".to_string(),
"# All fields below also work as task-level overrides.".to_string(),
String::new(),
];
lines.push("[defaults]".to_string());
if dir.is_empty() {
lines.push("# dir = \".\" # Working directory".to_string());
} else {
lines.push(format!("dir = \"{dir}\""));
}
lines.push("# agent = \"codex\" # Default agent for all tasks".to_string());
if team.is_empty() {
lines.push("# team = \"dev\" # Team knowledge context".to_string());
} else {
lines.push(format!("team = \"{team}\""));
}
lines.push("# model = \"o3\" # Model override".to_string());
if verify.is_empty() {
lines.push("# verify = \"cargo check\" # Auto-verify on completion".to_string());
} else {
lines.push(format!("verify = \"{verify}\""));
}
lines.push("# fallback = \"cursor,opencode\" # Comma-separated fallback agents".to_string());
lines.push(String::new());
lines.push("# --- Context & constraints ---".to_string());
lines.push("# context = [\"src/types.rs\"] # Files to inject as context".to_string());
lines.push("# skills = [\"implementer\"] # Methodology skills".to_string());
lines.push("# no_skill = false # Disable skill injection".to_string());
lines.push("# scope = [\"src/\"] # Restrict file access".to_string());
lines.push("# checklist = [\"must compile\", \"must have tests\"] # Quality checklist".to_string());
lines.push("# read_only = false # Read-only mode".to_string());
lines.push("# sandbox = false # Run agent in sandbox mode".to_string());
lines.push("# budget = false # Use budget-optimized model".to_string());
lines.push(String::new());
lines.push("# --- Execution ---".to_string());
lines.push("# max_duration_mins = 30 # Per-task hard timeout (minutes)".to_string());
lines.push("# max_wait_mins = 30 # Fail WAIT tasks after 30 minutes".to_string());
lines.push("# retry = 1 # Retry failed runs N times".to_string());
lines.push("# idle_timeout = 120 # Kill if idle for N seconds".to_string());
lines.push("# judge = true # AI judge evaluates output quality".to_string());
lines.push("# peer_review = \"gemini\" # Run a second-agent review".to_string());
lines.push("# best_of = 3 # Run N copies, pick best".to_string());
lines.push("# metric = \"cargo test\" # Best-of scoring command".to_string());
lines.push("# container = \"node:20\" # Run agent inside container".to_string());
lines.push("# on_done = \"notify done\" # Shell command after completion".to_string());
lines.push("# hooks = [\"pre:lint\"] # Hook specs".to_string());
lines.push(String::new());
lines.push("# --- Worktree & grouping ---".to_string());
lines.push("# worktree_prefix = \"feat/v9\" # Auto-generate worktree per task".to_string());
lines.push("# group = \"wg-abc1\" # Assign all tasks to workgroup".to_string());
lines.push("# shared_dir = false # Shared directory for inter-task files".to_string());
lines.push("# analyze = true # Warn about overlapping file edits".to_string());
lines.push(String::new());
lines.push("[[task]]".to_string());
lines.push("name = \"task-1\"".to_string());
lines.push("agent = \"codex\"".to_string());
lines.push("prompt = \"\"\"".to_string());
if language.is_empty() {
lines.push("<DESCRIBE_TASK_HERE>".to_string());
} else {
lines.push(format!("Implement <FEATURE> in {language}."));
}
lines.push("\"\"\"".to_string());
lines.push("# worktree = \"feat/<BRANCH>\" # Git worktree for isolation".to_string());
lines.push("# context = [\"src/types.rs\"] # Extra context files".to_string());
lines.push("# context_from = [\"task-0\"] # Inject output from previous tasks".to_string());
lines.push("# checklist = [\"no unwrap()\"] # Quality checklist items".to_string());
lines.push("# no_skill = true # Disable skill injection".to_string());
lines.push("# sandbox = true # Run in sandbox mode".to_string());
lines.push("# max_wait_mins = 30 # Fail WAIT tasks after 30 minutes".to_string());
lines.push("# idle_timeout = 120 # Kill if idle for N seconds".to_string());
lines.push("# retry = 2 # Retry the task on failure".to_string());
lines.push("# peer_review = \"gemini\" # Run a second-agent review".to_string());
lines.push("# best_of = 3 # Run N copies, pick best".to_string());
lines.push("# metric = \"cargo test\" # Best-of scoring command".to_string());
lines.push("# scope = [\"src/parser/\"] # Restrict file access".to_string());
lines.push("# depends_on = [\"other-task\"] # Run after named task(s)".to_string());
lines.push("# on_done = \"notify done\" # Shell command after completion".to_string());
lines.push("# on_success = \"deploy\" # Trigger conditional task on success".to_string());
lines.push("# on_fail = \"notify\" # Trigger conditional task on failure".to_string());
lines.push("# env = { RUST_LOG = \"debug\" } # Task-specific env vars".to_string());
lines.push(String::new());
lines.push("# [[task]]".to_string());
lines.push("# name = \"task-2\"".to_string());
lines.push("# agent = \"opencode\"".to_string());
lines.push("# prompt = \"<DESCRIBE_TASK_HERE>\"".to_string());
lines.push("# depends_on = [\"task-1\"]".to_string());
lines.push("# context_from = [\"task-1\"] # Use task-1 output as context".to_string());
lines.push(String::new());
lines.push("# [[task]]".to_string());
lines.push("# name = \"deploy\"".to_string());
lines.push("# conditional = true # Only runs when triggered".to_string());
lines.push("# agent = \"codex\"".to_string());
lines.push("# prompt = \"Run deploy script\"".to_string());
lines.push(String::new());
lines.join("\n") + "\n"
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn render_template_with_no_project() {
let template = render_template("", "", "", "");
assert!(template.contains("[defaults]"));
assert!(template.contains("[[task]]"));
assert!(template.contains("<DESCRIBE_TASK_HERE>"));
assert!(template.contains("# fallback"));
assert!(template.contains("# depends_on"));
assert!(template.contains("# idle_timeout"));
assert!(template.contains("# max_wait_mins"));
assert!(template.contains("# max_wait_mins = 30 # Fail WAIT tasks after 30 minutes"));
assert!(template.contains("# checklist"));
assert!(template.contains("# context_from"));
assert!(template.contains("# scope"));
assert!(template.contains("# conditional"));
assert!(template.contains("# env ="));
assert!(template.contains("# sandbox"));
assert!(template.contains("# retry"));
assert!(template.contains("# peer_review"));
assert!(template.contains("# metric"));
assert!(template.contains("# on_done"));
assert!(template.contains("# no_skill"));
}
#[test]
fn render_template_with_project_defaults() {
let template = render_template(".", "dev", "cargo test", "rust");
assert!(template.contains("dir = \".\""));
assert!(template.contains("team = \"dev\""));
assert!(template.contains("verify = \"cargo test\""));
assert!(template.contains("Implement <FEATURE> in rust."));
assert!(!template.contains("# dir ="));
assert!(!template.contains("# team ="));
}
#[test]
fn init_refuses_existing_file() {
let temp = tempfile::NamedTempFile::new().unwrap();
let path = temp.path().to_str().unwrap();
let result = init(Some(path));
assert!(result.is_err());
assert!(result.unwrap_err().to_string().contains("already exists"));
}
#[test]
fn init_creates_file() {
let dir = tempfile::tempdir().unwrap();
let path = dir.path().join("my-tasks.toml");
let path_str = path.to_str().unwrap();
init(Some(path_str)).unwrap();
let content = std::fs::read_to_string(&path).unwrap();
assert!(content.contains("[defaults]"));
assert!(content.contains("[[task]]"));
}
}