web.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#include <entropic/mcp/servers/web.h>
#include <entropic/mcp/tool_base.h>
#include <entropic/mcp/server_base.h>
#include <entropic/types/logging.h>
 
#include <nlohmann/json.hpp>
 
#include <algorithm>
#include <string>
 
static auto logger = entropic::log::get("mcp.web");
 
namespace entropic {
 
// ── WebFetchTool ────────────────────────────────────────────────
 
class WebFetchTool : public ToolBase {
public:
    explicit WebFetchTool(ToolDefinition def)
        : ToolBase(std::move(def)) {}
 
    MCPAccessLevel required_access_level() const override {
        return MCPAccessLevel::READ;
    }
 
    ServerResponse execute(const std::string& args_json) override;
};
 
ServerResponse WebFetchTool::execute(const std::string& args_json) {
    auto args = nlohmann::json::parse(args_json);
    std::string url = args.at("url").get<std::string>();
 
    static constexpr int default_max = 20000;
    static constexpr int cap_max = 50000;
    int max_length = args.value("max_length", default_max);
    max_length = std::min(max_length, cap_max);
 
    logger->info("[web_fetch] url='{}' max_length={}", url, max_length);
 
    return {
        "Web fetch not yet connected (requires cpp-httplib integration)",
        {}
    };
}
 
// ── WebSearchTool ───────────────────────────────────────────────
 
class WebSearchTool : public ToolBase {
public:
    explicit WebSearchTool(ToolDefinition def)
        : ToolBase(std::move(def)) {}
 
    MCPAccessLevel required_access_level() const override {
        return MCPAccessLevel::READ;
    }
 
    ServerResponse execute(const std::string& args_json) override;
};
 
ServerResponse WebSearchTool::execute(const std::string& args_json) {
    auto args = nlohmann::json::parse(args_json);
    std::string query = args.at("query").get<std::string>();
 
    static constexpr int default_max = 5;
    static constexpr int cap_max = 10;
    int max_results = args.value("max_results", default_max);
    max_results = std::min(max_results, cap_max);
 
    logger->info("[web_search] query='{}' max_results={}",
                 query, max_results);
 
    return {
        "Web search not yet connected (requires cpp-httplib integration)",
        {}
    };
}
 
// ── WebServer ───────────────────────────────────────────────────
 
WebServer::WebServer(const std::string& data_dir)
    : MCPServerBase("web") {
 
    std::string tools_dir = data_dir + "/tools";
 
    auto fetch_def = load_tool_definition(
        "web_fetch", "web", tools_dir);
    web_fetch_ = std::make_unique<WebFetchTool>(
        std::move(fetch_def));
 
    auto search_def = load_tool_definition(
        "web_search", "web", tools_dir);
    web_search_ = std::make_unique<WebSearchTool>(
        std::move(search_def));
 
    register_tool(web_fetch_.get());
    register_tool(web_search_.get());
 
    logger->info("WebServer initialized with 2 tools");
}
 
WebServer::~WebServer() = default;
 
} // namespace entropic