/* See LICENSE file for copyright and license details. */
#include <sys/select.h>
#include <sys/stat.h>
#include <sys/types.h>

#include <ctype.h>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <stdarg.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include <tox/tox.h>

#include "arg.h"
#include "queue.h"

#define LEN(x) (sizeof (x) / sizeof *(x))
#define DATAFILE ".ratox.data"

struct node {
	const char *addr;
	uint16_t port;
	uint8_t key[TOX_CLIENT_ID_SIZE];
};

#include "config.h"

struct file {
	int type;
	const char *name;
	int flags;
	mode_t mode;
};

enum {
	IN,
	OUT,
	ERR,
	NR_GFILES
};

struct slot {
	const char *name;
	void (*cb)(void *);
	int outtype;
	int outmode;
	int fd[NR_GFILES];
};

enum {
	NAME,
	STATUS,
	REQUEST,
};

enum {
	FIFO,
	OUT_F,
	STATIC,
	FOLDER
};

static void setname(void *);
static void setstatus(void *);
static void sendfriendreq(void *);

static struct slot gslots[] = {
	[NAME]    = { .name = "name",	 .cb = setname,	      .outtype = STATIC },
	[STATUS]  = { .name = "status",	 .cb = setstatus,     .outtype = STATIC },
	[REQUEST] = { .name = "request", .cb = sendfriendreq, .outtype = FOLDER, .outmode = 0755 },
};

static struct file gfiles[] = {
	{ .type = FIFO,  .name = "in",  .flags = O_RDWR | O_NONBLOCK,          .mode = 0644},
	{ .type = OUT_F, .name = "out", .flags = O_WRONLY | O_TRUNC | O_CREAT, .mode = 0644},
	{ .type = OUT_F, .name = "err", .flags = O_WRONLY | O_TRUNC | O_CREAT, .mode = 0644},
};

enum {
	FTEXT_IN,
	FFILE_IN,
	FONLINE,
	FNAME,
	FSTATUS,
	FTEXT_OUT,
	NR_FFILES
};

static struct file ffiles[] = {
	{ .type = FIFO,  .name = "text_in",  .flags = O_RDWR | O_NONBLOCK,          .mode = 0644 },
	{ .type = FIFO,  .name = "file_in",  .flags = O_RDWR | O_NONBLOCK,          .mode = 0644 },
	{ .type = OUT_F, .name = "online",   .flags = O_WRONLY | O_TRUNC | O_CREAT, .mode = 0644 },
	{ .type = OUT_F, .name = "name",     .flags = O_WRONLY | O_TRUNC | O_CREAT, .mode = 0644 },
	{ .type = OUT_F, .name = "status",   .flags = O_WRONLY | O_TRUNC | O_CREAT, .mode = 0644 },
	{ .type = OUT_F, .name = "text_out", .flags = O_APPEND | O_CREAT,           .mode = 0644 },
};

enum {
	TRANSFER_NONE,
	TRANSFER_INITIATED,
	TRANSFER_INPROGRESS,
	TRANSFER_DONE
};

struct transfer {
	uint8_t fnum;
	uint8_t *buf;
	int chunksz;
	ssize_t n;
	int pending;
	int state;
	time_t lastsent;
};

struct friend {
	/* null terminated name */
	char namestr[TOX_MAX_NAME_LENGTH + 1];
	int fid;
	uint8_t id[TOX_CLIENT_ID_SIZE];
	/* null terminated id */
	char idstr[2 * TOX_CLIENT_ID_SIZE + 1];
	int fd[NR_FFILES];
	struct transfer t;
	TAILQ_ENTRY(friend) entry;
};

struct request {
	uint8_t id[TOX_CLIENT_ID_SIZE];
	/* null terminated id */
	char idstr[2 * TOX_CLIENT_ID_SIZE + 1];
	/* null terminated friend request message */
	char *msgstr;
	int fd;
	TAILQ_ENTRY(request) entry;
};

char *argv0;

static TAILQ_HEAD(friendhead, friend) friendhead = TAILQ_HEAD_INITIALIZER(friendhead);
static TAILQ_HEAD(reqhead, request) reqhead = TAILQ_HEAD_INITIALIZER(reqhead);

