From 5fbbdc458236a17e1e1bbe676fe243fe15835c8d Mon Sep 17 00:00:00 2001 From: Robert Bragg Date: Sat, 7 Nov 2020 01:01:03 +0000 Subject: [PATCH] Check for FFMPEG_DIR in environment.json It's often impractical to use an environment variable to specify FFMPEG_DIR (such as when builds are run via rust-analyzer/RLE for editor diagnostics during development) so this enables support for parsing an environment.json file like: { "FFMPEG_DIR": "/path/to/ffmpeg-build/" } Addresses: rust-lang/cargo#4121 --- Cargo.toml | 1 + build.rs | 22 +++++++++++++++++++++- 2 files changed, 22 insertions(+), 1 deletion(-) diff --git a/Cargo.toml b/Cargo.toml index 638ca1e..1ec50f6 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -23,6 +23,7 @@ num_cpus = "1.11" cc = "1.0" pkg-config = "0.3" bindgen = { version = "0.54", default-features = false, features = ["runtime"] } +serde_json = "1.0" [target.'cfg(target_env = "msvc")'.build-dependencies] vcpkg = "0.2" diff --git a/build.rs b/build.rs index 9a560e8..a7a0e1c 100644 --- a/build.rs +++ b/build.rs @@ -2,11 +2,12 @@ extern crate bindgen; extern crate cc; extern crate num_cpus; extern crate pkg_config; +extern crate serde_json; use std::env; use std::fs::{self, File}; use std::io::{self, BufRead, BufReader, Write}; -use std::path::PathBuf; +use std::path::{Path, PathBuf}; use std::process::Command; use std::str; @@ -14,6 +15,8 @@ use bindgen::callbacks::{ EnumVariantCustomBehavior, EnumVariantValue, IntKind, MacroParsingBehavior, ParseCallbacks, }; +use serde_json::Value; + #[derive(Debug)] struct Library { name: &'static str, @@ -629,6 +632,23 @@ fn link_to_libraries(statik: bool) { fn main() { let statik = env::var("CARGO_FEATURE_STATIC").is_ok(); + // It's often impractical to use an environment variable to specify + // FFMPEG_DIR so we also support parsing an environment.json file + // Ref: rust-lang/cargo#4121 + let manifest_dir_env = env::var("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR evironment variable unset"); + let env_path = Path::new(&manifest_dir_env).join("environment.json"); + if let Ok(env) = fs::read_to_string(env_path) { + let env: Value = serde_json::from_str(&env).expect("Failed to parse environment.json"); + if let Some(env_map) = env.as_object() { + for key in env_map.keys() { + if let Some(env_value) = env[key].as_str() { + // Simply re-export as an actual environment variable for consistent handling below... + env::set_var(key, env_value); + } + } + } + } + let include_paths: Vec = if env::var("CARGO_FEATURE_BUILD").is_ok() { println!( "cargo:rustc-link-search=native={}",