use wasmer::{Array, Instance, Memory, NativeFunc, RuntimeError, WasmPtr, WasmTypeList};
use dprint_core::plugins::wasm::PLUGIN_SYSTEM_SCHEMA_VERSION;
use dprint_core::types::{ErrBox, Error};
pub enum FormatResult {
NoChange = 0,
Change = 1,
Error = 2,
}
impl From<u8> for FormatResult {
fn from(orig: u8) -> Self {
match orig {
0 => FormatResult::NoChange,
1 => FormatResult::Change,
2 => FormatResult::Error,
_ => unreachable!(),
}
}
}
pub struct WasmFunctions {
instance: Instance,
memory: Memory,
}
impl WasmFunctions {
pub fn new(instance: Instance) -> Result<Self, ErrBox> {
match get_plugin_schema_version(&instance) {
Ok(plugin_schema_version) => {
if plugin_schema_version != PLUGIN_SYSTEM_SCHEMA_VERSION {
return err!(
"Invalid schema version: {} -- Expected: {}. This may indicate you should upgrade your dprint CLI or plugin.",
plugin_schema_version,
PLUGIN_SYSTEM_SCHEMA_VERSION
);
}
}
Err(err) => {
return err!(
"Error determining plugin schema version. Are you sure this is a dprint plugin? {}",
err.to_string()
);
}
}
let memory = instance.exports.get_memory("memory")?.clone();
Ok(WasmFunctions { instance, memory })
}
#[inline]
pub fn set_global_config(&self) -> Result<(), ErrBox> {
let set_global_config_func = self.get_export::<(), ()>("set_global_config")?;
wasm_runtime_error_to_err_box(set_global_config_func.call())
}
#[inline]
pub fn set_plugin_config(&self) -> Result<(), ErrBox> {
let set_plugin_config_func = self.get_export::<(), ()>("set_plugin_config")?;
wasm_runtime_error_to_err_box(set_plugin_config_func.call())
}
#[inline]
pub fn get_plugin_info(&self) -> Result<usize, ErrBox> {
let get_plugin_info_func = self.get_export::<(), u32>("get_plugin_info")?;
wasm_runtime_error_to_err_box(get_plugin_info_func.call()).map(|value| value as usize)
}
#[inline]
pub fn get_license_text(&self) -> Result<usize, ErrBox> {
let get_license_text_func = self.get_export::<(), u32>("get_license_text")?;
wasm_runtime_error_to_err_box(get_license_text_func.call()).map(|value| value as usize)
}
#[inline]
pub fn get_resolved_config(&self) -> Result<usize, ErrBox> {
let get_resolved_config_func = self.get_export::<(), u32>("get_resolved_config")?;
wasm_runtime_error_to_err_box(get_resolved_config_func.call()).map(|value| value as usize)
}
#[inline]
pub fn get_config_diagnostics(&self) -> Result<usize, ErrBox> {
let get_config_diagnostics_func = self.get_export::<(), u32>("get_config_diagnostics")?;
wasm_runtime_error_to_err_box(get_config_diagnostics_func.call()).map(|value| value as usize)
}
#[inline]
pub fn set_override_config(&self) -> Result<(), ErrBox> {
let set_override_config_func = self.get_export::<(), ()>("set_override_config")?;
wasm_runtime_error_to_err_box(set_override_config_func.call())
}
#[inline]
pub fn set_file_path(&self) -> Result<(), ErrBox> {
let set_file_path_func = self.get_export::<(), ()>("set_file_path")?;
wasm_runtime_error_to_err_box(set_file_path_func.call())
}
#[inline]
pub fn format(&self) -> Result<FormatResult, ErrBox> {
let format_func = self.get_export::<(), u8>("format")?;
wasm_runtime_error_to_err_box(format_func.call()).map(|value| value.into())
}
#[inline]
pub fn get_formatted_text(&self) -> Result<usize, ErrBox> {
let get_formatted_text_func = self.get_export::<(), u32>("get_formatted_text")?;
wasm_runtime_error_to_err_box(get_formatted_text_func.call()).map(|value| value as usize)
}
#[inline]
pub fn get_error_text(&self) -> Result<usize, ErrBox> {
let get_error_text_func = self.get_export::<(), u32>("get_error_text")?;
wasm_runtime_error_to_err_box(get_error_text_func.call()).map(|value| value as usize)
}
#[inline]
pub fn get_memory(&self) -> &Memory {
&self.memory
}
#[inline]
pub fn clear_shared_bytes(&self, capacity: usize) -> Result<(), ErrBox> {
let clear_shared_bytes_func = self.get_export::<u32, ()>("clear_shared_bytes")?;
wasm_runtime_error_to_err_box(clear_shared_bytes_func.call(capacity as u32))
}
#[inline]
pub fn get_wasm_memory_buffer_size(&self) -> Result<usize, ErrBox> {
let get_wasm_memory_buffer_size_func = self.get_export::<(), u32>("get_wasm_memory_buffer_size")?;
wasm_runtime_error_to_err_box(get_wasm_memory_buffer_size_func.call()).map(|value| value as usize)
}
#[inline]
pub fn get_wasm_memory_buffer_ptr(&self) -> Result<WasmPtr<u8, Array>, ErrBox> {
let get_wasm_memory_buffer_func = self.get_export::<(), WasmPtr<u8, Array>>("get_wasm_memory_buffer")?;
wasm_runtime_error_to_err_box(get_wasm_memory_buffer_func.call())
}
#[inline]
pub fn set_buffer_with_shared_bytes(&self, offset: usize, length: usize) -> Result<(), ErrBox> {
let set_buffer_with_shared_bytes_func = self.get_export::<(u32, u32), ()>("set_buffer_with_shared_bytes")?;
wasm_runtime_error_to_err_box(set_buffer_with_shared_bytes_func.call(offset as u32, length as u32))
}
#[inline]
pub fn add_to_shared_bytes_from_buffer(&self, length: usize) -> Result<(), ErrBox> {
let add_to_shared_bytes_from_buffer_func = self.get_export::<u32, ()>("add_to_shared_bytes_from_buffer")?;
wasm_runtime_error_to_err_box(add_to_shared_bytes_from_buffer_func.call(length as u32))
}
fn get_export<'a, Args, Rets>(&'a self, name: &str) -> Result<NativeFunc<Args, Rets>, ErrBox>
where
Args: WasmTypeList,
Rets: WasmTypeList,
{
match self.instance.exports.get_function(name) {
Ok(func) => match func.native::<Args, Rets>() {
Ok(native_func) => Ok(native_func),
Err(err) => err!("Error creating function '{}'. Message: {}", name, err.to_string()),
},
Err(err) => err!("Could not find export in plugin with name '{}'. Message: {}", name, err.to_string()),
}
}
}
fn get_plugin_schema_version(instance: &Instance) -> Result<u32, ErrBox> {
let plugin_schema_version_func = instance.exports.get_function("get_plugin_schema_version")?;
let plugin_schema_version_func = plugin_schema_version_func.native::<(), u32>()?;
Ok(plugin_schema_version_func.call()?)
}
#[inline]
fn wasm_runtime_error_to_err_box<T>(result: Result<T, RuntimeError>) -> Result<T, ErrBox> {
match result {
Ok(value) => Ok(value),
Err(err) => Err(Error::new(err.to_string())),
}
}