/** Run with a `KJV` sub-directory,
 <https://github.com/scrollmapper/bible_databases/master/txt/KJV/>, outputs a
 `gperf` file that has all the the words of all the verses in `kjv.h` format.
 @license 2022 Neil Edelman, distributed under the terms of the
 [MIT License](https://opensource.org/licenses/MIT). Uses the KJV at
 [bible databases](https://github.com/scrollmapper/bible_databases/tree/master),
 "All included Bible translations are in the public domain."
 @std C11 */

#include "../src/text.h"
#define KJV_OMIT_VERSES
#define KJV_OMIT_PROTO
#include "../src/kjv.h" /* Just the base data. */
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <assert.h>
#include <dirent.h> /* opendir readdir closedir */
#include <unistd.h> /* chdir (POSIX) (because I'm lazy) */
/* #include <cmph.h> No; overkill. */


/* Reversible hash map. */
/** <https://nullprogram.com/blog/2018/07/31/>
 <https://github.com/skeeto/hash-prospector> on `x`. */
static uint32_t lowbias32(uint32_t x) {
	x ^= x >> 16;
	x *= 0x7feb352dU;
	x ^= x >> 15;
	x *= 0x846ca68bU;
	x ^= x >> 16;
	return x;
}
/* Inverts `x`. */
static uint32_t lowbias32_r(uint32_t x) {
	x ^= x >> 16;
	x *= 0x43021123U;
	x ^= x >> 15 ^ x >> 30;
	x *= 0x1d69e2a5U;
	x ^= x >> 16;
	return x;
}

static uint32_t kjv_hash(const union kjvcite x) { return lowbias32(x.u32); }
static union kjvcite kjv_unhash(const uint32_t x)
	{ union kjvcite k; k.u32 = lowbias32_r(x); return k; }
static void kjv_to_string(const union kjvcite x, char (*const a)[12])
	{ sprintf(*a, "%.4s%u:%u", kjv_book_string[x.book],
	(x.chapter + 1) % 1000, (x.verse + 1) % 1000); }
#define TABLE_NAME kjv
#define TABLE_KEY union kjvcite
#define TABLE_UINT uint32_t
#define TABLE_INVERSE
#define TABLE_TO_STRING
#include "../src/table.h"

/* Derived information on verse word count. */
static uint32_t verse_hash(const union kjvcite x) { return kjv_hash(x); }
static union kjvcite verse_unhash(const uint32_t x) { return kjv_unhash(x); }
static void verse_to_string(const union kjvcite x, const unsigned count,
	char (*const a)[12]) { (void)count; kjv_to_string(x, a); }
#define TABLE_NAME verse
#define TABLE_KEY union kjvcite
#define TABLE_UINT uint32_t
#define TABLE_VALUE unsigned /* Count words. */
#define TABLE_INVERSE
#define TABLE_DEFAULT 0
#define TABLE_TO_STRING
#include "../src/table.h"


/** Helper to parse unsigned; [`s`,`e`) => `n`. */
static int parse_natural(const char *s, const char *const e, unsigned *const n) {
	unsigned accum = 0;
	while(s < e) {
		unsigned next = accum * 10 + (unsigned)(*s - '0');
		if(accum >= next) return errno = ERANGE, 0;
		accum = next;
		s++;
	}
	*n = accum;
	return 1;
}


/* Parse filename of books. */
/*!re2c /**/
re2c:yyfill:enable = 0;
re2c:define:YYCTYPE = char;
natural = [1-9][0-9]*;
whitespace = [ \t\v\f];
word = [^ \t\v\f\n\x00]+;
*/
/** `fn` contains "<number>[*].txt", sticks that in `book_no`, otherwise
 returns false. */
static int looks_like_book_fn(const char *fn, unsigned *const book_no) {
	const char *YYCURSOR = fn, *YYMARKER, *yyt1, *yyt2, *s0, *s1;
	assert(fn && book_no);
	/*!re2c /**/
	*
		{ return 0; }
	@s0 natural @s1 [^.\x00]* ".txt" "\x00"
		{ return parse_natural(s0, s1, book_no); }
	*/
}


