/*
	(c) Copyright  2012 - 2018 Anton Sviridenko
	https://picapica.im

	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, version 3.

	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, see <https://www.gnu.org/licenses/>.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sqlite3.h>
#include "PICA_nodeaddrlist.h"
#include "PICA_log.h"


struct cb_param
{
	struct PICA_nodeaddr **n_addr;
	int *N;
};

static int nodeaddr_load_sqlite_cb(void *p, int argc, char **argv, char **ColName)
{
	struct cb_param *cp = (struct cb_param*)p;
	struct PICA_nodeaddr *na;

	if (argc != 4)
		return 1;

	na = (struct PICA_nodeaddr *)malloc(sizeof(struct PICA_nodeaddr));

	if (!na)
		return 1;

	PICA_debug1("node address loaded: addr: %s port %s\n", argv[0], argv[1]);//debug

	strncpy(na->addr, argv[0], 255);
	na->addr[255] = '\0';

	na->port = atoi(argv[1]);

	na->inactive_count = atoi(argv[2]);

	na->last_active = atoi(argv[3]);

	na->next = *(cp->n_addr);

	*(cp->n_addr) = na;
	(*(cp->N))++;

	return 0;
}

int PICA_nodeaddr_list_load(const char* dbfilename, struct PICA_nodeaddr **list_head)
{
	int N = 0, ret;
	sqlite3 *db;
	char *zErrMsg = 0;
	const char query[] = "select address, port, inactive_count, last_active  from nodes order by inactive_count desc, last_active asc";
	struct cb_param p = {list_head, &N};


	ret = sqlite3_open_v2(dbfilename, &db, SQLITE_OPEN_READWRITE, NULL);

	if (ret != SQLITE_OK)
	{
		PICA_fatal( "Can't open nodelist database (%s): %s\n", dbfilename, sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_exec(db, query, nodeaddr_load_sqlite_cb, &p, &zErrMsg);

	if( ret != SQLITE_OK )
	{
		PICA_error("SQL error: %s\n", zErrMsg);
		sqlite3_free(zErrMsg);
	}

	sqlite3_close(db);
	return N;
}

int PICA_nodeaddr_update(const char* dbfilename, struct PICA_nodeaddr *naddr, int is_alive)
{
	sqlite3 *db;
	sqlite3_stmt *stmt;
	int ret;
	const char q1[] = "update nodes set last_active = strftime('%s','now'), inactive_count = 0 where address = :addr and port = :port";
	const char q2[] = "update nodes set inactive_count = inactive_count + 1 where address = :addr and port = :port";
	char *query;

	query = is_alive ? q1 : q2;

	ret = sqlite3_open(dbfilename, &db);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't open nodelist database: %s\n", sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't prepare SQLite statement: %s\n", sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_bind_text(stmt, 1, naddr->addr, -1, SQLITE_STATIC);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't bind address value: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_bind_int(stmt, 2, naddr->port);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't bind port value: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_step(stmt);

	if (ret != SQLITE_DONE)
	{
		PICA_error( "Can't execute query: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	sqlite3_finalize(stmt);
	sqlite3_close(db);

	return 1;
}

int PICA_nodeaddr_cleanold(const char* dbfilename, int treshold)
{
	sqlite3 *db;
	sqlite3_stmt *stmt;
	int ret;
	const char query[] = "delete from nodes where inactive_count > :treshold";

	ret = sqlite3_open(dbfilename, &db);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't open nodelist database: %s\n", sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't prepare SQLite statement: %s\n", sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_bind_int(stmt, 1, treshold);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't bind treshold value: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_step(stmt);

	if (ret != SQLITE_DONE)
	{
		PICA_error( "Can't execute query: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	sqlite3_finalize(stmt);
	sqlite3_close(db);

	return 1;

}

int PICA_nodeaddr_save(const char* dbfilename, struct PICA_nodeaddr *naddr)
{
	sqlite3 *db;
	sqlite3_stmt *stmt;
	int ret;
	const char query[] = "insert into nodes (address, port, last_active, inactive_count) values(:addr, :port, strftime('%s','now'), 0);";

	ret = sqlite3_open(dbfilename, &db);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't open nodelist database: %s\n", sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't prepare SQLite statement: %s\n", sqlite3_errmsg(db));
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_bind_text(stmt, 1, naddr->addr, -1, SQLITE_STATIC);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't bind address value: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_bind_int(stmt, 2, naddr->port);

	if (ret != SQLITE_OK)
	{
		PICA_error("Can't bind port value: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	ret = sqlite3_step(stmt);

	if (ret != SQLITE_DONE)
	{
		PICA_error( "Can't execute query: %s\n", sqlite3_errmsg(db));
		sqlite3_finalize(stmt);
		sqlite3_close(db);
		return -1;
	}

	sqlite3_finalize(stmt);
	sqlite3_close(db);

	return 1;
}

void PICA_nodeaddr_list_free(struct PICA_nodeaddr *list_head)
{
	struct PICA_nodeaddr *pp, *pn;
	pp = list_head;

	while(pp)
	{
		pn = pp->next;
		free(pp);
		pp = pn;
	}
}