MCPProcessor.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 <atomic>
#include <memory>
 
#include <QBuffer>
#include <QByteArray>
#include <QHash>
#include <QJsonDocument>
#include <QJsonObject>
#include <QObject>
#include <QStringList>
#include <QTextStream>
#include <QThread>
#include <QTimer>
 
class BaseHandler;
class ConnectionHandler;
class ElementHandler;
class FileHandler;
class HistoryHandler;
class ICHandler;
class MCPValidator;
class MainWindow;
class QSocketNotifier;
class ServerInfoHandler;
class SimulationHandler;
class ThemeHandler;
 
#ifdef Q_OS_WIN
class StdinReader : public QThread
{
    Q_OBJECT
 
public:
    explicit StdinReader(QObject *parent = nullptr);
    void requestStop();
 
protected:
    void run() override;
 
signals:
    void dataReceived(const QString &line);
 
private:
    std::atomic<bool> m_stopRequested{false};
};
#endif

class MCPProcessor : public QObject
{
    Q_OBJECT
 
public:
    explicit MCPProcessor(MainWindow *mainWindow, QObject *parent = nullptr);
    ~MCPProcessor();
 
    void startProcessing();
    void stopProcessing();

    static constexpr qint64 kMaxStdinLineBytes = 16 * 1024 * 1024; // 16 MB

    static QStringList extractStdinLines(QByteArray &buffer, const QByteArray &data);
 
private slots:
    void processIncomingData(const QString &line);
#ifndef Q_OS_WIN
    void onStdinReadable();
#endif
 
private:
    void processCommand(const QString &line);
    void sendResponse(const QJsonObject &response);
 
    MainWindow *m_mainWindow;
    std::unique_ptr<MCPValidator> m_validator;
    QTextStream m_stdout;
#ifdef Q_OS_WIN
    StdinReader *m_stdinReader = nullptr;
#else
    QSocketNotifier *m_stdinNotifier = nullptr;
    QByteArray m_stdinBuffer;
#endif
 
    // Specialized handlers for different command domains
    std::unique_ptr<ServerInfoHandler> m_serverInfoHandler;
    std::unique_ptr<FileHandler> m_fileHandler;
    std::unique_ptr<ElementHandler> m_elementHandler;
    std::unique_ptr<ConnectionHandler> m_connectionHandler;
    std::unique_ptr<SimulationHandler> m_simulationHandler;
    std::unique_ptr<ICHandler> m_icHandler;
    std::unique_ptr<HistoryHandler> m_historyHandler;
    std::unique_ptr<ThemeHandler> m_themeHandler;

    QHash<QString, BaseHandler *> m_dispatchMap;
};