McpServer.h

// -*- mode: c++; c-file-style: "linux"; c-basic-offset: 2; indent-tabs-mode: nil -*-
//
//  Copyright (C) 2026 Gunter Königsmann <wxMaxima@physikbuch.de>
//
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation; either version 2 of the License, or
//  (at your option) any later version.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//
//  You should have received a copy of the GNU General Public License
//  along with this program; if not, write to the Free Software
//  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
//
//  SPDX-License-Identifier: GPL-2.0+
 
#ifndef MCPSERVER_H
#define MCPSERVER_H
 
#include "precomp.h"
#include "McpTools.h"
#include <wx/socket.h>
#include <functional>
#include <memory>
#include <string>
#include <unordered_map>
 
class Configuration;
class Worksheet;
class Variablespane;
 
class McpServer : public wxEvtHandler {
public:
  McpServer(Worksheet *worksheet, Variablespane *variablesPane);
  ~McpServer() override;
 
  void ReconcileWithConfig(const Configuration &config);
 
  void SetConnectionCheck(std::function<bool()> isConnected)
    {
      m_tools.SetConnectionCheck(std::move(isConnected));
    }
 
  void SetSidebarAccess(std::function<std::vector<McpSidebarInfo>()> list,
                        std::function<bool(const wxString &, bool)> setVisible)
    {
      m_tools.SetSidebarAccess(std::move(list), std::move(setVisible));
    }
 
private:
  struct Connection {
    wxSocketBase *socket = nullptr;
    std::string buffer;
    long contentLength = -1;
    std::size_t bodyStart = 0;
    bool badRequest = false;
  };
 
  void Start(int port);
  void Stop();
  void OnServerEvent(wxSocketEvent &event);
  void OnClientEvent(wxSocketEvent &event);
  void PumpConnection(wxSocketBase *socket, Connection &conn);
  bool TryParseHeaders(Connection &conn, std::string &method,
                       std::string &path,
                       std::unordered_map<std::string, std::string> &headers);
  void HandleRequest(Connection &conn, const std::string &method,
                     const std::string &path,
                     const std::unordered_map<std::string, std::string> &headers,
                     const std::string &body);
  std::string HandleJsonRpc(const std::string &requestBody);
  void SendResponse(Connection &conn, int status, const char *statusText,
                    const std::string &contentType, const std::string &body);
  void CloseConnection(wxSocketBase *socket);
  static bool OriginAllowed(const std::string &origin);
 
  McpTools m_tools;
  std::unique_ptr<wxSocketServer> m_server;
  std::unordered_map<wxSocketBase *, std::unique_ptr<Connection>> m_connections;
  int m_listeningPort = -1;
};
 
#endif // MCPSERVER_H