46 lines
1.1 KiB
Rust
46 lines
1.1 KiB
Rust
|
//! 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");
|
||
|
|
|
||
|
|
Ok(format!("[{}] TODO added: {}", priority, item))
|
||
|
|
}
|
||
|
|
}
|