tools.py

"""
============================================================================
 TOOLS MODULE — Agent Capabilities
============================================================================
 Tools that agents can use during swarm execution.
 - File read/write
 - Command execution
============================================================================
"""

from pathlib import Path
from command_safety import CommandExecutor


class AgentTools:
    def __init__(self, workspace_path: Path):
        self.workspace = workspace_path
        # Auto-approve safe and moderate commands. Dangerous ones will need approval.
        self.executor = CommandExecutor(cwd=str(self.workspace))

    def write_file(self, rel_path: str, content: str) -> str:
        """Write a file inside the workspace."""
        path = self.workspace / rel_path
        
        # Security: Prevent writing outside workspace
        try:
            path.resolve().relative_to(self.workspace.resolve())
        except ValueError:
            return f"Error: Cannot write outside workspace ({rel_path})"

        path.parent.mkdir(parents=True, exist_ok=True)
        path.write_text(content, encoding="utf-8")
        return f"File written: {rel_path}"

    def read_file(self, rel_path: str) -> str:
        """Read a file from the workspace."""
        path = self.workspace / rel_path
        
        try:
            path.resolve().relative_to(self.workspace.resolve())
        except ValueError:
            return f"Error: Cannot read outside workspace ({rel_path})"

        if not path.exists():
            return f"Error: File not found ({rel_path})"
            
        try:
            return path.read_text(encoding="utf-8")
        except Exception as e:
            return f"Error reading file: {e}"

    def run_command(self, command: str) -> str:
        """Run a shell command inside the workspace."""
        result = self.executor.execute(command, timeout=300)
        
        out = []
        out.append(f"Command: {command}")
        out.append(f"Exit code: {result.returncode}")
        out.append(f"Safety: {result.safety} (Approved: {result.approved})")
        
        if result.stdout:
            out.append(f"STDOUT:\n{result.stdout}")
        if result.stderr:
            out.append(f"STDERR:\n{result.stderr}")
            
        return "\n".join(out)