vsag 0.1.0

VSAG Rust binding
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

// Copyright 2023 Greptime Team
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

fn main() {
    println!("cargo:rerun-if-env-changed=VSAG_LIB_PATH");
    println!("cargo:rerun-if-changed=build.rs");
    println!("cargo:rustc-link-lib=dylib=vsag");

    if let Some(lib_path) = vsag_lib_path() {
        println!("cargo:rustc-link-search=native={lib_path}",);
    }
}

/// `some-feature` becomes `SOME_FEATURE` options in cmake.
#[cfg(feature = "vendored")]
macro_rules! define_config_based_on_features {
    ($config:ident, $($feature:expr),*) => {
        $(
            $config.define(
                $feature.to_uppercase().replace("-", "_"),
                if cfg!(feature = $feature) {
                    "ON"
                } else {
                    "OFF"
                },
            );
        )*
    };
}

fn vsag_lib_path() -> Option<String> {
    #[cfg(feature = "vendored")]
    {
        let mut config = cmake::Config::new("vsag-sys");

        // Cargo sets TARGET to the target triple
        // but building openblas via cmake will fail if it's set
        // ```plaintext
        // The TARGET specified on the command line or in Makefile.rule is not supported. Please choose a target from TargetList.txt
        // ```
        config.env("TARGET", "");

        define_config_based_on_features!(
            config,
            "enable-intel-mkl",
            "enable-libcxx",
            "enable-cxx11-abi"
        );

        let dst = config.build();

        // centos use `lib64`, ubuntu use `lib` convention.
        for path in ["lib64", "lib"] {
            let lib = dst.join(path);
            if lib.join("libvsag.so").exists() {
                return Some(lib.display().to_string());
            }
        }
    }

    std::env::var("VSAG_LIB_PATH").map_or(None, |v| Some(v.to_string()))
}