libpsyc/test/test_parser.c

#include <unistd.h>
#include <stdio.h>
#include <fcntl.h>

#include <psyc.h>
#include <psyc/parse.h>

int
main (int argc, char **argv)
{
    uint8_t routing_only = argc > 2 && memchr(argv[2], (int)'r', strlen(argv[2]));
    uint8_t verbose = argc > 2 && memchr(argv[2], (int)'v', strlen(argv[2]));
    int idx, ret;
    char buffer[2048], oper;
    PsycString name, value, elem;
    PsycParseState state;
    PsycParseListState listState;

    if (argc < 2)
	return -1;
    int file = open(argv[1],O_RDONLY);
    if (file < 0)
	return -1;
    idx = read(file,(void*)buffer,sizeof(buffer));

    if (verbose) {
	printf(">> INPUT\n");
	printf("%.*s\n", (int)idx, buffer);
	printf(">> PARSE\n");
    }

    psyc_parse_state_init(&state, routing_only ?
			  PSYC_PARSE_ROUTING_ONLY : PSYC_PARSE_ALL);
    psyc_parse_buffer_set(&state, buffer, idx);

    // try parsing that now
    do {
	oper = 0;
	name.length = 0;
	value.length = 0;

	ret = psyc_parse(&state, &oper, &name, &value);
	if (verbose)
	    printf(">> ret = %d\n", ret);

	switch (ret) {
	case PSYC_PARSE_ROUTING:
	case PSYC_PARSE_ENTITY:
	    if (verbose)
		printf("%c", oper);
	case PSYC_PARSE_BODY:
	    // printf("the string is '%.*s'\n", name);
	    if (verbose)
		printf("%.*s = %.*s\n",
		       (int)name.length, name.data,
		       (int)value.length, value.data);

	    if (psyc_var_is_list(PSYC_S2ARG(name))) {
		if (verbose)
		    printf(">> LIST START\n");

		psyc_parse_list_state_init(&listState);
		psyc_parse_list_buffer_set(&listState, PSYC_S2ARG(value));

		while ((ret = psyc_parse_list(&listState, &elem))) {
		    switch (ret) {
		    case PSYC_PARSE_LIST_END:
		    case PSYC_PARSE_LIST_ELEM:
			if (verbose)
			    printf("|%.*s\n", (int)elem.length, elem.data);
			break;
		    default:
			printf("Error while parsing list: %i\n", ret);
			return 1;
		    }

		    if (ret == PSYC_PARSE_LIST_END) {
			if (verbose)
			    printf(">> LIST END\n");
			break;
		    }
		}
	    }
	    break;
	case PSYC_PARSE_COMPLETE:
	    // printf("Done parsing.\n");
	    ret = 0;
	    continue;
	case PSYC_PARSE_INSUFFICIENT:
	    printf("Insufficient data.\n");
	    return 1;
	default:
	    printf("Error while parsing: %i\n", ret);
	    return 1;
	}
    }
    while (ret);

    return 0;
}