ToolDeclaration.h

/* Copyright (C) 2020-2026 by Marc Boris Duerner
   SPDX-License-Identifier: LGPL-2.1-or-later WITH mif-exception
*/
 
#ifndef PT_MCP_TOOLDECLARATION_H
#define PT_MCP_TOOLDECLARATION_H
 
#include <Pt/Mcp/Api.h>
#include <Pt/Mcp/Type.h>
#include <Pt/Mcp/ContentType.h>
#include <Pt/NonCopyable.h>
#include <string>
#include <vector>
 
namespace Pt {
 
namespace Mcp {

class PT_MCP_API Tool : private NonCopyable
{
  public:
    Tool(const std::string& name, const std::string& description);

    ~Tool();

    Tool& addParam(const std::string& name, const Type& type,
                   const std::string& description = "");

    Tool& setOptional(const std::string& paramName);

    Tool& setContent(const ContentType& content);

    const std::string& name() const
    { return _name; }

    const std::string& description() const
    { return _description; }

    const std::vector<Property>& params() const
    { return _params; }

    std::size_t paramCount() const
    { return _params.size(); }

    int getParamIndex(const std::string& name) const;

    const ContentType& content() const;
 
  private:
    std::string _name;
    std::string _description;
    std::vector<Property> _params;
    const ContentType* _content;
};
 

class PT_MCP_API ToolDeclaration : private NonCopyable
{
  public:
    ToolDeclaration(const std::string& serverName,
                    const std::string& serverVersion);

    ~ToolDeclaration();

    Tool& addTool(const std::string& name, const std::string& description);

    const Tool* getTool(const std::string& name) const;

    const std::string& serverName() const
    { return _serverName; }

    const std::string& serverVersion() const
    { return _serverVersion; }

    static std::string preferredVersion(const std::string& requested);

    static bool isSupportedVersion(const std::string& version);

    void toToolsList(std::ostream& os) const;

    void toInitializeResult(std::ostream& os,
                            const char* protocolVersion = 0) const;
 
  private:
    std::string _serverName;
    std::string _serverVersion;
    std::vector<Tool*> _tools;
};
 
} // namespace Mcp
 
} // namespace Pt
 
#endif // PT_MCP_TOOLDECLARATION_H