Pretend remote derives are not dead code

2026-04-29 01:27:56 +00:00 · 2018-05-02 13:29:44 -07:00
parent 893c0578dd
commit b78f434086
6 changed files with 152 additions and 4 deletions
@@ -0,0 +1,145 @@
+use proc_macro2::Span;
+use quote::Tokens;
+use syn::Ident;
+
+use internals::ast::{Container, Data, Field, Style};
+
+// Suppress dead_code warnings that would otherwise appear when using a remote
+// derive. Other than this pretend code, a struct annotated with remote derive
+// never has its fields referenced and an enum annotated with remote derive
+// never has its variants constructed.
+//
+//     warning: field is never used: `i`
+//      --> src/main.rs:4:20
+//       |
+//     4 | struct StructDef { i: i32 }
+//       |                    ^^^^^^
+//
+//     warning: variant is never constructed: `V`
+//      --> src/main.rs:8:16
+//       |
+//     8 | enum EnumDef { V }
+//       |                ^
+//
+pub fn pretend_used(cont: &Container) -> Tokens {
+    let pretend_fields = pretend_fields_used(cont);
+    let pretend_variants = pretend_variants_used(cont);
+
+    quote! {
+        #pretend_fields
+        #pretend_variants
+    }
+}
+
+// For structs with named fields, expands to:
+//
+//     match None::<T> {
+//         Some(T { a: ref __v0, b: ref __v1 }) => {}
+//         _ => {}
+//     }
+//
+// For enums, expands to the following but only including struct variants:
+//
+//     match None::<T> {
+//         Some(T::A { a: ref __v0 }) => {}
+//         Some(T::B { b: ref __v0 }) => {}
+//         _ => {}
+//     }
+//
+// The `ref` is important in case the user has written a Drop impl on their
+// type. Rust does not allow destructuring a struct or enum that has a Drop
+// impl.
+fn pretend_fields_used(cont: &Container) -> Tokens {
+    let type_ident = cont.ident;
+    let (_, ty_generics, _) = cont.generics.split_for_impl();
+
+    let patterns = match cont.data {
+        Data::Enum(ref variants) => {
+            variants.iter()
+                .filter_map(|variant| match variant.style {
+                    Style::Struct => {
+                        let variant_ident = variant.ident;
+                        let pat = struct_pattern(&variant.fields);
+                        Some(quote!(#type_ident::#variant_ident #pat))
+                    }
+                    _ => None,
+                })
+                .collect::<Vec<_>>()
+        }
+        Data::Struct(Style::Struct, ref fields) => {
+            let pat = struct_pattern(fields);
+            vec![quote!(#type_ident #pat)]
+        }
+        Data::Struct(_, _) => {
+            return quote!();
+        }
+    };
+
+    quote! {
+        match _serde::export::None::<#type_ident #ty_generics> {
+            #(
+                _serde::export::Some(#patterns) => {}
+            )*
+            _ => {}
+        }
+    }
+}
+
+// Expands to one of these per enum variant:
+//
+//     match None {
+//         Some((__v0, __v1,)) => {
+//             let _ = E::V { a: __v0, b: __v1 };
+//         }
+//         _ => {}
+//     }
+//
+fn pretend_variants_used(cont: &Container) -> Tokens {
+    let variants = match cont.data {
+        Data::Enum(ref variants) => variants,
+        Data::Struct(_, _) => {
+            return quote!();
+        }
+    };
+
+    let type_ident = cont.ident;
+    let (_, ty_generics, _) = cont.generics.split_for_impl();
+    let turbofish = ty_generics.as_turbofish();
+
+    let cases = variants.iter()
+        .map(|variant| {
+            let variant_ident = variant.ident;
+            let ref placeholders = (0..variant.fields.len())
+                .map(|i| Ident::new(&format!("__v{}", i), Span::call_site()))
+                .collect::<Vec<_>>();
+
+            let pat = match variant.style {
+                Style::Struct => {
+                    let names = variant.fields.iter().map(|field| field.ident);
+                    quote!({ #(#names: #placeholders),* })
+                }
+                Style::Tuple | Style::Newtype => {
+                    quote!(( #(#placeholders),* ))
+                }
+                Style::Unit => quote!(),
+            };
+
+            quote! {
+                match _serde::export::None {
+                    _serde::export::Some((#(#placeholders,)*)) => {
+                        let _ = #type_ident::#variant_ident #turbofish #pat;
+                    }
+                    _ => {}
+                }
+            }
+        });
+
+    quote!(#(#cases)*)
+}
+
+fn struct_pattern(fields: &[Field]) -> Tokens {
+    let names = fields.iter().map(|field| field.ident);
+    let placeholders = (0..fields.len())
+        .map(|i| Ident::new(&format!("__v{}", i), Span::call_site()));
+    quote!({ #(#names: ref #placeholders),* })
+}