McpServer.hpp

浏览该文件的文档.

#pragma once
 
#include "common/Export.h"
 
#include <atomic>
#include <cstdint>
#include <iostream>
#include <istream>
#include <memory>
#include <mutex>
#include <ostream>
#include <string>
#include <thread>
#include <vector>
 
namespace Poco::Net {
class ServerSocket;
class StreamSocket;
}
 
namespace eve::dev {
 
class EVENGINE_API McpServer {
public:
    static McpServer& instance();
 
    McpServer(const McpServer&)            = delete;
    McpServer& operator=(const McpServer&) = delete;
 
    enum class Transport { None, Tcp, Stdio };
 
    int  listen(uint16_t port);
    bool listenStdio(std::istream& in = std::cin, std::ostream& out = std::cout);
    void stop();
    bool isListening() const { return listening_.load(); }
    int  port() const { return port_.load(); }
    bool hasClient() const { return hasClient_.load(); }
    Transport transport() const { return transport_.load(); }
    bool stdinClosed() const { return stdinClosed_.load(); }
 
    void poll();
 
    void setGameRoot(std::string root);
    const std::string& gameRoot() const { return gameRoot_; }
 
private:
    // Immortal<McpServer> constructs the singleton (devtools/Immortal.hpp).
    template <typename> friend struct Immortal;
    McpServer();
    ~McpServer();
 
    void acceptNonBlocking();
    void readAndDispatch();
    bool sendLine(const std::string& json);
    void handleMessage(const std::string& json);
 
    std::atomic<bool> listening_{false};
    std::atomic<bool> hasClient_{false};
    std::atomic<int>  port_{0};
    std::atomic<Transport> transport_{Transport::None};
    std::atomic<bool> stdinClosed_{false};
    std::mutex        ioMu_;
    std::unique_ptr<Poco::Net::ServerSocket> server_;
    std::unique_ptr<Poco::Net::StreamSocket> client_;
    std::istream*     stdioIn_  = nullptr;
    std::ostream*     stdioOut_ = nullptr;
    std::vector<std::string> stdioQueue_;
    std::thread       stdioReader_;
    bool              joinReader_ = false;
    std::string       recvBuf_;
    std::string       gameRoot_;
    bool              initialized_ = false;
    std::string       protocolVersion_ = "2025-06-18";
};
 
}  // namespace eve::dev