// Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
#pragma once
#include <boost/beast/core/flat_buffer.hpp>
#include <boost/asio.hpp>
#include <boost/asio/ip/tcp.hpp>
#include "Message.pb.h"

namespace aws { namespace iot { namespace securedtunneling { namespace connection {
    class tcp_client
    {
    public:
        typedef boost::shared_ptr<tcp_client> pointer;
        tcp_client(boost::asio::io_context & io_context, std::size_t write_buf_size, std::size_t read_buf_size,  std::size_t ws_write_buf_size)
                : resolver_(io_context)
        {

        }
        static pointer create(boost::asio::io_context& io_context, std::size_t const & write_buf_size, std::size_t const & read_buf_size, std::size_t const & ws_write_buf_size)
        {
            return pointer(new tcp_client(io_context, write_buf_size, read_buf_size, ws_write_buf_size));
        }

        tcp::resolver                           resolver_;

        std::unordered_map<uint32_t, tcp_connection::pointer> connectionId_to_tcp_connection_map;

        // function object defines what to do after set up a tcp socket
        std::function<void()>                   after_setup_tcp_socket = nullptr;

        // function object defines what to do receiving control message: stream start
        std::function<void()>                   on_receive_stream_start = nullptr;
    };
}}}}