#include "db.h"
#include "invoice.h"
#include "jsonrpc.h"
#include "lightningd.h"
#include <ccan/str/hex/hex.h>
#include <ccan/structeq/structeq.h>
#include <ccan/tal/str/str.h>
#include <sodium/randombytes.h>

static struct invoice *find_inv(const struct list_head *list,
				const struct sha256 *rhash)
{
	struct invoice *i;

	list_for_each(list, i, list) {
		if (structeq(rhash, &i->rhash))
			return i;
	}
	return NULL;
}

struct invoice *find_unpaid(struct lightningd_state *dstate,
			     const struct sha256 *rhash)
{
	return find_inv(&dstate->unpaid, rhash);
}

static struct invoice *find_paid(struct lightningd_state *dstate,
				 const struct sha256 *rhash)
{
	return find_inv(&dstate->paid, rhash);
}

static struct invoice *find_invoice_by_label(const struct list_head *list,
					     const char *label)
{
	struct invoice *i;

	list_for_each(list, i, list) {
		if (streq(i->label, label))
			return i;
	}
	return NULL;
}

void invoice_add(struct lightningd_state *dstate,
		 const struct rval *r,
		 u64 msatoshi,
		 const char *label,
		 u64 paid_num)
{
	struct invoice *invoice = tal(dstate, struct invoice);

	invoice->msatoshi = msatoshi;
	invoice->r = *r;
	invoice->paid_num = paid_num;
	invoice->label = tal_strdup(invoice, label);
	sha256(&invoice->rhash, invoice->r.r, sizeof(invoice->r.r));

	if (paid_num) {
		list_add(&dstate->paid, &invoice->list);
		if (paid_num > dstate->invoices_completed)
			dstate->invoices_completed = paid_num;
	} else
		list_add(&dstate->unpaid, &invoice->list);
}

bool resolve_invoice(struct lightningd_state *dstate,
		     struct invoice *invoice)
{
	invoice->paid_num = ++dstate->invoices_completed;
	list_del_from(&dstate->unpaid, &invoice->list);
	list_add_tail(&dstate->paid, &invoice->list);
	return db_resolve_invoice(dstate, invoice->label, invoice->paid_num);
}
	
static void json_invoice(struct command *cmd,
			 const char *buffer, const jsmntok_t *params)
{
	struct invoice *invoice;
	jsmntok_t *msatoshi, *r, *label;
	struct json_result *response = new_json_result(cmd);	

	if (!json_get_params(buffer, params,
			     "amount", &msatoshi,
			     "label", &label,
			     "?r", &r,
			     NULL)) {
		command_fail(cmd, "Need {amount} and {label}");
		return;
	}

	invoice = tal(cmd, struct invoice);
	if (r) {
		if (!hex_decode(buffer + r->start, r->end - r->start,
				invoice->r.r, sizeof(invoice->r.r))) {
			command_fail(cmd, "Invalid hex r '%.*s'",
				     r->end - r->start, buffer + r->start);
			return;
		}
	} else
		randombytes_buf(invoice->r.r, sizeof(invoice->r.r));

	sha256(&invoice->rhash, invoice->r.r, sizeof(invoice->r.r));
	if (find_unpaid(cmd->dstate, &invoice->rhash)
	    || find_paid(cmd->dstate, &invoice->rhash)) {
		command_fail(cmd, "Duplicate r value '%.*s'",
			     r->end - r->start, buffer + r->start);
		return;
	}

	if (!json_tok_u64(buffer, msatoshi, &invoice->msatoshi)
	    || invoice->msatoshi == 0) {
		command_fail(cmd, "'%.*s' is not a valid positive number",
			     msatoshi->end - msatoshi->start,
			     buffer + msatoshi->start);
		return;
	}

	invoice->label = tal_strndup(invoice, buffer + label->start,
				     label->end - label->start);
	if (find_invoice_by_label(&cmd->dstate->paid, invoice->label)
	    || find_invoice_by_label(&cmd->dstate->unpaid, invoice->label)) {
		command_fail(cmd, "Duplicate label '%s'", invoice->label);
		return;
	}
	if (strlen(invoice->label) > INVOICE_MAX_LABEL_LEN) {
		command_fail(cmd, "label '%s' over %u bytes", invoice->label,
			     INVOICE_MAX_LABEL_LEN);
		return;
	}
	invoice->paid_num = 0;

	if (!db_new_invoice(cmd->dstate, invoice->msatoshi, invoice->label,
			    &invoice->r)) {
		command_fail(cmd, "database error");
		return;
	}		
	/* OK, connect it to main state, respond with hash */
	tal_steal(cmd->dstate, invoice);
	list_add(&cmd->dstate->unpaid, &invoice->list);

	json_object_start(response, NULL);
	json_add_hex(response, "rhash",
		     &invoice->rhash, sizeof(invoice->rhash));
	json_object_end(response);

	command_success(cmd, response);
}

