103 lines
2.5 KiB
Rust
103 lines
2.5 KiB
Rust
use std::any::TypeId;
|
|
use value::{RuntimeValue, TryInto};
|
|
use Error;
|
|
|
|
/// Safe wrapper for list of arguments
|
|
#[derive(Debug)]
|
|
pub struct RuntimeArgs<'a>(&'a [RuntimeValue]);
|
|
|
|
impl<'a> From<&'a [RuntimeValue]> for RuntimeArgs<'a> {
|
|
fn from(inner: &'a [RuntimeValue]) -> Self {
|
|
RuntimeArgs(inner)
|
|
}
|
|
}
|
|
|
|
impl<'a> RuntimeArgs<'a> {
|
|
|
|
/// Extract argument by index `idx` returning error if cast is invalid or not enough arguments
|
|
pub fn nth<T>(&self, idx: usize) -> Result<T, Error> where RuntimeValue: TryInto<T, Error> {
|
|
Ok(self.nth_value(idx)?.try_into().map_err(|_| Error::Value("Invalid argument cast".to_owned()))?)
|
|
}
|
|
|
|
/// Extract argument as a runtime value by index `idx` returning error is not enough arguments
|
|
pub fn nth_value(&self, idx: usize) -> Result<RuntimeValue, Error> {
|
|
if self.0.len() <= idx {
|
|
return Err(Error::Value("Invalid argument index".to_owned()));
|
|
}
|
|
Ok(self.0[idx])
|
|
}
|
|
|
|
/// Total number of arguments
|
|
pub fn len(&self) -> usize {
|
|
self.0.len()
|
|
}
|
|
}
|
|
|
|
/// Custom user error.
|
|
pub trait HostError: 'static + ::std::fmt::Display + ::std::fmt::Debug + Send + Sync {
|
|
#[doc(hidden)]
|
|
fn __private_get_type_id__(&self) -> TypeId {
|
|
TypeId::of::<Self>()
|
|
}
|
|
}
|
|
|
|
impl HostError {
|
|
/// Attempt to downcast this `HostError` to a concrete type by reference.
|
|
pub fn downcast_ref<T: HostError>(&self) -> Option<&T> {
|
|
if self.__private_get_type_id__() == TypeId::of::<T>() {
|
|
unsafe { Some(&*(self as *const HostError as *const T)) }
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
/// Attempt to downcast this `HostError` to a concrete type by mutable
|
|
/// reference.
|
|
pub fn downcast_mut<T: HostError>(&mut self) -> Option<&mut T> {
|
|
if self.__private_get_type_id__() == TypeId::of::<T>() {
|
|
unsafe { Some(&mut *(self as *mut HostError as *mut T)) }
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
pub trait Externals {
|
|
fn invoke_index(
|
|
&mut self,
|
|
index: usize,
|
|
args: RuntimeArgs,
|
|
) -> Result<Option<RuntimeValue>, Error>;
|
|
}
|
|
|
|
pub struct NopExternals;
|
|
|
|
impl Externals for NopExternals {
|
|
fn invoke_index(
|
|
&mut self,
|
|
_index: usize,
|
|
_args: RuntimeArgs,
|
|
) -> Result<Option<RuntimeValue>, Error> {
|
|
Err(Error::Trap("invoke index on no-op externals".into()))
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use value::RuntimeValue;
|
|
use super::RuntimeArgs;
|
|
|
|
#[test]
|
|
fn i32_runtime_args() {
|
|
let args: RuntimeArgs = (&[RuntimeValue::I32(0)][..]).into();
|
|
let val: i32 = args.nth(0).unwrap();
|
|
assert_eq!(val, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn i64_invalid_arg_cast() {
|
|
let args: RuntimeArgs = (&[RuntimeValue::I64(90534534545322)][..]).into();
|
|
assert!(args.nth::<i32>(0).is_err());
|
|
}
|
|
} |