#include <psyc.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <lib.h>

uint8_t verbose;

int
test_update (const char *buf, size_t buflen,
	     const char *r_idx, size_t r_idxlen, char r_oper,
	     const char *r_typ, size_t r_typlen,
	     const char *r_val, size_t r_vallen)
{
    char *idx = malloc(r_idxlen * 2);
    char *typ = NULL, *val = NULL;
    size_t idxlen = 0, typlen = 0, vallen = 0;

    int ret, reti, len = 0;
    char oper;
    PsycString value;
    PsycParseUpdateState state;
    psyc_parse_update_state_init(&state);
    psyc_parse_update_buffer_set(&state, buf, buflen);

    if (verbose)
	printf(">> %.*s\n", (int)buflen, buf);

    do {
	ret = reti = psyc_parse_update(&state, &oper, &value);
	len = 0;

	switch (ret) {
	case PSYC_PARSE_INDEX_LIST:
	    len = sprintf(idx + idxlen, "#%ld", value.length);
	    break;
	case PSYC_PARSE_INDEX_STRUCT:
	    len = sprintf(idx + idxlen, ".%.*s", PSYC_S2ARGP(value));
	    break;
	case PSYC_PARSE_INDEX_DICT:
	    if (state.elemlen_found)
		len = sprintf(idx + idxlen, "{%ld %.*s}",
			      value.length, PSYC_S2ARGP(value));
	    else
		len = sprintf(idx + idxlen, "{%.*s}", PSYC_S2ARGP(value));
	    break;
	case PSYC_PARSE_UPDATE_TYPE_END:
	    ret = 0;
	case PSYC_PARSE_UPDATE_TYPE:
	    typ = value.data;
	    typlen = value.length;
	    if (verbose)
		printf("%c[%.*s]\n", oper, PSYC_S2ARGP(value));
	    break;
	case PSYC_PARSE_UPDATE_VALUE:
	    val = value.data;
	    vallen = value.length;
	    ret = 0;
	    if (verbose)
		printf("[%.*s]\n", PSYC_S2ARGP(value));
	    break;
	case PSYC_PARSE_UPDATE_END:
	    ret = 0;
	    break;
	default:
	    ret = -1;
	}
	if (verbose && len)
		printf("%2d: %.*s\n", reti, len, idx + idxlen);
	idxlen += len;
    } while (ret > 0);

    if (ret == 0) {
	if (idxlen != r_idxlen || oper != r_oper
	    || typlen != r_typlen || vallen != r_vallen
	    || memcmp(r_idx, idx, idxlen) != 0
	    || memcmp(r_typ, typ, typlen) != 0
	    || memcmp(r_val, val, vallen) != 0) {
	    printf("ERROR: got\n\[%.*s] %c[%.*s]:[%ld] [%.*s] instead of\n[%.*s]\n",
		   (int)idxlen, idx, oper, (int)typlen, typ, vallen, (int)vallen, val,
		   (int)buflen, buf);
	    ret = 1;
	} else
	    ret = 0;
    }

    free(idx);
    return ret;
}

int
main (int argc, char **argv)
{
    verbose = argc > 1;

    if (test_update(PSYC_C2ARG("#1{foo}._bar =_foo:3 bar"),
		    PSYC_C2ARG("#1{foo}._bar"), '=',
		    PSYC_C2ARG("_foo"),
		    PSYC_C2ARG("bar")) != 0)
	return 1;

    if (test_update(PSYC_C2ARG("{3 foo}._bar#0 +:3 baz"),
		    PSYC_C2ARG("{3 foo}._bar#0"), '+',
		    PSYC_C2ARG(""),
		    PSYC_C2ARG("baz")) != 0)
	return 2;

    if (test_update(PSYC_C2ARG("{foo}#2._bar - 1337"),
		    PSYC_C2ARG("{foo}#2._bar"), '-',
		    PSYC_C2ARG(""),
		    PSYC_C2ARG("1337")) != 0)
	return 3;

    if (test_update(PSYC_C2ARG("._bar#1{3 foo} ="),
		    PSYC_C2ARG("._bar#1{3 foo}"), '=',
		    PSYC_C2ARG(""),
		    PSYC_C2ARG("")) != 0)
	return 4;

    if (test_update(PSYC_C2ARG("#1{3 foo}._bar =_list"),
		    PSYC_C2ARG("#1{3 foo}._bar"), '=',
		    PSYC_C2ARG("_list"),
		    PSYC_C2ARG("")) != 0)
	return 5;

    if (test_update(PSYC_C2ARG("#1{3 foo}._bar =_list "),
		    PSYC_C2ARG("#1{3 foo}._bar"), '=',
		    PSYC_C2ARG("_list"),
		    PSYC_C2ARG("")) != 0)
	return 6;

    if (test_update(PSYC_C2ARG("#1{3 foo}._bar =_list"),
		    PSYC_C2ARG("#1{3 foo}._bar"), '=',
		    PSYC_C2ARG("_list"),
		    PSYC_C2ARG("")) != 0)
	return 7;

    printf("test_update passed all tests.\n");
    return 0; // passed all tests
}