#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include "net.h"

void *net_recv(int sock, msg_type_t type)
{
	msg_t m;
	void *result;
	ssize_t len = recv(sock, &m.hdr, sizeof(msg_header_t), MSG_PEEK); // just peek into packet to determine message header

	assert(len != -1);

	if(m.hdr.type != type)
	{
		printf("net_recv: received message type %d instead of %d", m.hdr.type, type);
		return NULL;
	}

	m.payload = malloc(m.hdr.payload_length);	// Allocate space for message payload
	recv(sock, &m.hdr, sizeof(msg_header_t), 0);	// Remove message header from socket
	recv(sock, m.payload, m.hdr.payload_length, 0);// And then receive the payload

	switch(type)
	{
		case msg_type_hello:
			result = server_recv_hello(m.payload, m.hdr.payload_length);
			break;
		case msg_type_start_game:
			result = client_recv_player_list(m.payload, m.hdr.payload_length);
			break;
		case msg_type_deal_cards:
			result = client_recv_deal_cards(m.payload, m.hdr.payload_length);
			break;
		case msg_type_init_stacks:
			printf("not yet implemented: msg_type_init_stacks\n");
			exit(EXIT_FAILURE);
			break;
		case msg_type_selected_card:
			result = server_recv_selected_card(m.payload, m.hdr.payload_length);
			break;
		case msg_type_selected_stack_c:
			result = server_recv_selected_stack(m.payload, m.hdr.payload_length);
			break;
		case msg_type_selected_stack_s:
			result = client_recv_selected_stack(m.payload, m.hdr.payload_length);
			break;
		default:
			printf("net_recv: Unknown message type %d received!\n", type);
			exit(EXIT_FAILURE);
			break;
	}

	free(m.payload);

	return result;
}