static Tox *tox;

static void printrat(void);
static void cbconnstatus(Tox *, int32_t, uint8_t, void *);
static void cbfriendmessage(Tox *, int32_t, const uint8_t *, uint16_t, void *);
static void cbfriendrequest(Tox *, const uint8_t *, const uint8_t *, uint16_t, void *);
static void cbnamechange(Tox *, int32_t, const uint8_t *, uint16_t, void *);
static void cbstatusmessage(Tox *, int32_t, const uint8_t *, uint16_t, void *);
static void cbuserstatus(Tox *, int32_t, uint8_t, void *);
static void cbfilecontrol(Tox *, int32_t, uint8_t, uint8_t, uint8_t, const uint8_t *, uint16_t, void *);
static void completefile(struct friend *);
static void sendfriendfile(struct friend *);
static void dataload(void);
static void datasave(void);
static int localinit(void);
static int toxinit(void);
static int toxconnect(void);
static void id2str(uint8_t *, char *);
static void str2id(char *, uint8_t *);
static struct friend *friendcreate(int32_t);
static void friendload(void);
static void loop(void);

static void
printrat(void)
{
	printf("\033[31m");
	printf("                       ,     .\n");
	printf("                       (\\,;,/)\n");
	printf("                        (o o)\\//,\n");
	printf("                         \\ /     \\,\n");
	printf("                         `+'(  (   \\    )\n");
	printf("                            //  \\   |_./\n");
	printf("                          '~' '~----'\tratox v"VERSION"\n");
	printf("\033[0m");
}

static void
printout(const char *fmt, ...)
{
	va_list ap;
	char buft[64];
	time_t t;

	va_start(ap, fmt);
	t = time(NULL);
	strftime(buft, sizeof(buft), "%F %R", localtime(&t));
	printf("%s ", buft);
	vfprintf(stdout, fmt, ap);
	va_end(ap);
}

static void
cbconnstatus(Tox *m, int32_t fid, uint8_t status, void *udata)
{
	struct friend *f;
	uint8_t name[TOX_MAX_NAME_LENGTH + 1];
	int r;

	r = tox_get_name(tox, fid, name);
	if (r < 0) {
		fprintf(stderr, "tox_get_name() on fid %d failed\n", fid);
		exit(EXIT_FAILURE);
	}
	name[r] = '\0';

	printout("%s %s\n", r == 0 ? (uint8_t *)"Anonymous" : name,
		 status == 0 ? "went offline" : "came online");

	TAILQ_FOREACH(f, &friendhead, entry) {
		if (f->fid == fid) {
			ftruncate(f->fd[FONLINE], 0);
			dprintf(f->fd[FONLINE], status == 0 ? "0\n" : "1\n");
			return;
		}
	}

	friendcreate(fid);
}

static void
cbfriendmessage(Tox *m, int32_t fid, const uint8_t *data, uint16_t len, void *udata)
{
	struct friend *f;
	uint8_t msg[len + 1];
	char buft[64];
	time_t t;

	memcpy(msg, data, len);
	msg[len] = '\0';

	TAILQ_FOREACH(f, &friendhead, entry) {
		if (f->fid == fid) {
			t = time(NULL);
			strftime(buft, sizeof(buft), "%F %R", localtime(&t));
			dprintf(f->fd[FTEXT_OUT], "%s %s\n", buft, msg);
			printout("%s %s\n",
				 f->namestr[0] == '\0' ? "Anonymous" : f->namestr, msg);
			break;
		}
	}
}

static void
cbfriendrequest(Tox *m, const uint8_t *id, const uint8_t *data, uint16_t len, void *udata)
{
	struct request *req;
	int r;

	req = calloc(1, sizeof(*req));
	if (!req) {
		perror("calloc");
		exit(EXIT_FAILURE);
	}
	memcpy(req->id, id, TOX_CLIENT_ID_SIZE);
	id2str(req->id, req->idstr);

	if (len > 0) {
		req->msgstr = malloc(len + 1);
		if (!req->msgstr) {
			perror("malloc");
			exit(EXIT_FAILURE);
		}
		memcpy(req->msgstr, data, len);
		req->msgstr[len] = '\0';
	}

	r = mkfifoat(gslots[REQUEST].fd[OUT], req->idstr, 0644);
	if (r < 0 && errno != EEXIST) {
		perror("mkfifoat");
		exit(EXIT_FAILURE);
	}
	r = openat(gslots[REQUEST].fd[OUT], req->idstr, O_RDWR | O_NONBLOCK);
	if (r < 0) {
		perror("open");
		exit(EXIT_FAILURE);
	}
	req->fd = r;

	TAILQ_INSERT_TAIL(&reqhead, req, entry);

	printout("Pending request from %s with message: %s\n",
		 req->idstr, req->msgstr);
}

