#include <errno.h>
#include <netinet/in.h>

#include <wasi/descriptor_table.h>
#include <wasi/sockets_utils.h>

static int tcp_socket(network_ip_address_family_t family, bool blocking)
{
	tcp_create_socket_error_code_t error;
	tcp_own_tcp_socket_t socket;
	if (!tcp_create_socket_create_tcp_socket(family, &socket, &error)) {
		errno = __wasi_sockets_utils__map_error(error);
		return -1;
	}

	tcp_borrow_tcp_socket_t socket_borrow = tcp_borrow_tcp_socket(socket);
	poll_own_pollable_t socket_pollable =
		tcp_method_tcp_socket_subscribe(socket_borrow);

	descriptor_table_entry_t
		entry = { .tag = DESCRIPTOR_TABLE_ENTRY_TCP_SOCKET,
			  .tcp_socket = {
				  .socket = socket,
				  .socket_pollable = socket_pollable,
				  .blocking = blocking,
				  .fake_nodelay = false,
				  .family = family,
				  .state = { .tag = TCP_SOCKET_STATE_UNBOUND,
					     .unbound = {
						     /* No additional state. */ } },
			  } };

	int fd;
	if (!descriptor_table_insert(entry, &fd)) {
		errno = EMFILE;
		return -1;
	}
	return fd;
}

static int udp_socket(network_ip_address_family_t family, bool blocking)
{
	udp_create_socket_error_code_t error;
	udp_own_udp_socket_t socket;
	if (!udp_create_socket_create_udp_socket(family, &socket, &error)) {
		errno = __wasi_sockets_utils__map_error(error);
		return -1;
	}

	udp_borrow_udp_socket_t socket_borrow = udp_borrow_udp_socket(socket);
	poll_own_pollable_t socket_pollable =
		udp_method_udp_socket_subscribe(socket_borrow);

	descriptor_table_entry_t
		entry = { .tag = DESCRIPTOR_TABLE_ENTRY_UDP_SOCKET,
			  .udp_socket = {
				  .socket = socket,
				  .socket_pollable = socket_pollable,
				  .blocking = blocking,
				  .family = family,
				  .state = { .tag = UDP_SOCKET_STATE_UNBOUND,
					     .unbound = {
						     /* No additional state. */ } },
			  } };

	int fd;
	if (!descriptor_table_insert(entry, &fd)) {
		errno = EMFILE;
		return -1;
	}
	return fd;
}

int socket(int domain, int type, int protocol)
{
	network_ip_address_family_t family;
	switch (domain) {
	case PF_INET:
		family = NETWORK_IP_ADDRESS_FAMILY_IPV4;
		break;

	case PF_INET6:
		family = NETWORK_IP_ADDRESS_FAMILY_IPV6;
		break;

	default:
		errno = EAFNOSUPPORT;
		return -1;
	}

	int real_type = type & ~(SOCK_NONBLOCK | SOCK_CLOEXEC);
	bool blocking = (type & SOCK_NONBLOCK) == 0;
	// Ignore SOCK_CLOEXEC flag. That concept does not exist in WASI.

	if (real_type == SOCK_STREAM &&
	    (protocol == 0 || protocol == IPPROTO_TCP)) {
		return tcp_socket(family, blocking);

	} else if (real_type == SOCK_DGRAM &&
		   (protocol == 0 || protocol == IPPROTO_UDP)) {
		return udp_socket(family, blocking);

	} else {
		errno = EPROTONOSUPPORT;
		return -1;
	}
}