feat: add base response (#6)

Author: DanielleHuisman

packages/breach-macros/src/http/attribute.rs

@@ -1,11 +1,12 @@
 use proc_macro2::TokenStream;
 use quote::quote;
-use syn::{Attribute, Error, Result, spanned::Spanned};
+use syn::{Attribute, Error, Result, Type, spanned::Spanned};
 
 use crate::status::Status;
 
 pub struct HttpErrorAttribute {
     pub status: Option<Status>,
+    pub base: Option<Type>,
     pub axum: bool,
     pub utoipa: bool,
 }
@@ -34,13 +35,18 @@ impl<'a> HttpErrorAttribute {
 
     pub fn parse(attribute: &'a Attribute) -> Result<Self> {
         let mut status = None;
+        let mut base = None;
         let mut axum = false;
         let mut utoipa = false;
 
         attribute.parse_nested_meta(|meta| {
             if meta.path.is_ident("status") {
                 status = Some(meta.value()?.parse()?);
 
+                Ok(())
+            } else if meta.path.is_ident("base") {
+                base = Some(meta.value()?.parse()?);
+
                 Ok(())
             } else if meta.path.is_ident("axum") {
                 axum = true;
@@ -57,6 +63,7 @@ impl<'a> HttpErrorAttribute {
 
         Ok(Self {
             status,
+            base,
             axum,
             utoipa,
         })

packages/breach-macros/src/http/enum.rs

@@ -42,10 +42,15 @@ impl<'a> HttpErrorEnum<'a> {
     }
 
     pub fn responses(&self) -> TokenStream {
-        let mut responses = self
-            .variants
-            .iter()
-            .map(|variant| variant.responses())
+        let base = self
+            .attribute
+            .as_ref()
+            .and_then(|attribute| attribute.base.as_ref())
+            .map(|r#type| quote!(<#r#type as ::utoipa::IntoResponses>::responses()));
+
+        let mut responses = base
+            .into_iter()
+            .chain(self.variants.iter().map(|variant| variant.responses()))
             .collect::<Vec<_>>();
 
         if responses.is_empty() {