#include "server.hpp"

using namespace boost::asio::ip;
using namespace boost::asio;

Server::Server(io_service& io_service, const ListenSettings& listen_settings)
: listen_settings(listen_settings)
, v4_acceptor(io_service, address_v4::from_string(listen_settings.ipv4_inaddr), listen_settings.port)
, v6_acceptor(io_service, address_v6::from_string(listen_settings.ipv6_inaddr), listen_settings.port)
{
	/*
	 * We can't bind both a v4 and v6 socket to both inaddr4_any and inaddr6_any.
	 * So in case both ipv4 and ipv6 are enabled and both want to bind to their 
	 * respective inaddr_any we need to bind to v6 any and disable ipv6 only on
	 * the acceptor socket, else we can just bind to the interfaces normally.
	 */
	bool bind_v4_any = v4_acceptor.get_address().to_v4() == address_v4::any() && listen_settings.enable_ipv4;
	bool bind_v6_any = v6_acceptor.get_address().to_v6() == address_v6::any() && listen_settings.enable_ipv6;
	
	if(bind_v4_any && bind_v6_any) {
		v6_acceptor.listen_v6_and_v4_any();
	} else if((bind_v4_any ^ bind_v6_any) && listen_settings.enable_ipv4 && listen_settings.enable_ipv6) {
		throw std::runtime_error("Cannot bind an INADDR_ANY and a non INADDR_ANY address on ipv4 and ipv6");
	} else {
		if(listen_settings.enable_ipv4) {
			v4_acceptor.listen_socket();
		}
		if(listen_settings.enable_ipv6) {
			v6_acceptor.listen_socket();
		}
	}
}

Server::Server(io_service& io_service, const ListenSettings& listen_settings, AcceptHandler accept_handler)
: Server(io_service, listen_settings)
{
	if(v4_acceptor.is_open()) {
		v4_acceptor.start_accepting(accept_handler);
	}
	if(v6_acceptor.is_open()) {
		v6_acceptor.start_accepting(accept_handler);
	}
}

Server::Server(io_service& io_service, const ListenSettings& listen_settings, std::shared_ptr<ssl::context> ctx, SSLAcceptHandler accept_handler)
: Server(io_service, listen_settings)
{
	if(v4_acceptor.is_open()) {
		v4_acceptor.start_accepting(ctx, accept_handler);
	}
	if(v6_acceptor.is_open()) {
		v6_acceptor.start_accepting(ctx, accept_handler);
	}
}