module tests

import os
import time

const root = @VMODROOT
const binary = os.join_path(root, 'build', 'koh3')

struct Result {
mut:
	code int
	out  string
	err  string
}

fn testsuite_begin() {
	os.mkdir_all(os.join_path(root, 'build')) or { panic(err) }
	build := os.execute('v -o ${q(binary)} ${q(os.join_path(root, 'src', 'koh3.v'))}')
	if build.exit_code != 0 {
		panic('could not build koh3: ${build.output}')
	}
}

fn q(s string) string {
	return "'${s.replace("'", "'\\''")}'"
}

fn temp_repo(name string) string {
	dir := os.join_path(os.temp_dir(), 'koh3-cli-${name}-${time.now().unix_nano()}')
	os.mkdir_all(dir) or { panic(err) }
	return dir
}

fn run(repo string, args ...string) Result {
	stamp := '${os.getpid()}-${time.now().unix_nano()}'
	base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')
	out_path := '${base}.out'
	err_path := '${base}.err'
	code_path := '${base}.code'
	mut command := '${q(binary)}'
	for arg in args {
		command += ' ${q(arg)}'
	}
	shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" $? >${q(code_path)}'
	result := os.execute(shell)
	if result.exit_code != 0 {
		panic('could not run test command: ${result.output}')
	}
	code := os.read_file(code_path) or { panic(err) }
	out := os.read_file(out_path) or { '' }
	err := os.read_file(err_path) or { '' }
	os.rm(out_path) or {}
	os.rm(err_path) or {}
	os.rm(code_path) or {}
	return Result{
		code: code.int()
		out:  out
		err:  err
	}
}

fn init_repo(repo string) {
	r := run(repo, 'init')
	assert r.code == 0
	assert os.exists(os.join_path(repo, '.koh'))
	assert os.exists(os.join_path(repo, '.kohignore'))
}

fn write(repo string, path string, text string) {
	full := os.join_path(repo, path)
	os.mkdir_all(os.dir(full)) or { panic(err) }
	os.write_file(full, text) or { panic(err) }
}

fn read_json_has(text string, key string) {
	if !text.contains(key) {
		panic('missing JSON key ${key}: ${text}')
	}
}

fn test_init_and_usage() {
	dir := temp_repo('init')
	defer { os.rmdir_all(dir) or {} }
	r := run(dir, 'init', '--json')
	assert r.code == 0
	read_json_has(r.out, 'repo')
	read_json_has(r.out, 'ignore_created')
	assert run(dir, 'init').code == 1
	assert run(dir, 'unknown-command').code == 2
	assert run(dir, 'unknown-command').err.contains('unknown command')
}

fn test_save_checkpoint_and_status() {
	dir := temp_repo('save')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'tracked.txt', 'one\n')
	r := run(dir, 'save', 'first message', '--agent', 'agent-7', '--session', 'session-a', '--json')
	assert r.code == 0
	for key in ['id', 'parent', 'message', 'created_at', 'machine', 'is_checkpoint', 'agent',
		'session', 'files', 'size_bytes'] {
		read_json_has(r.out, key)
	}
	assert r.out.contains('"message":"first message"')
	assert r.out.contains('"agent":"agent-7"')
	assert r.out.contains('"session":"session-a"')
	assert r.out.contains('"is_checkpoint":false')
	assert run(dir, 'save', 'again').code == 1
	assert run(dir, 'save', 'again').err.contains('nothing has changed since the last save')
	assert run(dir, 'save').code == 2
	assert run(dir, 'save', 'x', '--bogus').code == 2
	cp := run(dir, 'checkpoint', 'still first', '--json')
	assert cp.code == 0
	assert cp.out.contains('"is_checkpoint":true')
	assert cp.out.contains('"parent":')
	clean := run(dir, 'status')
	assert clean.code == 0
	assert clean.out.contains('no changes')
	write(dir, 'tracked.txt', 'two\n')
	write(dir, 'new.txt', 'new\n')
	os.rm(os.join_path(dir, 'tracked.txt')) or { panic(err) }
	write(dir, '.kohignore', '*.log\n')
	status := run(dir, 'status')
	assert status.code == 0
	assert status.out.contains('+')
	assert status.out.contains('~') || status.out.contains('-')
	assert status.out.contains('added')
	js := run(dir, 'status', '--json')
	assert js.code == 0
	for key in ['head_id', 'head_msg', 'dirty', 'modified', 'added', 'deleted', 'ignore_changed'] {
		read_json_has(js.out, key)
	}
	assert js.out.contains('"ignore_changed":true')
}

