MCPValidator.h

Go to the documentation of this file.

// Copyright 2015 - 2026, GIBIS-UNIFESP and the wiRedPanda contributors
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
 
#include <memory>
#include <string>
#include <unordered_map>
 
#include <nlohmann/json-schema.hpp>
#include <nlohmann/json.hpp>
#include <QJsonObject>
#include <QString>
 
using json = nlohmann::json;
using nlohmann::json_schema::json_validator;

struct ValidationResult
{
    bool isValid = false;
    QString errorMessage;
    QString schemaUsed;
    QString commandType;
    QString errorPath;
 
    ValidationResult() = default;
    ValidationResult(bool valid, const QString &error = QString(), const QString &schema = QString(),
                     const QString &command = QString(), const QString &path = QString())
        : isValid(valid)
        , errorMessage(error)
        , schemaUsed(schema)
        , commandType(command)
        , errorPath(path)
    {
    }
};

class MCPValidator
{
public:
    explicit MCPValidator(const QString &schemaPath);
    ~MCPValidator();
 
    // Core validation methods
    ValidationResult validateCommand(const QJsonObject &command);
    ValidationResult validateCommand(const json &command);
    ValidationResult validateResponse(const QJsonObject &response, const QString &expectedCommand = QString());
    ValidationResult validateResponse(const json &response, const QString &expectedCommand = QString());
 
    // Schema management
    bool isSchemaLoaded() const;
    QString schemaPath() const;
 
    // Utility methods
    static json qjsonToNlohmann(const QJsonObject &qjson);
    static QJsonObject nlohmannToQJson(const json &nlohmannJson);
 
    // Schema introspection (used by describe_command)
    json findCommandSchema(const QString &commandName) const;
    json findResponseSchema(const QString &commandName) const;
 
private:
    bool loadSchema(const QString &schemaPath);
 
    // Internal validation methods
    ValidationResult validateAgainstSchema(const json &data, const json &schema, const QString &commandType, const std::string &cacheKey);
    QString extractErrorPath(const QString &errorMsg);
 
    // Member variables
    json m_schema;
    json m_commandSchemas;
    json m_responseSchemas;
    std::unique_ptr<json_validator> m_validator;
    std::unordered_map<std::string, std::unique_ptr<json_validator>> m_validatorCache;
    QString m_schemaPath;
    bool m_schemaLoaded = false;
};