static void
cbnamechange(Tox *m, int32_t fid, const uint8_t *data, uint16_t len, void *user)
{
	struct friend *f;
	uint8_t name[len + 1];

	memcpy(name, data, len);
	name[len] = '\0';

	TAILQ_FOREACH(f, &friendhead, entry) {
		if (f->fid == fid) {
			ftruncate(f->fd[FNAME], 0);
			dprintf(f->fd[FNAME], "%s\n", name);
			if (memcmp(f->namestr, name, len + 1) == 0)
				break;
			printout("%s -> %s\n", f->namestr[0] == '\0' ?
				 "Anonymous" : f->namestr, name);
			memcpy(f->namestr, name, len + 1);
			break;
		}
	}
	datasave();
}

static void
cbstatusmessage(Tox *m, int32_t fid, const uint8_t *data, uint16_t len, void *udata)
{
	struct friend *f;
	uint8_t status[len + 1];

	memcpy(status, data, len);
	status[len] = '\0';

	TAILQ_FOREACH(f, &friendhead, entry) {
		if (f->fid == fid) {
			ftruncate(f->fd[FSTATUS], 0);
			dprintf(f->fd[FSTATUS], "%s\n", status);
			printout("%s changed status: %s\n",
				 f->namestr[0] == '\0' ? "Anonymous" : f->namestr, status);
			break;
		}
	}
	datasave();
}

static void
cbuserstatus(Tox *m, int32_t fid, uint8_t status, void *udata)
{
	struct friend *f;
	char *statusstr[] = { "none", "away", "busy" };

	if (status >= LEN(statusstr)) {
		fprintf(stderr, "received invalid user status: %d\n", status);
		return;
	}

	TAILQ_FOREACH(f, &friendhead, entry) {
		if (f->fid == fid) {
			printout("%s changed user status: %s\n",
				 f->namestr[0] == '\0' ? "Anonymous" : f->namestr,
			         statusstr[status]);
			break;
		}
	}
}

static void
cbfilecontrol(Tox *m, int32_t fid, uint8_t rec_sen, uint8_t fnum, uint8_t ctrltype,
	const uint8_t *data, uint16_t len, void *udata)
{
	struct friend *f;

	switch (ctrltype) {
	case TOX_FILECONTROL_ACCEPT:
		if (rec_sen == 1) {
			TAILQ_FOREACH(f, &friendhead, entry) {
				if (f->fid != fid)
					continue;
				f->t.fnum = fnum;
				f->t.chunksz = tox_file_data_size(tox, fnum);
				f->t.buf = malloc(f->t.chunksz);
				if (!f->t.buf) {
					perror("malloc");
					exit(EXIT_FAILURE);
				}
				f->t.n = 0;
				f->t.pending = 0;
				f->t.state = TRANSFER_INPROGRESS;
				f->t.lastsent = time(NULL);
				break;
			}
		}
		break;
	case TOX_FILECONTROL_FINISHED:
		if (rec_sen == 1) {
			TAILQ_FOREACH(f, &friendhead, entry) {
				if (f->fid != fid)
					continue;
				f->t.state = TRANSFER_DONE;
				break;
			}
		}
		break;
	default:
		fprintf(stderr, "Unhandled file control type: %d\n", ctrltype);
		break;
	};
}

static void
completefile(struct friend *f)
{
	tox_file_send_control(tox, f->fid, 0, f->t.fnum,
			      TOX_FILECONTROL_FINISHED, NULL, 0);
}

