/*
 * Copyright (C) 2016 Charybdis Development Team
 * Copyright (C) 2016 Jason Volk <jason@zemos.net>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice is present in all copies.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#pragma once
#define HAVE_IRCD_CLIENT_SOCKET_H

#include <boost/asio.hpp>
#include <boost/asio/ssl.hpp>
#include <boost/asio/steady_timer.hpp>
#include "ctx/continuation.h"

namespace ircd {

namespace asio = boost::asio;
namespace ip = asio::ip;

using boost::system::error_code;
using asio::steady_timer;

IRCD_EXCEPTION(error, nxdomain)

extern asio::ssl::context sslv23_client;

std::string string(const ip::address &);
ip::address address(const ip::tcp::endpoint &);
std::string hostaddr(const ip::tcp::endpoint &);
uint16_t port(const ip::tcp::endpoint &);

} // namespace ircd

namespace ircd {

struct socket
:std::enable_shared_from_this<socket>
{
	struct init;
	struct stat;
	struct scope_timeout;
	struct io;

	enum dc
	{
		RST,                // hardest disconnect
		FIN,                // graceful shutdown both directions
		FIN_SEND,           // graceful shutdown send side
		FIN_RECV,           // graceful shutdown recv side
		SSL_NOTIFY,         // SSL close_notify (async, errors ignored)
		SSL_NOTIFY_YIELD,   // SSL close_notify (yields context, throws)
	};

	struct stat
	{
		size_t bytes {0};
		size_t calls {0};
	};

	using message_flags = boost::asio::socket_base::message_flags;
	using handshake_type = asio::ssl::stream<ip::tcp::socket>::handshake_type;
	using handler = std::function<void (const error_code &) noexcept>;

	asio::ssl::stream<ip::tcp::socket> ssl;
	ip::tcp::socket &sd;
	steady_timer timer;
	stat in, out;
	bool timedout;

	void call_user(const handler &, const error_code &) noexcept;
	bool handle_error(const error_code &ec);
	void handle_timeout(std::weak_ptr<socket> wp, const error_code &ec);
	void handle(std::weak_ptr<socket>, handler, const error_code &, const size_t &) noexcept;

  public:
	operator const ip::tcp::socket &() const     { return sd;                                      }
	operator ip::tcp::socket &()                 { return sd;                                      }

	ip::tcp::endpoint remote() const             { return sd.remote_endpoint();                    }
	ip::tcp::endpoint local() const              { return sd.local_endpoint();                     }

	bool connected() const noexcept;
	auto available() const                       { return sd.available();                          }

	template<class duration> void set_timeout(const duration &, handler);
	template<class duration> void set_timeout(const duration &);

	template<class iov> auto read_some(const iov &, error_code &);
	template<class iov> auto read_some(const iov &);
	template<class iov> auto read(const iov &, error_code &);
	template<class iov> auto read(const iov &);

	template<class iov> auto write_some(const iov &, error_code &);
	template<class iov> auto write_some(const iov &);
	template<class iov> auto write(const iov &, error_code &);
	template<class iov> auto write(const iov &);

	// Asynchronous 'ready' closure
	void operator()(const milliseconds &timeout, handler);
	void operator()(handler);
	void cancel();

	void disconnect(const dc &type = dc::SSL_NOTIFY);
	void connect(const ip::tcp::endpoint &ep, const milliseconds &timeout = -1ms);

	socket(const std::string &host,
	       const uint16_t &port,
	       const milliseconds &timeout           = -1ms,
	       asio::ssl::context &ssl               = sslv23_client,
	       boost::asio::io_service *const &ios   = ircd::ios);

	socket(const ip::tcp::endpoint &remote,
	       const milliseconds &timeout           = -1ms,
	       asio::ssl::context &ssl               = sslv23_client,
	       boost::asio::io_service *const &ios   = ircd::ios);

	socket(asio::ssl::context &ssl               = sslv23_client,
	       boost::asio::io_service *const &ios   = ircd::ios);

	socket(socket &&) = delete;
	socket(const socket &) = delete;
	~socket() noexcept;
};

class socket::scope_timeout
{
	socket *s;

  public:
	scope_timeout(socket &, const milliseconds &timeout, const socket::handler &handler);
	scope_timeout(socket &, const milliseconds &timeout);
	scope_timeout(const scope_timeout &) = delete;
	scope_timeout &operator=(const scope_timeout &) = delete;
	~scope_timeout() noexcept;
};

class socket::io
{
	struct socket &sock;
	struct stat &stat;
	size_t bytes;

  public:
	operator size_t() const;

	io(struct socket &, struct stat &, const std::function<size_t ()> &closure);
};

struct socket::init
{
	init();
	~init() noexcept;
};

template<class iov> size_t write(socket &, const iov &);               // write_all
template<class iov> size_t write(socket &, iov &);                     // write_some
size_t write(socket &, const char *const &buf, const size_t &size);
size_t write(socket &, const string_view &);

template<class iov> size_t read(socket &, const iov &);                // read_all
template<class iov> size_t read(socket &, iov &);                      // read_some
size_t read(socket &, char *const &buf, const size_t &max);


template<class iov>
size_t
read(socket &socket,
     iov &bufs)
{
	const auto read(socket.read_some(bufs));
	const auto consumed(consume(bufs, read));
	assert(read == consumed);
	return read;
}

template<class iov>
size_t
read(socket &socket,
     const iov &bufs)
{
	return socket.read(bufs);
}

template<class iov>
size_t
write(socket &socket,
      iov &bufs)
{
	const auto wrote(socket.write_some(bufs));
	const auto consumed(consume(bufs, wrote));
	assert(wrote == consumed);
	return consumed;
}

template<class iov>
size_t
write(socket &socket,
      const iov &bufs)
{
	const auto wrote(socket.write(bufs));
	assert(wrote == size(bufs));
	return wrote;
}

inline
socket::io::io(struct socket &sock,
               struct stat &stat,
               const std::function<size_t ()> &closure)
:sock{sock}
,stat{stat}
,bytes{closure()}
{
	stat.bytes += bytes;
	stat.calls++;
}

inline
socket::io::operator size_t()
const
{
	return bytes;
}

template<class iov>
auto
socket::write(const iov &bufs)
{
	return io(*this, out, [&]
	{
		return async_write(ssl, bufs, asio::transfer_all(), yield(continuation()));
	});
}

template<class iov>
auto
socket::write(const iov &bufs,
              error_code &ec)
{
	return io(*this, out, [&]
	{
		return async_write(ssl, bufs, asio::transfer_all(), yield(continuation())[ec]);
	});
}

template<class iov>
auto
socket::write_some(const iov &bufs)
{
	return io(*this, out, [&]
	{
		return ssl.async_write_some(bufs, yield(continuation()));
	});
}

template<class iov>
auto
socket::write_some(const iov &bufs,
                   error_code &ec)
{
	return io(*this, out, [&]
	{
		return ssl.async_write_some(bufs, yield(continuation())[ec]);
	});
}

template<class iov>
auto
socket::read(const iov &bufs)
{
	return io(*this, in, [&]
	{
		const auto ret(async_read(ssl, bufs, yield(continuation())));

		if(unlikely(!ret))
			throw boost::system::system_error(boost::asio::error::eof);

		return ret;
	});
}

template<class iov>
auto
socket::read(const iov &bufs,
             error_code &ec)
{
	return io(*this, in, [&]
	{
		return async_read(ssl, bufs, yield(continuation())[ec]);
	});
}

template<class iov>
auto
socket::read_some(const iov &bufs)
{
	return io(*this, in, [&]
	{
		const auto ret(ssl.async_read_some(bufs, yield(continuation())));

		if(unlikely(!ret))
			throw boost::system::system_error(boost::asio::error::eof);

		return ret;
	});
}

template<class iov>
auto
socket::read_some(const iov &bufs,
                  error_code &ec)
{
	return io(*this, in, [&]
	{
		return ssl.async_read_some(bufs, yield(continuation())[ec]);
	});
}

template<class duration>
void
socket::set_timeout(const duration &t)
{
	if(t < duration(0))
		return;

	timer.expires_from_now(t);
	timer.async_wait(std::bind(&socket::handle_timeout, this, weak_from(*this), ph::_1));
}

template<class duration>
void
socket::set_timeout(const duration &t,
                    handler h)
{
	if(t < duration(0))
		return;

	timer.expires_from_now(t);
	timer.async_wait(std::move(h));
}

} // namespace ircd

inline uint16_t
ircd::port(const ip::tcp::endpoint &ep)
{
	return ep.port();
}

inline std::string
ircd::hostaddr(const ip::tcp::endpoint &ep)
{
	return string(address(ep));
}

inline std::string
ircd::string(const ip::address &addr)
{
	return addr.to_string();
}

inline boost::asio::ip::address
ircd::address(const ip::tcp::endpoint &ep)
{
	return ep.address();
}

inline
ircd::buffer::mutable_buffer::operator boost::asio::mutable_buffer()
const
{
	return boost::asio::mutable_buffer
	{
		data(*this), size(*this)
	};
}

inline
ircd::buffer::const_buffer::operator boost::asio::const_buffer()
const
{
	return boost::asio::const_buffer
	{
		data(*this), size(*this)
	};
}