/*
 * sophia database
 * sphia.org
 *
 * Copyright (c) Dmitry Simonenko
 * BSD License
*/

#include <sophia.h>
#include <libss.h>
#include <libsf.h>
#include <libsr.h>
#include <libsv.h>
#include <libsw.h>
#include <libsd.h>
#include <libst.h>

static void
alloclogv(svlog *log, sr *r, uint8_t flags, int key)
{
	sfv pv[8];
	memset(pv, 0, sizeof(pv));
	pv[0].pointer = (char*)&key;
	pv[0].size = sizeof(uint32_t);
	pv[1].pointer = NULL;
	pv[1].size = 0;
	svv *v = sv_vbuild(r, pv);
	sf_flagsset(r->scheme, sv_vpointer(v), flags);
	svlogv logv;
	logv.index_id = 0;
	logv.next = 0;
	logv.v = v;
	logv.ptr = NULL;
	sv_logadd(log, r, &logv);
}

static void
freelog(svlog *log, sr *c)
{
	ssiter i;
	ss_iterinit(ss_bufiter, &i);
	ss_iteropen(ss_bufiter, &i, &log->buf, sizeof(svlogv));
	for (; ss_iteratorhas(&i); ss_iteratornext(&i)) {
		svlogv *v = ss_iteratorof(&i);
		ss_free(c->a, v->v);
	}
	sv_logfree(log, c);
}

static void
sw_itertx(void)
{
	swmanager lp;
	t( sw_managerinit(&lp, &st_r.r) == 0 );
	swconf *conf = sw_conf(&lp);
	conf->path     = strdup(st_r.conf->log_dir);
	conf->enable   = 1;
	conf->rotatewm = 1000;
	t( sw_manageropen(&lp) == 0 );
	t( sw_managerrotate(&lp) == 0 );

	svlog log;
	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);

	alloclogv(&log, &st_r.r, 0, 7);

	swtx ltx;
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );

	freelog(&log, &st_r.r);
	t( sw_managershutdown(&lp) == 0 );
}

static void
sw_itertx_read_empty(void)
{
	swmanager lp;
	t( sw_managerinit(&lp, &st_r.r) == 0 );
	swconf *conf = sw_conf(&lp);
	conf->path     = strdup(st_r.conf->log_dir);
	conf->enable   = 1;
	conf->rotatewm = 1000;
	t( sw_manageropen(&lp) == 0 );
	t( sw_managerrotate(&lp) == 0 );
	svlog log;
	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	freelog(&log, &st_r.r);

	sw *current = sscast(lp.list.prev, sw, link);
	ssiter li;
	ss_iterinit(sw_iter, &li);
	t( ss_iteropen(sw_iter, &li, &st_r.r, &current->file, 1) == 0 );
	for (;;) {
		// begin
		while (ss_iteratorhas(&li)) {
			swv *v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 7 );
			ss_iteratornext(&li);
		}
		t( sw_iter_error(&li) == 0 );
		// commit
		if (! sw_iter_continue(&li) )
			break;
	}
	ss_iteratorclose(&li);

	t( sw_managershutdown(&lp) == 0 );
}

static void
sw_itertx_read0(void)
{
	swmanager lp;
	t( sw_managerinit(&lp, &st_r.r) == 0 );
	swconf *conf = sw_conf(&lp);
	conf->path     = strdup(st_r.conf->log_dir);
	conf->enable   = 1;
	conf->rotatewm = 1000;
	t( sw_manageropen(&lp) == 0 );
	t( sw_managerrotate(&lp) == 0 );
	svlog log;
	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 7);
	swtx ltx;
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	sw *current = sscast(lp.list.prev, sw, link);
	ssiter li;
	ss_iterinit(sw_iter, &li);
	t( ss_iteropen(sw_iter, &li, &st_r.r, &current->file, 1) == 0 );
	for (;;) {
		// begin
		while (ss_iteratorhas(&li)) {
			swv *v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 7 );
			ss_iteratornext(&li);
		}
		t( sw_iter_error(&li) == 0 );
		// commit
		if (! sw_iter_continue(&li) )
			break;
	}
	ss_iteratorclose(&li);

	t( sw_managershutdown(&lp) == 0 );
}

static void
sw_itertx_read1(void)
{
	swmanager lp;
	t( sw_managerinit(&lp, &st_r.r) == 0 );
	swconf *conf = sw_conf(&lp);
	conf->path     = strdup(st_r.conf->log_dir);
	conf->enable   = 1;
	conf->rotatewm = 1000;
	t( sw_manageropen(&lp) == 0 );
	t( sw_managerrotate(&lp) == 0 );
	svlog log;
	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 7);
	alloclogv(&log, &st_r.r, 0, 8);
	alloclogv(&log, &st_r.r, 0, 9);
	swtx ltx;
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	sw *current = sscast(lp.list.prev, sw, link);
	ssiter li;
	ss_iterinit(sw_iter, &li);
	t( ss_iteropen(sw_iter, &li, &st_r.r, &current->file, 1) == 0 );
	for (;;) {
		// begin
		t( ss_iteratorhas(&li) == 1 );
		swv *v = ss_iteratorof(&li);
		t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 7 );
		ss_iteratornext(&li);
		t( ss_iteratorhas(&li) == 1 );
		v = ss_iteratorof(&li);
		t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 8 );
		ss_iteratornext(&li);
		t( ss_iteratorhas(&li) == 1 );
		v = ss_iteratorof(&li);
		t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 9 );
		ss_iteratornext(&li);
		t( ss_iteratorhas(&li) == 0 );

		t( sw_iter_error(&li) == 0 );
		// commit
		if (! sw_iter_continue(&li) )
			break;
	}
	ss_iteratorclose(&li);

	t( sw_managershutdown(&lp) == 0 );
}

