851 lines
31 KiB
Rust
851 lines
31 KiB
Rust
#[cfg(no_std)]
|
|
use alloc::format;
|
|
#[cfg(not(no_std))]
|
|
use std::format;
|
|
|
|
use std::borrow::Cow;
|
|
|
|
use quote::{quote, quote_spanned, ToTokens};
|
|
use syn::{
|
|
parse::{Parse, ParseStream},
|
|
spanned::Spanned,
|
|
};
|
|
|
|
use crate::attrs::{ExportInfo, ExportScope, ExportedParams};
|
|
|
|
#[derive(Clone, Debug, Eq, PartialEq, Copy, Hash)]
|
|
pub enum FnNamespaceAccess {
|
|
Unset,
|
|
Global,
|
|
Internal,
|
|
}
|
|
|
|
impl Default for FnNamespaceAccess {
|
|
fn default() -> Self {
|
|
Self::Unset
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Eq, PartialEq, Copy, Hash)]
|
|
pub enum Index {
|
|
Get,
|
|
Set,
|
|
}
|
|
|
|
#[derive(Clone, Debug, Eq, PartialEq, Hash)]
|
|
pub enum Property {
|
|
Get(syn::Ident),
|
|
Set(syn::Ident),
|
|
}
|
|
|
|
#[derive(Clone, Debug, Eq, PartialEq, Hash)]
|
|
pub enum FnSpecialAccess {
|
|
None,
|
|
Index(Index),
|
|
Property(Property),
|
|
}
|
|
|
|
impl Default for FnSpecialAccess {
|
|
fn default() -> FnSpecialAccess {
|
|
FnSpecialAccess::None
|
|
}
|
|
}
|
|
|
|
impl FnSpecialAccess {
|
|
pub fn get_fn_name(&self) -> Option<(String, String, proc_macro2::Span)> {
|
|
match self {
|
|
FnSpecialAccess::None => None,
|
|
FnSpecialAccess::Property(Property::Get(ref g)) => {
|
|
Some((format!("{}{}", FN_GET, g), g.to_string(), g.span()))
|
|
}
|
|
FnSpecialAccess::Property(Property::Set(ref s)) => {
|
|
Some((format!("{}{}", FN_SET, s), s.to_string(), s.span()))
|
|
}
|
|
FnSpecialAccess::Index(Index::Get) => Some((
|
|
FN_IDX_GET.to_string(),
|
|
"index_get".to_string(),
|
|
proc_macro2::Span::call_site(),
|
|
)),
|
|
FnSpecialAccess::Index(Index::Set) => Some((
|
|
FN_IDX_SET.to_string(),
|
|
"index_set".to_string(),
|
|
proc_macro2::Span::call_site(),
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn flatten_type_groups(ty: &syn::Type) -> &syn::Type {
|
|
match ty {
|
|
syn::Type::Group(syn::TypeGroup { ref elem, .. })
|
|
| syn::Type::Paren(syn::TypeParen { ref elem, .. }) => flatten_type_groups(elem.as_ref()),
|
|
_ => ty,
|
|
}
|
|
}
|
|
|
|
pub fn print_type(ty: &syn::Type) -> String {
|
|
ty.to_token_stream()
|
|
.to_string()
|
|
.replace(" , ", ", ")
|
|
.replace("& ", "&")
|
|
.replace(" :: ", "::")
|
|
.replace(" ( ", "(")
|
|
.replace(" ) ", ")")
|
|
.replace(" < ", "<")
|
|
.replace(" > ", ">")
|
|
}
|
|
|
|
#[derive(Debug, Default)]
|
|
pub struct ExportedFnParams {
|
|
pub name: Vec<String>,
|
|
pub return_raw: Option<proc_macro2::Span>,
|
|
pub pure: Option<proc_macro2::Span>,
|
|
pub skip: bool,
|
|
pub special: FnSpecialAccess,
|
|
pub namespace: FnNamespaceAccess,
|
|
pub span: Option<proc_macro2::Span>,
|
|
}
|
|
|
|
pub const FN_GET: &str = "get$";
|
|
pub const FN_SET: &str = "set$";
|
|
pub const FN_IDX_GET: &str = "index$get$";
|
|
pub const FN_IDX_SET: &str = "index$set$";
|
|
|
|
impl Parse for ExportedFnParams {
|
|
fn parse(args: ParseStream) -> syn::Result<Self> {
|
|
if args.is_empty() {
|
|
return Ok(ExportedFnParams::default());
|
|
}
|
|
|
|
let info = crate::attrs::parse_attr_items(args)?;
|
|
Self::from_info(info)
|
|
}
|
|
}
|
|
|
|
impl ExportedParams for ExportedFnParams {
|
|
fn parse_stream(args: ParseStream) -> syn::Result<Self> {
|
|
Self::parse(args)
|
|
}
|
|
|
|
fn no_attrs() -> Self {
|
|
Default::default()
|
|
}
|
|
|
|
fn from_info(info: crate::attrs::ExportInfo) -> syn::Result<Self> {
|
|
let ExportInfo {
|
|
item_span: span,
|
|
items: attrs,
|
|
} = info;
|
|
let mut name = Vec::new();
|
|
let mut return_raw = None;
|
|
let mut pure = None;
|
|
let mut skip = false;
|
|
let mut namespace = FnNamespaceAccess::Unset;
|
|
let mut special = FnSpecialAccess::None;
|
|
for attr in attrs {
|
|
let crate::attrs::AttrItem {
|
|
key,
|
|
value,
|
|
span: item_span,
|
|
} = attr;
|
|
match (key.to_string().as_ref(), value) {
|
|
("get", None) | ("set", None) | ("name", None) => {
|
|
return Err(syn::Error::new(key.span(), "requires value"))
|
|
}
|
|
("name", Some(s)) if &s.value() == FN_IDX_GET => {
|
|
return Err(syn::Error::new(
|
|
item_span,
|
|
"use attribute 'index_get' instead",
|
|
))
|
|
}
|
|
("name", Some(s)) if &s.value() == FN_IDX_SET => {
|
|
return Err(syn::Error::new(
|
|
item_span,
|
|
"use attribute 'index_set' instead",
|
|
))
|
|
}
|
|
("name", Some(s)) if s.value().starts_with(FN_GET) => {
|
|
return Err(syn::Error::new(
|
|
item_span,
|
|
format!(
|
|
"use attribute 'getter = \"{}\"' instead",
|
|
&s.value()[FN_GET.len()..]
|
|
),
|
|
))
|
|
}
|
|
("name", Some(s)) if s.value().starts_with(FN_SET) => {
|
|
return Err(syn::Error::new(
|
|
item_span,
|
|
format!(
|
|
"use attribute 'setter = \"{}\"' instead",
|
|
&s.value()[FN_SET.len()..]
|
|
),
|
|
))
|
|
}
|
|
("name", Some(s)) => name.push(s.value()),
|
|
|
|
("index_get", Some(s))
|
|
| ("index_set", Some(s))
|
|
| ("return_raw", Some(s))
|
|
| ("pure", Some(s))
|
|
| ("skip", Some(s))
|
|
| ("global", Some(s))
|
|
| ("internal", Some(s)) => {
|
|
return Err(syn::Error::new(s.span(), "extraneous value"))
|
|
}
|
|
|
|
("pure", None) => pure = Some(item_span),
|
|
("return_raw", None) => return_raw = Some(item_span),
|
|
("skip", None) => skip = true,
|
|
("global", None) => match namespace {
|
|
FnNamespaceAccess::Unset => namespace = FnNamespaceAccess::Global,
|
|
FnNamespaceAccess::Global => (),
|
|
FnNamespaceAccess::Internal => {
|
|
return Err(syn::Error::new(
|
|
key.span(),
|
|
"namespace is already set to 'internal'",
|
|
))
|
|
}
|
|
},
|
|
("internal", None) => match namespace {
|
|
FnNamespaceAccess::Unset => namespace = FnNamespaceAccess::Internal,
|
|
FnNamespaceAccess::Internal => (),
|
|
FnNamespaceAccess::Global => {
|
|
return Err(syn::Error::new(
|
|
key.span(),
|
|
"namespace is already set to 'global'",
|
|
))
|
|
}
|
|
},
|
|
|
|
("get", Some(s)) => {
|
|
special = match special {
|
|
FnSpecialAccess::None => FnSpecialAccess::Property(Property::Get(
|
|
syn::Ident::new(&s.value(), s.span()),
|
|
)),
|
|
_ => return Err(syn::Error::new(item_span.span(), "conflicting getter")),
|
|
}
|
|
}
|
|
("set", Some(s)) => {
|
|
special = match special {
|
|
FnSpecialAccess::None => FnSpecialAccess::Property(Property::Set(
|
|
syn::Ident::new(&s.value(), s.span()),
|
|
)),
|
|
_ => return Err(syn::Error::new(item_span.span(), "conflicting setter")),
|
|
}
|
|
}
|
|
("index_get", None) => {
|
|
special = match special {
|
|
FnSpecialAccess::None => FnSpecialAccess::Index(Index::Get),
|
|
_ => {
|
|
return Err(syn::Error::new(item_span.span(), "conflicting index_get"))
|
|
}
|
|
}
|
|
}
|
|
("index_set", None) => {
|
|
special = match special {
|
|
FnSpecialAccess::None => FnSpecialAccess::Index(Index::Set),
|
|
_ => {
|
|
return Err(syn::Error::new(item_span.span(), "conflicting index_set"))
|
|
}
|
|
}
|
|
}
|
|
|
|
(attr, _) => {
|
|
return Err(syn::Error::new(
|
|
key.span(),
|
|
format!("unknown attribute '{}'", attr),
|
|
))
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(ExportedFnParams {
|
|
name,
|
|
return_raw,
|
|
pure,
|
|
skip,
|
|
special,
|
|
namespace,
|
|
span: Some(span),
|
|
..Default::default()
|
|
})
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct ExportedFn {
|
|
entire_span: proc_macro2::Span,
|
|
signature: syn::Signature,
|
|
visibility: syn::Visibility,
|
|
pass_context: bool,
|
|
mut_receiver: bool,
|
|
params: ExportedFnParams,
|
|
}
|
|
|
|
impl Parse for ExportedFn {
|
|
fn parse(input: ParseStream) -> syn::Result<Self> {
|
|
let fn_all: syn::ItemFn = input.parse()?;
|
|
let entire_span = fn_all.span();
|
|
let str_type_path = syn::parse2::<syn::Path>(quote! { str }).unwrap();
|
|
|
|
let context_type_path1 = syn::parse2::<syn::Path>(quote! { NativeCallContext }).unwrap();
|
|
let context_type_path2 =
|
|
syn::parse2::<syn::Path>(quote! { rhai::NativeCallContext }).unwrap();
|
|
let mut pass_context = false;
|
|
|
|
// #[cfg] attributes are not allowed on functions due to what is generated for them
|
|
crate::attrs::deny_cfg_attr(&fn_all.attrs)?;
|
|
|
|
let visibility = fn_all.vis;
|
|
|
|
// Determine if the function requires a call context
|
|
match fn_all.sig.inputs.first() {
|
|
Some(syn::FnArg::Typed(syn::PatType { ref ty, .. })) => {
|
|
match flatten_type_groups(ty.as_ref()) {
|
|
syn::Type::Path(p)
|
|
if p.path == context_type_path1 || p.path == context_type_path2 =>
|
|
{
|
|
pass_context = true;
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
|
|
let skip_slots = if pass_context { 1 } else { 0 };
|
|
|
|
// Determine whether function generates a special calling convention for a mutable receiver.
|
|
let mut_receiver = match fn_all.sig.inputs.iter().skip(skip_slots).next() {
|
|
Some(syn::FnArg::Receiver(syn::Receiver {
|
|
reference: Some(_), ..
|
|
})) => true,
|
|
Some(syn::FnArg::Typed(syn::PatType { ref ty, .. })) => {
|
|
match flatten_type_groups(ty.as_ref()) {
|
|
syn::Type::Reference(syn::TypeReference {
|
|
mutability: Some(_),
|
|
..
|
|
}) => true,
|
|
syn::Type::Reference(syn::TypeReference {
|
|
mutability: None,
|
|
ref elem,
|
|
..
|
|
}) => match flatten_type_groups(elem.as_ref()) {
|
|
syn::Type::Path(ref p) if p.path == str_type_path => false,
|
|
_ => {
|
|
return Err(syn::Error::new(
|
|
ty.span(),
|
|
"references from Rhai in this position must be mutable",
|
|
))
|
|
}
|
|
},
|
|
_ => false,
|
|
}
|
|
}
|
|
_ => false,
|
|
};
|
|
|
|
// All arguments after the first must be moved except for &str.
|
|
for arg in fn_all.sig.inputs.iter().skip(skip_slots + 1) {
|
|
let ty = match arg {
|
|
syn::FnArg::Typed(syn::PatType { ref ty, .. }) => ty,
|
|
_ => panic!("internal error: receiver argument outside of first position!?"),
|
|
};
|
|
let is_ok = match flatten_type_groups(ty.as_ref()) {
|
|
syn::Type::Reference(syn::TypeReference {
|
|
mutability: Some(_),
|
|
..
|
|
}) => false,
|
|
syn::Type::Reference(syn::TypeReference {
|
|
mutability: None,
|
|
ref elem,
|
|
..
|
|
}) => {
|
|
matches!(flatten_type_groups(elem.as_ref()), syn::Type::Path(ref p) if p.path == str_type_path)
|
|
}
|
|
syn::Type::Verbatim(_) => false,
|
|
_ => true,
|
|
};
|
|
if !is_ok {
|
|
return Err(syn::Error::new(
|
|
ty.span(),
|
|
"this type in this position passes from Rhai by value",
|
|
));
|
|
}
|
|
}
|
|
|
|
// Check return type.
|
|
match fn_all.sig.output {
|
|
syn::ReturnType::Type(_, ref ret_type) => {
|
|
match flatten_type_groups(ret_type.as_ref()) {
|
|
syn::Type::Ptr(_) => {
|
|
return Err(syn::Error::new(
|
|
fn_all.sig.output.span(),
|
|
"Rhai functions cannot return pointers",
|
|
))
|
|
}
|
|
syn::Type::Reference(_) => {
|
|
return Err(syn::Error::new(
|
|
fn_all.sig.output.span(),
|
|
"Rhai functions cannot return references",
|
|
))
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
Ok(ExportedFn {
|
|
entire_span,
|
|
signature: fn_all.sig,
|
|
visibility,
|
|
pass_context,
|
|
mut_receiver,
|
|
params: Default::default(),
|
|
})
|
|
}
|
|
}
|
|
|
|
impl ExportedFn {
|
|
#![allow(unused)]
|
|
|
|
pub fn params(&self) -> &ExportedFnParams {
|
|
&self.params
|
|
}
|
|
|
|
pub fn update_scope(&mut self, parent_scope: &ExportScope) {
|
|
let keep = match (self.params.skip, parent_scope) {
|
|
(true, _) => false,
|
|
(_, ExportScope::PubOnly) => self.is_public(),
|
|
(_, ExportScope::Prefix(s)) => self.name().to_string().starts_with(s),
|
|
(_, ExportScope::All) => true,
|
|
};
|
|
self.params.skip = !keep;
|
|
}
|
|
|
|
pub fn skipped(&self) -> bool {
|
|
self.params.skip
|
|
}
|
|
|
|
pub fn pass_context(&self) -> bool {
|
|
self.pass_context
|
|
}
|
|
|
|
pub fn signature(&self) -> &syn::Signature {
|
|
&self.signature
|
|
}
|
|
|
|
pub fn mutable_receiver(&self) -> bool {
|
|
self.mut_receiver
|
|
}
|
|
|
|
pub fn is_public(&self) -> bool {
|
|
!matches!(self.visibility, syn::Visibility::Inherited)
|
|
}
|
|
|
|
pub fn span(&self) -> &proc_macro2::Span {
|
|
&self.entire_span
|
|
}
|
|
|
|
pub fn name(&self) -> &syn::Ident {
|
|
&self.signature.ident
|
|
}
|
|
|
|
pub fn exported_names(&self) -> Vec<syn::LitStr> {
|
|
let mut literals: Vec<_> = self
|
|
.params
|
|
.name
|
|
.iter()
|
|
.map(|s| syn::LitStr::new(s, proc_macro2::Span::call_site()))
|
|
.collect();
|
|
|
|
if let Some((s, _, span)) = self.params.special.get_fn_name() {
|
|
literals.push(syn::LitStr::new(&s, span));
|
|
}
|
|
|
|
if literals.is_empty() {
|
|
literals.push(syn::LitStr::new(
|
|
&self.signature.ident.to_string(),
|
|
self.signature.ident.span(),
|
|
));
|
|
}
|
|
|
|
literals
|
|
}
|
|
|
|
pub fn exported_name<'n>(&'n self) -> Cow<'n, str> {
|
|
self.params
|
|
.name
|
|
.last()
|
|
.map_or_else(|| self.signature.ident.to_string().into(), |s| s.into())
|
|
}
|
|
|
|
pub fn arg_list(&self) -> impl Iterator<Item = &syn::FnArg> {
|
|
let skip = if self.pass_context { 1 } else { 0 };
|
|
self.signature.inputs.iter().skip(skip)
|
|
}
|
|
|
|
pub fn arg_count(&self) -> usize {
|
|
let skip = if self.pass_context { 1 } else { 0 };
|
|
self.signature.inputs.len() - skip
|
|
}
|
|
|
|
pub fn return_type(&self) -> Option<&syn::Type> {
|
|
match self.signature.output {
|
|
syn::ReturnType::Type(_, ref ret_type) => Some(flatten_type_groups(ret_type)),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn set_params(&mut self, mut params: ExportedFnParams) -> syn::Result<()> {
|
|
// Several issues are checked here to avoid issues with diagnostics caused by raising them later.
|
|
//
|
|
// 1a. Do not allow non-returning raw functions.
|
|
//
|
|
if params.return_raw.is_some() && self.return_type().is_none() {
|
|
return Err(syn::Error::new(
|
|
params.return_raw.unwrap(),
|
|
"functions marked with 'return_raw' must return Result<T, Box<EvalAltResult>>",
|
|
));
|
|
}
|
|
|
|
// 1b. Do not allow non-method pure functions.
|
|
//
|
|
if params.pure.is_some() && !self.mutable_receiver() {
|
|
return Err(syn::Error::new(
|
|
params.pure.unwrap(),
|
|
"'pure' is not necessary on functions without a &mut first parameter",
|
|
));
|
|
}
|
|
|
|
match params.special {
|
|
// 2a. Property getters must take only the subject as an argument.
|
|
FnSpecialAccess::Property(Property::Get(_)) if self.arg_count() != 1 => {
|
|
return Err(syn::Error::new(
|
|
self.signature.inputs.span(),
|
|
"property getter requires exactly 1 parameter",
|
|
))
|
|
}
|
|
// 2b. Property getters must return a value.
|
|
FnSpecialAccess::Property(Property::Get(_)) if self.return_type().is_none() => {
|
|
return Err(syn::Error::new(
|
|
self.signature.span(),
|
|
"property getter must return a value",
|
|
))
|
|
}
|
|
// 3a. Property setters must take the subject and a new value as arguments.
|
|
FnSpecialAccess::Property(Property::Set(_)) if self.arg_count() != 2 => {
|
|
return Err(syn::Error::new(
|
|
self.signature.inputs.span(),
|
|
"property setter requires exactly 2 parameters",
|
|
))
|
|
}
|
|
// 3b. Property setters must return nothing.
|
|
FnSpecialAccess::Property(Property::Set(_)) if self.return_type().is_some() => {
|
|
return Err(syn::Error::new(
|
|
self.signature.output.span(),
|
|
"property setter cannot return any value",
|
|
))
|
|
}
|
|
// 4a. Index getters must take the subject and the accessed "index" as arguments.
|
|
FnSpecialAccess::Index(Index::Get) if self.arg_count() != 2 => {
|
|
return Err(syn::Error::new(
|
|
self.signature.inputs.span(),
|
|
"index getter requires exactly 2 parameters",
|
|
))
|
|
}
|
|
// 4b. Index getters must return a value.
|
|
FnSpecialAccess::Index(Index::Get) if self.return_type().is_none() => {
|
|
return Err(syn::Error::new(
|
|
self.signature.span(),
|
|
"index getter must return a value",
|
|
))
|
|
}
|
|
// 5a. Index setters must take the subject, "index", and new value as arguments.
|
|
FnSpecialAccess::Index(Index::Set) if self.arg_count() != 3 => {
|
|
return Err(syn::Error::new(
|
|
self.signature.inputs.span(),
|
|
"index setter requires exactly 3 parameters",
|
|
))
|
|
}
|
|
// 5b. Index setters must return nothing.
|
|
FnSpecialAccess::Index(Index::Set) if self.return_type().is_some() => {
|
|
return Err(syn::Error::new(
|
|
self.signature.output.span(),
|
|
"index setter cannot return any value",
|
|
))
|
|
}
|
|
_ => {}
|
|
}
|
|
|
|
self.params = params;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn generate(self) -> proc_macro2::TokenStream {
|
|
let name: syn::Ident =
|
|
syn::Ident::new(&format!("rhai_fn_{}", self.name()), self.name().span());
|
|
let impl_block = self.generate_impl("Token");
|
|
let dyn_result_fn_block = self.generate_dynamic_fn();
|
|
let vis = self.visibility;
|
|
quote! {
|
|
#[automatically_derived]
|
|
#vis mod #name {
|
|
use super::*;
|
|
pub struct Token();
|
|
#impl_block
|
|
#dyn_result_fn_block
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn generate_dynamic_fn(&self) -> proc_macro2::TokenStream {
|
|
let name = self.name().clone();
|
|
|
|
let mut dynamic_signature = self.signature.clone();
|
|
dynamic_signature.ident =
|
|
syn::Ident::new("dynamic_result_fn", proc_macro2::Span::call_site());
|
|
dynamic_signature.output = syn::parse2::<syn::ReturnType>(quote! {
|
|
-> RhaiResult
|
|
})
|
|
.unwrap();
|
|
let arguments: Vec<syn::Ident> = dynamic_signature
|
|
.inputs
|
|
.iter()
|
|
.filter_map(|fn_arg| match fn_arg {
|
|
syn::FnArg::Typed(syn::PatType { ref pat, .. }) => match pat.as_ref() {
|
|
syn::Pat::Ident(ref ident) => Some(ident.ident.clone()),
|
|
_ => None,
|
|
},
|
|
_ => None,
|
|
})
|
|
.collect();
|
|
|
|
let return_span = self
|
|
.return_type()
|
|
.map(|r| r.span())
|
|
.unwrap_or_else(|| proc_macro2::Span::call_site());
|
|
if self.params.return_raw.is_some() {
|
|
quote_spanned! { return_span =>
|
|
pub #dynamic_signature {
|
|
#name(#(#arguments),*).map(Dynamic::from)
|
|
}
|
|
}
|
|
} else {
|
|
quote_spanned! { return_span =>
|
|
#[allow(unused)]
|
|
#[inline(always)]
|
|
pub #dynamic_signature {
|
|
Ok(Dynamic::from(#name(#(#arguments),*)))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn generate_impl(&self, on_type_name: &str) -> proc_macro2::TokenStream {
|
|
let sig_name = self.name().clone();
|
|
let arg_count = self.arg_count();
|
|
let is_method_call = self.mutable_receiver();
|
|
|
|
let mut unpack_statements: Vec<syn::Stmt> = Vec::new();
|
|
let mut unpack_exprs: Vec<syn::Expr> = Vec::new();
|
|
#[cfg(feature = "metadata")]
|
|
let mut input_type_names: Vec<String> = Vec::new();
|
|
let mut input_type_exprs: Vec<syn::Expr> = Vec::new();
|
|
|
|
let return_type = self
|
|
.return_type()
|
|
.map(print_type)
|
|
.unwrap_or_else(|| "()".to_string());
|
|
|
|
let skip_first_arg;
|
|
|
|
if self.pass_context {
|
|
unpack_exprs.push(syn::parse2::<syn::Expr>(quote! { context }).unwrap());
|
|
}
|
|
|
|
// Handle the first argument separately if the function has a "method like" receiver
|
|
if is_method_call {
|
|
skip_first_arg = true;
|
|
let first_arg = self.arg_list().next().unwrap();
|
|
let var = syn::Ident::new("arg0", proc_macro2::Span::call_site());
|
|
match first_arg {
|
|
syn::FnArg::Typed(syn::PatType { pat, ty, .. }) => {
|
|
#[cfg(feature = "metadata")]
|
|
let arg_name = format!("{}: {}", pat.to_token_stream(), print_type(ty));
|
|
let arg_type = match flatten_type_groups(ty.as_ref()) {
|
|
syn::Type::Reference(syn::TypeReference { ref elem, .. }) => elem.as_ref(),
|
|
p => p,
|
|
};
|
|
let downcast_span = quote_spanned!(arg_type.span() =>
|
|
&mut args[0usize].write_lock::<#arg_type>().unwrap()
|
|
);
|
|
unpack_statements.push(
|
|
syn::parse2::<syn::Stmt>(quote! {
|
|
let #var = #downcast_span;
|
|
})
|
|
.unwrap(),
|
|
);
|
|
if !self.params().pure.is_some() {
|
|
let arg_lit_str =
|
|
syn::LitStr::new(&pat.to_token_stream().to_string(), pat.span());
|
|
unpack_statements.push(
|
|
syn::parse2::<syn::Stmt>(quote! {
|
|
if args[0usize].is_read_only() {
|
|
return Err(Box::new(
|
|
EvalAltResult::ErrorAssignmentToConstant(#arg_lit_str.to_string(), Position::NONE)
|
|
));
|
|
}
|
|
})
|
|
.unwrap(),
|
|
);
|
|
}
|
|
#[cfg(feature = "metadata")]
|
|
input_type_names.push(arg_name);
|
|
input_type_exprs.push(
|
|
syn::parse2::<syn::Expr>(quote_spanned!(arg_type.span() =>
|
|
TypeId::of::<#arg_type>()
|
|
))
|
|
.unwrap(),
|
|
);
|
|
}
|
|
syn::FnArg::Receiver(_) => todo!("true self parameters not implemented yet"),
|
|
}
|
|
unpack_exprs.push(syn::parse2::<syn::Expr>(quote! { #var }).unwrap());
|
|
} else {
|
|
skip_first_arg = false;
|
|
}
|
|
|
|
// Handle the rest of the arguments, which all are passed by value.
|
|
//
|
|
// The only exception is strings, which need to be downcast to ImmutableString to enable a
|
|
// zero-copy conversion to &str by reference, or a cloned String.
|
|
let str_type_path = syn::parse2::<syn::Path>(quote! { str }).unwrap();
|
|
let string_type_path = syn::parse2::<syn::Path>(quote! { String }).unwrap();
|
|
for (i, arg) in self.arg_list().enumerate().skip(skip_first_arg as usize) {
|
|
let var = syn::Ident::new(&format!("arg{}", i), proc_macro2::Span::call_site());
|
|
let is_string;
|
|
let is_ref;
|
|
match arg {
|
|
syn::FnArg::Typed(syn::PatType { pat, ty, .. }) => {
|
|
#[cfg(feature = "metadata")]
|
|
let arg_name = format!("{}: {}", pat.to_token_stream(), print_type(ty));
|
|
let arg_type = ty.as_ref();
|
|
let downcast_span = match flatten_type_groups(arg_type) {
|
|
syn::Type::Reference(syn::TypeReference {
|
|
mutability: None,
|
|
ref elem,
|
|
..
|
|
}) => match flatten_type_groups(elem.as_ref()) {
|
|
syn::Type::Path(ref p) if p.path == str_type_path => {
|
|
is_string = true;
|
|
is_ref = true;
|
|
quote_spanned!(arg_type.span() =>
|
|
mem::take(args[#i]).take_immutable_string().unwrap()
|
|
)
|
|
}
|
|
_ => panic!("internal error: why wasn't this found earlier!?"),
|
|
},
|
|
syn::Type::Path(ref p) if p.path == string_type_path => {
|
|
is_string = true;
|
|
is_ref = false;
|
|
quote_spanned!(arg_type.span() =>
|
|
mem::take(args[#i]).take_string().unwrap()
|
|
)
|
|
}
|
|
_ => {
|
|
is_string = false;
|
|
is_ref = false;
|
|
quote_spanned!(arg_type.span() =>
|
|
mem::take(args[#i]).cast::<#arg_type>()
|
|
)
|
|
}
|
|
};
|
|
|
|
unpack_statements.push(
|
|
syn::parse2::<syn::Stmt>(quote! {
|
|
let #var = #downcast_span;
|
|
})
|
|
.unwrap(),
|
|
);
|
|
#[cfg(feature = "metadata")]
|
|
input_type_names.push(arg_name);
|
|
if !is_string {
|
|
input_type_exprs.push(
|
|
syn::parse2::<syn::Expr>(quote_spanned!(arg_type.span() =>
|
|
TypeId::of::<#arg_type>()
|
|
))
|
|
.unwrap(),
|
|
);
|
|
} else {
|
|
input_type_exprs.push(
|
|
syn::parse2::<syn::Expr>(quote_spanned!(arg_type.span() =>
|
|
TypeId::of::<ImmutableString>()
|
|
))
|
|
.unwrap(),
|
|
);
|
|
}
|
|
}
|
|
syn::FnArg::Receiver(_) => panic!("internal error: how did this happen!?"),
|
|
}
|
|
if !is_ref {
|
|
unpack_exprs.push(syn::parse2::<syn::Expr>(quote! { #var }).unwrap());
|
|
} else {
|
|
unpack_exprs.push(syn::parse2::<syn::Expr>(quote! { &#var }).unwrap());
|
|
}
|
|
}
|
|
|
|
// In method calls, the first argument will need to be mutably borrowed. Because Rust marks
|
|
// that as needing to borrow the entire array, all of the previous argument unpacking via
|
|
// clone needs to happen first.
|
|
if is_method_call {
|
|
let arg0 = unpack_statements.remove(0);
|
|
unpack_statements.push(arg0);
|
|
}
|
|
|
|
// Handle "raw returns", aka cases where the result is a dynamic or an error.
|
|
//
|
|
// This allows skipping the Dynamic::from wrap.
|
|
let return_span = self
|
|
.return_type()
|
|
.map(|r| r.span())
|
|
.unwrap_or_else(|| proc_macro2::Span::call_site());
|
|
let return_expr = if !self.params.return_raw.is_some() {
|
|
quote_spanned! { return_span =>
|
|
Ok(Dynamic::from(#sig_name(#(#unpack_exprs),*)))
|
|
}
|
|
} else {
|
|
quote_spanned! { return_span =>
|
|
#sig_name(#(#unpack_exprs),*).map(Dynamic::from)
|
|
}
|
|
};
|
|
|
|
let type_name = syn::Ident::new(on_type_name, proc_macro2::Span::call_site());
|
|
|
|
#[cfg(feature = "metadata")]
|
|
let param_names = quote! {
|
|
pub const PARAM_NAMES: &'static [&'static str] = &[#(#input_type_names,)* #return_type];
|
|
};
|
|
#[cfg(not(feature = "metadata"))]
|
|
let param_names = quote! {};
|
|
|
|
quote! {
|
|
impl #type_name {
|
|
#param_names
|
|
#[inline(always)] pub fn param_types() -> [TypeId; #arg_count] { [#(#input_type_exprs),*] }
|
|
}
|
|
impl PluginFunction for #type_name {
|
|
#[inline(always)]
|
|
fn call(&self, context: NativeCallContext, args: &mut [&mut Dynamic]) -> RhaiResult {
|
|
#(#unpack_statements)*
|
|
#return_expr
|
|
}
|
|
|
|
#[inline(always)] fn is_method_call(&self) -> bool { #is_method_call }
|
|
#[inline(always)] fn is_variadic(&self) -> bool { false }
|
|
}
|
|
}
|
|
}
|
|
}
|