flux_bin/
utils.rs

1use std::{
2    env,
3    ffi::OsString,
4    path::{Path, PathBuf},
5    process::Command,
6};
7
8use anyhow::{Result, anyhow};
9use serde::Deserialize;
10
11#[cfg(target_os = "windows")]
12pub const LIB_PATH: &str = "PATH";
13#[cfg(any(target_os = "linux", target_os = "freebsd"))]
14pub const LIB_PATH: &str = "LD_LIBRARY_PATH";
15#[cfg(target_os = "macos")]
16pub const LIB_PATH: &str = "DYLD_FALLBACK_LIBRARY_PATH";
17
18pub const EXIT_ERR: i32 = -1;
19
20pub const FLUX_SYSROOT: &str = "FLUX_SYSROOT";
21
22const FLUX_DRIVER: &str = "FLUX_DRIVER";
23
24/// The path of the flux sysroot lib containing precompiled libraries and the flux driver.
25pub fn flux_sysroot_dir() -> PathBuf {
26    env::var_os(FLUX_SYSROOT).map_or_else(default_flux_sysroot_dir, PathBuf::from)
27}
28
29#[derive(Deserialize)]
30pub struct ToolchainToml {
31    toolchain: ToolchainSpec,
32}
33
34#[derive(Deserialize)]
35pub struct ToolchainSpec {
36    channel: String,
37}
38
39/// Return the default sysroot
40fn default_flux_sysroot_dir() -> PathBuf {
41    home::home_dir()
42        .expect("Couldn't find home directory")
43        .join(".flux")
44}
45
46pub fn get_flux_driver_path(sysroot: &Path) -> Result<PathBuf> {
47    let path = if let Some(path) = env::var_os(FLUX_DRIVER) {
48        PathBuf::from(path)
49    } else {
50        let mut path = sysroot.join("flux-driver");
51        if cfg!(target_os = "windows") {
52            path.set_extension("exe");
53        }
54        path
55    };
56    if !path.is_file() {
57        return Err(anyhow!("path to flux-driver {:?} does not exist or is not a file", path));
58    }
59    Ok(path)
60}
61
62pub fn get_rust_toolchain() -> Result<String> {
63    let toolchain_str = include_str!("../../../rust-toolchain.toml");
64    let toolchain_file: ToolchainToml = toml::from_str(toolchain_str)?;
65    Ok(toolchain_file.toolchain.channel)
66}
67
68pub fn get_rust_sysroot(toolchain: &str) -> Result<PathBuf> {
69    let out = Command::new("rustc")
70        .arg(format!("+{toolchain}"))
71        .args(["--print", "sysroot"])
72        .output()?;
73    bytes_to_pathbuf(out.stdout)
74}
75
76/// Path from where to load the rustc-driver library from
77pub fn get_rust_lib_path(sysroot: &Path) -> PathBuf {
78    if cfg!(windows) { sysroot.join("bin") } else { sysroot.join("lib") }
79}
80
81/// Prepends the path so that it's the first checked.
82pub fn prepend_path_to_env_var(var_name: &str, new_path: PathBuf) -> Result<OsString> {
83    let mut paths = match env::var(var_name) {
84        Ok(paths) => env::split_paths(&paths).collect(),
85        Err(env::VarError::NotPresent) => vec![],
86        Err(e) => Err(e)?,
87    };
88    paths.insert(0, new_path);
89    env::join_paths(paths).map_err(anyhow::Error::from)
90}
91
92fn bytes_to_pathbuf(input: Vec<u8>) -> Result<PathBuf> {
93    Ok(PathBuf::from(String::from_utf8(input)?.trim()))
94}