forked from kent/consciousness
dd85a56902
ToolHandler is now Arc<dyn Fn(...)> supporting closures that capture state. The output tool is created during init_output_tool() as a closure capturing Arc<Mutex<Subconscious>>, writing directly to Subconscious.state. No more POC_AGENT_OUTPUT_DIR filesystem hack. - All tool handlers wrapped in Arc::new() - Tool is Clone (not Copy) — .copied() → .cloned() - Subconscious wrapped in Arc<Mutex<>> on Mind - Dead filesystem-based output() function removed - memory_tools returns 11 items (output removed from static list) Co-Authored-By: Proof of Concept <poc@bcachefs.org>
71 lines
2.3 KiB
Rust
71 lines
2.3 KiB
Rust
use std::sync::Arc;
|
|
// tools/glob_tool.rs — Find files by pattern
|
|
//
|
|
// Fast file discovery using glob patterns. Returns matching paths
|
|
// sorted by modification time (newest first), which is usually
|
|
// what you want when exploring a codebase.
|
|
|
|
use anyhow::{Context, Result};
|
|
use serde::Deserialize;
|
|
use std::path::PathBuf;
|
|
|
|
#[derive(Deserialize)]
|
|
struct Args {
|
|
pattern: String,
|
|
#[serde(default = "default_path")]
|
|
path: String,
|
|
}
|
|
|
|
fn default_path() -> String { ".".into() }
|
|
|
|
pub fn tool() -> super::Tool {
|
|
super::Tool {
|
|
name: "glob",
|
|
description: "Find files matching a glob pattern. Returns file paths sorted by modification time (newest first).",
|
|
parameters_json: r#"{"type":"object","properties":{"pattern":{"type":"string","description":"Glob pattern to match files (e.g. '**/*.rs')"},"path":{"type":"string","description":"Directory to search in (default: current directory)"}},"required":["pattern"]}"#,
|
|
handler: Arc::new(|_a, v| Box::pin(async move { glob_search(&v) })),
|
|
}
|
|
}
|
|
|
|
fn glob_search(args: &serde_json::Value) -> Result<String> {
|
|
let a: Args = serde_json::from_value(args.clone())
|
|
.context("invalid glob arguments")?;
|
|
|
|
let full_pattern = if a.pattern.starts_with('/') {
|
|
a.pattern.clone()
|
|
} else {
|
|
format!("{}/{}", a.path, a.pattern)
|
|
};
|
|
|
|
let mut entries: Vec<(PathBuf, std::time::SystemTime)> = Vec::new();
|
|
|
|
for entry in glob::glob(&full_pattern)
|
|
.with_context(|| format!("Invalid glob pattern: {}", full_pattern))?
|
|
{
|
|
if let Ok(path) = entry {
|
|
if path.is_file() {
|
|
let mtime = path
|
|
.metadata()
|
|
.and_then(|m| m.modified())
|
|
.unwrap_or(std::time::SystemTime::UNIX_EPOCH);
|
|
entries.push((path, mtime));
|
|
}
|
|
}
|
|
}
|
|
|
|
// Sort by modification time, newest first
|
|
entries.sort_by(|a, b| b.1.cmp(&a.1));
|
|
|
|
if entries.is_empty() {
|
|
return Ok("No files matched.".to_string());
|
|
}
|
|
|
|
let mut output = String::new();
|
|
for (path, _) in &entries {
|
|
output.push_str(&path.display().to_string());
|
|
output.push('\n');
|
|
}
|
|
|
|
output.push_str(&format!("\n({} files matched)", entries.len()));
|
|
Ok(super::truncate_output(output, 30000))
|
|
}
|