Os based lib import path string

hannobraun · Dec 17, 2021 · 6f8b538 · 6f8b538
1 parent aab78ec
commit 6f8b538
Showing 1 changed file with 21 additions and 5 deletions.
diff --git a/src/model.rs b/src/model.rs
@@ -39,11 +39,27 @@ impl Model {
         // TASK: Read up why those calls are unsafe. Make sure calling them is
         //       sound, and document why that is.
         let shape = unsafe {
-            let lib = libloading::Library::new(format!(
-                "{}/target/debug/lib{}.so",
-                self.path(),
-                self.name(),
-            ))?;
+            let lib = libloading::Library::new(
+                if cfg!(windows) {
+                    format!(
+                        "{}/target/debug/{}.dll",
+                        self.path(),
+                        self.name(),
+                    )
+                } else if cfg!(target_os = "macos") {
+                    format!(
+                        "{}/target/debug/lib{}.dylib",
+                        self.path(),
+                        self.name(),
+                    )
+                } else { //Unix
+                    format!(
+                        "{}/target/debug/lib{}.so",
+                        self.path(),
+                        self.name(),
+                    )
+                }
+            )?;
             let model: libloading::Symbol<ModelFn> = lib.get(b"model")?;
             model(&arguments)
         };