tori/src/tools.rs

use std::collections::HashMap;
use std::os::unix::fs::PermissionsExt;
use std::path::{Path, PathBuf};

use crate::llm::{Tool, ToolFunction};

struct ExternalTool {
    path: PathBuf,
    schema: Tool,
}

pub struct ExternalToolManager {
    tools: HashMap<String, ExternalTool>,
}

impl ExternalToolManager {
    /// Scan a tools/ directory, calling `--print-schema` on each executable to discover tools.
    pub async fn discover(tools_dir: &Path) -> Self {
        let mut tools = HashMap::new();

        let mut entries = match tokio::fs::read_dir(tools_dir).await {
            Ok(e) => e,
            Err(_) => return Self { tools },
        };

        while let Ok(Some(entry)) = entries.next_entry().await {
            let path = entry.path();

            // Skip non-files
            let meta = match tokio::fs::metadata(&path).await {
                Ok(m) => m,
                Err(_) => continue,
            };
            if !meta.is_file() {
                continue;
            }

            // Check executable bit
            if meta.permissions().mode() & 0o111 == 0 {
                tracing::debug!("Skipping non-executable: {}", path.display());
                continue;
            }

            // Call --print-schema
            let output = match tokio::process::Command::new(&path)
                .arg("--print-schema")
                .output()
                .await
            {
                Ok(o) => o,
                Err(e) => {
                    tracing::warn!("Failed to run --print-schema on {}: {}", path.display(), e);
                    continue;
                }
            };

            if !output.status.success() {
                tracing::warn!(
                    "--print-schema failed for {}: {}",
                    path.display(),
                    String::from_utf8_lossy(&output.stderr)
                );
                continue;
            }

            let schema: serde_json::Value = match serde_json::from_slice(&output.stdout) {
                Ok(v) => v,
                Err(e) => {
                    tracing::warn!("Invalid schema JSON from {}: {}", path.display(), e);
                    continue;
                }
            };

            let name = schema["name"]
                .as_str()
                .unwrap_or_else(|| {
                    path.file_name()
                        .and_then(|n| n.to_str())
                        .unwrap_or("unknown")
                })
                .to_string();

            let description = schema["description"].as_str().unwrap_or("").to_string();
            let parameters = schema["parameters"].clone();

            let tool = Tool {
                tool_type: "function".into(),
                function: ToolFunction {
                    name: name.clone(),
                    description,
                    parameters,
                },
            };

            tracing::info!("Discovered external tool: {}", name);
            tools.insert(
                name,
                ExternalTool {
                    path: path.clone(),
                    schema: tool,
                },
            );
        }

        Self { tools }
    }

    /// Return all discovered Tool definitions for LLM API calls.
    pub fn tool_definitions(&self) -> Vec<Tool> {
        self.tools.values().map(|t| t.schema.clone()).collect()
    }

    /// Invoke an external tool by name, passing JSON args as the first argv.
    pub async fn invoke(
        &self,
        name: &str,
        args_json: &str,
        workdir: &str,
    ) -> anyhow::Result<String> {
        let tool = self
            .tools
            .get(name)
            .ok_or_else(|| anyhow::anyhow!("External tool not found: {}", name))?;

        let output = tokio::process::Command::new(&tool.path)
            .arg(args_json)
            .current_dir(workdir)
            .output()
            .await?;

        let stdout = String::from_utf8_lossy(&output.stdout).to_string();
        let stderr = String::from_utf8_lossy(&output.stderr).to_string();

        if output.status.success() {
            Ok(stdout)
        } else {
            let mut result = stdout;
            if !stderr.is_empty() {
                result.push_str("\nSTDERR: ");
                result.push_str(&stderr);
            }
            result.push_str(&format!(
                "\n[exit code: {}]",
                output.status.code().unwrap_or(-1)
            ));
            Ok(result)
        }
    }

    /// Check if a tool with the given name exists.
    pub fn has_tool(&self, name: &str) -> bool {
        self.tools.contains_key(name)
    }

    /// Number of discovered tools.
    pub fn len(&self) -> usize {
        self.tools.len()
    }
}