// Silverkin Industries Comm-Link Server, Engineering Sample Alpha 0.1.
// PROJECT CODENAME: WHAT DO I PAY YOU FOR? | Level-3 Clearance.
// Barry Kane, 2021
#include <netdb.h>
#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include "misc/texteffects.h"
const int PORT = 5000;
const int MAX = 1024;
typedef struct sockaddr sockaddr;
	
int main()
{
	int socketFileDesc, connectionFileDesc, length, clientsAmount,
		socketCheck, activityCheck, readLength;
	int clientSockets[64];
	int maxClients = 64;
	char receiveBuffer[MAX];
	fd_set connectedClients;
	struct sockaddr_in serverAddress, clientAddress;	

	// Give an intro: Display the Silverkin Industries logo and splash text.
	slowprint(logostring, 3000);
	slowprint("\n--==== \033[33;40mSILVERKIN INDUSTRIES\033[0m COMM-LINK SERVER ====--\nVersion Alpha 0.1\n", 5000);
	 
	// Initialize the sockets to 0, so we don't crash.
	for (int index = 0; index < maxClients; index++)  
	{  
		clientSockets[index] = 0;  
	}
	
	// Get a socket and make sure we actually get one.
	socketFileDesc = socket(AF_INET, SOCK_STREAM, 0);
	if (socketFileDesc == -1)
	{
		perror("Socket creation is \033[33;40mRED.\033[0m Aborting launch.\n");
		exit(0);
	}

	else
	{
		slowprint(" Socket creation is \033[32;40mGREEN.\033[0m\n", 5000);
	}

	bzero(&serverAddress, sizeof(serverAddress));
  
	// Assign IP and port:
	serverAddress.sin_family = AF_INET;
	serverAddress.sin_addr.s_addr = htonl(INADDR_ANY);
	serverAddress.sin_port = htons(PORT);
  
	// Binding newly created socket to given IP, and checking it works:
	if ((bind(socketFileDesc, (sockaddr*)&serverAddress, sizeof(serverAddress))) != 0)
	{
		perror("Socket binding is \033[33;40mRED.\033[0m Aborting launch.\n");
		exit(0);
	}
	else
	{
		slowprint(" Socket binding is \033[32;40mGREEN.\033[0m\n", 5000);
	}
	
	// Let's start listening:
	if ((listen(socketFileDesc, 64)) != 0)
	{
		perror("Server listening is \033[33;40mRED.\033[0m Aborting launch.\n");
		exit(0);
	}
	else
	{
		slowprint(" Server listening is \033[32;40mGREEN.\033[0m\n", 5000);
	}
	length = sizeof(clientAddress);

	//connectionFileDesc = accept(socketFileDesc, (sockaddr*)&clientAddress, &length);
	// Accept the data packet from client and verification
	while (1)
	{
		FD_ZERO(&connectedClients);
		FD_SET(socketFileDesc, &connectedClients);
		clientsAmount = socketFileDesc;
		bzero(receiveBuffer, sizeof(receiveBuffer));
		for (int i = 0; i < maxClients; i++)  
		{  
			// Just get the one we're working with to another name:
			socketCheck = clientSockets[i];  
                 
			// If it's working, bang it into the list:
			if(socketCheck > 0)
			{
				FD_SET(socketCheck, &connectedClients);  
			}
			// The amount of clients is needed for select():
			if(socketCheck > clientsAmount)
			{
				clientsAmount = socketCheck;
			}
		}

		// See if a connection is ready to be interacted with:
		activityCheck = select((clientsAmount + 1), &connectedClients, NULL, NULL, NULL);

		// Check if select() worked:
	        if ((activityCheck < 0) && (errno != EINTR))  
		{ 
			perror("Error in select(), retrying.\n");  
		}

		// If it's the master socket selected, there is a new connection:
		if (FD_ISSET(socketFileDesc, &connectedClients))  
		{  
			if ((connectionFileDesc = accept(socketFileDesc, 
							 (struct sockaddr *)&clientAddress, (socklen_t*)&length))<0)  
			{  
				perror("Failed to accept connection. Aborting.\n");  
				exit(EXIT_FAILURE);  
			}  
             
			// Print new connection details:
			printf("Client connected: Socket file descriptor: #%d, IP address: %s, Port: %d.\n",
			       connectionFileDesc, inet_ntoa(clientAddress.sin_addr) , ntohs
			       (clientAddress.sin_port));  
           
			// See if we can put in the client:
			for (int i = 0; i < maxClients; i++)  
			{  
				// When there is an empty slot, pop it in:
				if( clientSockets[i] == 0 )  
				{  
					clientSockets[i] = connectionFileDesc;  
					printf("Adding to list of sockets as %d.\n" , i);                           
					break;  
				}  
			}  
		}
		else
		{
			// Otherwise, it's a client socket to be interacted with:
			for (int i = 0; i < maxClients; i++)  
			{  
				socketCheck = clientSockets[i];  
                 
				if (FD_ISSET(socketCheck, &connectedClients))  
				{  
					//Check if it was for closing, and also read the incoming message
					explicit_bzero(receiveBuffer, sizeof(receiveBuffer));
					readLength = read(socketCheck, receiveBuffer, sizeof(receiveBuffer));
					if (readLength == 0)
					{
						// Somebody disconnected , get his details and print:
						getpeername(socketCheck, (struct sockaddr*)&clientAddress, (socklen_t*)&length);  
						printf("Client disconnected: IP Address: %s, Port: %d.\n", 
						       inet_ntoa(clientAddress.sin_addr) , ntohs(clientAddress.sin_port));  
						
						// Close the socket and mark as 0 in list for reuse:
						close(socketCheck);  
						clientSockets[i] = 0;  
					}  
                     
					// Echo back the message that came in:
					else 
					{  
						printf("%d: %s", clientSockets[i], receiveBuffer);
						fflush(stdout);
						for (int sendIndex = 0; sendIndex < clientsAmount; sendIndex++)
						{
							if(sendIndex != i && clientSockets[sendIndex] != STDIN_FILENO && clientSockets[sendIndex] != STDOUT_FILENO && clientSockets[sendIndex] != STDERR_FILENO)
							{
								write(clientSockets[sendIndex], receiveBuffer, sizeof(receiveBuffer));
							}
						}
						bzero(receiveBuffer, sizeof(receiveBuffer));
					}  
				}
			}
		}
	}
}