#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/types.h>
#include <elwix.h>


int
run_simple(json_t *json)
{
	static const char *teststr = "{\"user\": \"johndoe\", { \"meow\", \"\", \"\aaa\"}, \"admin\": false, \"uid\": 1000, \"objects\" : { \"a\":\"1\", \"b\" : \"2\" ,\"c\" : \"3\", , \"oho\"},\n"
       				"\"groups\": [\"users\", \"\", \"wheel\", \"audio\", \"video\"], \"testend\":{\"x\" : \"8\", \"y\":\"7\", \"z\" : \"9\"}}";
//	static const char *teststr = "{}\n[]\n\"sdfsdf\"";
	int i, ret;
	jtok_t *tok, toks[36];
	ait_val_t *v;
	char *str;
	array_t *arr;

	json_init(json, 0);
	ret = json_parse(json, teststr, strlen(teststr), NULL, 0);
	if (ret == -1) {
		printf("Error:: #%d - %s\n", elwix_GetErrno(), elwix_GetError());
		return 1;
	} else
		printf("We are need from %d tokens\n", ret);
	json_init(json, 1);
	ret = json_parse(json, teststr, strlen(teststr), NULL, 0);
	if (ret == -1) {
		printf("Error:: #%d - %s\n", elwix_GetErrno(), elwix_GetError());
		return 1;
	} else
		printf("(strict) We are need from %d tokens\n", ret);
	json_init(json, 0);
	ret = json_parse(json, teststr, strlen(teststr), toks, sizeof toks/sizeof *toks);
	if (ret == -1) {
		printf("Error:: #%d - %s\n", elwix_GetErrno(), elwix_GetError());
		return 1;
	} else
		printf("We parsed %d tokens next=%lu\n", ret, json->h_next);
	memset(toks, 0, sizeof toks);
	json_init(json, 1);
	ret = json_parse(json, teststr, strlen(teststr), toks, sizeof toks/sizeof *toks);
	if (ret == -1) {
		printf("Error:: #%d - %s\n", elwix_GetErrno(), elwix_GetError());
		return 1;
	} else
		printf("(strict) We parsed %d tokens next=%lu\n", ret, json->h_next);

	if (!(tok = json_findbykey(teststr, "boza s kosmi", toks, ret)))
		printf("Key=\"boza s kosmi\" not found!\n");
	tok = json_findbykey(teststr, "user", toks, ret);
	if (tok) {
		v = json_token2val(teststr, tok);
		printf("Key=\"user\" data parent=%ld idx=%ld size=%ld type=%d %s\n", tok->tok_parent, tok->tok_idx, tok->tok_size, tok->tok_type, AIT_GET_STR(v));
		ait_freeVar(&v);
	}
	tok = json_findbykey(teststr, "admin", toks, ret);
	if (tok) {
		str = json_token2str(teststr, tok);
		printf("Key=\"admin\" data parent=%ld idx=%ld size=%ld type=%d %s\n", tok->tok_parent, tok->tok_idx, tok->tok_size, tok->tok_type, str);
		e_free(str);
	}
	tok = json_findbykey(teststr, "uid", toks, ret);
	if (tok) {
		printf("Key=\"uid\" data parent=%ld idx=%ld size=%ld type=%d %ld\n", tok->tok_parent, tok->tok_idx, tok->tok_size, tok->tok_type, 
				json_token2num(teststr, tok));
		arr = json_token2array(teststr, tok);
		for (i = 0; i < array_Size(arr); i++)
			printf("arr[%d]=%s\n", i, AIT_GET_STR(array(arr, i, ait_val_t*)));
		ait_freeVars(&arr);
	}

	tok = json_findbykey(teststr, "groups", toks, ret);
	if (tok) {
		str = json_token2str(teststr, tok);
		arr = json_token2array(teststr, tok);
		printf("Key=\"groups\" data parent=%ld idx=%ld size=%ld type=%d %s\n", tok->tok_parent, tok->tok_idx, tok->tok_size, tok->tok_type, str);
		e_free(str);
		for (i = 0; i < array_Size(arr); i++)
			printf("arr[%d]=%s\n", i, AIT_GET_STR(array(arr, i, ait_val_t*)));
		ait_freeVars(&arr);
	}

	tok = json_findbykey(teststr, "objects", toks, ret);
	if (tok) {
		str = json_token2str(teststr, tok);
		arr = json_token2array(teststr, tok);
		printf("Key=\"objects\" data parent=%ld idx=%ld size=%ld type=%d %s\n", tok->tok_parent, tok->tok_idx, tok->tok_size, tok->tok_type, str);
		e_free(str);
		for (i = 0; i < array_Size(arr); i++)
			printf("arr[%d]=%s\n", i, AIT_GET_STR(array(arr, i, ait_val_t*)));
		ait_freeVars(&arr);
	}

	tok = json_findbykey(teststr, "testend", toks, ret);
	if (tok) {
		str = json_token2str(teststr, tok);
		arr = json_token2array(teststr, tok);
		printf("Key=\"testend\" data parent=%ld idx=%ld size=%ld type=%d %s\n", tok->tok_parent, tok->tok_idx, tok->tok_size, tok->tok_type, str);
		e_free(str);
		for (i = 0; i < array_Size(arr); i++)
			printf("arr[%d]=%s\n", i, AIT_GET_STR(array(arr, i, ait_val_t*)));
		ait_freeVars(&arr);
	}

	return 0;
}