const struct json_command invoice_command = {
	"invoice",
	json_invoice,
	"Create invoice for {msatoshi} with {label} (with a set {r}, otherwise generate one)",
	"Returns the {rhash} on success. "
};

static void json_add_invoices(struct json_result *response,
			      const struct list_head *list,
			      const char *buffer, const jsmntok_t *label)
{
	struct invoice *i;

	list_for_each(list, i, list) {
		if (label && !json_tok_streq(buffer, label, i->label))
			continue;
		json_object_start(response, NULL);
		json_add_string(response, "label", i->label);
		json_add_hex(response, "rhash", &i->rhash, sizeof(i->rhash));
		json_add_u64(response, "msatoshi", i->msatoshi);
		json_add_bool(response, "complete", i->paid_num != 0);
		json_object_end(response);
	}
}

static void json_listinvoice(struct command *cmd,
			     const char *buffer, const jsmntok_t *params)
{
	jsmntok_t *label = NULL;
	struct json_result *response = new_json_result(cmd);	

	if (!json_get_params(buffer, params,
			     "?label", &label,
			     NULL)) {
		command_fail(cmd, "Invalid arguments");
		return;
	}

	
	json_object_start(response, NULL);
	json_array_start(response, NULL);
	json_add_invoices(response, &cmd->dstate->paid, buffer, label);
	json_add_invoices(response, &cmd->dstate->unpaid, buffer, label);
	json_array_end(response);
	json_object_end(response);
	command_success(cmd, response);
}

const struct json_command listinvoice_command = {
	"listinvoice",
	json_listinvoice,
	"Show invoice {label} (or all, if no {label}))",
	"Returns an array of {label}, {rhash}, {msatoshi} and {complete} on success. "
};

static void json_delinvoice(struct command *cmd,
			    const char *buffer, const jsmntok_t *params)
{
	struct invoice *i;
	jsmntok_t *labeltok;
	struct json_result *response = new_json_result(cmd);
	const char *label;

	if (!json_get_params(buffer, params,
			     "label", &labeltok,
			     NULL)) {
		command_fail(cmd, "Invalid arguments");
		return;
	}

	label = tal_strndup(cmd, buffer + labeltok->start,
			    labeltok->end - labeltok->start);
	i = find_invoice_by_label(&cmd->dstate->unpaid, label);
	if (!i) {
		command_fail(cmd, "Unknown invoice");
		return;
	}
	if (!db_remove_invoice(cmd->dstate, i->label)) {
		command_fail(cmd, "Database error");
		return;
	}
	list_del_from(&cmd->dstate->unpaid, &i->list);
	
	json_object_start(response, NULL);
	json_add_string(response, "label", i->label);
	json_add_hex(response, "rhash", &i->rhash, sizeof(i->rhash));
	json_add_u64(response, "msatoshi", i->msatoshi);
	json_object_end(response);
	command_success(cmd, response);
	tal_free(i);
}

const struct json_command delinvoice_command = {
	"delinvoice",
	json_delinvoice,
	"Delete unpaid invoice {label}))",
	"Returns {label}, {rhash} and {msatoshi} on success. "
};