bitwarden_error_macro/flat/
attribute.rs1use 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}