ClawdDotNet/src/ClawdDotNet.Core/Tools/ToolRegistry.cs

using ClawdDotNet.Core.Config;

namespace ClawdDotNet.Core.Tools;

public sealed class ToolRegistry
{
    private readonly Dictionary<string, IAgentTool> _tools = new();

    /// <summary>
    /// #689: Tools every agent has, without being granted them. Escalation is one: a case
    /// nobody can raise is a case nobody solves — being able to say "I cannot get on" is not a
    /// permission, it is part of working in an organisation.
    /// </summary>
    public static readonly IReadOnlySet<string> ImplicitTools =
        new HashSet<string>(StringComparer.OrdinalIgnoreCase) { Cases.CaseDesk.ToolName };

    private readonly Lock _lock = new();

    public void Register(IAgentTool tool)
    {
        lock (_lock)
            _tools[tool.Name] = tool;
    }

    /// <summary>#197: Takes a tool out again — an MCP server the human removed.</summary>
    public bool Unregister(string name)
    {
        lock (_lock)
            return _tools.Remove(name);
    }

    public IAgentTool? Get(string name)
    {
        lock (_lock)
            return _tools.GetValueOrDefault(name);
    }

    public IReadOnlyList<IAgentTool> GetForAgent(AgentConfig agent)
    {
        lock (_lock)
            return _tools.Values
                .Where(t => agent.Tools.ContainsKey(t.Name) || ImplicitTools.Contains(t.Name))
                .ToList();
    }

    public IReadOnlyList<IAgentTool> GetAll()
    {
        lock (_lock)
            return _tools.Values.ToList();
    }

    public IReadOnlyList<(IAgentTool Tool, IToolJobProvider Provider)> GetJobProviders()
    {
        lock (_lock)
            return _tools.Values
                .OfType<IToolJobProvider>()
                .Select(p => ((IAgentTool)p, p))
                .ToList();
    }
}