Support windows

Author: termoshtt

intel-mkl-tool/Cargo.toml

@@ -15,12 +15,12 @@ pkg-config = "0.3.17"
 tar = "0.4.26"
 xz2 = "0.1.6"
 log = "0.4.8"
-dirs = "*"
-glob = "*"
-zstd = "*"
+dirs = "2.0.2"
+glob = "0.3.0"
+zstd = "0.5.1"
 
 # CLI
-structopt  = { version = "0.3.4", optional = true }
+structopt  = { version = "0.3.5", optional = true }
 env_logger = { version = "0.7.1", optional = true }
 
 [[bin]]

intel-mkl-tool/src/lib.rs

@@ -122,12 +122,28 @@ pub fn package(mkl_path: &Path) -> Fallible<PathBuf> {
         bail!("MKL directory not found: {}", mkl_path.display());
     }
     let (year, update) = get_mkl_version(&mkl_path.join("include/mkl_version.h"))?;
-    let out = PathBuf::from(format!("mkl_linux64_{}_{}.tar.zst", year, update));
     info!("Intel MKL version: {}.{}", year, update);
-    info!("Create archive for Linux/64bit systems: {}", out.display());
-    let shared_libs: Vec<_> = glob(mkl_path.join("lib/intel64/*.so").to_str().unwrap())?
-        .map(|path| path.unwrap())
-        .collect();
+    let out = if cfg!(target_os = "Linux") {
+        let out = PathBuf::from(format!("mkl_linux64_{}_{}.tar.zst", year, update));
+        info!("Create archive for Linux/64bit systems: {}", out.display());
+        out
+    } else {
+        let out = PathBuf::from(format!("mkl_windows64_{}_{}.tar.zst", year, update));
+        info!(
+            "Create archive for Windows/64bit systems: {}",
+            out.display()
+        );
+        out
+    };
+
+    let shared_libs: Vec<_> = glob(
+        mkl_path
+            .join(format!("lib/intel64/*.{}", mkl::EXT))
+            .to_str()
+            .unwrap(),
+    )?
+    .map(|path| path.unwrap())
+    .collect();
     let f = fs::File::create(&out)?;
     let buf = io::BufWriter::new(f);
     let zstd = zstd::stream::write::Encoder::new(buf, 6)?;