import os
import rand

const vexe = @VEXE

fn test_shared_library_links_with_system_cc() {
    if os.user_os() != 'linux' {
        return
    }
    workdir := os.join_path(os.vtmp_dir(), 'v_shared_link_${rand.ulid()}')
    os.mkdir_all(workdir) or { panic(err) }
    defer {
        os.rmdir_all(workdir) or {}
    }
    lib_src := os.join_path(workdir, 'libfoo.v')
    lib_out := os.join_path(workdir, 'libfoo')
    lib_so := '${lib_out}.so'
    host_src := os.join_path(workdir, 'host.c')
    host_bin := os.join_path(workdir, 'host')
    os.write_file(lib_src, [
        'module libfoo',
        '',
        "@[export: 'libfoo_square']",
        'pub fn square(x int) int {',
        '\treturn x * x',
        '}',
    ].join('\n')) or { panic(err) }
    os.write_file(host_src, [
        '#include <stdio.h>',
        '',
        'int libfoo_square(int);',
        '',
        'int main(void) {',
        '\tprintf("%d\\n", libfoo_square(2));',
        '\treturn 0;',
        '}',
    ].join('\n')) or { panic(err) }
    run_cmd('${os.quoted_path(vexe)} -shared -o ${os.quoted_path(lib_out)} ${os.quoted_path(lib_src)}') or {
        panic(err)
    }
    assert os.exists(lib_so)
    run_cmd('cc ${os.quoted_path(host_src)} -L${os.quoted_path(workdir)} -lfoo -o ${os.quoted_path(host_bin)}') or {
        panic(err)
    }
    res := run_cmd('LD_LIBRARY_PATH=${os.quoted_path(workdir)} ${os.quoted_path(host_bin)}') or {
        panic(err)
    }
    assert res.output.trim_space() == '4'
}

fn test_shared_library_only_exports_tagged_symbols() {
    // Regression test for vlang/v#27167: `v -shared` used to export every
    // V runtime/stdlib symbol, not just functions tagged with `@[export: '…']`.
    uos := os.user_os()
    if uos !in ['linux', 'macos'] {
        return
    }
    // tcc does not honor `-fvisibility=hidden`, so the symbol-set check would be
    // noisy there; use a system compiler for Linux and clang on macOS.
    cc_flag := if uos == 'macos' { '-cc clang' } else { '-cc cc' }
    workdir := os.join_path(os.vtmp_dir(), 'v_shared_visibility_${rand.ulid()}')
    os.mkdir_all(workdir) or { panic(err) }
    defer {
        os.rmdir_all(workdir) or {}
    }
    lib_src := os.join_path(workdir, 'lib.v')
    lib_out := os.join_path(workdir, 'liblib')
    lib_so := if uos == 'macos' { '${lib_out}.dylib' } else { '${lib_out}.so' }
    os.write_file(lib_src, [
        'module main',
        '',
        "@[export: 'my_add']",
        'fn my_add(a int, b int) int {',
        '\treturn a + b',
        '}',
    ].join('\n')) or { panic(err) }
    run_cmd('${os.quoted_path(vexe)} ${cc_flag} -shared -o ${os.quoted_path(lib_out)} ${os.quoted_path(lib_src)}') or {
        panic(err)
    }
    assert os.exists(lib_so)
    nm_flag := if uos == 'macos' { '-gU' } else { '-D --defined-only' }
    res := run_cmd('nm ${nm_flag} ${os.quoted_path(lib_so)}') or { panic(err) }
    symbol_token := if uos == 'macos' { ' _my_add' } else { ' my_add' }
    assert res.output.contains(symbol_token), 'expected exported `my_add` in:\n${res.output}'
    // None of the V runtime/stdlib helpers should leak into the exported ABI.
    leaky_prefixes := ['Array_string_', 'main__', 'builtin__', 'GC_', '_vinit', '_vcleanup',
        '_const_']
    for line in res.output.split('\n') {
        for prefix in leaky_prefixes {
            needles := if uos == 'macos' { [' _${prefix}', ' T _${prefix}', ' D _${prefix}'] } else { [
                    ' ${prefix}',
                    ' T ${prefix}',
                    ' D ${prefix}',
                ] }
            for needle in needles {
                assert !line.contains(needle), 'unexpected exported symbol on line: ${line}'
            }
        }
    }
}

fn run_cmd(cmd string) !os.Result {
    res := os.execute(cmd)
    if res.exit_code != 0 {
        return error('command failed:\n${cmd}\n${res.output}')
    }
    return res
}