/*
 * Copyright (C) 2012 Timo Kokkonen <timo.t.kokkonen@iki.fi>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <linux/types.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>

#include "hiddev.h"
#include "utils.h"
#include "options.h"
#include "contour-protocol.h"

static char *token(char **str, char sep)
{
	char *start = *str;
	char *cur;

	for (cur = start; *cur && (*cur != sep); ++cur);

	*cur = 0;
	*str = cur+1;

	return start;
}

static void format_csv(struct user_options *opts, struct msg *msg)
{
	char *tok = (char *) msg->data;
			  token(&tok, '|');  // unknown
	char *seq	= token(&tok, '|');
	char *type	= token(&tok, '|');
	char *val	= token(&tok, '|');
	char *unit	= token(&tok, '|');
			  token(&tok, '|');// unknown
	char *notes	= token(&tok, '|');
			  token(&tok, '|');// unknown
	char *time	= token(&tok, '\r');

	unit[strlen(unit)-2] = 0;
	fprintf(opts->outf,
		"%s,\"%.4s-%.2s-%.2s %.2s:%.2s\",\"%s\",\"%s\",\"%s\","
		"%s,%s,%s,%s,%s,%s,%s\n",
		seq, &time[0], &time[4], &time[6], &time[8], &time[10],
		&type[3], val, unit,
		strchr(notes, 'B') ? "X" : "",
		strchr(notes, 'A') ? "X" : "",
		strchr(notes, 'S') ? "X" : "",
		strchr(notes, 'I') ? "X" : "",
		strchr(notes, 'D') ? "X" : "",
		strchr(notes, 'X') ? "X" : "",
		strchr(notes, 'C') ? "X" : ""
		);
}

static int format_message(struct user_options *opts, struct msg *msg, int len)
{
	switch (opts->output_format)
	{
	case CSV:
		format_csv(opts, msg);
		break;

	case CLEAN:
		sanitize_ascii(msg->data, len);
		fprintf(opts->outf, "%s\n", msg->data);
		break;

	case RAW:
		fprintf(opts->outf, "%s", msg->data);
		break;

	default:
		trace(0, "BUG: Invalid message format %d\n",
			opts->output_format);
		return -1;
	}

	fflush(opts->outf);

	return 0;
}

static int dump_entries(struct user_options *opts, int fd, int usage_code)
{
	struct msg msg;
	int ret;
	int entries = 0;

	trace(0, "Reading data ...\n");
	if (opts->output_format == CSV)
		fprintf(opts->outf, "#,Time,Type,Value,Unit,\"Before meal\","
			"\"After meal\",Stress,Sick,\"Dont feel right\","
			"Activity,\"Control test\"\n");

	while (1) {
		ret = contour_read_entry(fd, usage_code, &msg);
		if (ret < 45)
			return -1;

		ret = format_message(opts, &msg, ret);
		if (ret < 0)
			return ret;

		entries++;

		if ((opts->outf != stdout) || !isatty(fileno(stdout))) {
			trace(0, "\r%d entries", entries);
			fflush(stdout);
		}
	}
	trace(0, "\nDone.\n");

	return 0;
}

int main(int argc, char *argv[])
{
	struct user_options opts;
	int fd, usage_code, ret;

	bzero(&opts, sizeof(opts));
	opts.output_format = CLEAN;

	if ( read_args(argc, argv, &opts) )
		return -1;

	trace_level = opts.trace_level;

	if (opts.output_path) {
		opts.outf = fopen(opts.output_path, "w");
		if (opts.outf == NULL) {
			trace(0, "Failed to open output file \"%s\": %m\n",
				opts.output_path);
			return 1;
		}
	} else {
		opts.outf = stdout;
	}

	if (opts.usbdev == NULL)
		fd = wait_for_device(CONTOUR_USB_VENDOR_ID,
				CONTOUR_USB_PRODUCT_ID, &usage_code);
	else
		fd = hiddev_open(opts.usbdev, &usage_code);
	if (fd < 0)
		return 1;

	trace(0, "Initializing ...\n");
	contour_initialize(fd, usage_code);

	ret = dump_entries(&opts, fd, usage_code);

	return ret ? 1 : 0;
}