use crate::{imp::*, prelude::*};
#[derive(Debug, FromMeta)]
pub struct List {
#[darling(default, skip)]
pub(crate) def: Def,
#[darling(default)]
pub(crate) name: Option<LitStr>,
pub(crate) item: Item,
#[darling(default)]
pub(crate) ty: Type,
#[darling(default)]
pub(crate) traits: TraitBuilder,
}
impl HasDef for List {
fn def(&self) -> &Def {
&self.def
}
}
impl ValidateNode for List {
fn validate(&self) -> Result<(), DarlingError> {
self.validate_traits()?;
self.item.validate()?;
Ok(())
}
}
impl HasSchema for List {
fn schema_node_kind() -> SchemaNodeKind {
SchemaNodeKind::List
}
}
impl HasSchemaPart for List {
fn schema_part(&self, paths: &CratePaths) -> TokenStream {
let model = &paths.model;
let def = self.def.schema_part(paths);
let name = self.current_name_literal(self.name.as_ref());
let item = self.item.schema_part(paths);
let ty = self.ty.schema_part(paths);
quote! {
#model::node::List::new(#def, #name, #item, #ty)
}
}
}
impl_collection_has_traits!(List, List);
impl HasType for List {
fn type_part(&self, paths: &CratePaths) -> TokenStream {
let ident = self.def.ident();
let item = &self.item.type_expr(paths);
quote! {
#[repr(transparent)]
pub struct #ident(pub ::std::vec::Vec<#item>);
}
}
}
impl List {
pub(crate) fn generate_tokens(&self, paths: &CratePaths) -> TokenStream {
let mut tokens = TokenStream::new();
let base = self.all_tokens(paths);
let typed_adapter = crate::node::typed_adapter::list_adapter_tokens(self, paths);
tokens.extend(quote! {
#base
#typed_adapter
});
tokens
}
}
#[cfg(test)]
mod tests {
use super::List;
use crate::prelude::*;
use darling::{FromMeta, ast::NestedMeta};
use quote::quote;
#[test]
fn shape_baseline_precedes_trait_removal() {
let args = NestedMeta::parse_meta_list(quote!(item(prim = "Nat8"), traits(remove(Deref))))
.expect("list args should parse");
let mut node = List::from_list(&args).expect("list should lower");
node.def = Def::new(
syn::parse2(quote!(
pub struct WithoutDeref {}
))
.expect("list input should parse as a struct"),
crate::crate_path::CratePaths::test(),
);
node.validate().expect("shape trait should be removable");
let traits = node.traits();
assert!(!traits.contains(&TraitKind::Deref));
assert!(traits.contains(&TraitKind::DerefMut));
assert!(traits.contains(&TraitKind::From));
assert!(traits.contains(&TraitKind::FromIterator));
assert!(traits.contains(&TraitKind::IntoIterator));
}
}