#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>
#include <termios.h>
#include <time.h>

#include "Clean.h"

#define INITIAL_BUFFERSIZE 2
#define die(s) {perror(s);exit(EXIT_FAILURE);}

static speed_t baudrates[] = {B0, B50, B75, B110, B134, B150, B200, B300, B600,
	B1200, B1800, B2400, B4800, B9600, B19200, B38400, B57600, B115200,
	B230400};
static int bytesizes[4] = {CS5, CS6, CS7, CS8};
static char *error = "";

struct termioslist
{
	int fd;
	struct termios to;
	struct termioslist *next;
};

struct termioslist *head = NULL;

static struct termios *getTermios(int fd)
{
	struct termioslist *h = head;
	while(h != NULL)
		if(h->fd == fd)
			return &h->to;
	return NULL;
}

static void remTermios(int fd)
{
	struct termioslist *beforeit = NULL;
	struct termioslist *it = head;
	while(it != NULL){
		if(it->fd == fd){
			if(beforeit == NULL)
				head = it->next;
			else
				beforeit->next = it->next;
			free(it);
			break;
		}
		beforeit = it;
		it = it->next;
	}
}

static char *cleanStringToCString(CleanString s)
{
	unsigned long len = CleanStringLength(s);
	char *cs = (char *)malloc(len+1);
	if(cs == NULL)
		die("malloc");
	memcpy(cs, CleanStringCharacters(s), len);
	cs[len] = '\0';
	return cs;
}

static void addTermios(int fd, struct termios *t)
{
	struct termioslist *new = malloc(sizeof(struct termioslist));
	if(new == NULL)
		die("malloc");
	new->fd = fd;
	memcpy(&new->to, t, sizeof(struct termios));
	new->next = NULL;
	if(head == NULL){
		head = new;
	} else {
		struct termioslist *h = head;
		while(h->next != NULL)
			h = h->next;
		h->next = new;
	}
}

void ttyopen(CleanString fn, int baudrate, int bytesize, int parity,
	int stopbits, int xonoff, int *status, int *fd)
{
	struct termios tio;
	char *cs_fn = cleanStringToCString(fn);
	*fd = open(cs_fn, O_RDWR | O_NOCTTY);
	*status = 0;
	if(*fd < 0){
		error = strerror(errno);
	} else {
		//Get
		tcgetattr(*fd, &tio);
		addTermios(*fd, &tio);
		//Baudrate
		cfsetispeed(&tio, baudrates[baudrate]);
		//Bytesize
		tio.c_cflag &= ~CSIZE;
		tio.c_cflag |= bytesizes[bytesize];
		//Parity
		if(parity == 0) {
			tio.c_cflag &= ~PARENB | ~INPCK;
		} else if(parity == 1) {
			tio.c_cflag |= PARODD | PARENB;
		} else if(parity == 2) {
			tio.c_cflag |= PARENB;
			tio.c_cflag &= ~PARODD;
		} else if(parity == 3) {
			tio.c_cflag |= PARENB | CMSPAR;
			tio.c_cflag &= ~PARODD;
		} else if( parity == 4) {
			tio.c_cflag |= PARENB | CMSPAR | PARODD;
		}
		//Stopbits
		if(stopbits != 0)
			tio.c_cflag |= CSTOPB;
		else
			tio.c_cflag &= ~CSTOPB;
		//Xonoff
		if(xonoff == 1)
			tio.c_cflag |= IXON;
		else
			tio.c_cflag &= ~IXON;
		//Set
		tio.c_oflag = 0;
		tio.c_lflag |= ICANON;
		tcsetattr(*fd, TCSANOW, &tio);

		*status = 1;
		error = strerror(errno);
	}
	free(cs_fn);
}

void ttyerror(CleanString *result)
{
	CleanStringVariable(clean_string, strlen(error));
	*result = (CleanString) clean_string;
	memcpy(CleanStringCharacters(clean_string), error, strlen(error));
	CleanStringLength(clean_string) = strlen(error);
}

void ttyreadline(int fd, CleanString *result, int *fdo)
{
	size_t bufsize = INITIAL_BUFFERSIZE;
	char *buf = NULL;
	ssize_t cr, charsread = 0; 
	while(buf == NULL || buf[charsread-1] != '\n'){
		if((buf = realloc(buf, (bufsize*=2)+1)) == NULL)
			die("realloc");
		if((cr = read(fd, buf+charsread, bufsize-charsread)) < 0)
			die("read");
		charsread += cr;
	}
	buf[charsread] = '\0';

	CleanStringVariable(cleanOutput, charsread);
	*result = (CleanString) cleanOutput;
	memcpy(CleanStringCharacters(cleanOutput), buf, charsread);
	CleanStringLength(cleanOutput) = charsread;
	*fdo = fd;

	free(buf);
}

int ttywrite(int fd, CleanString s)
{
	write(fd, CleanStringCharacters(s), CleanStringLength(s));
	tcdrain(fd);
	return fd;
}

int ttyclose(int fd)
{
	struct termios *to = getTermios(fd);
	tcsetattr(fd, TCSANOW, to);
	remTermios(fd);
	int ret = close(fd);
	error = strerror(errno);
	return ret + 1;
}