100 lines
4.0 KiB
Rust
100 lines
4.0 KiB
Rust
use proc_macro::TokenStream;
|
|
use quote::quote;
|
|
use syn::ItemImpl;
|
|
|
|
use crate::{
|
|
args::{self, RenameTarget},
|
|
utils::{get_crate_name, get_rustdoc, get_type_path_and_name, visible_fn, GeneratorResult},
|
|
};
|
|
|
|
pub fn generate(
|
|
scalar_args: &args::Scalar,
|
|
item_impl: &mut ItemImpl,
|
|
) -> GeneratorResult<TokenStream> {
|
|
let crate_name = get_crate_name(scalar_args.internal);
|
|
let self_name = get_type_path_and_name(item_impl.self_ty.as_ref())?.1;
|
|
let gql_typename = scalar_args
|
|
.name
|
|
.clone()
|
|
.unwrap_or_else(|| RenameTarget::Type.rename(self_name.clone()));
|
|
|
|
let desc = if scalar_args.use_type_description {
|
|
quote! { ::std::option::Option::Some(<Self as #crate_name::Description>::description()) }
|
|
} else {
|
|
get_rustdoc(&item_impl.attrs)?
|
|
.map(|s| quote!(::std::option::Option::Some(#s)))
|
|
.unwrap_or_else(|| quote!(::std::option::Option::None))
|
|
};
|
|
|
|
let self_ty = &item_impl.self_ty;
|
|
let generic = &item_impl.generics;
|
|
let where_clause = &item_impl.generics.where_clause;
|
|
let visible = visible_fn(&scalar_args.visible);
|
|
let specified_by_url = match &scalar_args.specified_by_url {
|
|
Some(specified_by_url) => quote! { ::std::option::Option::Some(#specified_by_url) },
|
|
None => quote! { ::std::option::Option::None },
|
|
};
|
|
|
|
let expanded = quote! {
|
|
#item_impl
|
|
|
|
#[allow(clippy::all, clippy::pedantic)]
|
|
impl #generic #crate_name::InputType for #self_ty #where_clause {
|
|
type RawValueType = Self;
|
|
|
|
fn type_name() -> ::std::borrow::Cow<'static, ::std::primitive::str> {
|
|
::std::borrow::Cow::Borrowed(#gql_typename)
|
|
}
|
|
|
|
fn create_type_info(registry: &mut #crate_name::registry::Registry) -> ::std::string::String {
|
|
registry.create_input_type::<#self_ty, _>(#crate_name::registry::MetaTypeId::Scalar, |_| #crate_name::registry::MetaType::Scalar {
|
|
name: ::std::borrow::ToOwned::to_owned(#gql_typename),
|
|
description: #desc,
|
|
is_valid: |value| <#self_ty as #crate_name::ScalarType>::is_valid(value),
|
|
visible: #visible,
|
|
specified_by_url: #specified_by_url,
|
|
})
|
|
}
|
|
|
|
fn parse(value: ::std::option::Option<#crate_name::Value>) -> #crate_name::InputValueResult<Self> {
|
|
<#self_ty as #crate_name::ScalarType>::parse(value.unwrap_or_default())
|
|
}
|
|
|
|
fn to_value(&self) -> #crate_name::Value {
|
|
<#self_ty as #crate_name::ScalarType>::to_value(self)
|
|
}
|
|
|
|
fn as_raw_value(&self) -> ::std::option::Option<&Self::RawValueType> {
|
|
::std::option::Option::Some(self)
|
|
}
|
|
}
|
|
|
|
#[allow(clippy::all, clippy::pedantic)]
|
|
#[#crate_name::async_trait::async_trait]
|
|
impl #generic #crate_name::OutputType for #self_ty #where_clause {
|
|
fn type_name() -> ::std::borrow::Cow<'static, ::std::primitive::str> {
|
|
::std::borrow::Cow::Borrowed(#gql_typename)
|
|
}
|
|
|
|
fn create_type_info(registry: &mut #crate_name::registry::Registry) -> ::std::string::String {
|
|
registry.create_output_type::<#self_ty, _>(#crate_name::registry::MetaTypeId::Scalar, |_| #crate_name::registry::MetaType::Scalar {
|
|
name: ::std::borrow::ToOwned::to_owned(#gql_typename),
|
|
description: #desc,
|
|
is_valid: |value| <#self_ty as #crate_name::ScalarType>::is_valid(value),
|
|
visible: #visible,
|
|
specified_by_url: #specified_by_url,
|
|
})
|
|
}
|
|
|
|
async fn resolve(
|
|
&self,
|
|
_: &#crate_name::ContextSelectionSet<'_>,
|
|
_field: &#crate_name::Positioned<#crate_name::parser::types::Field>
|
|
) -> #crate_name::ServerResult<#crate_name::Value> {
|
|
::std::result::Result::Ok(#crate_name::ScalarType::to_value(self))
|
|
}
|
|
}
|
|
};
|
|
Ok(expanded.into())
|
|
}
|