#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