#include<sys/socket.h>
#include<sys/types.h>
#include<netinet/ip.h>
#include<arpa/inet.h>
#include<stdio.h>
#include<unistd.h>
#include<signal.h>
#include<stdlib.h>
#include<string.h>

int skt;
void shut_down_server(int signal){
	printf("\nDoing everything we need to do to shut down the server\n");
	close(skt);
	exit(0);
}

/* Protocol:
 * The server will send out:
 * 	(2 bytes, unsigned integer in little-endian order)   message length
 * 	(variable, message length bytes)                     ASCII string, not null terminated
 * 	(4 bytes, signed integer in little-endian order)     moosenum
 * 	(8 bytes, 64-bit float point, C double)              mooseweight
 */

void send_data_to_client(int client_fd){
	char* message = "There are moose swimming in the hot tub";
	size_t total_written = 0;
	uint16_t messagelen = strlen(message);
	write(client_fd, &messagelen, 2); // this'll be sent in little endian
	ssize_t writelen = write(client_fd, message, messagelen);
	total_written += writelen;
	printf("writelen = %ld\n", writelen);
	usleep(10000);
	if(writelen != strlen(message)){
		return;
	}
	int32_t moosenum = 35;
	writelen = write(client_fd, &moosenum, 4); // also little endian
	total_written += writelen;
	// we could, and probably should, check to see if writelen == sizeof(moosenum)
	double mooseweight = 1408.32;
	writelen = write(client_fd, &mooseweight, sizeof(mooseweight));
	total_written += writelen;
	// again, should check writelen == sizeof(mooseweight)
	printf("In total, we wrote %lu bytes\n", total_written);
}

int main(int argc, char ** argv){
	uint16_t listen_port = 5141; // 5141 is the default
	if(argc > 1){
		listen_port = atoi(argv[1]);
		if(listen_port < 1){
			printf("Invalid port:  %s\n", argv[1]);
			exit(1);
		}
		printf("Will listen on port %d\n", listen_port);
	} else {
		printf("No port specified, defaulting to %d\n", listen_port);
	}	
	struct sigaction sa;
	sa.sa_handler = shut_down_server;
	sigaction(SIGINT, &sa, 0);

	struct sockaddr_in sad;
	sad.sin_port = htons(listen_port);
	sad.sin_addr.s_addr = INADDR_ANY;
	sad.sin_family = AF_INET;

	skt = socket(AF_INET, SOCK_STREAM, 0); // Step 1
	if(skt == -1){
		perror("socket");
		return 1;
	}
	if( bind(skt, (struct sockaddr *)(&sad), sizeof(struct sockaddr_in)) ){ // step 2
		perror("bind");
		return 1;
	}
	if( listen(skt, 5) ){ // step 3
		perror("listen");
		return 1;
	}	
	int client_fd;
	struct sockaddr_in client_address;
	socklen_t address_size = sizeof(struct sockaddr_in);
	while(1){
		client_fd = accept(skt, (struct sockaddr *)(&client_address), &address_size); // step 4
		printf("Connection made from address %s\n", inet_ntoa(client_address.sin_addr));
		send_data_to_client(client_fd);
		close(client_fd);
	}
	return 0;
}