static void
sendfriendfile(struct friend *f)
{
	ssize_t n;

	while (1) {
		/* attempt to transmit the pending buffer */
		if (f->t.pending == 1) {
			if (tox_file_send_data(tox, f->fid, f->t.fnum, f->t.buf, f->t.n) == -1) {
				/* bad luck - we will try again later */
				break;
			}
			f->t.lastsent = time(NULL);
			f->t.pending = 0;
		}
		/* grab another buffer from the FIFO */
		n = read(f->fd[FFILE_IN], f->t.buf, f->t.chunksz);
		if (n < 0) {
			if (errno == EINTR)
				continue;
			/* go back to select() until the fd is readable */
			if (errno == EWOULDBLOCK)
				break;
			perror("read");
			exit(EXIT_FAILURE);
		}
		/* store transfer size in case we can't send it right now */
		f->t.n = n;
		if (tox_file_send_data(tox, f->fid, f->t.fnum, f->t.buf, f->t.n) == -1) {
			/* ok we will have to send it later, flip state */
			f->t.pending = 1;
			return;
		}
		f->t.lastsent = time(NULL);
	}
}

static void
sendfriendtext(struct friend *f)
{
	uint8_t buf[TOX_MAX_MESSAGE_LENGTH];
	ssize_t n;

again:
	n = read(f->fd[FTEXT_IN], buf, sizeof(buf));
	if (n < 0) {
		if (errno == EINTR)
			goto again;
		/* go back to select() until the fd is readable */
		if (errno == EWOULDBLOCK)
			return;
		perror("read");
		exit(EXIT_FAILURE);
	}
	if (buf[n - 1] == '\n')
		n--;
	tox_send_message(tox, f->fid, buf, n);
}

static void
dataload(void)
{
	FILE *fp;
	size_t sz;
	uint8_t *data;
	int r;

	fp = fopen(DATAFILE, "r");
	if (!fp)
		return;

	fseek(fp, 0, SEEK_END);
	sz = ftell(fp);
	rewind(fp);

	data = malloc(sz);
	if (!data) {
		perror("malloc");
		exit(EXIT_FAILURE);
	}

	if (fread(data, 1, sz, fp) != sz || ferror(fp)) {
		fprintf(stderr, "failed to read %s\n", DATAFILE);
		exit(EXIT_FAILURE);
	}
	r = tox_load(tox, data, sz);
	if (r < 0) {
		fprintf(stderr, "tox_load() failed\n");
		exit(EXIT_FAILURE);
	}
	if (r == 1)
		printf("Found encrypted data in %s\n", DATAFILE);

	free(data);
	fclose(fp);
}

static void
datasave(void)
{
	FILE *fp;
	size_t sz;
	uint8_t *data;

	fp = fopen(DATAFILE, "w");
	if (!fp) {
		fprintf(stderr, "can't open %s for writing\n", DATAFILE);
		exit(EXIT_FAILURE);
	}

	sz = tox_size(tox);
	data = malloc(sz);
	if (!data) {
		perror("malloc");
		exit(EXIT_FAILURE);
	}

	tox_save(tox, data);
	if (fwrite(data, 1, sz, fp) != sz || ferror(fp)) {
		fprintf(stderr, "failed to write %s\n", DATAFILE);
		exit(EXIT_FAILURE);
	}

	free(data);
	fclose(fp);
}

