Files
zesdex/apps/infrastructure/src/tools/utility/todowrite.rs
T

59 lines
1.6 KiB
Rust
Raw Normal View History

//! Write a TODO item.
use crate::tools::{Tool, ToolCtx};
use anyhow::Result;
use serde_json::{json, Value};
pub struct Todowrite;
impl Tool for Todowrite {
fn name(&self) -> &'static str {
"todowrite"
}
fn description(&self) -> &'static str {
"Add an item to the TODO list"
}
fn parameters(&self) -> Value {
json!({
"type": "object",
"properties": {
"item": {
"type": "string",
"description": "TODO item text"
},
"priority": {
"type": "string",
"enum": ["high", "medium", "low"],
"description": "Priority level"
}
},
"required": ["item"]
})
}
fn run(&self, ctx: &ToolCtx, args: &Value) -> Result<String> {
let item = crate::tools::arg_str(args, "item")?;
let priority = args
.get("priority")
.and_then(|v| v.as_str())
.unwrap_or("medium");
let todo_line = format!("- [{}] {}\n", priority, item);
let todo_path = ctx.session_dir.join("TODO.md");
let mut content = std::fs::read_to_string(&todo_path).unwrap_or_default();
content.push_str(&todo_line);
let _ = std::fs::write(&todo_path, &content);
if let Some(events) = &ctx.turn_events {
let mut q = events.lock().unwrap();
q.push_back(crate::TurnEvent::TodoUpdate(content));
}
Ok(format!("[{}] TODO added: {}", priority, item))
}
}