From d1b1fceb32118528cff3854fa84d5d0fdd12eccc Mon Sep 17 00:00:00 2001
From: Robert Bragg <robert@sixbynine.org>
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..2baa645 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(ffmpeg_dir) = env[key].as_str() {
+                    // Simply re-export as an actual environment variable for consistent handling below...
+                    env::set_var("FFMPEG_DIR", ffmpeg_dir);
+                }
+            }
+        }
+    }
+
     let include_paths: Vec<PathBuf> = if env::var("CARGO_FEATURE_BUILD").is_ok() {
         println!(
             "cargo:rustc-link-search=native={}",