use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, DeriveInput};
#[proc_macro_attribute]
pub fn derive_stack_error(_attr: TokenStream, item: TokenStream) -> TokenStream {
let input = parse_macro_input!(item as DeriveInput);
let name = &input.ident;
let first_field_type = if let syn::Data::Struct(data) = &input.data {
if let Some(field) = data.fields.iter().next() {
&field.ty
} else {
panic!("Expected at least one field");
}
} else {
panic!("Expected a struct");
};
let expanded = quote! {
#input
impl #name {
pub fn new() -> Self {
Self(#first_field_type::new())
}
pub fn from_msg(error: impl std::fmt::Display + Send + Sync + 'static) -> Self {
Self(#first_field_type::from_msg(error))
}
}
impl ErrorStacks<ErrorCode> for #name {
fn err_code(&self) -> Option<&ErrorCode> {
self.0.err_code()
}
fn with_err_code(self, code: ErrorCode) -> Self {
Self(self.0.with_err_code(code))
}
fn with_no_err_code(self) -> Self {
Self(self.0.with_no_err_code())
}
fn err_uri(&self) -> Option<&str> {
self.0.err_uri()
}
fn with_err_uri(self, uri: String) -> Self {
Self(self.0.with_err_uri(uri))
}
fn with_no_err_uri(self) -> Self {
Self(self.0.with_no_err_uri())
}
fn with_err_msg(self, error: impl std::fmt::Display + Send + Sync + 'static) -> Self {
Self(self.0.with_err_msg(error))
}
fn with_no_err_msg(self) -> Self {
Self(self.0.with_no_err_msg())
}
fn stack_err(self) -> Self {
Self(self.0.stack_err())
}
fn stack_err_msg(self, error: impl std::fmt::Display + Send + Sync + 'static) -> Self {
Self(self.0.stack_err_msg(error))
}
}
impl std::fmt::Display for #name {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
std::fmt::Display::fmt(&self.0, f)
}
}
impl std::fmt::Debug for #name {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
std::fmt::Debug::fmt(&self.0, f)
}
}
impl std::error::Error for #name {
fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
self.0.source()
}
}
};
TokenStream::from(expanded)
}