static int
localinit(void)
{
	uint8_t name[TOX_MAX_NAME_LENGTH + 1];
	uint8_t address[TOX_FRIEND_ADDRESS_SIZE];
	uint8_t status[TOX_MAX_STATUSMESSAGE_LENGTH + 1];
	FILE *fp;
	DIR *d;
	int r;
	size_t i, m;

	for (i = 0; i < LEN(gslots); i++) {
		r = mkdir(gslots[i].name, 0755);
		if (r < 0 && errno != EEXIST) {
			perror("mkdir");
			exit(EXIT_FAILURE);
		}
		r = chdir(gslots[i].name);
		if (r < 0) {
			perror("chdir");
			exit(EXIT_FAILURE);
		}
		for (m = 0; m < LEN(gfiles); m++) {
			if (gfiles[m].type == FIFO) {
				r = mkfifo(gfiles[m].name, gfiles[m].mode);
				if (r < 0 && errno != EEXIST) {
					perror("mkfifo");
					exit(EXIT_FAILURE);
				}
				r = open(gfiles[m].name, gfiles[m].flags, 0);
				if (r < 0) {
					perror("open");
					exit(EXIT_FAILURE);
				}
				gslots[i].fd[m] = r;
			} else if (gfiles[m].type == OUT_F) {
				if (gslots[i].outtype == STATIC) {
					r = open(gfiles[m].name, gfiles[m].flags, gfiles[m].mode);
					if (r < 0) {
						perror("open");
						exit(EXIT_FAILURE);
					}
					gslots[i].fd[m] = r;
				} else if (gslots[i].outtype == FOLDER) {
					r = mkdir(gfiles[m].name, gslots[i].outmode);
					if (r < 0 && errno != EEXIST) {
						perror("mkdir");
						exit(EXIT_FAILURE);
					}
					d = opendir(gfiles[m].name);
					if (!d) {
						perror("opendir");
						exit(EXIT_FAILURE);
					}
					r = dirfd(d);
					if (r < 0) {
						perror("dirfd");
						exit(EXIT_FAILURE);
					}
					gslots[i].fd[m] = r;
				}
			}
		}
		chdir("..");
	}

	/* Dump current name */
	r = tox_get_self_name(tox, name);
	if (r > sizeof(name) - 1)
		r = sizeof(name) - 1;
	name[r] = '\0';
	ftruncate(gslots[NAME].fd[OUT], 0);
	dprintf(gslots[NAME].fd[OUT], "%s\n", name);

	/* Dump status message */
	r = tox_get_self_status_message(tox, status,
					sizeof(status) - 1);
	if (r > sizeof(status) - 1)
		r = sizeof(status) - 1;
	status[r] = '\0';
	ftruncate(gslots[STATUS].fd[OUT], 0);
	dprintf(gslots[STATUS].fd[OUT], "%s\n", name);

	/* Dump ID */
	fp = fopen("id", "w");
	if (!fp) {
		perror("fopen");
		exit(EXIT_FAILURE);
	}
	tox_get_address(tox, address);
	for (i = 0; i < TOX_FRIEND_ADDRESS_SIZE; i++)
		fprintf(fp, "%02X", address[i]);
	fputc('\n', fp);
	fclose(fp);

	return 0;
}

static int
toxinit(void)
{
	/* IPv4 only */
	tox = tox_new(0);
	dataload();
	datasave();
	tox_callback_connection_status(tox, cbconnstatus, NULL);
	tox_callback_friend_message(tox, cbfriendmessage, NULL);
	tox_callback_friend_request(tox, cbfriendrequest, NULL);
	tox_callback_name_change(tox, cbnamechange, NULL);
	tox_callback_status_message(tox, cbstatusmessage, NULL);
	tox_callback_user_status(tox, cbuserstatus, NULL);
	tox_callback_file_control(tox, cbfilecontrol, NULL);
	return 0;
}

static int
toxconnect(void)
{
	struct node *bn;
	size_t i;

	for (i = 0; i < LEN(nodes); i++) {
		bn = &nodes[i];
		tox_bootstrap_from_address(tox, bn->addr, bn->port, bn->key);
	}
	return 0;
}

static void
id2str(uint8_t *id, char *idstr)
{
	char hex[] = "0123456789ABCDEF";
	int i;

	for (i = 0; i < TOX_CLIENT_ID_SIZE; i++) {
		*idstr++ = hex[(id[i] >> 4) & 0xf];
		*idstr++ = hex[id[i] & 0xf];
	}
	*idstr = '\0';
}

static void
str2id(char *idstr, uint8_t *id)
{
	size_t i, len = strlen(idstr) / 2;
	char *p = idstr;

	for (i = 0; i < len; ++i, p += 2)
		sscanf(p, "%2hhx", &id[i]);
}

