import os

const export_attr_vexe = @VEXE
const export_attr_tests_dir = os.dir(@FILE)
const export_attr_v3_dir = os.dir(export_attr_tests_dir)
const export_attr_vlib_dir = os.dir(export_attr_v3_dir)
const export_attr_v3_src = os.join_path(export_attr_v3_dir, 'v3.v')

fn export_attr_build_v3() string {
    v3_bin := os.join_path(os.temp_dir(), 'v3_export_attr_test_${os.getpid()}')
    os.rm(v3_bin) or {}
    build :=
        os.execute('${export_attr_vexe} -gc none -path "${export_attr_vlib_dir}|@vlib|@vmodules" -o ${v3_bin} ${export_attr_v3_src}')
    assert build.exit_code == 0, build.output
    return v3_bin
}

fn export_attr_project(name string, files map[string]string) string {
    root := os.join_path(os.temp_dir(), 'v3_export_attr_${name}_${os.getpid()}')
    os.rmdir_all(root) or {}
    mut rels := files.keys()
    rels.sort()
    for rel in rels {
        path := os.join_path(root, rel)
        os.mkdir_all(os.dir(path)) or { panic(err) }
        os.write_file(path, files[rel]) or { panic(err) }
    }
    return root
}

fn export_attr_compile(v3_bin string, main_file string, output string) os.Result {
    return os.execute('${v3_bin} ${main_file} -b c -o ${output}')
}

fn test_exported_imported_function_is_rooted_and_emitted_as_raw_symbol() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('imported_link', {
        'main.v':       'module main

import expmod

fn C.raw_exported_answer() int

fn take_callback(cb fn () int) int {
    return cb()
}

fn main() {
    println(C.raw_exported_answer().str())
    println(expmod.exported_answer().str())
    println(take_callback(expmod.exported_answer).str())
}
'
        'expmod/mod.v': "module expmod

@[export: 'raw_exported_answer']
pub fn exported_answer() int {
    return helper_used()
}

fn helper_used() int {
    return 41
}

fn helper_unused() int {
    return 99
}
"
    })
    bin_path := os.join_path(root, 'app')
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), bin_path)
    assert compile.exit_code == 0, compile.output
    run := os.execute(bin_path)
    assert run.exit_code == 0, run.output
    assert run.output.trim_space() == '41\n41\n41', run.output

    c_code := os.read_file(bin_path + '.c') or { panic(err) }
    assert c_code.contains('int expmod__exported_answer(void) {'), c_code
    assert c_code.contains('int raw_exported_answer(void) {'), c_code
    assert c_code.contains('return expmod__exported_answer();'), c_code
    assert c_code.contains('raw_exported_answer()'), c_code
    assert c_code.contains('take_callback(expmod__exported_answer)'), c_code
    assert c_code.contains('expmod__helper_used('), c_code
    assert !c_code.contains('expmod__helper_unused('), c_code
}

fn test_duplicate_export_name_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('duplicate', {
        'main.v': "module main

@[export: 'raw_duplicate']
fn one() {}

@[export: 'raw_duplicate']
fn two() {}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('duplicate export name `raw_duplicate`'), compile.output
}

fn test_disabled_export_attr_does_not_register_raw_symbol() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('disabled_export', {
        'main.v': "module main

@[if missing_export_flag ?]
@[export: 'raw_disabled_export']
fn disabled_export() int {
    return 2
}

@[export: 'raw_enabled_export']
fn enabled_export() int {
    return 7
}

fn main() {
    println(enabled_export().str())
}
"
    })
    bin_path := os.join_path(root, 'app')
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), bin_path)
    assert compile.exit_code == 0, compile.output
    run := os.execute(bin_path)
    assert run.exit_code == 0, run.output
    assert run.output.trim_space() == '7', run.output

    c_code := os.read_file(bin_path + '.c') or { panic(err) }
    assert c_code.contains('int raw_enabled_export(void) {'), c_code
    assert !c_code.contains('raw_disabled_export'), c_code
}

