import os
import rand
import test_utils

const vexe = os.quoted_path(@VEXE)
const test_path = os.join_path(os.vtmp_dir(), 'vpm_link_test_${rand.ulid()}')

fn testsuite_begin() {
    test_utils.set_test_env(test_path)
    os.mkdir_all(test_path) or {}
}

fn testsuite_end() {
    os.rmdir_all(test_path) or {}
}

fn execute_in_dir(dir string, cmd string) os.Result {
    old_dir := os.getwd()
    os.chdir(dir) or { return os.Result{
        exit_code: -1
        output:    'failed to chdir: ${err}'
    } }
    defer {
        os.chdir(old_dir) or {}
    }
    return os.execute(cmd)
}

fn test_link_and_unlink_current_project() {
    module_name := 'author.coollib'
    project_path := os.join_path(test_path, 'project')
    write_vmod(project_path, module_name) or {
        assert false, err.msg()
        return
    }
    project_subdir := os.join_path(project_path, 'src')
    os.mkdir_all(project_subdir) or {
        assert false, err.msg()
        return
    }
    link_path := os.join_path(test_path, 'author', 'coollib')

    link_res := execute_in_dir(project_subdir, '${vexe} link')
    if link_res.exit_code != 0 && is_symlink_privilege_error(link_res.output) {
        eprintln('Skipping symlink test due to missing privileges.')
        return
    }
    assert link_res.exit_code == 0, link_res.output
    assert os.is_link(link_path), 'expected `${link_path}` to be a symlink'
    assert os.real_path(link_path) == os.real_path(project_path)
    assert link_res.output.contains('Linked `${module_name}`'), link_res.output

    link_again_res := execute_in_dir(project_path, '${vexe} link')
    assert link_again_res.exit_code == 0, link_again_res.output
    assert link_again_res.output.contains('already linked')
        || link_again_res.output.contains('already available'), link_again_res.output

    unlink_res := execute_in_dir(project_subdir, '${vexe} unlink')
    assert unlink_res.exit_code == 0, unlink_res.output
    assert !os.exists(link_path) && !os.is_link(link_path)
    assert !os.exists(os.join_path(test_path, 'author'))
    assert unlink_res.output.contains('Unlinked `${module_name}`'), unlink_res.output
}

fn test_link_without_vmod() {
    path := os.join_path(test_path, 'no_manifest')
    os.mkdir_all(path) or {
        assert false, err.msg()
        return
    }
    res := execute_in_dir(path, '${vexe} link')
    assert res.exit_code == 1, res.output
    assert res.output.contains('no `v.mod` file found'), res.output
}

fn write_vmod(path string, module_name string) ! {
    os.mkdir_all(path)!
    vmod_path := os.join_path(path, 'v.mod')
    vmod_contents := "Module {\n\tname: '${module_name}'\n\tdescription: ''\n\tversion: '0.0.0'\n\tlicense: 'MIT'\n\tdependencies: []\n}\n"
    os.write_file(vmod_path, vmod_contents)!
}

fn is_symlink_privilege_error(output string) bool {
    lower := output.to_lower()
    return lower.contains('required privilege is not held') || lower.contains('symbolic link')
}