static void
sw_itertx_read2(void)
{
	swmanager lp;
	t( sw_managerinit(&lp, &st_r.r) == 0 );
	swconf *conf = sw_conf(&lp);
	conf->path     = strdup(st_r.conf->log_dir);
	conf->enable   = 1;
	conf->rotatewm = 1000;
	t( sw_manageropen(&lp) == 0 );
	t( sw_managerrotate(&lp) == 0 );
	svlog log;
	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 7);
	alloclogv(&log, &st_r.r, 0, 8);
	alloclogv(&log, &st_r.r, 0, 9);
	swtx ltx;
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );

	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	sw *current = sscast(lp.list.prev, sw, link);
	ssiter li;
	ss_iterinit(sw_iter, &li);
	t( ss_iteropen(sw_iter, &li, &st_r.r, &current->file, 1) == 0 );
	for (;;) {
		// begin
		t( ss_iteratorhas(&li) == 1 );
		swv *v = ss_iteratorof(&li);
		t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 7 );
		ss_iteratornext(&li);
		t( ss_iteratorhas(&li) == 1 );
		v = ss_iteratorof(&li);
		t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 8 );
		ss_iteratornext(&li);
		t( ss_iteratorhas(&li) == 1 );
		v = ss_iteratorof(&li);
		t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 9 );
		ss_iteratornext(&li);

		t( ss_iteratorhas(&li) == 0 );
		t( sw_iter_error(&li) == 0 );
		// commit
		if (! sw_iter_continue(&li) )
			break;
	}
	ss_iteratorclose(&li);

	t( sw_managershutdown(&lp) == 0 );
}

static void
sw_itertx_read3(void)
{
	swmanager lp;
	t( sw_managerinit(&lp, &st_r.r) == 0 );
	swconf *conf = sw_conf(&lp);
	conf->path     = strdup(st_r.conf->log_dir);
	conf->enable   = 1;
	conf->rotatewm = 1000;
	t( sw_manageropen(&lp) == 0 );
	t( sw_managerrotate(&lp) == 0 );
	svlog log;

	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 7); /* single stmt */
	swtx ltx;
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 8); /* multi stmt */
	alloclogv(&log, &st_r.r, 0, 9);
	alloclogv(&log, &st_r.r, 0, 10);
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 11); /* multi stmt */
	alloclogv(&log, &st_r.r, 0, 12);
	alloclogv(&log, &st_r.r, 0, 13);
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	sv_loginit(&log, &st_r.r, 1);
	sv_loginit_index(&log, 0, &st_r.r);
	alloclogv(&log, &st_r.r, 0, 14); /* single stmt */
	t( sw_begin(&lp, &ltx, 0, 0) == 0 );
	t( sw_write(&ltx, &log) == 0 );
	t( sw_commit(&ltx) == 0 );
	freelog(&log, &st_r.r);

	int state = 0;

	sw *current = sscast(lp.list.prev, sw, link);
	ssiter li;
	ss_iterinit(sw_iter, &li);
	t( ss_iteropen(sw_iter, &li, &st_r.r, &current->file, 1) == 0 );
	for (;;) {
		swv *v;
		// begin
		switch (state) {
		case 0:
			t( ss_iteratorhas(&li) == 1 );
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 7 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( v == NULL );
			break;
		case 1:
			t( ss_iteratorhas(&li) == 1 );
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 8 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 9 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 10 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( v == NULL );
			break;
		case 2:
			t( ss_iteratorhas(&li) == 1 );
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 11 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 12 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 13 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( v == NULL );
			break;
		case 3:
			t( ss_iteratorhas(&li) == 1 );
			v = ss_iteratorof(&li);
			t( *(int*)sf_field(st_r.r.scheme, 0, sw_vpointer(v), &st_r.size) == 14 );
			ss_iteratornext(&li);
			v = ss_iteratorof(&li);
			t( v == NULL );
			break;
		}
		// commit
		if (! sw_iter_continue(&li) )
			break;
		state++;
	}
	ss_iteratorclose(&li);
	t( state == 3);

	t( sw_managershutdown(&lp) == 0 );
}

stgroup *sw_iter_group(void)
{
	stgroup *group = st_group("switer");
	st_groupadd(group, st_test("tx", sw_itertx));
	st_groupadd(group, st_test("tx_read_empty", sw_itertx_read_empty));
	st_groupadd(group, st_test("tx_read0", sw_itertx_read0));
	st_groupadd(group, st_test("tx_read1", sw_itertx_read1));
	st_groupadd(group, st_test("tx_read2", sw_itertx_read2));
	st_groupadd(group, st_test("tx_read3", sw_itertx_read3));
	return group;
}