pylsp_jsonrpc_server_with_method_dispatcher_stdin_stdout.py

1import sys
2from pylsp_jsonrpc.dispatchers import MethodDispatcher
3from pylsp_jsonrpc.endpoint import Endpoint
4from pylsp_jsonrpc.streams import JsonRpcStreamReader, JsonRpcStreamWriter
5
6class ExampleDispatcher(MethodDispatcher):
7    def m_add(self, a, b):
8        return a + b
9
10def main():
11    # Read from stdin and write to stdout
12    reader = JsonRpcStreamReader(sys.stdin.buffer)
13    writer = JsonRpcStreamWriter(sys.stdout.buffer)
14    
15    # Create the dispatcher and the endpoint
16    dispatcher = ExampleDispatcher()
17    endpoint = Endpoint(dispatcher, writer.write)
18    
19    # Start the reader loop
20    reader.listen(endpoint.consume)
21
22if __name__ == "__main__":
23    main()