static struct friend *
friendcreate(int32_t fid)
{
	struct friend *f;
	uint8_t status[TOX_MAX_STATUSMESSAGE_LENGTH + 1];
	size_t i;
	int r;

	f = calloc(1, sizeof(*f));
	if (!f) {
		perror("calloc");
		exit(EXIT_FAILURE);
	}

	r = tox_get_name(tox, fid, (uint8_t *)f->namestr);
	if (r < 0) {
		fprintf(stderr, "tox_get_name() on fid %d failed\n", fid);
		exit(EXIT_FAILURE);
	}
	f->namestr[r] = '\0';

	f->fid = fid;
	tox_get_client_id(tox, f->fid, f->id);
	id2str(f->id, f->idstr);

	r = mkdir(f->idstr, 0755);
	if (r < 0 && errno != EEXIST) {
		perror("mkdir");
		exit(EXIT_FAILURE);
	}

	r = chdir(f->idstr);
	if (r < 0) {
		perror("chdir");
		exit(EXIT_FAILURE);
	}
	for (i = 0; i < LEN(ffiles); i++) {
		if (ffiles[i].type == FIFO) {
			r = mkfifo(ffiles[i].name, ffiles[i].mode);
			if (r < 0 && errno != EEXIST) {
				perror("mkfifo");
				exit(EXIT_FAILURE);
			}
			r = open(ffiles[i].name, ffiles[i].flags);
			if (r < 0) {
				perror("open");
				exit(EXIT_FAILURE);
			}
		} else if (ffiles[i].type == OUT_F) {
			r = open(ffiles[i].name, ffiles[i].flags, ffiles[i].mode);
			if (r < 0) {
				perror("open");
				exit(EXIT_FAILURE);
			}
		}
		f->fd[i] = r;
	}
	chdir("..");

	ftruncate(f->fd[FNAME], 0);
	dprintf(f->fd[FNAME], "%s\n", f->namestr);
	ftruncate(f->fd[FONLINE], 0);
	dprintf(f->fd[FONLINE], "%s\n",
		tox_get_friend_connection_status(tox, fid) == 0 ? "0" : "1");
	r = tox_get_status_message_size(tox, fid);
	if (r > sizeof(status) - 1)
		r = sizeof(status) - 1;
	status[r] = '\0';
	ftruncate(f->fd[FSTATUS], 0);
	dprintf(f->fd[FSTATUS], "%s\n", status);

	TAILQ_INSERT_TAIL(&friendhead, f, entry);

	return f;
}

static void
friendload(void)
{
	int32_t *fids;
	uint32_t sz;
	uint32_t i;

	sz = tox_count_friendlist(tox);
	fids = malloc(sz);
	if (!fids) {
		perror("malloc");
		exit(EXIT_FAILURE);
	}

	tox_get_friendlist(tox, fids, sz);

	for (i = 0; i < sz; i++)
		friendcreate(fids[i]);

	free(fids);
}

static void
setname(void *data)
{
	uint8_t name[TOX_MAX_NAME_LENGTH + 1];
	int r;

again:
	r = read(gslots[NAME].fd[IN], name, sizeof(name) - 1);
	if (r < 0) {
		if (errno == EINTR)
			goto again;
		if (errno == EWOULDBLOCK)
			return;
		perror("read");
		return;
	}
	if (name[r - 1] == '\n')
		r--;
	name[r] = '\0';
	tox_set_name(tox, name, r);
	datasave();
	printout("Changed name to %s\n", name);
	ftruncate(gslots[NAME].fd[OUT], 0);
	dprintf(gslots[NAME].fd[OUT], "%s\n", name);
}

static void
setstatus(void *data)
{
	uint8_t status[TOX_MAX_STATUSMESSAGE_LENGTH + 1];
	int r;

again:
	r = read(gslots[STATUS].fd[IN], status, sizeof(status) - 1);
	if (r < 0) {
		if (errno == EINTR)
			goto again;
		if (errno == EWOULDBLOCK)
			return;
		perror("read");
		return;
	}
	if (status[r - 1] == '\n')
		r--;
	status[r] = '\0';
	tox_set_status_message(tox, status, r);
	datasave();
	printout("Changed status message to %s\n", status);
	ftruncate(gslots[STATUS].fd[OUT], 0);
	dprintf(gslots[STATUS].fd[OUT], "%s\n", status);
}

