| 1 | //===--- Transport.h - sending and receiving LSP messages -------*- C++ -*-===// |
| 2 | // |
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | // |
| 9 | // The language server protocol is usually implemented by writing messages as |
| 10 | // JSON-RPC over the stdin/stdout of a subprocess. However other communications |
| 11 | // mechanisms are possible, such as XPC on mac. |
| 12 | // |
| 13 | // The Transport interface allows the mechanism to be replaced, and the JSONRPC |
| 14 | // Transport is the standard implementation. |
| 15 | // |
| 16 | //===----------------------------------------------------------------------===// |
| 17 | |
| 18 | #ifndef LLVM_CLANG_TOOLS_EXTRA_CLANGD_TRANSPORT_H |
| 19 | #define |
| 20 | |
| 21 | #include "Feature.h" |
| 22 | #include "llvm/ADT/StringRef.h" |
| 23 | #include "llvm/Support/JSON.h" |
| 24 | #include "llvm/Support/raw_ostream.h" |
| 25 | |
| 26 | namespace clang { |
| 27 | namespace clangd { |
| 28 | |
| 29 | // A transport is responsible for maintaining the connection to a client |
| 30 | // application, and reading/writing structured messages to it. |
| 31 | // |
| 32 | // Transports have limited thread safety requirements: |
| 33 | // - messages will not be sent concurrently |
| 34 | // - messages MAY be sent while loop() is reading, or its callback is active |
| 35 | class Transport { |
| 36 | public: |
| 37 | virtual ~Transport() = default; |
| 38 | |
| 39 | // Called by Clangd to send messages to the client. |
| 40 | virtual void notify(llvm::StringRef Method, llvm::json::Value Params) = 0; |
| 41 | virtual void call(llvm::StringRef Method, llvm::json::Value Params, |
| 42 | llvm::json::Value ID) = 0; |
| 43 | virtual void reply(llvm::json::Value ID, |
| 44 | llvm::Expected<llvm::json::Value> Result) = 0; |
| 45 | |
| 46 | // Implemented by Clangd to handle incoming messages. (See loop() below). |
| 47 | class MessageHandler { |
| 48 | public: |
| 49 | virtual ~MessageHandler() = default; |
| 50 | // Handler returns true to keep processing messages, or false to shut down. |
| 51 | virtual bool onNotify(llvm::StringRef Method, llvm::json::Value) = 0; |
| 52 | virtual bool onCall(llvm::StringRef Method, llvm::json::Value Params, |
| 53 | llvm::json::Value ID) = 0; |
| 54 | virtual bool onReply(llvm::json::Value ID, |
| 55 | llvm::Expected<llvm::json::Value> Result) = 0; |
| 56 | }; |
| 57 | // Called by Clangd to receive messages from the client. |
| 58 | // The transport should in turn invoke the handler to process messages. |
| 59 | // If handler returns false, the transport should immediately exit the loop. |
| 60 | // (This is used to implement the `exit` notification). |
| 61 | // Otherwise, it returns an error when the transport becomes unusable. |
| 62 | virtual llvm::Error loop(MessageHandler &) = 0; |
| 63 | }; |
| 64 | |
| 65 | // Controls the way JSON-RPC messages are encoded (both input and output). |
| 66 | enum JSONStreamStyle { |
| 67 | // Encoding per the LSP specification, with mandatory Content-Length header. |
| 68 | Standard, |
| 69 | // Messages are delimited by a '---' line. Comment lines start with #. |
| 70 | Delimited |
| 71 | }; |
| 72 | |
| 73 | // Returns a Transport that speaks JSON-RPC over a pair of streams. |
| 74 | // The input stream must be opened in binary mode. |
| 75 | // If InMirror is set, data read will be echoed to it. |
| 76 | // |
| 77 | // The use of C-style std::FILE* input deserves some explanation. |
| 78 | // Previously, std::istream was used. When a debugger attached on MacOS, the |
| 79 | // process received EINTR, the stream went bad, and clangd exited. |
| 80 | // A retry-on-EINTR loop around reads solved this problem, but caused clangd to |
| 81 | // sometimes hang rather than exit on other OSes. The interaction between |
| 82 | // istreams and signals isn't well-specified, so it's hard to get this right. |
| 83 | // The C APIs seem to be clearer in this respect. |
| 84 | std::unique_ptr<Transport> |
| 85 | newJSONTransport(std::FILE *In, llvm::raw_ostream &Out, |
| 86 | llvm::raw_ostream *InMirror, bool Pretty, |
| 87 | JSONStreamStyle = JSONStreamStyle::Standard); |
| 88 | |
| 89 | #if CLANGD_BUILD_XPC |
| 90 | // Returns a Transport for macOS based on XPC. |
| 91 | // Clangd with this transport is meant to be run as bundled XPC service. |
| 92 | std::unique_ptr<Transport> newXPCTransport(); |
| 93 | #endif |
| 94 | |
| 95 | } // namespace clangd |
| 96 | } // namespace clang |
| 97 | |
| 98 | #endif |
| 99 | |