rquickjs-macro 0.5.1

Procedural macros for rquickjs
Documentation
use proc_macro2::{Ident, TokenStream};
use proc_macro_error::emit_warning;
use quote::quote;

use crate::common::{Case, GET_PREFIX, SET_PREFIX};

use super::method::Method;

pub struct JsAccessor {
    get: Option<Method>,
    set: Option<Method>,
}

impl JsAccessor {
    pub fn new() -> Self {
        JsAccessor {
            get: None,
            set: None,
        }
    }

    pub(crate) fn define_get(&mut self, method: Method, rename: Option<Case>) {
        if let Some(first_getter) = self.get.as_ref() {
            let first_span = first_getter.attr_span;
            emit_warning!(
                method.attr_span, "Redefined a getter for `{:?}`.", method.name(rename);
                hint = first_span => "Getter first defined here."
            );
        }
        self.get = Some(method);
    }

    pub(crate) fn define_set(&mut self, method: Method, rename: Option<Case>) {
        if let Some(first_setter) = self.set.as_ref() {
            let first_span = first_setter.attr_span;
            emit_warning!(
                method.attr_span, "Redefined a setter for `{:?}`.", method.name(rename);
                hint = first_span => "Setter first defined here."
            );
        }
        self.set = Some(method);
    }

    pub fn expand_impl(&self) -> TokenStream {
        let mut res = TokenStream::new();
        if let Some(ref x) = self.get {
            res.extend(x.expand_impl());
        }
        if let Some(ref x) = self.set {
            res.extend(x.expand_impl());
        }
        res
    }

    pub fn expand_js_impl(&self, lib_crate: &Ident) -> TokenStream {
        let mut res = TokenStream::new();
        if let Some(ref g) = self.get {
            res.extend(g.expand_js_impl(GET_PREFIX, lib_crate));
        }
        if let Some(ref s) = self.set {
            res.extend(s.expand_js_impl(SET_PREFIX, lib_crate));
        }
        res
    }

    pub fn expand_apply_to_proto(&self, lib_crate: &Ident, case: Option<Case>) -> TokenStream {
        match (self.get.as_ref(), self.set.as_ref()) {
            (Some(get), Some(set)) => {
                let configurable = get.config.configurable || set.config.configurable;
                let enumerable = get.config.enumerable || set.config.enumerable;

                let name = get.name(case);

                let configurable = configurable
                    .then(|| quote!(.configurable()))
                    .unwrap_or_default();
                let enumerable = enumerable
                    .then(|| quote!(.enumerable()))
                    .unwrap_or_default();

                let get_name = get.function.expand_carry_type_name(GET_PREFIX);
                let set_name = set.function.expand_carry_type_name(SET_PREFIX);
                quote! {_proto.prop(#name,
                        #lib_crate::object::Accessor::new(#get_name,#set_name)
                        #configurable
                        #enumerable
                )?;}
            }
            (Some(get), None) => {
                let configurable = get.config.configurable;
                let enumerable = get.config.enumerable;

                let name = get.name(case);

                let configurable = configurable
                    .then(|| quote!(.configurable()))
                    .unwrap_or_default();
                let enumerable = enumerable
                    .then(|| quote!(.enumerable()))
                    .unwrap_or_default();

                let get_name = get.function.expand_carry_type_name(GET_PREFIX);
                quote! {_proto.prop(#name,
                        #lib_crate::object::Accessor::new_get(#get_name)
                        #configurable
                        #enumerable
                )?;}
            }
            (None, Some(set)) => {
                let configurable = set.config.configurable;
                let enumerable = set.config.enumerable;

                let name = set.name(case);

                let configurable = configurable
                    .then(|| quote!(.configurable()))
                    .unwrap_or_default();
                let enumerable = enumerable
                    .then(|| quote!(.enumerable()))
                    .unwrap_or_default();

                let set_name = set.function.expand_carry_type_name(GET_PREFIX);
                quote! {_proto.prop(#name,
                        #lib_crate::object::Accessor::new_set(#set_name)
                        #configurable
                        #enumerable
                )?;}
            }
            (None, None) => TokenStream::new(),
        }
    }
}