/* This is the contents of the <fn:looks_like_book_fn>. */
struct lex {
	size_t line;
	const char *cursor;
	int error;
	unsigned chapter, verse, words;
};
static struct lex lex(const char *cursor) {
	struct lex lex;
	assert(cursor);
	lex.line = 1;
	lex.cursor = cursor;
	lex.error = 0;
	lex.chapter = lex.verse = lex.words = 0;
	return lex;
}
/*!conditions:re2c*/
static int lex_next_verse(struct lex *const lex) {
	const char *YYMARKER, *yyt1 = 0, *yyt2 = 0, *s0, *s1, *t0, *t1;
	enum YYCONDTYPE condition = yycline;
	/*!re2c /**/
	re2c:define:YYCURSOR = lex->cursor;
	re2c:define:YYGETCONDITION = "condition";
	re2c:define:YYSETCONDITION = "condition = @@;";
	re2c:define:YYGETCONDITION:naked = 1;
	re2c:define:YYSETCONDITION:naked = 1; */
	assert(lex && lex->cursor);
	lex->error = 0;
scan:
	/*!re2c /**/
	<*> * { return errno = EILSEQ, lex->error = 1, 0; }
	<line> [^[\]\n\x00]* "\n" { lex->line++; goto scan; }
	<line> "\x00" { return 0; }
	<line> "[" @s0 natural @s1 ":" @t0 natural @t1 "]" => verse {
		if(!parse_natural(s0, s1, &lex->chapter)
			|| !parse_natural(t0, t1, &lex->verse))
			return errno = EILSEQ, lex->error = 1, 0;
		lex->words = 0;
		/*printf("%u:%u", lex->chapter, lex->verse);*/
		goto scan;
	}
	<verse> whitespace+ { goto scan; }
	<verse> @s0 word @s1 { lex->words++; goto scan; }
	<verse> "\n" { /*printf(" -> %u\n", lex->words);*/ lex->line++; return 1; }
	*/
}


#define KJV_OMIT_BASE
#define KJV_OMIT_VERSES
#include "../src/kjv.h" /* Just the kjv and prototypes. */

void kjv_(struct kjv *const kjv) {
	if(!kjv) return;
	verse_table_(&kjv->verses);
}

struct kjv kjv(void) {
	const char *const dir_kjv = "kjv";
	struct text backing;
	struct kjv kjv = { 0 };
	DIR *dir = 0;
	struct dirent *de = 0;
	struct { size_t offset; int is; } build[KJV_BOOK_SIZE] = { 0 };
	enum kjv_book b = 0;
	int attempted_closedir = 0;

	/* For all files in directory KJV with <#>*.txt, read into backing. */
	if(chdir(dir_kjv) == -1 || !(dir = opendir("."))) goto catch;
	while((de = readdir(dir))) {
		unsigned ordinal;
		char *unstable_book;
		if(!looks_like_book_fn(de->d_name, &ordinal)) continue;
		/*fprintf(stderr, "<%s> ordinal: %u\n", de->d_name, ordinal);*/
		if(ordinal < 1 || ordinal > KJV_BOOK_SIZE)
			{ errno = ERANGE; goto catch; } /* Not in range. */
		if(build[b = ordinal - 1].is) /* Convert to zero-based. */
			{ errno = EDOM; goto catch; } /* Is duplicate. */
		if(!(unstable_book = text_append_file(&backing, de->d_name)))
			goto catch;
		build[b].is = 1;
		build[b].offset = (size_t)(unstable_book - backing.a.data);
	}
	if(attempted_closedir = 1, closedir(dir) == -1) goto catch; dir = 0;

	/* Now backing is stable; count all the words for each verse. */
	for(b = 0; b < KJV_BOOK_SIZE; b++) {
		struct lex x;
		if(!build[b].is) { fprintf(stderr, "Missing book [%u]%s.\n",
			b + 1, kjv_book_string[b]); errno = EDOM; goto catch; }
		x = lex(backing.a.data + build[b].offset);
		while(lex_next_verse(&x)) {
			const union kjvcite cite
				= { .book = b, .chapter = x.chapter, .verse = x.verse };
			unsigned *words;
			switch(verse_table_assign(&kjv.verses, cite, &words)) {
			case TABLE_PRESENT: fprintf(stderr, "[%u]%s %u:%u duplicated.\n",
				b + 1, kjv_book_string[b], x.chapter, x.verse); errno = EDOM;
			case TABLE_ERROR: goto catch;
			case TABLE_ABSENT: break;
			}
			*words = x.words, kjv.words += x.words;
		}
		if(x.error) { fprintf(stderr, "[%u]%s on line %zu\n",
			b + 1, kjv_book_string[b], x.line); goto catch; }
	}
	goto finally;
catch:
	if(de) fprintf(stderr, "While reading %s.\n", de->d_name);
	perror(de ? de->d_name : dir_kjv);
	if(dir && !attempted_closedir && closedir(dir) == -1) perror(dir_kjv);
	kjv_(&kjv);
finally:
	text_(&backing);
	return kjv;
}

unsigned kjv_words(/*const*/ struct kjv *const kjv, const union kjvcite cite) {
	return kjv && verse_table_get(&kjv->verses, cite);
}

const char *kjv_entire_to_string(const struct kjv *const kjv)
	{ return kjv ? verse_table_to_string(&kjv->verses) : ""; }