#include "ipf.h"
#include <err.h>

extern int nohdrfields;

wordtab_t *parsefields(wordtab_t *table, char *arg)
{
	wordtab_t *f, *fields;
	char *s, *t;
	int num;

	fields = NULL;
	num = 0;

	for (s = strtok(arg, ","); s != NULL; s = strtok(NULL, ",")) {
		t = strchr(s, '=');
		if (t != NULL) {
			*t++ = '\0';
			if (*t == '\0')
				nohdrfields = 1;
		}

		f = findword(table, s);
		if (f == NULL) {
			fprintf(stderr, "Unknown field '%s'\n", s);
			exit(1);
		}

		num++;
		if (fields == NULL) {
			fields = malloc(2 * sizeof(*fields));
		} else {
			fields = reallocarray(fields, num + 1, sizeof(*fields));
			if (fields == NULL) {
				warnx("memory allocation error at %d in %s in %s", __LINE__, __FUNCTION__, __FILE__);
				abort();
			}
		}

		if (t == NULL) {
			fields[num - 1].w_word = f->w_word;
		} else {
			fields[num - 1].w_word = t;
		}
		fields[num - 1].w_value = f->w_value;
		fields[num].w_word = NULL;
		fields[num].w_value = 0;
	}

	return (fields);
}
