#ifndef WEBCC_HTTP_SOCKET_H_
#define WEBCC_HTTP_SOCKET_H_

#include <vector>

#include "boost/asio/ip/tcp.hpp"
#include "boost/asio/ssl.hpp"

#include "webcc/http_request.h"

namespace webcc {

// -----------------------------------------------------------------------------

class HttpSocketBase {
public:
  virtual ~HttpSocketBase() = default;

  typedef boost::asio::ip::tcp::resolver::results_type Endpoints;

  typedef std::function<void(boost::system::error_code, std::size_t)>
      ReadHandler;

  // TODO: Remove |host|
  virtual void Connect(const std::string& host,
                       const Endpoints& endpoints,
                       boost::system::error_code* ec) = 0;

  virtual void Write(const HttpRequest& request,
                     boost::system::error_code* ec) = 0;

  virtual void AsyncReadSome(ReadHandler&& handler,
                             std::vector<char>* buffer) = 0;

  virtual void Close(boost::system::error_code* ec) = 0;
};

// -----------------------------------------------------------------------------

class HttpSocket : public HttpSocketBase {
public:
  explicit HttpSocket(boost::asio::io_context& io_context);

  void Connect(const std::string& host,
               const Endpoints& endpoints,
               boost::system::error_code* ec) final;

  void Write(const HttpRequest& request,
             boost::system::error_code* ec) final;

  void AsyncReadSome(ReadHandler&& handler, std::vector<char>* buffer) final;

  void Close(boost::system::error_code* ec) final;

private:
  boost::asio::ip::tcp::socket socket_;
};

// -----------------------------------------------------------------------------

class HttpSslSocket : public HttpSocketBase {
public:
  explicit HttpSslSocket(boost::asio::io_context& io_context,
                         bool ssl_verify = true);

  void Connect(const std::string& host,
               const Endpoints& endpoints,
               boost::system::error_code* ec) final;

  void Write(const HttpRequest& request,
             boost::system::error_code* ec) final;

  void AsyncReadSome(ReadHandler&& handler, std::vector<char>* buffer) final;

  void Close(boost::system::error_code* ec) final;

private:
  void Handshake(const std::string& host, boost::system::error_code* ec);

  boost::asio::ssl::context ssl_context_;

  boost::asio::ssl::stream<boost::asio::ip::tcp::socket> ssl_socket_;

  // Verify the certificate of the peer (remote server) or not.
  bool ssl_verify_;
};

}  // namespace webcc

#endif  // WEBCC_HTTP_SOCKET_H_