build-probe-mpi 0.1.4

#![deny(missing_docs)]
#![warn(missing_copy_implementations)]
#![warn(trivial_casts)]
#![warn(trivial_numeric_casts)]
#![warn(unused_extern_crates)]
#![warn(unused_import_braces)]
#![warn(unused_qualifications)]

use super::super::Library;
use std::{env, error::Error, fmt, path::PathBuf};

#[derive(Debug)]
struct VarError {
    key: String,
    origin: env::VarError,
}

impl fmt::Display for VarError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}: {}", self.key, self.origin)
    }
}

impl Error for VarError {
    fn source(&self) -> Option<&(dyn Error + 'static)> {
        Some(&self.origin)
    }
}

fn var(key: &str) -> Result<String, VarError> {
    env::var(key).map_err(|e| VarError {
        key: key.to_owned(),
        origin: e,
    })
}

/// Probe the environment for MS-MPI
pub fn probe() -> Result<Library, Vec<Box<dyn Error>>> {
    let mut errs: Vec<Box<dyn Error>> = vec![];

    let include_path = match var("MSMPI_INC") {
        Ok(include_path) => Some(include_path),
        Err(err) => {
            errs.push(Box::new(err));
            None
        }
    };

    let lib_env = if env::var("CARGO_CFG_TARGET_ARCH") == Ok("x86".to_string()) {
        "MSMPI_LIB32"
    } else if env::var("CARGO_CFG_TARGET_ARCH") == Ok("x86_64".to_string()) {
        "MSMPI_LIB64"
    } else {
        panic!("rsmpi does not support your windows architecture!");
    };

    let lib_path = match var(lib_env) {
        Ok(lib_path) => Some(lib_path),
        Err(err) => {
            errs.push(Box::new(err));
            None
        }
    };

    if errs.len() > 0 {
        return Err(errs);
    }

    Ok(Library {
        mpicc: None,
        libs: vec!["msmpi".to_owned()],
        lib_paths: vec![lib_path.map(PathBuf::from).unwrap()],
        include_paths: vec![include_path.map(PathBuf::from).unwrap()],
        version: String::from("MS-MPI"),
        _priv: (),
    })
}