summaryrefslogtreecommitdiff
path: root/graphics/asymptote/LspCpp/include/LibLsp/JsonRpc/WebSocketServer.h
blob: 5206525cebea1f8d51f2ee6ed7b27a7e1d9346cb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#pragma once

#include <boost/asio.hpp>
#include <string>
#include <boost/beast/core/tcp_stream.hpp>
#include <boost/beast/websocket/stream.hpp>


#include "RemoteEndPoint.h"
#include "stream.h"
#include "threaded_queue.h"

namespace lsp {
    class Log;
}


namespace lsp
{



        class websocket_stream_wrapper :public istream, public ostream
        {
        public:

            websocket_stream_wrapper(boost::beast::websocket::stream<boost::beast::tcp_stream>& _w);

            boost::beast::websocket::stream<boost::beast::tcp_stream>& ws_;
            std::atomic<bool> quit{};
            std::shared_ptr < MultiQueueWaiter> request_waiter;
            ThreadedQueue< char > on_request;
            std::string error_message;
            bool fail() override;

            bool eof() override;

            bool good() override;

            websocket_stream_wrapper& read(char* str, std::streamsize count) override;

            int get() override;

            bool bad() override;

            websocket_stream_wrapper& write(const std::string& c) override;

            websocket_stream_wrapper& write(std::streamsize _s) override;

            websocket_stream_wrapper& flush() override;

            void clear() override;

            std::string what() override;
        };

        /// The top-level class of the HTTP server.
        class WebSocketServer
        {
        public:
            WebSocketServer(const WebSocketServer&) = delete;
            WebSocketServer& operator=(const WebSocketServer&) = delete;
            ~WebSocketServer();
            /// Construct the server to listen on the specified TCP address and port, and
            /// serve up files from the given directory.
            explicit WebSocketServer(const std::string& user_agent, const std::string& address, const std::string& port,
                std::shared_ptr < MessageJsonHandler> json_handler,
                std::shared_ptr < Endpoint> localEndPoint, lsp::Log& ,uint32_t _max_workers = 2);

            /// Run the server's io_context loop.
            void run();
            void stop();

            RemoteEndPoint point;
        private:
            struct Data;
            /// Perform an asynchronous accept operation.
            void do_accept();

            /// Wait for a request to stop the server.
            void do_stop();
            Data* d_ptr = nullptr;


        };

    } // namespace