import os

@[markused]
const turn_off_vcolors = os.setenv('VCOLORS', 'never', true)

const vexe = @VEXE

fn temp_output_path(name string) string {
    $if windows {
        return os.join_path(os.vtmp_dir(), '${name}.exe')
    }
    return os.join_path(os.vtmp_dir(), name)
}

fn project_path() string {
    return os.join_path(os.real_path(os.vtmp_dir()), 'local_module_submodules')
}

fn issue_24649_project_path() string {
    return os.join_path(os.real_path(os.vtmp_dir()), 'issue_24649_local_module_submodules')
}

fn write_local_module_submodules_project() {
    basepath := project_path()
    os.rmdir_all(basepath) or {}
    os.mkdir_all(os.join_path(basepath, 'confy', 'marshalers')) or { panic(err) }
    os.write_file(os.join_path(basepath, 'main.v'),
        ['module main', '', 'import confy', 'import confy.marshalers', '', 'fn main() {', '\tmut marshaler := marshalers.JSON{}', '\tload(mut marshaler)', '}', '', 'fn load(mut marshaler confy.Marshaler) {', '\tmarshaler.load()', '}'].join('\n') +
        '\n') or { panic(err) }
    os.write_file(os.join_path(basepath, 'confy', 'confy.v'),
        ['module confy', '', 'pub interface Marshaler {', '\tmut:', '\t\tload()', '}'].join('\n') +
        '\n') or { panic(err) }
    os.write_file(os.join_path(basepath, 'confy', 'marshalers', 'json.v'),
        ['module marshalers', '', 'pub struct JSON {}', '', 'pub fn (mut j JSON) load() {}'].join('\n') +
        '\n') or { panic(err) }
}

fn write_issue_24649_project() {
    basepath := issue_24649_project_path()
    os.rmdir_all(basepath) or {}
    os.mkdir_all(os.join_path(basepath, 'my_math', 'my_integer')) or { panic(err) }
    os.write_file(os.join_path(basepath, 'running_my_math.v'),
        ['module main', '', 'import my_math', 'import my_math.my_integer', '', 'fn main() {', '\ta := 34.5', '\tb := 13.2', '', "\tprintln('La suma de a y b es \${my_math.addition(a, b)}')", '', "\tprintln('La suma entera de a y b es \${my_integer.addition(a, b)}')", '}'].join('\n') +
        '\n') or { panic(err) }
    os.write_file(os.join_path(basepath, 'my_math', 'floating.v'),
        ['module my_math', '', 'pub fn addition(x f64, y f64) f64 {', '\treturn x +
        y', '}'].join('\n') +
        '\n') or { panic(err) }
    os.write_file(os.join_path(basepath, 'my_math', 'my_integer', 'integer.v'),
        ['module my_integer', '', 'pub fn addition(x f64, y f64) int {', '\treturn int(x +
        y)', '}'].join('\n') +
        '\n') or { panic(err) }
}

fn compile_local_module_submodules(target string, out_name string) os.Result {
    basepath := project_path()
    write_local_module_submodules_project()
    old_wd := os.getwd()
    os.chdir(basepath) or { panic(err) }
    defer {
        os.chdir(old_wd) or {}
        os.rmdir_all(basepath) or {}
    }
    return os.execute('${os.quoted_path(vexe)} -o ${os.quoted_path(out_name)} ${os.quoted_path(target)}')
}

fn compile_issue_24649_project(target string, out_name string, cwd string) os.Result {
    basepath := issue_24649_project_path()
    write_issue_24649_project()
    old_wd := os.getwd()
    os.chdir(cwd) or { panic(err) }
    defer {
        os.chdir(old_wd) or {}
        os.rmdir_all(basepath) or {}
    }
    return os.execute('${os.quoted_path(vexe)} -o ${os.quoted_path(out_name)} ${os.quoted_path(target)}')
}

fn test_local_module_submodules_compile_from_main_file() {
    out_name := temp_output_path('local_module_submodules_main')
    os.rm(out_name) or {}
    defer {
        os.rm(out_name) or {}
    }
    res := compile_local_module_submodules('main.v', out_name)
    assert res.exit_code == 0, res.output
}

fn test_local_module_submodules_compile_from_project_dir() {
    out_name := temp_output_path('local_module_submodules_dir')
    os.rm(out_name) or {}
    defer {
        os.rm(out_name) or {}
    }
    res := compile_local_module_submodules('.', out_name)
    assert res.exit_code == 0, res.output
}

fn test_issue_24649_submodules_compile_from_main_file() {
    basepath := issue_24649_project_path()
    out_name := temp_output_path('issue_24649_local_module_submodules_main')
    os.rm(out_name) or {}
    defer {
        os.rm(out_name) or {}
    }
    res := compile_issue_24649_project('running_my_math.v', out_name, basepath)
    assert res.exit_code == 0, res.output
}

fn test_issue_24649_submodules_compile_from_dot_slash_main_file() {
    basepath := issue_24649_project_path()
    out_name := temp_output_path('issue_24649_local_module_submodules_dot_main')
    os.rm(out_name) or {}
    defer {
        os.rm(out_name) or {}
    }
    res := compile_issue_24649_project('./running_my_math.v', out_name, basepath)
    assert res.exit_code == 0, res.output
}

fn test_issue_24649_submodules_compile_from_project_dir() {
    basepath := issue_24649_project_path()
    out_name := temp_output_path('issue_24649_local_module_submodules_dir')
    os.rm(out_name) or {}
    defer {
        os.rm(out_name) or {}
    }
    res := compile_issue_24649_project('.', out_name, basepath)
    assert res.exit_code == 0, res.output
}

fn test_issue_24649_submodules_compile_from_absolute_main_file() {
    basepath := issue_24649_project_path()
    out_name := temp_output_path('issue_24649_local_module_submodules_abs_main')
    os.rm(out_name) or {}
    defer {
        os.rm(out_name) or {}
    }
    target := os.join_path(basepath, 'running_my_math.v')
    res := compile_issue_24649_project(target, out_name, os.dir(basepath))
    assert res.exit_code == 0, res.output
}