Merge pull request #8 from shigedangao/fix/documentation

Fix documentation generation due to missing headers

Author: shigedangao

.cargo/config.toml

@@ -3,3 +3,5 @@ rustflags = ["-C", "link-arg=-Wl,-rpath,@executable_path/../lib"]
 
 [env]
 INSIGHTFACE_PATH = "/Users/marcintha/workspace/baklava/insightface/cpp-package/inspireface/build/inspireface-macos/InspireFace"
+# Enable the setting below to generate documentation locally
+# DOCS_RS = "true"

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 name = "baklava"
 description = "A rust wrapper around the InsightFace library to perform face comparison"
-version = "0.1.0"
+version = "0.1.1"
 edition = "2021"
 authors = ["Marc Intha-amnouay <marc.inthaamnouay@gmail.com>"]
 keywords = ["insightface", "face-comparison", "inspireface"]

Cargo.toml

@@ -1,17 +1,31 @@
 use std::env;
 use std::path::PathBuf;
+use std::process::Command;
 
 fn main() {
     let lib_path = match env::var("INSIGHTFACE_PATH") {
         Ok(path) => PathBuf::from(path),
         Err(_) => {
             let current_dir = env::current_dir().unwrap();
-            let local_path = format!(
-                "insightface/cpp-package/inspireface/build/inspireface-{}/InspireFace",
-                env::consts::OS
-            );
+            match env::var("DOCS_RS").is_ok() {
+                true => {
+                    // The headers can be passed from the "zipped" file "insightface_headers.zip"
+                    Command::new("unzip")
+                        .arg("insightface_headers.zip")
+                        .spawn()
+                        .expect("Expect unzip command to work");
 
-            current_dir.join(local_path)
+                    current_dir
+                }
+                false => {
+                    let local_path = format!(
+                        "insightface/cpp-package/inspireface/build/inspireface-{}/InspireFace",
+                        env::consts::OS
+                    );
+
+                    current_dir.join(local_path)
+                }
+            }
         }
     };