#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>

#include <dynarr.h>
#include <errors.h>
#include <macros.h>
#include <mbstring.h>
#include <unicode/string.h>

#define TESTFILE "wbrk-human.in"

static bool test(struct u8view, int);

int
main(int, char **argv)
{
	int rv;
	size_t n;
	ssize_t nr;
	char *line;
	FILE *fp;

	rv = EXIT_SUCCESS;
	line = nullptr;
	mlib_setprogname(argv[0]);

	if ((fp = fopen(TESTFILE, "r")) == nullptr)
		err("fopen: %s:", TESTFILE);

	for (int id = 1; (nr = getline(&line, &n, fp)) > 0; id++) {
		if (line[nr - 1] == '\n')
			line[--nr] = '\0';

		if (!test((struct u8view){line, nr}, id))
			rv = EXIT_FAILURE;
	}
	if (ferror(fp))
		err("getline: %s:", TESTFILE);

	free(line);
	fclose(fp);
	return rv;
}

bool
test(struct u8view sv, int id)
{
	struct u8view src;
	ucscut(&src, &sv, U";", 1);

	struct u8view w;
	dynarr(struct u8view) ws = {.alloc = alloc_heap};

	while (ucscut(&w, &sv, U"|", 1) != MBEND)
		DAPUSH(&ws, w);
	if (w.len > 0)
		DAPUSH(&ws, w);

	/* Assert the word count is correct */
	size_t n;
	if ((n = ucswcnt_human(src)) != ws.len) {
		warn("case %d: expected %zu words but got %zu", id, ws.len, n);
		return false;
	}

	/* Assert the individual words are correct */
	for (size_t i = 0; ucswnext_human(&w, &src) != 0; i++) {
		if (!ucseq(w, ws.buf[i])) {
			warn("case %d: expected word %zu to be ‘%.*s’ but got ‘%.*s’", id,
			     i, SV_PRI_ARGS(ws.buf[i]), SV_PRI_ARGS(w));
			return false;
		}
	}

	free(ws.buf);
	return true;
}