aboutsummaryrefslogtreecommitdiff
path: root/mingling_macros/src/func/pack_err_structural.rs
blob: 7d3a6f80b0ba93b9c151e9f8510793d85faf0fe6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
use just_fmt::snake_case;
use proc_macro::TokenStream;
use quote::quote;
use syn::{Ident, Token, Type, parse_macro_input};

/// `pack_err_structural!` — like `pack_err!` but also marks the type as
/// supporting structured output via `StructuralData`.
pub(crate) fn pack_err_structural(input: TokenStream) -> TokenStream {
    let parsed = parse_macro_input!(input as PackErrInput);

    let type_name = match &parsed {
        PackErrInput::Simple { type_name } => type_name.clone(),
        PackErrInput::Typed { type_name, .. } => type_name.clone(),
    };

    // Register in STRUCTURED_TYPES
    let type_name_str = type_name.to_string();
    crate::get_global_set(&crate::STRUCTURED_TYPES)
        .lock()
        .unwrap()
        .insert(type_name_str);

    let structural_data = quote! {
        impl ::mingling::__private::StructuralDataSealed<crate::ThisProgram> for #type_name {}
        impl ::mingling::__private::StructuralData<crate::ThisProgram> for #type_name {}
    };

    // Generate the struct + impls (same as pack_err! but with Serialize derive + sealed)
    match parsed {
        PackErrInput::Simple { type_name } => {
            let name_str = type_name.to_string();
            let snake_name = snake_case!(&name_str);

            let expanded = quote! {
                #[derive(::mingling::Grouped, ::serde::Serialize)]
                pub struct #type_name {
                    /// The snake_case name of this error, automatically set at compile time.
                    pub name: String,
                }

                impl ::std::default::Default for #type_name {
                    fn default() -> Self {
                        Self {
                            name: #snake_name.into(),
                        }
                    }
                }

                ::mingling::macros::register_type!(#type_name);

                #structural_data
            };

            expanded.into()
        }
        PackErrInput::Typed {
            type_name,
            inner_type,
        } => {
            let name_str = type_name.to_string();
            let snake_name = snake_case!(&name_str);

            let expanded = quote! {
                #[derive(::mingling::Grouped, ::serde::Serialize)]
                pub struct #type_name {
                    /// The snake_case name of this error, automatically set at compile time.
                    pub name: String,
                    /// Additional context info for this error.
                    pub info: #inner_type,
                }

                impl #type_name {
                    /// Creates a new error with the given info.
                    /// The `name` field is automatically set to the snake_case of the struct name.
                    pub fn new(info: #inner_type) -> Self {
                        Self {
                            name: #snake_name.into(),
                            info,
                        }
                    }
                }

                ::mingling::macros::register_type!(#type_name);

                #structural_data
            };

            expanded.into()
        }
    }
}

// Re-use pack_err's input parser
enum PackErrInput {
    Simple {
        type_name: Ident,
    },
    Typed {
        type_name: Ident,
        inner_type: Box<Type>,
    },
}

impl syn::parse::Parse for PackErrInput {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        let type_name: Ident = input.parse()?;

        if input.peek(Token![=]) {
            input.parse::<Token![=]>()?;
            let inner_type: Type = input.parse()?;
            Ok(PackErrInput::Typed {
                type_name,
                inner_type: Box::new(inner_type),
            })
        } else {
            Ok(PackErrInput::Simple { type_name })
        }
    }
}