import os

const wait_header_vexe = @VEXE
const wait_header_tests_dir = os.dir(@FILE)
const wait_header_v3_dir = os.dir(wait_header_tests_dir)
const wait_header_vlib_dir = os.dir(wait_header_v3_dir)
const wait_header_v3_src = os.join_path(wait_header_v3_dir, 'v3.v')

struct WaitHeaderProgram {
	c_code string
	out    string
}

fn wait_header_build_v3() string {
	pid := os.getpid()
	v3_bin := os.join_path(os.temp_dir(), 'v3_wait_header_test_${pid}')
	os.rm(v3_bin) or {}
	build :=
		os.execute('${wait_header_vexe} -gc none -path "${wait_header_vlib_dir}|@vlib|@vmodules" -o ${v3_bin} ${wait_header_v3_src}')
	assert build.exit_code == 0, build.output
	return v3_bin
}

fn wait_header_compile(v3_bin string, name string, source string) WaitHeaderProgram {
	pid := os.getpid()
	src := os.join_path(os.temp_dir(), 'v3_wait_header_${name}_${pid}.v')
	out := os.join_path(os.temp_dir(), 'v3_wait_header_${name}_${pid}')
	os.write_file(src, source) or { panic(err) }
	os.rm(out) or {}
	os.rm(out + '.c') or {}
	compile := os.execute('${v3_bin} ${src} -b c -o ${out}')
	assert compile.exit_code == 0, compile.output
	return WaitHeaderProgram{
		c_code: os.read_file(out + '.c') or { panic(err) }
		out:    out
	}
}

fn test_os_import_emits_sys_wait_header() {
	$if windows {
		return
	}
	v3_bin := wait_header_build_v3()
	with_os := wait_header_compile(v3_bin, 'with_os_execute', 'module main

import os

fn main() {
	result := os.execute(\'/bin/sh -c "printf waitpid-ok"\')
	assert result.exit_code == 0
	println(result.output)
}
')
	assert with_os.c_code.contains('#include <sys/wait.h>'), with_os.c_code
	assert with_os.c_code.contains('waitpid('), with_os.c_code
	run := os.execute(with_os.out)
	assert run.exit_code == 0, run.output
	assert run.output.trim_space() == 'waitpid-ok', run.output

	hello := wait_header_compile(v3_bin, 'hello', "module main

fn main() {
	println('hello')
}
")
	assert !hello.c_code.contains('#include <sys/wait.h>'), hello.c_code
}