Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add basic support for precompiling plugins

Browse source
This commit is contained in:
Isaac Clayton 2022-07-07 11:34:12 +02:00
parent 895747476f
commit a16fc2ba0c
5 changed files with 51 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

34
crates/plugin_runtime/build.rs
View file

@ -1,4 +1,5 @@
use std::path::Path;
use std::{io::Write, path::Path};
use wasmtime::{Config, Engine};
fn main() {
let base = Path::new("../../plugins");
@ -28,6 +29,8 @@ fn main() {
.expect("Could not find compiled plugins in target");
println!("cargo:warning={:?}", binaries);
let engine = create_engine();
for file in binaries {
let is_wasm = || {
let path = file.ok()?.path();
@ -39,11 +42,30 @@ fn main() {
};
if let Some(path) = is_wasm() {
std::fs::copy(&path, base.join("bin").join(path.file_name().unwrap()))
.expect("Could not copy compiled plugin to bin");
let out_path = base.join("bin").join(path.file_name().unwrap());
std::fs::copy(&path, &out_path).expect("Could not copy compiled plugin to bin");
precompile(&out_path, &engine);
}
}
// TODO: create .wat versions
// TODO: optimize with wasm-opt
}
fn create_engine() -> Engine {
let mut config = Config::default();
config.async_support(true);
// config.epoch_interruption(true);
Engine::new(&config).expect("Could not create engine")
}
fn precompile(path: &Path, engine: &Engine) {
let bytes = std::fs::read(path).expect("Could not read wasm module");
let compiled = engine
.precompile_module(&bytes)
.expect("Could not precompile module");
let out_path = path.parent().unwrap().join(&format!(
"{}.pre",
path.file_name().unwrap().to_string_lossy()
));
let mut out_file = std::fs::File::create(out_path)
.expect("Could not create output file for precompiled module");
out_file.write_all(&compiled).unwrap();
}