bitwarden_error_macro/flat/
attribute.rs

1use quote::quote;
2use syn::Data;
3
4pub(crate) fn bitwarden_error_flat(
5    input: &syn::DeriveInput,
6    type_identifier: &proc_macro2::Ident,
7    export_as_identifier: &proc_macro2::Ident,
8) -> proc_macro::TokenStream {
9    match &input.data {
10        Data::Enum(data) => {
11            let variant_names = data.variants.iter().map(|variant| &variant.ident);
12            let match_arms = data.variants.iter().map(|variant| {
13                let variant_ident = &variant.ident;
14                let variant_str = variant_ident.to_string();
15
16                match variant.fields {
17                    syn::Fields::Unit => {
18                        quote! {
19                            #type_identifier::#variant_ident => #variant_str
20                        }
21                    }
22                    syn::Fields::Named(_) => {
23                        quote! {
24                            #type_identifier::#variant_ident { .. } => #variant_str
25                        }
26                    }
27                    syn::Fields::Unnamed(_) => {
28                        quote! {
29                            #type_identifier::#variant_ident(..) => #variant_str
30                        }
31                    }
32                }
33            });
34
35            let wasm = cfg!(feature = "wasm").then(|| {
36                flat_error_wasm(
37                    type_identifier,
38                    export_as_identifier,
39                    &variant_names.collect::<Vec<_>>(),
40                )
41            });
42
43            quote! {
44                #[cfg_attr(feature = "uniffi", derive(uniffi::Error), uniffi(flat_error))]
45                #input
46                #wasm
47
48                #[automatically_derived]
49                impl ::bitwarden_error::flat_error::FlatError for #type_identifier {
50                    fn error_variant(&self) -> &'static str {
51                        match &self {
52                            #(#match_arms), *
53                        }
54                    }
55                }
56            }
57            .into()
58        }
59        _ => syn::Error::new_spanned(input, "bitwarden_error can only be used with enums")
60            .to_compile_error()
61            .into(),
62    }
63}
64
65fn flat_error_wasm(
66    type_identifier: &proc_macro2::Ident,
67    export_as_identifier: &proc_macro2::Ident,
68    variant_names: &[&proc_macro2::Ident],
69) -> proc_macro2::TokenStream {
70    let export_as_identifier_str = export_as_identifier.to_string();
71    let is_error_function_name = format!("is{export_as_identifier}");
72    let ts_variant_names = variant_names
73        .iter()
74        .map(|vn| format!(r#""{vn}""#))
75        .collect::<Vec<String>>()
76        .join("|");
77    let ts_code_str = format!(
78        r##"r#"
79            export interface {export_as_identifier_str} extends Error {{
80                name: "{export_as_identifier_str}";
81                variant: {ts_variant_names};
82            }};
83
84            export function {is_error_function_name}(error: any): error is {export_as_identifier_str};
85        "#"##,
86    );
87    let ts_code: proc_macro2::TokenStream = ts_code_str
88        .parse()
89        .expect("Could not generate TypeScript code");
90
91    quote! {
92        const _: () = {
93            use bitwarden_error::wasm_bindgen::prelude::*;
94            use bitwarden_error::wasm_bindgen as wasm_bindgen;
95
96            #[wasm_bindgen(typescript_custom_section)]
97            const TS_APPEND_CONTENT: &'static str = #ts_code;
98
99            #[wasm_bindgen(js_name = #is_error_function_name, skip_typescript)]
100            pub fn is_error(error: &JsValue) -> bool {
101                let name_js_value = ::bitwarden_error::js_sys::Reflect::get(&error, &JsValue::from_str("name")).unwrap_or(JsValue::NULL);
102                let name = name_js_value.as_string().unwrap_or_default();
103                name == #export_as_identifier_str
104            }
105
106            #[automatically_derived]
107            impl From<#type_identifier> for JsValue {
108                fn from(error: #type_identifier) -> Self {
109                    let js_error = ::bitwarden_error::wasm::SdkJsError::new(error.to_string());
110                    js_error.set_name(#export_as_identifier_str.to_owned());
111                    js_error.set_variant(::bitwarden_error::flat_error::FlatError::error_variant(&error).to_owned());
112                    js_error.into()
113                }
114            }
115        };
116    }
117}