// vtest build: !sanitized_job?
import os
import term
import benchmark
import v.ast
import v.fmt
import v.parser
import v.pref
import v.util.diff
import v.util.vtest

const vroot = @VEXEROOT
const tdir = os.join_path(vroot, 'vlib/v/fmt')
const fpref = &pref.Preferences{
	is_fmt: true
}

fn run_fmt(mut input_files []string) {
	fmt_message := 'vfmt tests'
	eprintln(term.header(fmt_message, '-'))
	tmpfolder := os.temp_dir()
	assert input_files.len > 0
	input_files = vtest.filter_vtest_only(input_files)
	if input_files.len == 0 {
		// No need to produce a failing test here.
		eprintln('no tests found with VTEST_ONLY filter set to: ' + os.getenv('VTEST_ONLY'))
		exit(0)
	}
	mut fmt_bench := benchmark.new_benchmark()
	fmt_bench.set_total_expected_steps(input_files.len)
	for istep, ipath in input_files {
		fmt_bench.cstep = istep
		fmt_bench.step()
		opath := ipath.replace('_input.vv', '_expected.vv')
		if !os.exists(opath) {
			fmt_bench.fail()
			eprintln(fmt_bench.step_message_fail('missing file ${opath}'))
			continue
		}
		expected_ocontent := os.read_file(opath) or {
			fmt_bench.fail()
			eprintln(fmt_bench.step_message_fail('cannot read from ${opath}'))
			continue
		}
		mut table := ast.new_table()
		file_ast := parser.parse_file(ipath, mut table, .parse_comments, fpref)
		result_ocontent := fmt.fmt(file_ast, mut table, fpref, false)
		if expected_ocontent != result_ocontent {
			fmt_bench.fail()
			eprintln(fmt_bench.step_message_fail('file ${ipath} after formatting, does not look as expected.'))
			vfmt_result_file := os.join_path(tmpfolder, 'vfmt_run_over_${os.file_name(ipath)}')
			os.write_file(vfmt_result_file, result_ocontent) or { panic(err) }
			println(diff.compare_files(opath, vfmt_result_file) or { err.msg() })
			continue
		}
		fmt_bench.ok()
		eprintln(fmt_bench.step_message_ok(ipath))
	}
	fmt_bench.stop()
	eprintln(term.h_divider('-'))
	eprintln(fmt_bench.total_message(fmt_message))
	assert fmt_bench.nfail == 0
}

fn test_fmt() {
	mut input_files := os.walk_ext(os.join_path(tdir, 'tests'), '_input.vv')
	run_fmt(mut input_files)

	input_files = os.walk_ext(os.join_path(tdir, 'testdata', 'vmodules'), '_input.vv')
	run_fmt(mut input_files)
}

fn test_fmt_vmodules() {
	mut vmodules_tdir := os.join_path(tdir, 'tests')
	os.setenv('VMODULES', vmodules_tdir, true)
	mut input_files := os.walk_ext(vmodules_tdir, '_input.vv')
	run_fmt(mut input_files)

	vmodules_tdir = os.join_path(tdir, 'testdata', 'vmodules')
	os.setenv('VMODULES', vmodules_tdir, true)
	input_files = os.walk_ext(vmodules_tdir, '_input.vv')
	run_fmt(mut input_files)
}