rhai/src/fn_call.rs

1437 lines
52 KiB
Rust
Raw Normal View History

2020-11-20 09:52:28 +01:00
//! Implement function-calling mechanism for [`Engine`].
2017-12-20 22:16:53 +01:00
2021-03-08 08:30:32 +01:00
use crate::ast::FnHash;
use crate::engine::{
2021-03-01 07:54:20 +01:00
Imports, State, KEYWORD_DEBUG, KEYWORD_EVAL, KEYWORD_FN_PTR, KEYWORD_FN_PTR_CALL,
KEYWORD_FN_PTR_CURRY, KEYWORD_IS_DEF_VAR, KEYWORD_PRINT, KEYWORD_TYPE_OF,
2021-02-24 15:40:18 +01:00
MAX_DYNAMIC_PARAMETERS,
};
2021-03-02 16:08:54 +01:00
use crate::fn_builtin::{get_builtin_binary_op_fn, get_builtin_op_assignment_fn};
use crate::fn_native::{FnAny, FnCallArgs};
2020-11-16 16:10:14 +01:00
use crate::module::NamespaceRef;
2020-11-16 16:32:44 +01:00
use crate::optimize::OptimizationLevel;
use crate::stdlib::{
any::{type_name, TypeId},
boxed::Box,
convert::TryFrom,
format,
iter::{empty, once},
mem,
2021-03-01 02:30:23 +01:00
string::{String, ToString},
vec::Vec,
};
2021-03-01 09:53:03 +01:00
use crate::{
ast::{Expr, Stmt},
fn_native::CallableFunction,
2021-03-02 08:02:28 +01:00
RhaiResult,
2021-03-01 09:53:03 +01:00
};
2020-11-16 16:10:14 +01:00
use crate::{
2021-03-08 08:30:32 +01:00
calc_fn_hash, calc_fn_params_hash, combine_hashes, Dynamic, Engine, EvalAltResult, FnPtr,
2021-03-07 15:10:54 +01:00
ImmutableString, Module, ParseErrorType, Position, Scope, StaticVec,
2020-11-16 16:10:14 +01:00
};
#[cfg(not(feature = "no_object"))]
use crate::Map;
/// Extract the property name from a getter function name.
#[cfg(not(feature = "no_object"))]
2020-07-26 09:53:22 +02:00
#[inline(always)]
fn extract_prop_from_getter(_fn_name: &str) -> Option<&str> {
2020-11-16 09:28:04 +01:00
if _fn_name.starts_with(crate::engine::FN_GET) {
Some(&_fn_name[crate::engine::FN_GET.len()..])
} else {
None
}
}
/// Extract the property name from a setter function name.
#[cfg(not(feature = "no_object"))]
2020-07-26 09:53:22 +02:00
#[inline(always)]
fn extract_prop_from_setter(_fn_name: &str) -> Option<&str> {
2020-11-16 09:28:04 +01:00
if _fn_name.starts_with(crate::engine::FN_SET) {
Some(&_fn_name[crate::engine::FN_SET.len()..])
} else {
None
}
}
2020-07-31 06:11:16 +02:00
/// A type that temporarily stores a mutable reference to a `Dynamic`,
/// replacing it with a cloned copy.
#[derive(Debug, Default)]
struct ArgBackup<'a> {
orig_mut: Option<&'a mut Dynamic>,
value_copy: Dynamic,
}
impl<'a> ArgBackup<'a> {
/// This function replaces the first argument of a method call with a clone copy.
/// This is to prevent a pure function unintentionally consuming the first argument.
///
/// `restore_first_arg` must be called before the end of the scope to prevent the shorter lifetime from leaking.
///
/// # Safety
///
/// This method blindly casts a reference to another lifetime, which saves allocation and string cloning.
///
/// If `restore_first_arg` is called before the end of the scope, the shorter lifetime will not leak.
2020-12-29 03:41:20 +01:00
#[inline(always)]
2021-03-02 07:44:21 +01:00
fn change_first_arg_to_copy(&mut self, args: &mut FnCallArgs<'a>) {
2020-07-31 06:11:16 +02:00
// Clone the original value.
self.value_copy = args[0].clone();
// Replace the first reference with a reference to the clone, force-casting the lifetime.
// Must remember to restore it later with `restore_first_arg`.
//
// # Safety
//
// Blindly casting a reference to another lifetime saves allocation and string cloning,
// but must be used with the utmost care.
//
// We can do this here because, before the end of this scope, we'd restore the original reference
// via `restore_first_arg`. Therefore this shorter lifetime does not leak.
self.orig_mut = Some(mem::replace(args.get_mut(0).unwrap(), unsafe {
mem::transmute(&mut self.value_copy)
}));
}
2020-07-31 06:11:16 +02:00
/// This function restores the first argument that was replaced by `change_first_arg_to_copy`.
///
/// # Safety
///
/// If `change_first_arg_to_copy` has been called, this function **MUST** be called _BEFORE_ exiting
/// the current scope. Otherwise it is undefined behavior as the shorter lifetime will leak.
2020-12-29 03:41:20 +01:00
#[inline(always)]
2021-03-02 07:44:21 +01:00
fn restore_first_arg(mut self, args: &mut FnCallArgs<'a>) {
2020-07-31 06:11:16 +02:00
if let Some(this_pointer) = self.orig_mut.take() {
args[0] = this_pointer;
}
}
2017-12-20 22:16:53 +01:00
}
2020-07-31 06:11:16 +02:00
impl Drop for ArgBackup<'_> {
2020-12-29 03:41:20 +01:00
#[inline(always)]
2020-07-31 06:11:16 +02:00
fn drop(&mut self) {
// Panic if the shorter lifetime leaks.
assert!(
self.orig_mut.is_none(),
"ArgBackup::restore_first_arg has not been called prior to existing this scope"
2020-07-31 06:11:16 +02:00
);
}
2017-12-20 22:16:53 +01:00
}
2020-08-02 07:33:51 +02:00
#[inline(always)]
pub fn ensure_no_data_race(
fn_name: &str,
args: &FnCallArgs,
is_ref: bool,
) -> Result<(), Box<EvalAltResult>> {
2020-08-03 06:10:20 +02:00
if cfg!(not(feature = "no_closure")) {
2020-08-02 07:33:51 +02:00
let skip = if is_ref { 1 } else { 0 };
if let Some((n, _)) = args
.iter()
.skip(skip)
.enumerate()
.find(|(_, a)| a.is_locked())
{
2020-08-06 04:17:32 +02:00
return EvalAltResult::ErrorDataRace(
2020-08-02 07:33:51 +02:00
format!("argument #{} of function '{}'", n + 1 + skip, fn_name),
2020-11-20 09:52:28 +01:00
Position::NONE,
2020-08-06 04:17:32 +02:00
)
.into();
2020-08-02 07:33:51 +02:00
}
}
Ok(())
}
impl Engine {
2021-02-27 08:07:16 +01:00
/// Generate the signature for a function call.
fn gen_call_signature(
&self,
namespace: Option<&NamespaceRef>,
fn_name: &str,
args: &[&mut Dynamic],
) -> String {
format!(
"{}{} ({})",
namespace.map_or(String::new(), |ns| ns.to_string()),
fn_name,
args.iter()
.map(|a| if a.is::<ImmutableString>() {
"&str | ImmutableString | String"
} else {
self.map_type_name((*a).type_name())
})
.collect::<Vec<_>>()
.join(", ")
)
}
2021-03-01 09:53:03 +01:00
/// Resolve a function call.
///
2021-03-01 09:53:03 +01:00
/// Search order:
/// 1) AST - script functions in the AST
/// 2) Global namespace - functions registered via Engine::register_XXX
/// 3) Global modules - packages
/// 4) Imported modules - functions marked with global namespace
/// 5) Global sub-modules - functions marked with global namespace
2021-03-04 11:13:47 +01:00
#[inline(always)]
2021-03-01 09:53:03 +01:00
fn resolve_function<'s>(
&self,
2021-03-01 10:17:13 +01:00
mods: &Imports,
2021-03-01 09:53:03 +01:00
state: &'s mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
fn_name: &str,
2021-03-08 08:30:32 +01:00
hash_script: u64,
args: Option<&mut FnCallArgs>,
2021-03-01 09:53:03 +01:00
allow_dynamic: bool,
is_op_assignment: bool,
2021-03-01 09:53:03 +01:00
) -> &'s Option<(CallableFunction, Option<ImmutableString>)> {
2021-03-08 08:30:32 +01:00
let mut hash = if let Some(ref args) = args {
let hash_params = calc_fn_params_hash(args.iter().map(|a| a.type_id()));
combine_hashes(hash_script, hash_params)
} else {
hash_script
};
2021-03-01 09:53:03 +01:00
&*state
.fn_resolution_cache_mut()
2021-03-01 09:53:03 +01:00
.entry(hash)
.or_insert_with(|| {
2021-03-08 08:30:32 +01:00
let num_args = args.as_ref().map(|a| a.len()).unwrap_or(0);
2021-03-01 09:53:03 +01:00
let max_bitmask = if !allow_dynamic {
0
} else {
2021-03-08 08:30:32 +01:00
1usize << num_args.min(MAX_DYNAMIC_PARAMETERS)
2021-03-01 09:53:03 +01:00
};
let mut bitmask = 1usize; // Bitmask of which parameter to replace with `Dynamic`
loop {
let func = lib
.iter()
.find_map(|m| {
m.get_fn(hash, false)
.map(|f| (f.clone(), m.id_raw().cloned()))
})
.or_else(|| {
self.global_namespace
.get_fn(hash, false)
.cloned()
.map(|f| (f, None))
})
.or_else(|| {
self.global_modules.iter().find_map(|m| {
m.get_fn(hash, false)
.map(|f| (f.clone(), m.id_raw().cloned()))
})
})
.or_else(|| {
mods.get_fn(hash)
.map(|(f, source)| (f.clone(), source.cloned()))
})
.or_else(|| {
self.global_sub_modules.values().find_map(|m| {
m.get_qualified_fn(hash)
.map(|f| (f.clone(), m.id_raw().cloned()))
})
});
match func {
// Specific version found
Some(f) => return Some(f),
2021-02-24 15:40:18 +01:00
// Stop when all permutations are exhausted
None if bitmask >= max_bitmask => {
2021-03-02 16:08:54 +01:00
return if num_args != 2 {
None
2021-03-08 08:30:32 +01:00
} else if let Some(ref args) = args {
if !is_op_assignment {
if let Some(f) =
get_builtin_binary_op_fn(fn_name, &args[0], &args[1])
{
Some((
CallableFunction::from_method(Box::new(f) as Box<FnAny>),
None,
))
} else {
None
}
} else {
2021-03-08 08:30:32 +01:00
let (first, second) = args.split_first().unwrap();
if let Some(f) =
get_builtin_op_assignment_fn(fn_name, *first, second[0])
{
Some((
CallableFunction::from_method(Box::new(f) as Box<FnAny>),
None,
))
} else {
None
}
}
} else {
2021-03-08 08:30:32 +01:00
None
}
}
// Try all permutations with `Dynamic` wildcards
2021-03-01 09:53:03 +01:00
None => {
2021-03-08 08:30:32 +01:00
let hash_params = calc_fn_params_hash(
args.as_ref().unwrap().iter().enumerate().map(|(i, a)| {
let mask = 1usize << (num_args - i - 1);
if bitmask & mask != 0 {
// Replace with `Dynamic`
TypeId::of::<Dynamic>()
} else {
a.type_id()
}
}),
2021-03-08 08:30:32 +01:00
);
hash = combine_hashes(hash_script, hash_params);
bitmask += 1;
}
}
}
2021-03-01 09:53:03 +01:00
})
}
/// Call a native Rust function registered with the [`Engine`].
///
/// # WARNING
///
/// Function call arguments be _consumed_ when the function requires them to be passed by value.
/// All function arguments not in the first position are always passed by value and thus consumed.
/// **DO NOT** reuse the argument values unless for the first `&mut` argument - all others are silently replaced by `()`!
pub(crate) fn call_native_fn(
&self,
mods: &Imports,
state: &mut State,
lib: &[&Module],
fn_name: &str,
2021-03-08 08:30:32 +01:00
hash_native: u64,
2021-03-01 09:53:03 +01:00
args: &mut FnCallArgs,
is_ref: bool,
is_op_assignment: bool,
pos: Position,
) -> Result<(Dynamic, bool), Box<EvalAltResult>> {
self.inc_operations(state, pos)?;
let source = state.source.clone();
// Check if function access already in the cache
let func = self.resolve_function(
mods,
state,
lib,
fn_name,
2021-03-08 08:30:32 +01:00
hash_native,
Some(args),
true,
is_op_assignment,
);
2020-12-30 14:12:51 +01:00
if let Some((func, src)) = func {
2020-07-31 06:11:16 +02:00
assert!(func.is_native());
// Calling pure function but the first argument is a reference?
2021-03-02 07:44:21 +01:00
let mut backup: Option<ArgBackup> = None;
if is_ref && func.is_pure() && !args.is_empty() {
backup = Some(Default::default());
backup.as_mut().unwrap().change_first_arg_to_copy(args);
}
// Run external function
let source = src.as_ref().or_else(|| source.as_ref()).map(|s| s.as_str());
2020-08-02 12:53:25 +02:00
let result = if func.is_plugin_fn() {
2020-12-21 16:12:45 +01:00
func.get_plugin_fn()
.call((self, fn_name, source, mods, lib).into(), args)
2020-08-02 12:53:25 +02:00
} else {
func.get_native_fn()((self, fn_name, source, mods, lib).into(), args)
2020-08-02 12:53:25 +02:00
};
// Restore the original reference
2021-03-02 07:44:21 +01:00
if let Some(backup) = backup {
backup.restore_first_arg(args);
}
2020-07-31 06:11:16 +02:00
let result = result.map_err(|err| err.fill_position(pos))?;
// See if the function match print/debug (which requires special processing)
return Ok(match fn_name {
2020-12-22 08:27:27 +01:00
KEYWORD_PRINT => {
2021-03-05 16:00:27 +01:00
let text = result.take_immutable_string().map_err(|typ| {
2020-08-06 04:17:32 +02:00
EvalAltResult::ErrorMismatchOutputType(
self.map_type_name(type_name::<ImmutableString>()).into(),
typ.into(),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
2020-12-22 08:27:27 +01:00
})?;
2021-03-05 16:00:27 +01:00
((self.print)(&text).into(), false)
2020-12-22 08:27:27 +01:00
}
2020-12-21 15:04:46 +01:00
KEYWORD_DEBUG => {
2021-03-05 16:00:27 +01:00
let text = result.take_immutable_string().map_err(|typ| {
2020-12-21 15:04:46 +01:00
EvalAltResult::ErrorMismatchOutputType(
self.map_type_name(type_name::<ImmutableString>()).into(),
typ.into(),
pos,
)
})?;
let source = state.source.as_ref().map(|s| s.as_str());
2021-03-05 16:00:27 +01:00
((self.debug)(&text, source, pos).into(), false)
2020-12-21 15:04:46 +01:00
}
_ => (result, func.is_method()),
});
}
// Getter function not found?
#[cfg(not(feature = "no_object"))]
if let Some(prop) = extract_prop_from_getter(fn_name) {
2020-08-06 04:17:32 +02:00
return EvalAltResult::ErrorDotExpr(
format!(
2021-01-15 10:13:04 +01:00
"Unknown property '{}' - a getter is not registered for type '{}'",
prop,
self.map_type_name(args[0].type_name())
),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
.into();
}
// Setter function not found?
#[cfg(not(feature = "no_object"))]
if let Some(prop) = extract_prop_from_setter(fn_name) {
2020-08-06 04:17:32 +02:00
return EvalAltResult::ErrorDotExpr(
format!(
2021-01-15 10:13:04 +01:00
"No writable property '{}' - a setter is not registered for type '{}' to handle '{}'",
prop,
2020-09-26 05:35:18 +02:00
self.map_type_name(args[0].type_name()),
self.map_type_name(args[1].type_name()),
),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
.into();
}
// index getter function not found?
2020-07-26 09:53:22 +02:00
#[cfg(not(feature = "no_index"))]
2020-11-16 09:28:04 +01:00
if fn_name == crate::engine::FN_IDX_GET && args.len() == 2 {
2020-08-06 04:17:32 +02:00
return EvalAltResult::ErrorFunctionNotFound(
format!(
"{} [{}]",
self.map_type_name(args[0].type_name()),
self.map_type_name(args[1].type_name()),
),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
.into();
}
// index setter function not found?
2020-07-26 09:53:22 +02:00
#[cfg(not(feature = "no_index"))]
2020-11-16 09:28:04 +01:00
if fn_name == crate::engine::FN_IDX_SET {
2020-08-06 04:17:32 +02:00
return EvalAltResult::ErrorFunctionNotFound(
format!(
"{} [{}]=",
self.map_type_name(args[0].type_name()),
self.map_type_name(args[1].type_name()),
),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
.into();
}
// Raise error
2020-08-06 04:17:32 +02:00
EvalAltResult::ErrorFunctionNotFound(
2021-02-27 08:07:16 +01:00
self.gen_call_signature(None, fn_name, args.as_ref()),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
.into()
}
/// Call a script-defined function.
///
2021-01-02 16:30:10 +01:00
/// # WARNING
///
/// Function call arguments may be _consumed_ when the function requires them to be passed by value.
/// All function arguments not in the first position are always passed by value and thus consumed.
/// **DO NOT** reuse the argument values unless for the first `&mut` argument - all others are silently replaced by `()`!
2020-07-26 07:51:09 +02:00
#[cfg(not(feature = "no_function"))]
pub(crate) fn call_script_fn(
&self,
scope: &mut Scope,
mods: &mut Imports,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
this_ptr: &mut Option<&mut Dynamic>,
2020-11-16 09:28:04 +01:00
fn_def: &crate::ast::ScriptFnDef,
args: &mut FnCallArgs,
2020-12-12 04:15:09 +01:00
pos: Position,
level: usize,
2021-03-02 08:02:28 +01:00
) -> RhaiResult {
#[inline(always)]
fn make_error(
2021-02-21 07:41:20 +01:00
name: crate::stdlib::string::String,
fn_def: &crate::ast::ScriptFnDef,
state: &State,
err: Box<EvalAltResult>,
pos: Position,
2021-03-02 08:02:28 +01:00
) -> RhaiResult {
Err(Box::new(EvalAltResult::ErrorInFunctionCall(
name,
fn_def
.lib
.as_ref()
.and_then(|m| m.id())
.unwrap_or_else(|| state.source.as_ref().map_or_else(|| "", |s| s.as_str()))
.to_string(),
err,
pos,
)))
}
2020-12-20 16:25:11 +01:00
self.inc_operations(state, pos)?;
2020-07-31 06:11:16 +02:00
// Check for stack overflow
#[cfg(not(feature = "no_function"))]
#[cfg(not(feature = "unchecked"))]
if level > self.max_call_levels() {
2020-12-12 04:15:09 +01:00
return Err(Box::new(EvalAltResult::ErrorStackOverflow(pos)));
2020-07-31 06:11:16 +02:00
}
let orig_scope_level = state.scope_level;
state.scope_level += 1;
let prev_scope_len = scope.len();
let prev_mods_len = mods.len();
// Put arguments into scope as variables
// Actually consume the arguments instead of cloning them
scope.extend(
fn_def
.params
.iter()
.zip(args.iter_mut().map(|v| mem::take(*v)))
.map(|(name, value)| {
2020-11-16 09:28:04 +01:00
let var_name: crate::stdlib::borrow::Cow<'_, str> =
crate::r#unsafe::unsafe_cast_var_name_to_lifetime(name).into();
2020-12-08 15:47:38 +01:00
(var_name, value)
}),
);
// Merge in encapsulated environment, if any
let lib_merged;
let (unified_lib, unified) = if let Some(ref env_lib) = fn_def.lib {
state.push_fn_resolution_cache();
lib_merged = once(env_lib.as_ref())
.chain(lib.iter().cloned())
.collect::<StaticVec<_>>();
(lib_merged.as_ref(), true)
} else {
(lib, false)
};
2020-11-09 14:52:23 +01:00
#[cfg(not(feature = "no_module"))]
if !fn_def.mods.is_empty() {
2020-12-21 15:04:46 +01:00
mods.extend(fn_def.mods.iter_raw().map(|(n, m)| (n.clone(), m.clone())));
2020-11-09 14:52:23 +01:00
}
2020-12-29 05:29:45 +01:00
// Evaluate the function
let stmt = &fn_def.body;
2020-11-20 09:52:28 +01:00
let result = self
2020-12-29 05:29:45 +01:00
.eval_stmt(scope, mods, state, unified_lib, this_ptr, stmt, level)
2020-11-20 09:52:28 +01:00
.or_else(|err| match *err {
// Convert return statement to return value
EvalAltResult::Return(x, _) => Ok(x),
// Error in sub function call
2020-12-30 14:12:51 +01:00
EvalAltResult::ErrorInFunctionCall(name, src, err, _) => {
let fn_name = if src.is_empty() {
format!("{} < {}", name, fn_def.name)
} else {
format!("{} @ '{}' < {}", name, src, fn_def.name)
};
make_error(fn_name, fn_def, state, err, pos)
2020-11-20 09:52:28 +01:00
}
// System errors are passed straight-through
mut err if err.is_system_exception() => Err(Box::new({
err.set_position(pos);
err
})),
2020-11-20 09:52:28 +01:00
// Other errors are wrapped in `ErrorInFunctionCall`
_ => make_error(fn_def.name.to_string(), fn_def, state, err, pos),
2020-11-20 09:52:28 +01:00
});
// Remove all local variables
scope.rewind(prev_scope_len);
mods.truncate(prev_mods_len);
state.scope_level = orig_scope_level;
if unified {
state.pop_fn_resolution_cache();
}
result
}
2021-03-08 11:40:23 +01:00
// Does a scripted function exist?
2020-12-29 03:41:20 +01:00
#[inline(always)]
2021-03-08 11:40:23 +01:00
pub(crate) fn has_script_fn(
&self,
2020-11-22 08:41:55 +01:00
mods: Option<&Imports>,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
2021-03-08 11:40:23 +01:00
hash_script: u64,
) -> bool {
let cache = state.fn_resolution_cache_mut();
2021-03-08 11:40:23 +01:00
if let Some(result) = cache.get(&hash_script).map(|v| v.is_some()) {
return result;
}
// First check script-defined functions
2021-03-08 11:40:23 +01:00
let result = lib.iter().any(|&m| m.contains_fn(hash_script, false))
// Then check registered functions
2021-03-08 11:40:23 +01:00
|| self.global_namespace.contains_fn(hash_script, false)
// Then check packages
2021-03-08 11:40:23 +01:00
|| self.global_modules.iter().any(|m| m.contains_fn(hash_script, false))
// Then check imported modules
2021-03-08 11:40:23 +01:00
|| mods.map_or(false, |m| m.contains_fn(hash_script))
2021-03-01 08:39:49 +01:00
// Then check sub-modules
2021-03-08 11:40:23 +01:00
|| self.global_sub_modules.values().any(|m| m.contains_qualified_fn(hash_script));
if !result {
cache.insert(hash_script, None);
}
2021-03-08 11:40:23 +01:00
result
}
2020-07-31 06:11:16 +02:00
/// Perform an actual function call, native Rust or scripted, taking care of special functions.
///
2021-01-02 16:30:10 +01:00
/// # WARNING
///
/// Function call arguments may be _consumed_ when the function requires them to be passed by value.
/// All function arguments not in the first position are always passed by value and thus consumed.
/// **DO NOT** reuse the argument values unless for the first `&mut` argument - all others are silently replaced by `()`!
pub(crate) fn exec_fn_call(
&self,
mods: &mut Imports,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
fn_name: &str,
2021-03-08 08:30:32 +01:00
hash: FnHash,
args: &mut FnCallArgs,
is_ref: bool,
2020-08-05 16:53:01 +02:00
_is_method: bool,
2020-12-12 04:15:09 +01:00
pos: Position,
2020-10-22 06:26:44 +02:00
_capture_scope: Option<Scope>,
2020-08-05 16:53:01 +02:00
_level: usize,
) -> Result<(Dynamic, bool), Box<EvalAltResult>> {
2020-08-02 07:33:51 +02:00
// Check for data race.
2020-08-03 06:10:20 +02:00
if cfg!(not(feature = "no_closure")) {
2020-08-02 07:33:51 +02:00
ensure_no_data_race(fn_name, args, is_ref)?;
}
2021-03-01 15:44:56 +01:00
// These may be redirected from method style calls.
match fn_name {
2021-03-01 15:44:56 +01:00
// Handle type_of()
KEYWORD_TYPE_OF if args.len() == 1 => {
return Ok((
self.map_type_name(args[0].type_name()).to_string().into(),
false,
));
}
2021-03-01 15:44:56 +01:00
// Handle is_def_fn()
2020-07-30 12:18:28 +02:00
#[cfg(not(feature = "no_function"))]
2021-03-01 15:44:56 +01:00
crate::engine::KEYWORD_IS_DEF_FN
2021-03-07 15:10:54 +01:00
if args.len() == 2 && args[0].is::<FnPtr>() && args[1].is::<crate::INT>() =>
2021-03-01 15:44:56 +01:00
{
let fn_name = args[0].read_lock::<ImmutableString>().unwrap();
2021-03-01 15:44:56 +01:00
let num_params = args[1].as_int().unwrap();
2020-09-25 13:07:24 +02:00
2021-03-01 15:44:56 +01:00
return Ok((
if num_params < 0 {
Dynamic::FALSE
} else {
2021-03-08 08:30:32 +01:00
let hash_script = calc_fn_hash(empty(), &fn_name, num_params as usize);
2021-03-08 11:40:23 +01:00
self.has_script_fn(Some(mods), state, lib, hash_script)
.into()
2021-03-01 15:44:56 +01:00
},
false,
));
}
2020-12-21 15:04:46 +01:00
2021-03-01 15:44:56 +01:00
// Handle is_shared()
#[cfg(not(feature = "no_closure"))]
crate::engine::KEYWORD_IS_SHARED if args.len() == 1 => {
return Err(Box::new(EvalAltResult::ErrorRuntime(
format!(
"'{}' should not be called this way. Try {}(...);",
fn_name, fn_name
)
.into(),
pos,
)))
}
2020-12-21 15:04:46 +01:00
2021-03-01 15:44:56 +01:00
KEYWORD_FN_PTR | KEYWORD_EVAL | KEYWORD_IS_DEF_VAR if args.len() == 1 => {
return Err(Box::new(EvalAltResult::ErrorRuntime(
format!(
"'{}' should not be called this way. Try {}(...);",
fn_name, fn_name
)
.into(),
pos,
)))
}
2020-12-29 05:29:45 +01:00
2021-03-01 15:44:56 +01:00
KEYWORD_FN_PTR_CALL | KEYWORD_FN_PTR_CURRY if !args.is_empty() => {
return Err(Box::new(EvalAltResult::ErrorRuntime(
format!(
"'{}' should not be called this way. Try {}(...);",
fn_name, fn_name
)
.into(),
pos,
)))
}
2020-12-21 15:04:46 +01:00
2021-03-01 15:44:56 +01:00
_ => (),
}
2020-12-21 15:04:46 +01:00
2021-03-08 08:30:32 +01:00
// Scripted function call?
let hash_script = if hash.is_native_only() {
None
} else {
Some(hash.script_hash())
};
2021-03-01 15:44:56 +01:00
#[cfg(not(feature = "no_function"))]
2021-03-08 08:30:32 +01:00
if let Some((func, source)) = hash_script.and_then(|hash| {
self.resolve_function(mods, state, lib, fn_name, hash, None, false, false)
2021-03-01 15:44:56 +01:00
.as_ref()
.map(|(f, s)| (f.clone(), s.clone()))
}) {
// Script function call
assert!(func.is_script());
2020-07-30 12:18:28 +02:00
2021-03-01 15:44:56 +01:00
let func = func.get_fn_def();
2020-12-21 15:04:46 +01:00
2021-03-01 15:44:56 +01:00
let scope: &mut Scope = &mut Default::default();
2020-12-29 05:29:45 +01:00
2021-03-01 15:44:56 +01:00
// Move captured variables into scope
#[cfg(not(feature = "no_closure"))]
if let Some(captured) = _capture_scope {
if !func.externals.is_empty() {
captured
.into_iter()
.filter(|(name, _, _)| func.externals.iter().any(|ex| ex == name))
.for_each(|(name, value, _)| {
// Consume the scope values.
scope.push_dynamic(name, value);
});
}
}
2020-07-30 12:18:28 +02:00
2021-03-01 15:44:56 +01:00
let result = if _is_method {
// Method call of script function - map first argument to `this`
let (first, rest) = args.split_first_mut().unwrap();
2020-12-21 15:04:46 +01:00
2021-03-01 15:44:56 +01:00
let orig_source = mem::take(&mut state.source);
state.source = source;
2020-09-25 13:07:24 +02:00
2021-03-01 15:44:56 +01:00
let level = _level + 1;
2020-07-30 12:18:28 +02:00
2021-03-01 15:44:56 +01:00
let result = self.call_script_fn(
scope,
mods,
state,
lib,
&mut Some(*first),
func,
rest,
pos,
level,
);
// Restore the original source
state.source = orig_source;
result?
} else {
// Normal call of script function
// The first argument is a reference?
2021-03-02 07:44:21 +01:00
let mut backup: Option<ArgBackup> = None;
if is_ref && !args.is_empty() {
backup = Some(Default::default());
backup.as_mut().unwrap().change_first_arg_to_copy(args);
}
2020-09-11 16:32:59 +02:00
2021-03-01 15:44:56 +01:00
let orig_source = mem::take(&mut state.source);
state.source = source;
let level = _level + 1;
let result =
self.call_script_fn(scope, mods, state, lib, &mut None, func, args, pos, level);
// Restore the original source
state.source = orig_source;
// Restore the original reference
2021-03-02 07:44:21 +01:00
if let Some(backup) = backup {
backup.restore_first_arg(args);
}
2021-03-01 15:44:56 +01:00
result?
};
return Ok((result, false));
}
2021-03-01 15:44:56 +01:00
// Native function call
2021-03-08 08:30:32 +01:00
self.call_native_fn(
mods,
state,
lib,
fn_name,
hash.native_hash(),
args,
is_ref,
false,
pos,
)
}
/// Evaluate a list of statements with no `this` pointer.
2020-11-20 09:52:28 +01:00
/// This is commonly used to evaluate a list of statements in an [`AST`] or a script function body.
2021-03-04 11:13:47 +01:00
#[inline(always)]
pub(crate) fn eval_global_statements(
2020-10-20 04:54:32 +02:00
&self,
scope: &mut Scope,
mods: &mut Imports,
2020-12-20 16:25:11 +01:00
state: &mut State,
statements: &[Stmt],
2020-10-20 04:54:32 +02:00
lib: &[&Module],
2020-12-29 05:29:45 +01:00
level: usize,
2021-03-02 08:02:28 +01:00
) -> RhaiResult {
self.eval_stmt_block(scope, mods, state, lib, &mut None, statements, false, level)
2020-10-20 04:54:32 +02:00
.or_else(|err| match *err {
EvalAltResult::Return(out, _) => Ok(out),
EvalAltResult::LoopBreak(_, _) => {
unreachable!("no outer loop scope to break out of")
}
2020-10-20 04:54:32 +02:00
_ => Err(err),
})
}
2020-12-20 16:25:11 +01:00
/// Evaluate a text script in place - used primarily for 'eval'.
#[inline]
2020-12-20 16:25:11 +01:00
fn eval_script_expr_in_place(
&self,
scope: &mut Scope,
mods: &mut Imports,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
2020-10-03 17:27:30 +02:00
script: &str,
2020-12-20 16:25:11 +01:00
pos: Position,
2020-12-29 05:29:45 +01:00
level: usize,
2021-03-02 08:02:28 +01:00
) -> RhaiResult {
2020-12-20 16:25:11 +01:00
self.inc_operations(state, pos)?;
2020-07-31 06:11:16 +02:00
2020-10-27 04:30:38 +01:00
let script = script.trim();
if script.is_empty() {
2020-11-15 16:14:29 +01:00
return Ok(Dynamic::UNIT);
2020-10-27 04:30:38 +01:00
}
// Compile the script text
// No optimizations because we only run it once
2020-10-20 04:54:32 +02:00
let ast = self.compile_with_scope_and_optimization_level(
&Default::default(),
&[script],
OptimizationLevel::None,
)?;
// If new functions are defined within the eval string, it is an error
2020-10-05 15:52:39 +02:00
if ast.lib().count().0 != 0 {
return Err(ParseErrorType::FnWrongDefinition.into());
}
// Evaluate the AST
let mut new_state: State = Default::default();
new_state.source = state.source.clone();
new_state.operations = state.operations;
2020-12-29 05:29:45 +01:00
let result =
self.eval_global_statements(scope, mods, &mut new_state, ast.statements(), lib, level);
2020-12-20 16:46:46 +01:00
state.operations = new_state.operations;
2020-12-21 10:39:37 +01:00
result
}
/// Call a dot method.
2020-07-26 09:53:22 +02:00
#[cfg(not(feature = "no_object"))]
pub(crate) fn make_method_call(
&self,
mods: &mut Imports,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
2020-11-10 16:26:50 +01:00
fn_name: &str,
2021-03-08 08:30:32 +01:00
mut hash: FnHash,
2020-11-16 09:28:04 +01:00
target: &mut crate::engine::Target,
2021-03-09 11:11:43 +01:00
(call_args, call_arg_positions): &mut (StaticVec<Dynamic>, StaticVec<Position>),
2020-12-12 04:15:09 +01:00
pos: Position,
level: usize,
) -> Result<(Dynamic, bool), Box<EvalAltResult>> {
let is_ref = target.is_ref();
// Get a reference to the mutation target Dynamic
let obj = target.as_mut();
2020-11-10 16:26:50 +01:00
let mut fn_name = fn_name;
2021-03-01 15:44:56 +01:00
let (result, updated) = match fn_name {
KEYWORD_FN_PTR_CALL if obj.is::<FnPtr>() => {
// FnPtr call
let fn_ptr = obj.read_lock::<FnPtr>().unwrap();
// Redirect function name
let fn_name = fn_ptr.fn_name();
let args_len = call_args.len() + fn_ptr.curry().len();
2021-03-08 08:30:32 +01:00
// Recalculate hashes
2021-03-09 16:30:48 +01:00
let new_hash = FnHash::from_script(calc_fn_hash(empty(), fn_name, args_len));
2021-03-01 15:44:56 +01:00
// Arguments are passed as-is, adding the curried arguments
let mut curry = fn_ptr.curry().iter().cloned().collect::<StaticVec<_>>();
let mut arg_values = curry
.iter_mut()
2021-03-09 11:11:43 +01:00
.chain(call_args.iter_mut())
2021-03-01 15:44:56 +01:00
.collect::<StaticVec<_>>();
let args = arg_values.as_mut();
// Map it to name(args) in function-call style
self.exec_fn_call(
2021-03-09 16:30:48 +01:00
mods, state, lib, fn_name, new_hash, args, false, false, pos, None, level,
)
2020-10-03 10:25:58 +02:00
}
KEYWORD_FN_PTR_CALL => {
if call_args.len() > 0 {
2021-03-09 11:11:43 +01:00
if !call_args[0].is::<FnPtr>() {
return Err(self.make_type_mismatch_err::<FnPtr>(
self.map_type_name(obj.type_name()),
2021-03-09 11:11:43 +01:00
call_arg_positions[0],
));
}
} else {
return Err(self.make_type_mismatch_err::<FnPtr>(
self.map_type_name(obj.type_name()),
pos,
));
}
2021-03-01 15:44:56 +01:00
// FnPtr call on object
2021-03-09 11:11:43 +01:00
let fn_ptr = call_args.remove(0).cast::<FnPtr>();
call_arg_positions.remove(0);
2021-03-01 15:44:56 +01:00
// Redirect function name
let fn_name = fn_ptr.fn_name();
let args_len = call_args.len() + fn_ptr.curry().len();
// Recalculate hash
2021-03-09 16:30:48 +01:00
let new_hash = FnHash::from_script_and_native(
2021-03-08 08:30:32 +01:00
calc_fn_hash(empty(), fn_name, args_len),
calc_fn_hash(empty(), fn_name, args_len + 1),
);
2021-03-01 15:44:56 +01:00
// Replace the first argument with the object pointer, adding the curried arguments
let mut curry = fn_ptr.curry().iter().cloned().collect::<StaticVec<_>>();
let mut arg_values = once(obj)
.chain(curry.iter_mut())
2021-03-09 11:11:43 +01:00
.chain(call_args.iter_mut())
2021-03-01 15:44:56 +01:00
.collect::<StaticVec<_>>();
let args = arg_values.as_mut();
// Map it to name(args) in function-call style
self.exec_fn_call(
2021-03-09 16:30:48 +01:00
mods, state, lib, fn_name, new_hash, args, is_ref, true, pos, None, level,
2021-03-01 15:44:56 +01:00
)
}
KEYWORD_FN_PTR_CURRY => {
if !obj.is::<FnPtr>() {
return Err(self.make_type_mismatch_err::<FnPtr>(
self.map_type_name(obj.type_name()),
pos,
));
}
2021-03-01 15:44:56 +01:00
let fn_ptr = obj.read_lock::<FnPtr>().unwrap();
// Curry call
2021-03-01 15:44:56 +01:00
Ok((
if call_args.is_empty() {
fn_ptr.clone()
} else {
FnPtr::new_unchecked(
fn_ptr.get_fn_name().clone(),
fn_ptr
.curry()
.iter()
.cloned()
2021-03-09 11:11:43 +01:00
.chain(call_args.iter_mut().map(|v| mem::take(v)))
.collect(),
)
}
2021-03-01 15:44:56 +01:00
.into(),
false,
))
}
2021-03-01 15:44:56 +01:00
// Handle is_shared()
#[cfg(not(feature = "no_closure"))]
crate::engine::KEYWORD_IS_SHARED if call_args.is_empty() => {
return Ok((target.is_shared().into(), false));
2020-07-30 12:18:28 +02:00
}
2021-03-01 15:44:56 +01:00
_ => {
let _redirected;
// Check if it is a map method call in OOP style
#[cfg(not(feature = "no_object"))]
if let Some(map) = obj.read_lock::<Map>() {
if let Some(val) = map.get(fn_name) {
if let Some(fn_ptr) = val.read_lock::<FnPtr>() {
// Remap the function name
_redirected = fn_ptr.get_fn_name().clone();
fn_name = &_redirected;
// Add curried arguments
fn_ptr
.curry()
.iter()
.cloned()
.enumerate()
2021-03-09 11:11:43 +01:00
.for_each(|(i, v)| {
call_args.insert(i, v);
call_arg_positions.insert(i, Position::NONE);
});
2021-03-01 15:44:56 +01:00
// Recalculate the hash based on the new function name and new arguments
2021-03-08 08:30:32 +01:00
hash = FnHash::from_script_and_native(
calc_fn_hash(empty(), fn_name, call_args.len()),
calc_fn_hash(empty(), fn_name, call_args.len() + 1),
);
2021-03-01 15:44:56 +01:00
}
}
};
2021-03-01 15:44:56 +01:00
// Attached object pointer in front of the arguments
let mut arg_values = once(obj)
2021-03-09 11:11:43 +01:00
.chain(call_args.iter_mut())
2021-03-01 15:44:56 +01:00
.collect::<StaticVec<_>>();
let args = arg_values.as_mut();
self.exec_fn_call(
mods, state, lib, fn_name, hash, args, is_ref, true, pos, None, level,
)
}
}?;
2020-10-04 04:40:44 +02:00
// Propagate the changed value back to the source if necessary
if updated {
target.propagate_changed_value();
}
Ok((result, updated))
}
/// Call a function in normal function-call style.
pub(crate) fn make_function_call(
&self,
scope: &mut Scope,
mods: &mut Imports,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
this_ptr: &mut Option<&mut Dynamic>,
2020-11-10 16:26:50 +01:00
fn_name: &str,
args_expr: &[Expr],
2021-03-08 08:30:32 +01:00
mut hash: FnHash,
2020-12-12 04:15:09 +01:00
pos: Position,
2020-10-22 06:26:44 +02:00
capture_scope: bool,
level: usize,
2021-03-02 08:02:28 +01:00
) -> RhaiResult {
let args_expr = args_expr.as_ref();
2021-02-03 12:14:26 +01:00
// Handle call() - Redirect function call
let redirected;
let mut args_expr = args_expr.as_ref();
let mut curry = StaticVec::new();
let mut name = fn_name;
2021-03-01 15:44:56 +01:00
match name {
// Handle call()
KEYWORD_FN_PTR_CALL if args_expr.len() >= 1 => {
let fn_ptr =
self.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[0], level)?;
if !fn_ptr.is::<FnPtr>() {
return Err(self.make_type_mismatch_err::<FnPtr>(
self.map_type_name(fn_ptr.type_name()),
args_expr[0].position(),
));
}
2021-02-03 12:14:26 +01:00
2021-03-01 15:44:56 +01:00
let fn_ptr = fn_ptr.cast::<FnPtr>();
curry.extend(fn_ptr.curry().iter().cloned());
// Redirect function name
redirected = fn_ptr.take_data().0;
name = &redirected;
// Skip the first argument
args_expr = &args_expr.as_ref()[1..];
// Recalculate hash
let args_len = args_expr.len() + curry.len();
2021-03-08 08:30:32 +01:00
hash = if !hash.is_native_only() {
FnHash::from_script(calc_fn_hash(empty(), name, args_len))
} else {
FnHash::from_native(calc_fn_hash(empty(), name, args_len))
};
2021-02-03 12:14:26 +01:00
}
2021-03-01 15:44:56 +01:00
// Handle Fn()
KEYWORD_FN_PTR if args_expr.len() == 1 => {
// Fn - only in function call style
return self
.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[0], level)?
.take_immutable_string()
.map_err(|typ| {
self.make_type_mismatch_err::<ImmutableString>(typ, args_expr[0].position())
})
.and_then(|s| FnPtr::try_from(s))
.map(Into::<Dynamic>::into)
.map_err(|err| err.fill_position(args_expr[0].position()));
}
2021-02-03 12:14:26 +01:00
2021-03-01 15:44:56 +01:00
// Handle curry()
KEYWORD_FN_PTR_CURRY if args_expr.len() > 1 => {
let fn_ptr =
self.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[0], level)?;
2021-02-03 12:14:26 +01:00
2021-03-01 15:44:56 +01:00
if !fn_ptr.is::<FnPtr>() {
return Err(self.make_type_mismatch_err::<FnPtr>(
self.map_type_name(fn_ptr.type_name()),
args_expr[0].position(),
));
}
2021-02-03 12:14:26 +01:00
2021-03-01 15:44:56 +01:00
let (name, mut fn_curry) = fn_ptr.cast::<FnPtr>().take_data();
2021-02-03 12:14:26 +01:00
2021-03-01 15:44:56 +01:00
// Append the new curried arguments to the existing list.
2021-03-01 15:44:56 +01:00
args_expr.iter().skip(1).try_for_each(
|expr| -> Result<(), Box<EvalAltResult>> {
fn_curry
.push(self.eval_expr(scope, mods, state, lib, this_ptr, expr, level)?);
Ok(())
},
)?;
2021-03-01 15:44:56 +01:00
return Ok(FnPtr::new_unchecked(name, fn_curry).into());
}
2021-03-01 15:44:56 +01:00
// Handle is_shared()
#[cfg(not(feature = "no_closure"))]
crate::engine::KEYWORD_IS_SHARED if args_expr.len() == 1 => {
let value =
self.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[0], level)?;
return Ok(value.is_shared().into());
}
2021-03-01 15:44:56 +01:00
// Handle is_def_fn()
#[cfg(not(feature = "no_function"))]
crate::engine::KEYWORD_IS_DEF_FN if args_expr.len() == 2 => {
let fn_name = self
.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[0], level)?
.take_immutable_string()
.map_err(|err| {
self.make_type_mismatch_err::<ImmutableString>(err, args_expr[0].position())
})?;
let num_params = self
.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[1], level)?
.as_int()
.map_err(|err| {
2021-03-07 15:10:54 +01:00
self.make_type_mismatch_err::<crate::INT>(err, args_expr[0].position())
})?;
return Ok(if num_params < 0 {
Dynamic::FALSE
2021-03-01 15:44:56 +01:00
} else {
2021-03-08 08:30:32 +01:00
let hash_script = calc_fn_hash(empty(), &fn_name, num_params as usize);
2021-03-08 11:40:23 +01:00
self.has_script_fn(Some(mods), state, lib, hash_script)
.into()
});
2021-03-01 15:44:56 +01:00
}
2021-03-01 15:44:56 +01:00
// Handle is_def_var()
KEYWORD_IS_DEF_VAR if args_expr.len() == 1 => {
let var_name = self
.eval_expr(scope, mods, state, lib, this_ptr, &args_expr[0], level)?
.take_immutable_string()
.map_err(|err| {
self.make_type_mismatch_err::<ImmutableString>(err, args_expr[0].position())
})?;
2021-03-05 16:00:27 +01:00
return Ok(scope.contains(&var_name).into());
2021-03-01 15:44:56 +01:00
}
2021-03-01 15:44:56 +01:00
// Handle eval()
KEYWORD_EVAL if args_expr.len() == 1 => {
let script_expr = &args_expr[0];
let script_pos = script_expr.position();
// eval - only in function call style
let prev_len = scope.len();
let script = self
.eval_expr(scope, mods, state, lib, this_ptr, script_expr, level)?
.take_immutable_string()
.map_err(|typ| {
self.make_type_mismatch_err::<ImmutableString>(typ, script_pos)
})?;
2021-03-01 15:44:56 +01:00
let result = self.eval_script_expr_in_place(
scope,
mods,
state,
lib,
2021-03-05 16:00:27 +01:00
&script,
2021-03-01 15:44:56 +01:00
script_pos,
level + 1,
);
// IMPORTANT! If the eval defines new variables in the current scope,
// all variable offsets from this point on will be mis-aligned.
if scope.len() != prev_len {
state.always_search = true;
}
2020-10-03 10:25:58 +02:00
2021-03-01 15:44:56 +01:00
return result.map_err(|err| {
Box::new(EvalAltResult::ErrorInFunctionCall(
KEYWORD_EVAL.to_string(),
state
.source
.as_ref()
.map_or_else(|| "", |s| s.as_str())
.to_string(),
err,
pos,
))
});
2020-07-30 12:18:28 +02:00
}
2021-03-01 10:17:13 +01:00
2021-03-01 15:44:56 +01:00
_ => (),
2020-07-30 12:18:28 +02:00
}
// Normal function call - except for Fn, curry, call and eval (handled above)
let mut arg_values: StaticVec<_>;
let mut args: StaticVec<_>;
let mut is_ref = false;
2020-10-22 06:26:44 +02:00
let capture = if capture_scope && !scope.is_empty() {
2020-10-12 11:00:58 +02:00
Some(scope.clone_visible())
2020-07-30 12:18:28 +02:00
} else {
None
};
if args_expr.is_empty() && curry.is_empty() {
// No arguments
args = Default::default();
} else {
2021-02-03 12:14:26 +01:00
// If the first argument is a variable, and there is no curried arguments,
// convert to method-call style in order to leverage potential &mut first argument and
// avoid cloning the value
2020-10-25 15:08:02 +01:00
if curry.is_empty() && args_expr[0].get_variable_access(false).is_some() {
// func(x, ...) -> x.func(...)
arg_values = args_expr
.iter()
.skip(1)
.map(|expr| self.eval_expr(scope, mods, state, lib, this_ptr, expr, level))
.collect::<Result<_, _>>()?;
2020-12-26 06:05:57 +01:00
let (mut target, pos) =
self.search_namespace(scope, mods, state, lib, this_ptr, &args_expr[0])?;
2020-12-08 16:09:12 +01:00
if target.as_ref().is_read_only() {
2020-12-08 15:47:38 +01:00
target = target.into_owned();
}
2020-12-20 16:25:11 +01:00
self.inc_operations(state, pos)?;
args = if target.is_shared() || target.is_value() {
arg_values.insert(0, target.take_or_clone().flatten());
arg_values.iter_mut().collect()
} else {
// Turn it into a method call only if the object is not shared and not a simple value
is_ref = true;
once(target.take_ref().unwrap())
.chain(arg_values.iter_mut())
.collect()
};
} else {
// func(..., ...)
arg_values = args_expr
.iter()
.map(|expr| self.eval_expr(scope, mods, state, lib, this_ptr, expr, level))
.collect::<Result<_, _>>()?;
args = curry.iter_mut().chain(arg_values.iter_mut()).collect();
}
}
let args = args.as_mut();
2020-07-30 12:18:28 +02:00
self.exec_fn_call(
2021-03-08 08:30:32 +01:00
mods, state, lib, name, hash, args, is_ref, false, pos, capture, level,
)
.map(|(v, _)| v)
}
2020-11-10 16:26:50 +01:00
/// Call a namespace-qualified function in normal function-call style.
pub(crate) fn make_qualified_function_call(
&self,
scope: &mut Scope,
mods: &mut Imports,
state: &mut State,
2020-10-20 04:54:32 +02:00
lib: &[&Module],
this_ptr: &mut Option<&mut Dynamic>,
2020-11-10 16:26:50 +01:00
namespace: Option<&NamespaceRef>,
fn_name: &str,
args_expr: &[Expr],
2021-03-08 08:30:32 +01:00
hash: u64,
2020-12-12 04:15:09 +01:00
pos: Position,
level: usize,
2021-03-02 08:02:28 +01:00
) -> RhaiResult {
let args_expr = args_expr.as_ref();
2021-02-27 08:07:16 +01:00
let namespace = namespace.unwrap();
let mut arg_values: StaticVec<_>;
let mut first_arg_value = None;
let mut args: StaticVec<_>;
if args_expr.is_empty() {
// No arguments
args = Default::default();
} else {
2020-11-10 16:26:50 +01:00
// See if the first argument is a variable (not namespace-qualified).
// If so, convert to method-call style in order to leverage potential
// &mut first argument and avoid cloning the value
if args_expr[0].get_variable_access(true).is_some() {
// func(x, ...) -> x.func(...)
arg_values = args_expr
.iter()
.enumerate()
.map(|(i, expr)| {
// Skip the first argument
if i == 0 {
Ok(Default::default())
} else {
self.eval_expr(scope, mods, state, lib, this_ptr, expr, level)
}
})
.collect::<Result<_, _>>()?;
// Get target reference to first argument
2020-12-26 06:05:57 +01:00
let (target, pos) =
self.search_scope_only(scope, mods, state, lib, this_ptr, &args_expr[0])?;
2020-12-20 16:25:11 +01:00
self.inc_operations(state, pos)?;
if target.is_shared() || target.is_value() {
arg_values[0] = target.take_or_clone().flatten();
args = arg_values.iter_mut().collect();
} else {
let (first, rest) = arg_values.split_first_mut().unwrap();
first_arg_value = Some(first);
args = once(target.take_ref().unwrap())
.chain(rest.iter_mut())
.collect();
}
} else {
// func(..., ...) or func(mod::x, ...)
arg_values = args_expr
.iter()
.map(|expr| self.eval_expr(scope, mods, state, lib, this_ptr, expr, level))
.collect::<Result<_, _>>()?;
args = arg_values.iter_mut().collect();
}
}
2021-03-03 15:49:57 +01:00
let module = self.search_imports(mods, state, namespace).ok_or_else(|| {
EvalAltResult::ErrorModuleNotFound(namespace[0].name.to_string(), namespace[0].pos)
})?;
// First search in script-defined functions (can override built-in)
2021-03-08 08:30:32 +01:00
let func = match module.get_qualified_fn(hash) {
// Then search in Rust functions
None => {
2020-12-20 16:25:11 +01:00
self.inc_operations(state, pos)?;
2021-03-08 08:30:32 +01:00
let hash_params = calc_fn_params_hash(args.iter().map(|a| a.type_id()));
let hash_qualified_fn = combine_hashes(hash, hash_params);
module.get_qualified_fn(hash_qualified_fn)
}
r => r,
};
// Clone first argument if the function is not a method after-all
if let Some(first) = first_arg_value {
if !func.map(|f| f.is_method()).unwrap_or(true) {
let first_val = args[0].clone();
args[0] = first;
*args[0] = first_val;
}
}
match func {
#[cfg(not(feature = "no_function"))]
Some(f) if f.is_script() => {
let args = args.as_mut();
2020-10-20 04:54:32 +02:00
let new_scope = &mut Default::default();
2020-11-07 16:33:21 +01:00
let fn_def = f.get_fn_def().clone();
2020-12-21 15:04:46 +01:00
let mut source = module.id_raw().cloned();
2020-12-21 15:04:46 +01:00
mem::swap(&mut state.source, &mut source);
2020-12-29 05:29:45 +01:00
let level = level + 1;
2020-12-21 15:04:46 +01:00
let result = self.call_script_fn(
2020-12-12 04:15:09 +01:00
new_scope, mods, state, lib, &mut None, &fn_def, args, pos, level,
2020-12-21 15:04:46 +01:00
);
state.source = source;
result
}
Some(f) if f.is_plugin_fn() => f
.get_plugin_fn()
.clone()
.call(
(self, fn_name, module.id(), &*mods, lib).into(),
args.as_mut(),
)
.map_err(|err| err.fill_position(pos)),
Some(f) if f.is_native() => f.get_native_fn()(
(self, fn_name, module.id(), &*mods, lib).into(),
args.as_mut(),
)
.map_err(|err| err.fill_position(pos)),
Some(f) => unreachable!("unknown function type: {:?}", f),
2020-08-06 04:17:32 +02:00
None => EvalAltResult::ErrorFunctionNotFound(
2021-02-27 08:07:16 +01:00
self.gen_call_signature(Some(namespace), fn_name, args.as_ref()),
2020-12-12 04:15:09 +01:00
pos,
2020-08-06 04:17:32 +02:00
)
.into(),
}
}
}