harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add more automation to build script

Browse source
This commit is contained in:
0/0 2022-10-10 11:05:03 -06:00
parent 6df5ed5ab6
commit cc2822e7e5
No known key found for this signature in database
GPG key ID: DE8D5010C0AAA3DC
2 changed files with 35 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

40
sys/build.rs
View file

@ -12,11 +12,39 @@ fn main() {
.header("wrapper.h")
.clang_arg("-I./whisper.cpp")
.parse_callbacks(Box::new(bindgen::CargoCallbacks))
.generate()
.expect("Unable to generate bindings");
.generate();
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
bindings
.write_to_file(out_path.join("bindings.rs"))
.expect("Couldn't write bindings!");
match bindings {
Ok(b) => {
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
b.write_to_file(out_path.join("bindings.rs"))
.expect("Couldn't write bindings!");
}
Err(e) => {
println!("cargo:warning=Unable to generate bindings: {}", e);
println!("cargo:warning=Using bundled bindings.rs, which may be out of date");
// copy src/bindings.rs to OUT_DIR
std::fs::copy(
"src/bindings.rs",
env::var("OUT_DIR").unwrap() + "/bindings.rs",
)
.expect("Unable to copy bindings.rs");
}
}
// build libwhisper.a
env::set_current_dir("whisper.cpp").expect("Unable to change directory");
let code = std::process::Command::new("make")
.arg("libwhisper.a")
.status()
.expect("Failed to build libwhisper.a");
if code.code() != Some(0) {
panic!("Failed to build libwhisper.a");
}
// move libwhisper.a to where Cargo expects it (OUT_DIR)
std::fs::copy(
"libwhisper.a",
format!("{}/libwhisper.a", env::var("OUT_DIR").unwrap()),
)
.expect("Failed to copy libwhisper.a");
}