static void
sendfriendreq(void *data)
{
	char *p;
	uint8_t id[TOX_FRIEND_ADDRESS_SIZE];
	char buf[BUFSIZ], *msg = "ratox is awesome!";
	int r;

again:
	r = read(gslots[REQUEST].fd[IN], buf, sizeof(buf) - 1);
	if (r < 0) {
		if (errno == EINTR)
			goto again;
		if (errno == EWOULDBLOCK)
			return;
		perror("read");
		return;
	}
	buf[r] = '\0';

	for (p = buf; *p && isspace(*p) == 0; p++)
		;
	if (*p != '\0') {
		*p = '\0';
		while (isspace(*p++) != 0)
			;
		if (*p != '\0')
			msg = p;
		if (msg[strlen(msg) - 1] == '\n')
			msg[strlen(msg) - 1] = '\0';
	}
	str2id(buf, id);

	r = tox_add_friend(tox, id, (uint8_t *)buf, strlen(buf));
	if (r < 0)
		ftruncate(gslots[REQUEST].fd[ERR], 0);
	switch (r) {
	case TOX_FAERR_TOOLONG:
		dprintf(gslots[REQUEST].fd[ERR], "Message is too long\n");
		break;
	case TOX_FAERR_NOMESSAGE:
		dprintf(gslots[REQUEST].fd[ERR], "Please add a message to your request\n");
		break;
	case TOX_FAERR_OWNKEY:
		dprintf(gslots[REQUEST].fd[ERR], "That appears to be your own ID\n");
		break;
	case TOX_FAERR_ALREADYSENT:
		dprintf(gslots[REQUEST].fd[ERR], "Friend request already sent\n");
		break;
	case TOX_FAERR_UNKNOWN:
		dprintf(gslots[REQUEST].fd[ERR], "Unknown error while sending your request\n");
		break;
	case TOX_FAERR_BADCHECKSUM:
		dprintf(gslots[REQUEST].fd[ERR], "Bad checksum while verifying address\n");
		break;
	case TOX_FAERR_SETNEWNOSPAM:
		dprintf(gslots[REQUEST].fd[ERR], "Friend already added but nospam doesn't match\n");
		break;
	default:
		printout("Friend request sent\n");
		break;
	}
	datasave();
}

