module lorem

const lorem_corpora = {
	'lorem':  lorem_corpus
	'poe':    poe_corpus
	'darwin': darwin_corpus
	'bard':   shakespeare_corpus
}

const lorem_corpus = '
lorem ipsum dolor sit amet consectetur adipiscing elit sed do eiusmod tempor
incididunt ut labore et dolore magna aliqua Ut enim ad minim veniam quis
nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat
Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore
eu fugiat nulla pariatur Excepteur sint occaecat cupidatat non proident sunt
in culpa qui officia deserunt mollit anim id est laborum
'

const poe_corpus = '
once upon a midnight dreary while I pondered weak and weary
over many a quaint and curious volume of forgotten lore
while I nodded nearly napping suddenly there came a tapping
as of someone gently rapping at my chamber door
'

const darwin_corpus = '
when we look to the individuals of the same variety or sub variety of our
older cultivated plants and animals one of the first points which strikes
us is that they generally differ much more from each other than do the
individuals of any one species or variety in a state of nature
'

const shakespeare_corpus = '
to be or not to be that is the question
all the worlds a stage and all the men and women merely players
the lady doth protest too much methinks
a rose by any other name would smell as sweet
et tu brute
if music be the food of love play on
now is the winter of our discontent
we are such stuff as dreams are made on
brevity is the soul of wit
some are born great some achieve greatness and some have greatness thrust upon them
cry havoc and let slip the dogs of war
all that glisters is not gold
the fault dear brutus is not in our stars but in ourselves
to thine own self be true
lord what fools these mortals be
shall i compare thee to a summers day
'

// ---------------- Seed Phrases ----------------

const lorem_seed_phrases = [
	'in the beginning',
	'once upon a time',
	'it was the first',
	'when we consider',
	'there was a moment',
]

// ---------------- Text Generation ----------------

// LoremCfg
// - `corpus_name` random if not specified
// - `seed_text`   random if not specified
// - `rng_seed`    random if not specified
@[params]
pub struct LoremCfg {
pub:
	words_per_sentence      int = 10
	sentences_per_paragraph int = 5
	paragraphs              int = 3
	markov_order            int = 2
	corpus_name             string
	seed_text               string
	rng_seed                int
}

// lorum - pseudo-random text using a Markov chain built from
// one of several embedded corpora. It produces structured output in the
// form of paragraphs and sentences, with configurable parameters for:
//
// - Markov order (n-gram size)
// - Words per sentence
// - Sentences per paragraph
// - Paragraph count
// - Optional seed phrases and RNG seed
// - Optional corpus selection
//
pub fn generate(cfg LoremCfg) string {
	// Initialize LCG with provided seed or default
	// If rng_seed is 0, we use a default seed (making it deterministic by default)
	initial_seed := if cfg.rng_seed != 0 { u32(cfg.rng_seed) } else { u32(123456789) }
	mut rng := LorumRNG{
		seed: initial_seed
	}

	seed := match cfg.seed_text != '' {
		true { cfg.seed_text }
		else { lorem_random_seed_phrase(mut rng) }
	}

	corpus := lorem_select_corpus(mut rng, cfg.corpus_name)
	tokens := lorem_tokenize(corpus)

	if tokens.len <= cfg.markov_order {
		eprintln('corpus too small for selected order')
		return ''
	}

	model := lorem_build_markov(tokens, cfg.markov_order)

	mut state := lorem_tokenize(seed)
	if state.len < cfg.markov_order {
		start := rng.intn(tokens.len - cfg.markov_order)
		state = tokens[start..start + cfg.markov_order].clone()
	}

	mut out := []string{}

	for pi in 0 .. cfg.paragraphs {
		if pi != 0 {
			out << '\n\n'
		}
		sentences := lorem_vary(mut rng, cfg.sentences_per_paragraph, 1)

		for si in 0 .. sentences {
			if si != 0 {
				out << ' '
			}
			words := lorem_vary(mut rng, cfg.words_per_sentence, 3)
			mut sentence := []string{}

			for _ in 0 .. words {
				key := state.join('\u0001')
				nexts := model[key] or {
					start := rng.intn(tokens.len - cfg.markov_order)
					state = tokens[start..start + cfg.markov_order].clone()
					continue
				}

				next := nexts[rng.intn(nexts.len)]
				sentence << next

				state = state[1..].clone()
				state << next
			}

			if sentence.len > 0 {
				out << sentence.join(' ').capitalize()
				out << '.'
			}
		}
	}

	return out.join('')
}

// ---------------- Utilities ----------------

// LorumRNG is a simple internal PRNG used to avoid importing the `rand` module,
// which would cause an import cycle (strings -> rand -> time -> ... -> strings).
struct LorumRNG {
mut:
	seed u32
}

// intn returns a pseudo-random number in [0, max).
// It is a minimal implementation sufficient for text generation.
fn (mut r LorumRNG) intn(max int) int {
	if max <= 0 {
		return 0
	}
	r.seed = r.seed * 1664525 + 1013904223
	return int(r.seed % u32(max))
}

fn lorem_vary(mut rng LorumRNG, base int, min int) int {
	delta := int(f32(base) * 0.2)
	if delta == 0 {
		return base
	}
	offset := rng.intn(delta * 2 + 1) - delta
	val := base + offset
	return if val < min { min } else { val }
}

fn lorem_select_corpus(mut rng LorumRNG, name string) string {
	if name != '' {
		if corpus := lorem_corpora[name] {
			return corpus
		}
		eprintln('unknown corpus: ${name}')
		exit(1)
	}

	keys := lorem_corpora.keys()
	key := keys[rng.intn(keys.len)]
	return lorem_corpora[key]
}

fn lorem_random_seed_phrase(mut rng LorumRNG) string {
	return lorem_seed_phrases[rng.intn(lorem_seed_phrases.len)]
}

fn lorem_tokenize(text string) []string {
	return text
		.replace_each(['\n', ' ', '\t', ' '])
		.split(' ')
		.filter(it.len > 0)
}

fn lorem_build_markov(tokens []string, order int) map[string][]string {
	mut model := map[string][]string{}
	for i in 0 .. tokens.len - order {
		key := tokens[i..i + order].join('\u0001')
		if key !in model {
			model[key] = []string{}
		}
		model[key] << tokens[i + order]
	}
	return model
}