fn test_status_requires_repo_and_ignore_rules() {
	dir := temp_repo('status')
	defer { os.rmdir_all(dir) or {} }
	assert run(dir, 'status').code == 1
	init_repo(dir)
	write(dir, '.kohignore', '*.log\nbuild/\n')
	write(dir, 'keep.txt', 'keep\n')
	write(dir, 'ignored.log', 'ignore\n')
	write(dir, 'build/output.txt', 'ignore\n')
	assert run(dir, 'save', 'keep').code == 0
	assert run(dir, 'status').out.contains('no changes')
	assert !run(dir, 'status').out.contains('ignored.log')
	assert !run(dir, 'status').out.contains('build/output.txt')
	assert !run(dir, 'status').out.contains('LOG.md')
}

fn test_log_and_diff() {
	dir := temp_repo('history')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'a.txt', 'one\n')
	first := run(dir, 'save', 'one', '--session', 'alpha', '--json')
	id1 := first.out.all_after('"id":"')[..12]
	write(dir, 'a.txt', 'two\n')
	second := run(dir, 'save', 'two', '--session', 'beta', '--json')
	id2 := second.out.all_after('"id":"')[..12]
	cp := run(dir, 'checkpoint', 'checkpoint', '--json')
	assert cp.code == 0
	log := run(dir, 'log')
	assert log.code == 0
	assert log.out.contains('▸')
	assert log.out.contains('◐')
	lj := run(dir, 'log', '--json')
	assert lj.code == 0
	assert lj.out.contains('"head_id"')
	assert lj.out.contains('"saves"')
	assert run(dir, 'log', '--since', 'today').out.contains('two')
	session := run(dir, 'log', '--session', 'alpha')
	assert session.out.contains('one')
	assert !session.out.contains('two')
	write(dir, 'a.txt', 'three\n')
	d := run(dir, 'diff')
	assert d.code == 0
	assert d.out.contains('--- a/')
	assert d.out.contains('+++ b/')
	assert run(dir, 'diff', id1).code == 0
	assert run(dir, 'diff', id1, id2).code == 0
	dj := run(dir, 'diff', '--json')
	assert dj.code == 0
	assert dj.out.contains('"from"')
	assert dj.out.contains('"to"')
	assert dj.out.contains('"files"')
	assert dj.out.contains('"change":"modified"')
	assert run(dir, 'diff', 'not-a-real-id').code == 1
	assert run(dir, 'diff', id1, id2, 'third').code == 1
	write(dir, 'blob.bin', '\x00\x01\x02')
	diff_binary := run(dir, 'diff', '--json')
	assert diff_binary.out.contains('"binary":true')
	assert diff_binary.out.contains('Binary files differ')
}

fn test_undo_revert_and_load() {
	dir := temp_repo('restore')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'a.txt', 'one\n')
	one := run(dir, 'save', 'one', '--json').out
	id1 := one.all_after('"id":"')[..12]
	write(dir, 'a.txt', 'two\n')
	write(dir, 'gone.txt', 'gone\n')
	second := run(dir, 'save', 'two', '--json').out
	id2 := second.all_after('"id":"')[..12]
	write(dir, 'a.txt', 'three\n')
	third := run(dir, 'save', 'three', '--json').out
	id3 := third.all_after('"id":"')[..12]
	write(dir, 'never.txt', 'keep me\n')
	undo := run(dir, 'undo', '--yes', '--json')
	assert undo.code == 0
	if !undo.out.contains(id2) {
		panic('undo JSON did not name ${id2}: ${undo.out}')
	}
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'
	assert os.exists(os.join_path(dir, 'gone.txt'))
	assert os.exists(os.join_path(dir, 'never.txt'))
	assert run(dir, 'load', id3, '--yes').code == 0
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'
	assert run(dir, 'undo', '2', '--yes').code == 0
	assert run(dir, 'load', id2, '--yes').code == 0
	assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')
	write(dir, 'a.txt', 'dirty\n')
	before := run(dir, 'log')
	refuse := run(dir, 'revert')
	assert refuse.code == 1
	assert refuse.err.contains('needs --yes when not interactive')
	assert run(dir, 'log').out == before.out
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'
	write(dir, 'untracked.txt', 'remove\n')
	rv := run(dir, 'revert', '--yes')
	assert rv.code == 0
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'
	assert !os.exists(os.join_path(dir, 'untracked.txt'))
	assert run(dir, 'revert', '--yes').out.contains('working tree already matches')
	assert run(dir, 'load', '--checkpoint', '--yes').code == 0
	assert run(dir, 'load', id2, '--json').code == 1
	assert run(dir, 'load', id2, '--json').err.contains('error')
	assert run(dir, 'load').code == 2
	assert run(dir, 'load', 'unknown-id', '--yes').code == 1
}

