dbaa82b384
This moves all the generic MCP code into the new ProtocolMCP library so it can be shared between the MCP implementation in LLDB (built on the private API) and lldb-mcp (built on the public API). This PR doesn't include the actual MCP server. The reason is that the transport mechanism will be different between the LLDB implementation (using sockets) and the lldb-mcp implementation (using stdio). The goal s to abstract away that difference by making the server use JSONTransport (again) but I'm saving that for a separate PR.
113 lines
3.7 KiB
C++
113 lines
3.7 KiB
C++
//===- ProtocolServerMCP.h ------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLDB_PLUGINS_PROTOCOL_MCP_PROTOCOLSERVERMCP_H
|
|
#define LLDB_PLUGINS_PROTOCOL_MCP_PROTOCOLSERVERMCP_H
|
|
|
|
#include "lldb/Core/ProtocolServer.h"
|
|
#include "lldb/Host/MainLoop.h"
|
|
#include "lldb/Host/Socket.h"
|
|
#include "lldb/Protocol/MCP/Protocol.h"
|
|
#include "lldb/Protocol/MCP/Resource.h"
|
|
#include "lldb/Protocol/MCP/Tool.h"
|
|
#include "llvm/ADT/StringMap.h"
|
|
#include <thread>
|
|
|
|
namespace lldb_private::mcp {
|
|
|
|
class ProtocolServerMCP : public ProtocolServer {
|
|
public:
|
|
ProtocolServerMCP();
|
|
virtual ~ProtocolServerMCP() override;
|
|
|
|
virtual llvm::Error Start(ProtocolServer::Connection connection) override;
|
|
virtual llvm::Error Stop() override;
|
|
|
|
static void Initialize();
|
|
static void Terminate();
|
|
|
|
static llvm::StringRef GetPluginNameStatic() { return "MCP"; }
|
|
static llvm::StringRef GetPluginDescriptionStatic();
|
|
|
|
static lldb::ProtocolServerUP CreateInstance();
|
|
|
|
llvm::StringRef GetPluginName() override { return GetPluginNameStatic(); }
|
|
|
|
Socket *GetSocket() const override { return m_listener.get(); }
|
|
|
|
protected:
|
|
using RequestHandler =
|
|
std::function<llvm::Expected<lldb_protocol::mcp::Response>(
|
|
const lldb_protocol::mcp::Request &)>;
|
|
using NotificationHandler =
|
|
std::function<void(const lldb_protocol::mcp::Notification &)>;
|
|
|
|
void AddTool(std::unique_ptr<lldb_protocol::mcp::Tool> tool);
|
|
void AddResourceProvider(
|
|
std::unique_ptr<lldb_protocol::mcp::ResourceProvider> resource_provider);
|
|
|
|
void AddRequestHandler(llvm::StringRef method, RequestHandler handler);
|
|
void AddNotificationHandler(llvm::StringRef method,
|
|
NotificationHandler handler);
|
|
|
|
private:
|
|
void AcceptCallback(std::unique_ptr<Socket> socket);
|
|
|
|
llvm::Expected<std::optional<lldb_protocol::mcp::Message>>
|
|
HandleData(llvm::StringRef data);
|
|
|
|
llvm::Expected<lldb_protocol::mcp::Response>
|
|
Handle(lldb_protocol::mcp::Request request);
|
|
void Handle(lldb_protocol::mcp::Notification notification);
|
|
|
|
llvm::Expected<lldb_protocol::mcp::Response>
|
|
InitializeHandler(const lldb_protocol::mcp::Request &);
|
|
|
|
llvm::Expected<lldb_protocol::mcp::Response>
|
|
ToolsListHandler(const lldb_protocol::mcp::Request &);
|
|
llvm::Expected<lldb_protocol::mcp::Response>
|
|
ToolsCallHandler(const lldb_protocol::mcp::Request &);
|
|
|
|
llvm::Expected<lldb_protocol::mcp::Response>
|
|
ResourcesListHandler(const lldb_protocol::mcp::Request &);
|
|
llvm::Expected<lldb_protocol::mcp::Response>
|
|
ResourcesReadHandler(const lldb_protocol::mcp::Request &);
|
|
|
|
lldb_protocol::mcp::Capabilities GetCapabilities();
|
|
|
|
llvm::StringLiteral kName = "lldb-mcp";
|
|
llvm::StringLiteral kVersion = "0.1.0";
|
|
|
|
bool m_running = false;
|
|
|
|
MainLoop m_loop;
|
|
std::thread m_loop_thread;
|
|
|
|
std::unique_ptr<Socket> m_listener;
|
|
std::vector<MainLoopBase::ReadHandleUP> m_listen_handlers;
|
|
|
|
struct Client {
|
|
lldb::IOObjectSP io_sp;
|
|
MainLoopBase::ReadHandleUP read_handle_up;
|
|
std::string buffer;
|
|
};
|
|
llvm::Error ReadCallback(Client &client);
|
|
std::vector<std::unique_ptr<Client>> m_clients;
|
|
|
|
std::mutex m_server_mutex;
|
|
llvm::StringMap<std::unique_ptr<lldb_protocol::mcp::Tool>> m_tools;
|
|
std::vector<std::unique_ptr<lldb_protocol::mcp::ResourceProvider>>
|
|
m_resource_providers;
|
|
|
|
llvm::StringMap<RequestHandler> m_request_handlers;
|
|
llvm::StringMap<NotificationHandler> m_notification_handlers;
|
|
};
|
|
} // namespace lldb_private::mcp
|
|
|
|
#endif
|