static void
loop(void)
{
	struct friend *f;
	struct request *r, *rtmp;
	time_t t0, t1, now;
	int connected = 0;
	int i, n;
	int fdmax;
	char c;
	fd_set rfds;
	struct timeval tv;

	t0 = time(NULL);
	printout("Connecting to DHT...\n");
	toxconnect();
	while (1) {
		if (tox_isconnected(tox) == 1) {
			if (connected == 0) {
				printout("Connected to DHT\n");
				connected = 1;
			}
		} else {
			connected = 0;
			t1 = time(NULL);
			if (t1 > t0 + 5) {
				t0 = time(NULL);
				printout("Connecting to DHT...\n");
				toxconnect();
			}
		}
		tox_do(tox);

		FD_ZERO(&rfds);

		for (i = 0; i < LEN(gslots); i++) {
			FD_SET(gslots[i].fd[IN], &rfds);
			if (gslots[i].fd[IN] > fdmax)
				fdmax = gslots[i].fd[IN];
		}

		TAILQ_FOREACH(r, &reqhead, entry) {
			FD_SET(r->fd, &rfds);
			if(r->fd > fdmax)
				fdmax = r->fd;
		}

		TAILQ_FOREACH(f, &friendhead, entry) {
			/* Only monitor friends that are online */
			if (tox_get_friend_connection_status(tox, f->fid) == 1) {
				FD_SET(f->fd[FTEXT_IN], &rfds);
				if (f->fd[FTEXT_IN] > fdmax)
					fdmax = f->fd[FTEXT_IN];
				/* If the transfer has just been initiated
				 * wait until we have a state change before we start
				 * polling.  Avoids spinning endlessly while waiting
				 * for the transfer to be accepted by the other
				 * party.
				 */
				if (f->t.state == TRANSFER_INITIATED)
					continue;
				FD_SET(f->fd[FFILE_IN], &rfds);
				if (f->fd[FFILE_IN] > fdmax)
					fdmax = f->fd[FFILE_IN];
			}
		}

		tv.tv_sec = 0;
		tv.tv_usec = tox_do_interval(tox) * 1000;
		n = select(fdmax + 1, &rfds, NULL, NULL, &tv);
		if (n < 0) {
			if (errno == EINTR)
				continue;
			perror("select");
			exit(EXIT_FAILURE);
		}

		/* Check for broken transfers, i.e. the friend went offline
		 * in the middle of a transfer.
		 */
		TAILQ_FOREACH(f, &friendhead, entry) {
			if (tox_get_friend_connection_status(tox, f->fid) == 0) {
				if (f->t.state != TRANSFER_NONE) {
					printout("Stale transfer detected, friend offline\n");
					f->t.state = TRANSFER_NONE;
					free(f->t.buf);
					f->t.buf = NULL;
				}
			}
		}

		/* If we hit the receiver too hard, we will run out of
		 * local buffer slots.  In that case tox_file_send_data()
		 * will return -1 and we will have to queue the buffer to
		 * send it later.  If this is the last buffer read from
		 * the FIFO, then select() won't make the fd readable again
		 * so we have to check if there's anything pending to be
		 * sent.
		 */
		TAILQ_FOREACH(f, &friendhead, entry) {
			if (tox_get_friend_connection_status(tox, f->fid) == 0)
				continue;
			if (f->t.state != TRANSFER_INPROGRESS)
				continue;
			if (f->t.pending == 1)
				sendfriendfile(f);
		}

		/* Check for completed file transfers */
		now = time(NULL);
		TAILQ_FOREACH(f, &friendhead, entry) {
			if (tox_get_friend_connection_status(tox, f->fid) == 0)
				continue;
			switch (f->t.state) {
			case TRANSFER_NONE:
			case TRANSFER_INITIATED:
				break;
			case TRANSFER_INPROGRESS:
				/* We use a 1 second grace period, if no reads
				 * happen during that period we consider the transfer
				 * to have completed and signal the end of transmission
				 * control packet.  If we knew the size of the file
				 * it would have been easier to do this but since we are
				 * reading from a FIFO this is not really a possibility.
				 */
				if (now > f->t.lastsent + 1)
					completefile(f);
				break;
			case TRANSFER_DONE:
				printout("Transfer complete\n");
				f->t.state = TRANSFER_NONE;
				free(f->t.buf);
				f->t.buf = NULL;
				break;
			}
		}

		if (n == 0)
			continue;

		for (i = 0; i < LEN(gslots); i++) {
			if (FD_ISSET(gslots[i].fd[IN], &rfds) == 0)
				continue;
			(*gslots[i].cb)(NULL);
		}

		for (r = TAILQ_FIRST(&reqhead); r; r = rtmp) {
			rtmp = TAILQ_NEXT(r, entry);
			if (FD_ISSET(r->fd, &rfds) == 0)
				continue;
			if (read(r->fd, &c, 1) != 1 || c != '1')
				continue;
			tox_add_friend_norequest(tox, r->id);
			printout("Accepted friend request for %s\n", r->idstr);
			datasave();
			unlinkat(gslots[REQUEST].fd[OUT], r->idstr, 0);
			close(r->fd);
			TAILQ_REMOVE(&reqhead, r, entry);
			free(r->msgstr);
			free(r);
		}

		TAILQ_FOREACH(f, &friendhead, entry) {
			for (i = 0; i < NR_FFILES; i++) {
				if (FD_ISSET(f->fd[i], &rfds) == 0)
					continue;
				switch (i) {
				case FTEXT_IN:
					sendfriendtext(f);
					break;
				case FFILE_IN:
					switch (f->t.state) {
					case TRANSFER_NONE:
						/* prepare a new transfer */
						f->t.state = TRANSFER_INITIATED;
						tox_new_file_sender(tox, f->fid,
							0, (uint8_t *)"file", strlen("file") + 1);
						printout("Initiated transfer to %s\n",
							 f->namestr[0] == '\0' ? "Anonymous" : f->namestr);
						break;
					case TRANSFER_INPROGRESS:
						sendfriendfile(f);
						break;
					}
					break;
				default:
					fprintf(stderr, "Unhandled FIFO read\n");
				}
			}
		}
	}
}

int
main(int argc, char *argv[])
{
	printrat();
	toxinit();
	localinit();
	friendload();
	loop();
	return EXIT_SUCCESS;
}