fn test_flag_and_flagged_log() {
	dir := temp_repo('flag')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'flag.txt', 'one\n')
	saved := run(dir, 'save', 'flag me', '--json')
	id := saved.out.all_after('"id":"')[..12]
	flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')
	assert flagged.code == 0
	assert flagged.out.contains('"flagged":true')
	assert flagged.out.contains('"reason":"unsafe"')
	list := run(dir, 'flag', '--json')
	assert list.code == 0
	assert list.out.contains(id)
	assert list.out.contains('unsafe')
	assert run(dir, 'log').out.contains('⚑')
	assert run(dir, 'flag', '--clear', id).code == 0
	assert run(dir, 'flag').out.contains('nothing is flagged')
	assert run(dir, 'flag', 'unknown-id').code == 1
	assert run(dir, 'flag').code == 0
}

fn test_pluck_forward_reverse_dry_run_and_conflict() {
	dir := temp_repo('pluck')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'a.txt', 'one\n')
	first := run(dir, 'save', 'one', '--json').out
	id1 := first.all_after('"id":"')[..12]
	write(dir, 'a.txt', 'two\n')
	second := run(dir, 'save', 'two', '--json').out
	id2 := second.all_after('"id":"')[..12]
	assert run(dir, 'load', id1, '--yes').code == 0
	dry := run(dir, 'pluck', id2, '--dry-run')
	assert dry.code == 0
	assert dry.out.contains('a.txt')
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'
	assert run(dir, 'pluck', id2, '--yes').code == 0
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'
	assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0
	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'
	write(dir, 'a.txt', 'different\n')
	conflict := run(dir, 'pluck', id2, '--yes')
	assert conflict.code == 1
	assert conflict.err.contains('a.txt')
	assert run(dir, 'pluck').code == 2
	assert run(dir, 'pluck', 'unknown-id').code == 1
}

fn test_peek_and_doctor() {
	dir := temp_repo('peek-doctor')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'peek.txt', 'peek me\n')
	assert run(dir, 'save', 'peekable').code == 0
	id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]
	before_log := run(dir, 'log', '--json').out
	peek := run(dir, 'peek', id, '--json')
	assert peek.code == 0
	peek_path := peek.out.all_after('"path":"').before('"')
	assert peek.out.contains('"id":"${id}"')
	assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'
	assert run(dir, 'status').out.contains('no changes')
	assert run(dir, 'log', '--json').out == before_log
	assert run(dir, 'peek').code == 2
	assert run(dir, 'peek', 'unknown-id').code == 1
	backup := os.join_path(dir, 'doctor.backup')
	healthy := run(dir, 'doctor')
	assert healthy.code == 0
	assert healthy.out.contains('is sound')
	report := run(dir, 'doctor', '--json')
	assert report.code == 0
	for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',
		'backup'] {
		read_json_has(report.out, key)
	}
	backed := run(dir, 'doctor', '--backup', backup)
	assert backed.code == 0
	assert os.exists(backup)
	assert backed.out.contains(backup)
	assert run(dir, 'doctor', '--backup', backup).code == 1
	assert run(dir, 'peek', id, '--json').code == 0
	removed := run(dir, 'doctor', '--json')
	assert removed.code == 0
	assert removed.out.contains('"peek_folders_removed":1')
	assert !os.exists(peek_path)
}

fn test_strip_refuses_without_terminal() {
	dir := temp_repo('strip')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'secret.txt', 'keep\n')
	assert run(dir, 'save', 'secret').code == 0
	before_log := run(dir, 'log', '--json').out
	refused := run(dir, 'strip', 'secret.txt')
	assert refused.code == 1
	assert refused.err.contains('needs a terminal')
	assert run(dir, 'log', '--json').out == before_log
	assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'
	assert run(dir, 'log', '--json').out.contains('secret')
}

fn test_saves_and_checkpoints_aliases() {
	dir := temp_repo('aliases')
	defer { os.rmdir_all(dir) or {} }
	init_repo(dir)
	write(dir, 'alias.txt', 'one\n')
	assert run(dir, 'save', 'plain save').code == 0
	assert run(dir, 'checkpoint', 'checkpoint save').code == 0
	saves := run(dir, 'saves')
	assert saves.code == 0
	assert saves.out.contains('plain save')
	assert saves.out.contains('checkpoint save')
	checkpoints := run(dir, 'checkpoints')
	assert checkpoints.code == 0
	assert checkpoints.out.contains('checkpoint save')
	assert !checkpoints.out.contains('plain save')
	json_result := run(dir, 'checkpoints', '--json')
	assert json_result.code == 0
	assert json_result.out.contains('"head_id"')
	assert json_result.out.contains('"saves"')
	assert json_result.out.contains('"is_checkpoint":true')
	assert !json_result.out.contains('"is_checkpoint":false')
}