fn test_export_name_collision_with_runtime_symbol_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('runtime_collision', {
        'main.v': "module main

@[export: 'v_free']
fn collides_with_runtime_free() {}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('export name `v_free`'), compile.output
}

fn test_export_name_collision_with_natural_symbol_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('natural_collision', {
        'main.v': "module main

@[export: 'natural_name']
fn natural_name() int {
    return 1
}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('export name `natural_name`'), compile.output
}

fn test_export_name_collision_with_libc_remapped_natural_symbol_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('libc_remapped_natural_collision', {
        'main.v': "module main

fn rint() int {
    return 1
}

@[export: 'v_rint']
fn exported_rint_collision() int {
    return 2
}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('export name `v_rint`'), compile.output
    assert compile.output.contains('collides with `rint`'), compile.output
    assert !compile.output.contains('C compilation failed'), compile.output
    assert !compile.output.contains('redefinition'), compile.output
}

fn test_export_main_collision_with_top_level_script_entry_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('top_level_script_export_main_collision', {
        'main.v': "
@[export: 'main']
fn exported_entry() {}

println('script')
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('export name `main` for `exported_entry`'), compile.output
    assert compile.output.contains('synthetic entry point `main`'), compile.output
    assert !compile.output.contains('C compilation failed'), compile.output
    assert !compile.output.contains('redefinition'), compile.output
}

fn test_export_main_collision_with_top_level_match_entry_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('top_level_match_export_main_collision', {
        'main.v': "
@[export: 'main']
fn exported_entry() {}

match 1 {
    1 { println('one') }
    else { println('other') }
}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('export name `main` for `exported_entry`'), compile.output
    assert compile.output.contains('synthetic entry point `main`'), compile.output
    assert !compile.output.contains('C compilation failed'), compile.output
    assert !compile.output.contains('redefinition'), compile.output
}

fn test_export_main_collision_with_test_harness_entry_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('test_harness_export_main_collision', {
        'main_test.v': "
@[export: 'main']
fn exported_entry() {}

fn test_ok() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main_test.v'), os.join_path(root,
        'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('export name `main` for `exported_entry`'), compile.output
    assert compile.output.contains('synthetic entry point `main`'), compile.output
    assert !compile.output.contains('C compilation failed'), compile.output
    assert !compile.output.contains('redefinition'), compile.output
}

fn test_export_main_in_non_main_test_module_does_not_report_synthetic_collision() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('test_harness_non_main_module_export_main', {
        'foo_test.v': "module foo

@[export: 'main']
fn exported_entry() {}

fn test_ok() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'foo_test.v'), os.join_path(root,
        'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('only module main test files are supported'), compile.output
    assert !compile.output.contains('synthetic entry point `main`'), compile.output
    assert !compile.output.contains('C compilation failed'), compile.output
    assert !compile.output.contains('redefinition'), compile.output
}

fn test_invalid_export_names_are_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('invalid_names', {
        'main.v': "module main

@[export: '1bad']
fn bad_digit() {}

@[export: 'for']
fn bad_keyword() {}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('invalid export name `1bad`'), compile.output
    assert compile.output.contains('invalid export name `for`'), compile.output
}

fn test_invalid_imported_export_name_is_rejected_before_cgen() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('invalid_imported_name', {
        'main.v':          'module main

import badexp

fn main() {
    println(badexp.answer().str())
}
'
        'badexp/badexp.v': "module badexp

@[export: '1bad']
pub fn answer() int {
    return 1
}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0
    assert compile.output.contains('invalid export name `1bad` for `badexp.answer`'), compile.output
}

fn test_export_name_reserved_by_v3_c_preamble_is_rejected() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('preamble_reserved_name', {
        'main.v': "module main

@[export: 'bool']
fn collides_with_preamble_bool() bool {
    return true
}

@[export: 'string']
fn collides_with_preamble_string() {}

@[export: 'voidptr']
fn collides_with_preamble_voidptr() {}

@[export: 'i8']
fn collides_with_preamble_i8() {}

@[export: 'true']
fn collides_with_preamble_true() {}

@[export: 'Array']
fn collides_with_runtime_array() {}

@[export: 'map']
fn collides_with_runtime_map() int {
    values := map[string]int{
        'a': 1
    }
    return values['a'] or { 0 }
}

@[export: 'DenseArray']
fn collides_with_runtime_dense_array() {}

@[export: 'SortedMap']
fn collides_with_runtime_sorted_map() {}

@[export: 'Optional']
fn collides_with_runtime_optional() {}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('invalid export name `bool`'), compile.output
    assert compile.output.contains('invalid export name `string`'), compile.output
    assert compile.output.contains('invalid export name `voidptr`'), compile.output
    assert compile.output.contains('invalid export name `i8`'), compile.output
    assert compile.output.contains('invalid export name `true`'), compile.output
    assert compile.output.contains('invalid export name `Array`'), compile.output
    assert compile.output.contains('invalid export name `map`'), compile.output
    assert compile.output.contains('invalid export name `DenseArray`'), compile.output
    assert compile.output.contains('invalid export name `SortedMap`'), compile.output
    assert compile.output.contains('invalid export name `Optional`'), compile.output
}

fn test_generic_export_is_rejected_fail_closed() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('generic_export', {
        'main.v': "module main

@[export: 'raw_generic']
fn raw_generic[T](value T) T {
    return value
}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('generic function `raw_generic` cannot be exported'), compile.output
}

fn test_exported_function_must_name_params() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('unnamed_param', {
        'main.v': "module main

@[export: 'raw_unnamed']
fn unnamed(int) int {
    return 0
}

fn main() {}
"
    })
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), os.join_path(root, 'app'))
    assert compile.exit_code != 0, compile.output
    assert compile.output.contains('must name all parameters'), compile.output
}

fn test_export_wrapper_fixed_array_return_uses_return_wrapper_type() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('fixed_array_return', {
        'main.v': "module main

@[export: 'raw_numbers']
fn numbers() [3]int {
    mut values := [3]int{}
    values[0] = 1
    values[1] = 2
    values[2] = 3
    return values
}

fn main() {}
"
    })
    bin_path := os.join_path(root, 'app')
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), bin_path)
    assert compile.exit_code == 0, compile.output
    c_code := os.read_file(bin_path + '.c') or { panic(err) }
    assert c_code.contains('_v_ret_Array_fixed_int_3 numbers(void);'), c_code
    assert c_code.contains('_v_ret_Array_fixed_int_3 raw_numbers(void);'), c_code
    assert c_code.contains('_v_ret_Array_fixed_int_3 numbers(void) {'), c_code
    assert c_code.contains('_v_ret_Array_fixed_int_3 raw_numbers(void) {'), c_code
    assert c_code.contains('return numbers();'), c_code
    assert !c_code.contains('\nArray_fixed_int_3 raw_numbers(void)'), c_code
}

fn test_export_wrapper_fn_pointer_return_uses_fn_pointer_typedef() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('fn_pointer_return', {
        'main.v': "module main

type Callback = fn () int

fn callback_value() int {
    return 7
}

@[export: 'raw_callback']
fn callback() Callback {
    return callback_value
}

fn main() {}
"
    })
    bin_path := os.join_path(root, 'app')
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), bin_path)
    assert compile.exit_code == 0, compile.output
    c_code := os.read_file(bin_path + '.c') or { panic(err) }
    assert c_code.contains('typedef int (*_fn_ptr_'), c_code
    assert c_code.contains(' callback(void);'), c_code
    assert c_code.contains(' raw_callback(void);'), c_code
    assert c_code.contains(' raw_callback(void) {'), c_code
    assert c_code.contains('return callback();'), c_code
    assert !c_code.contains('fn_ptr:'), c_code
}

fn test_export_wrapper_for_veb_handler_forwards_implicit_ctx() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('veb_implicit_ctx_export_wrapper', {
        'main.v': "module main

import veb

pub struct Context {
    veb.Context
}

pub struct App {}

@[export: 'raw_show']
pub fn (app &App) show(id int) veb.Result {
    _ = id
    return veb.Result{}
}

fn main() {}
"
    })
    c_path := os.join_path(root, 'app.c')
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), c_path)
    assert compile.exit_code == 0, compile.output
    c_code := os.read_file(c_path) or { panic(err) }
    assert c_code.contains('veb__Result raw_show(App* app, Context* ctx, int id);'), c_code
    assert c_code.contains('veb__Result raw_show(App* app, Context* ctx, int id) {'), c_code
    assert c_code.contains('return App__show(app, ctx, id);'), c_code
}

fn test_export_wrapper_for_veb_handler_underscore_param_matches_implicit_ctx_position() {
    v3_bin := export_attr_build_v3()
    root := export_attr_project('veb_implicit_ctx_underscore_export_wrapper', {
        'main.v': "module main

import veb

pub struct Context {
    veb.Context
}

pub struct App {}

@[export: 'raw_show_underscore']
pub fn (app &App) show(_ int) veb.Result {
    return veb.Result{}
}

fn main() {}
"
    })
    c_path := os.join_path(root, 'app.c')
    compile := export_attr_compile(v3_bin, os.join_path(root, 'main.v'), c_path)
    assert compile.exit_code == 0, compile.output
    c_code := os.read_file(c_path) or { panic(err) }
    assert c_code.contains('veb__Result raw_show_underscore(App* app, Context* ctx, int _2);'), c_code
    assert c_code.contains('veb__Result raw_show_underscore(App* app, Context* ctx, int _2) {'), c_code
    assert c_code.contains('return App__show(app, ctx, _2);'), c_code
    assert !c_code.contains('return App__show(app, ctx, _1);'), c_code
}