mod config;
mod item;
#[cfg(test)]
mod tests;
pub use self::{
config::TraitDefinitionConfig,
item::{
InkItemTrait,
InkTraitItem,
InkTraitMessage,
IterInkTraitItems,
},
};
use super::attrs::InkAttribute;
use proc_macro2::TokenStream as TokenStream2;
use syn::Result;
#[derive(Debug, PartialEq, Eq)]
pub struct InkTraitDefinition {
config: TraitDefinitionConfig,
item: InkItemTrait,
}
impl InkTraitDefinition {
pub fn new(config: TokenStream2, input: TokenStream2) -> Result<Self> {
let parsed_config = syn::parse2::<crate::ast::AttributeArgs>(config)?;
let parsed_item = syn::parse2::<syn::ItemTrait>(input)?;
let config = TraitDefinitionConfig::try_from(parsed_config)?;
let item = InkItemTrait::new(&config, parsed_item)?;
Ok(Self { config, item })
}
#[cfg(test)]
pub fn from_raw_parts(config: TraitDefinitionConfig, item: InkItemTrait) -> Self {
Self { config, item }
}
}
impl InkTraitDefinition {
pub fn config(&self) -> &TraitDefinitionConfig {
&self.config
}
pub fn item(&self) -> &InkItemTrait {
&self.item
}
}