mcp_tool_registry.h

Go to the documentation of this file.

#pragma once
 
#include <functional>
#include <string>
#include <unordered_map>
#include <vector>
 
#include <context/context.hpp>
#include <ser20/external/simdjson/simdjson.h>
 
namespace unravel::mcp
{
 
struct tool_result
{
    std::string text;
    bool is_error{false};
    std::string image_base64;
    std::string image_mime{"image/png"};
};
 
using tool_handler = std::function<tool_result(rtti::context& ctx, const simdjson::dom::object& args)>;
 
struct mcp_tool
{
    std::string name;
    std::string description;
    std::string input_schema_json;
    tool_handler handler;
    bool mutates_scene{false};
    bool requires_main_thread{true};
};
 
class mcp_tool_registry
{
public:
    void add(mcp_tool tool);
    auto find(const std::string& name) const -> const mcp_tool*;
    auto tools() const -> const std::vector<mcp_tool>&;
    auto list_tools_json() const -> std::string;
 
private:
    std::vector<mcp_tool> tools_;
    std::unordered_map<std::string, size_t> index_;
};
 
void register_scene_tools(mcp_tool_registry& registry);
void register_scene_batch_tools(mcp_tool_registry& registry);
void register_ops_batch_tools(mcp_tool_registry& registry);
void register_asset_tools(mcp_tool_registry& registry);
void register_viewport_tools(mcp_tool_registry& registry);
void register_material_tools(mcp_tool_registry& registry);
void register_project_tools(mcp_tool_registry& registry);
void register_script_tools(mcp_tool_registry& registry);
void register_editor_tools(mcp_tool_registry& registry);
 
} // namespace unravel::mcp