static int
dump(const char *jstr, jtok_t *toks, int toksnum, int indent)
{
	register int i, j, k;

	if (!toksnum)
		return 0;

	if (toks->tok_type == J_VALUE) {
		printf("%.*s", (int) json_toklen(toks), json_tokstr(jstr, toks));
		return 1;
	} else if (toks->tok_type == J_STRING) {
		printf("%.*s", (int) json_toklen(toks), json_tokstr(jstr, toks));
		return 1;
	} else if (toks->tok_type == J_OBJECT) {
		printf("\n");
		for (j = i = 0; i < json_toksize(toks); i++) {
			for (k = 0; k < indent; k++)
				printf("  ");
			j += dump(jstr, toks + j + 1, toksnum - j, indent + 1);
			printf(": ");
			j += dump(jstr, toks + j + 1, toksnum - j, indent + 1);
			printf("\n");
		}
		return j + 1;
	} else if (toks->tok_type == J_ARRAY) {
		printf("\n");
		for (j = i = 0; i < json_toksize(toks); i++) {
			for (k = 0; k < indent - 1; k++)
				printf("  ");
			printf("   - ");
			j += dump(jstr, toks + j + 1, toksnum - j, indent + 1);
			printf("\n");
		}
		return j + 1;
	}

	return 0;
}

int
run_dump(json_t *json, char **args)
{
	int rlen, toksnum, wait4eof = 0;
	char buf[BUFSIZ];
	ait_val_t v;
	jtok_t *toks, *t;
	FILE *f;

	json_init(json, 0);
	f = fopen(args[1], "r");
	if (!f) {
		printf("fopen(%s) #%d - %s\n", *args, errno, strerror(errno));
		return 1;
	}

	toks = e_malloc(sizeof(jtok_t));
	if (!toks) {
		fclose(f);
		return 1;
	} else
		toksnum = 1;
	AIT_SET_STRSIZ(&v, 1);
	while (42) {
		memset(buf, 0, sizeof buf);
		rlen = fread(buf, 1, sizeof buf, f);
		switch (rlen) {
			case -1:
				printf("Error:: fread() #%d - %s\n", errno, strerror(errno));
				AIT_FREE_VAL(&v);
				e_free(toks);
				fclose(f);
				return 2;
			case 0:
				if (wait4eof)
					return 0;
				printf("Error:: Unexpected termination!\n");
				AIT_FREE_VAL(&v);
				e_free(toks);
				fclose(f);
				return 3;
			default:
				AIT_SET_STRCAT(&v, buf);
				break;
		}
again:
		rlen = json_parse(json, AIT_GET_STR(&v), AIT_LEN(&v), toks, toksnum);
		if (rlen == (u_int) -1) {
		       	if (elwix_GetErrno() == J_ERR_NOMEM) {
				toksnum *= 2;
				t = e_realloc(toks, sizeof(jtok_t) * toksnum);
				if (t) {
					toks = t;
					goto again;
				}
			}
		} else {
			dump(AIT_GET_STR(&v), toks, json->h_next, 0);
			wait4eof = 1;
		}
	}
	AIT_FREE_VAL(&v);
	e_free(toks);
	fclose(f);

	return 0;
}

int
main(int argc, char **argv)
{
	json_t json;

	if (argc < 2)
		return run_simple(&json);
	else
		return run_dump(&json, argv);

	return 0;
}