Skip to content

Commit

Permalink
avoid requiring struct/enum definitions to constrain any type params
Browse files Browse the repository at this point in the history
- add doctest and FAQ
- clarify that if fields impl Debug, they don't need to impl Display
- make pedantic spelling change
- add note on how destructuring is used for format strings
- clarify the mechanism of adding Display only when otherwise unconstrained and add TODO
  • Loading branch information
cosmicexplorer committed Oct 8, 2021
1 parent 2d321d7 commit f6433f4
Show file tree
Hide file tree
Showing 3 changed files with 345 additions and 13 deletions.
2 changes: 1 addition & 1 deletion src/attr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ impl AttrsHelper {
_ => unimplemented!(),
};

// Make an attempt and cleaning up multiline doc comments
// Make an attempt at cleaning up multiline doc comments.
let doc_str = lit
.value()
.lines()
Expand Down
274 changes: 272 additions & 2 deletions src/expand.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,15 @@
use super::attr::AttrsHelper;
use proc_macro2::TokenStream;
use proc_macro2::{Span, TokenStream};
use quote::{format_ident, quote};
use syn::{Data, DataEnum, DataStruct, DeriveInput, Error, Fields, Result};
use syn::{
punctuated::Punctuated,
token::{Add, Colon, Colon2, Comma, Where},
Data, DataEnum, DataStruct, DeriveInput, Error, Fields, Generics, Ident, Path, PathArguments,
PathSegment, PredicateType, Result, TraitBound, TraitBoundModifier, Type, TypeParam,
TypeParamBound, TypePath, WhereClause, WherePredicate,
};

use std::collections::HashMap;

pub(crate) fn derive(input: &DeriveInput) -> Result<TokenStream> {
let impls = match &input.data {
Expand Down Expand Up @@ -66,6 +74,7 @@ fn specialization() -> TokenStream {
fn impl_struct(input: &DeriveInput, data: &DataStruct) -> Result<TokenStream> {
let ty = &input.ident;
let (impl_generics, ty_generics, where_clause) = input.generics.split_for_impl();
let where_clause = generate_where_clause(&input.generics, where_clause);

let helper = AttrsHelper::new(&input.attrs);

Expand All @@ -84,6 +93,10 @@ fn impl_struct(input: &DeriveInput, data: &DataStruct) -> Result<TokenStream> {
quote! {
impl #impl_generics core::fmt::Display for #ty #ty_generics #where_clause {
fn fmt(&self, formatter: &mut core::fmt::Formatter) -> core::fmt::Result {
// NB: This destructures the fields of `self` into named variables (for unnamed
// fields, it uses _0, _1, etc as above). The `#[allow(unused_variables)]`
// section means it doesn't have to parse the individual field references out of
// the docstring.
#[allow(unused_variables)]
let #pat = self;
#display
Expand All @@ -95,9 +108,266 @@ fn impl_struct(input: &DeriveInput, data: &DataStruct) -> Result<TokenStream> {
Ok(quote! { #display })
}

/// Create a `where` predicate for `ident`, without any [bound][TypeParamBound]s yet.
fn new_empty_where_type_predicate(ident: Ident) -> PredicateType {
let mut path_segments = Punctuated::<PathSegment, Colon2>::new();
path_segments.push_value(PathSegment {
ident,
arguments: PathArguments::None,
});
PredicateType {
lifetimes: None,
bounded_ty: Type::Path(TypePath {
qself: None,
path: Path {
leading_colon: None,
segments: path_segments,
},
}),
colon_token: Colon {
spans: [Span::call_site()],
},
bounds: Punctuated::<TypeParamBound, Add>::new(),
}
}

/// Create a `where` clause that we can add [WherePredicate]s to.
fn new_empty_where_clause() -> WhereClause {
WhereClause {
where_token: Where {
span: Span::call_site(),
},
predicates: Punctuated::<WherePredicate, Comma>::new(),
}
}

enum UseGlobalPrefix {
LeadingColon,
#[allow(dead_code)]
NoLeadingColon,
}

/// Create a path with segments composed of [Idents] *without* any [PathArguments].
fn join_paths(name_segments: &[&str], use_global_prefix: UseGlobalPrefix) -> Path {
let mut segments = Punctuated::<PathSegment, Colon2>::new();
assert!(name_segments.len() >= 1);
segments.push_value(PathSegment {
ident: Ident::new(name_segments[0], Span::call_site()),
arguments: PathArguments::None,
});
for name in name_segments[1..].iter() {
segments.push_punct(Colon2 {
spans: [Span::call_site(), Span::mixed_site()],
});
segments.push_value(PathSegment {
ident: Ident::new(name, Span::call_site()),
arguments: PathArguments::None,
});
}
Path {
leading_colon: match use_global_prefix {
UseGlobalPrefix::LeadingColon => Some(Colon2 {
spans: [Span::call_site(), Span::mixed_site()],
}),
UseGlobalPrefix::NoLeadingColon => None,
},
segments,
}
}

/// Push `new_type_predicate` onto the end of `where_clause`.
fn append_where_clause_type_predicate(
where_clause: &mut WhereClause,
new_type_predicate: PredicateType,
) {
// Push a comma at the end if there are already any `where` predicates.
if !where_clause.predicates.is_empty() {
where_clause.predicates.push_punct(Comma {
spans: [Span::call_site()],
});
}
where_clause
.predicates
.push_value(WherePredicate::Type(new_type_predicate));
}

/// Add a requirement for [core::fmt::Display] to a `where` predicate for some type.
fn add_display_constraint_to_type_predicate(
predicate_that_needs_a_display_impl: &mut PredicateType,
) {
// Create a `Path` of `::core::fmt::Display`.
let display_path = join_paths(&["core", "fmt", "Display"], UseGlobalPrefix::LeadingColon);

let display_bound = TypeParamBound::Trait(TraitBound {
paren_token: None,
modifier: TraitBoundModifier::None,
lifetimes: None,
path: display_path,
});
if !predicate_that_needs_a_display_impl.bounds.is_empty() {
predicate_that_needs_a_display_impl.bounds.push_punct(Add {
spans: [Span::call_site()],
});
}

predicate_that_needs_a_display_impl
.bounds
.push_value(display_bound);
}

/// Map each declared generic type parameter to the set of all trait boundaries declared on it.
///
/// These boundaries may come from the declaration site:
/// pub enum E<T: MyTrait> { ... }
/// or a `where` clause after the parameter declarations:
/// pub enum E<T> where T: MyTrait { ... }
/// This method will return the boundaries from both of those cases.
fn extract_trait_constraints_from_source(
where_clause: &WhereClause,
type_params: &[&TypeParam],
) -> HashMap<Ident, Vec<TraitBound>> {
// Add trait bounds provided at the declaration site of type parameters for the struct/enum.
let mut param_constraint_mapping: HashMap<Ident, Vec<TraitBound>> = type_params
.iter()
.map(|type_param| {
let trait_bounds: Vec<TraitBound> = type_param
.bounds
.iter()
.flat_map(|bound| match bound {
TypeParamBound::Trait(trait_bound) => Some(trait_bound),
_ => None,
})
.cloned()
.collect();
(type_param.ident.clone(), trait_bounds)
})
.collect();

// Add trait bounds from `where` clauses, which may be type parameters or types containing
// those parameters.
for predicate in where_clause.predicates.iter() {
match predicate {
WherePredicate::Type(ref pred_ty) => {
let ident = match &pred_ty.bounded_ty {
Type::Path(TypePath { path, qself: None }) => match path.get_ident() {
None => continue,
Some(ident) => ident,
},
_ => continue,
};
// We ignore any type constraints that aren't direct references to type
// parameters of the current enum of struct definition. No types can be
// constrained in a `where` clause unless they are a type parameter or a generic
// type instantiated with one of the type parameters, so by only allowing single
// identifiers, we can be sure that the constrained type is a type parameter
// that is contained in `param_constraint_mapping`.
if let Some((_, ref mut known_bounds)) = param_constraint_mapping
.iter_mut()
.find(|(id, _)| *id == ident)
{
for bound in pred_ty.bounds.iter() {
match bound {
TypeParamBound::Trait(ref bound) => {
known_bounds.push(bound.clone());
}
// We only care about trait bounds here.
_ => (),
}
}
}
}
// We only care about type and not lifetime constraints here.
_ => (),
}
}

param_constraint_mapping
}

/// Hygienically add `where _: Display` to the set of [TypeParamBound]s for `ident`, creating such
/// a set if necessary.
fn ensure_display_in_where_clause_for_type(where_clause: &mut WhereClause, ident: Ident) {
for pred_ty in where_clause
.predicates
.iter_mut()
// Find the `where` predicate constraining the current type param, if it exists.
.flat_map(|predicate| match predicate {
WherePredicate::Type(pred_ty) => Some(pred_ty),
// We're looking through type constraints, not lifetime constraints.
_ => None,
})
{
// Do a complicated destructuring in order to check if the type being constrained in this
// `where` clause is the type we're looking for, so we can use the mutable reference to
// `pred_ty` if so.
let matches_desired_type = match &pred_ty.bounded_ty {
Type::Path(TypePath { path, .. }) if Some(&ident) == path.get_ident() => true,
_ => false,
};
if matches_desired_type {
add_display_constraint_to_type_predicate(pred_ty);
return;
}
}

// If there is no `where` predicate for the current type param, we will construct one.
let mut new_type_predicate = new_empty_where_type_predicate(ident.clone());
add_display_constraint_to_type_predicate(&mut new_type_predicate);
append_where_clause_type_predicate(where_clause, new_type_predicate);
}

/// Given all declared type parameters, add a [core::fmt::Display] constraint unless the type
/// parameter already has any type constraint.
fn ensure_where_clause_has_display_for_all_unconstrained_members(
where_clause: &mut WhereClause,
type_params: &[&TypeParam],
) {
let param_constraint_mapping = extract_trait_constraints_from_source(where_clause, type_params);

for (ident, known_bounds) in param_constraint_mapping.into_iter() {
// If the type parameter has any constraints already, we don't want to touch it, to avoid
// breaking use cases where a type parameter only needs to impl `Debug`, for example.
if known_bounds.is_empty() {
ensure_display_in_where_clause_for_type(where_clause, ident);
}
}
}

/// Generate a `where` clause that ensures all generic type parameters `impl`
/// [core::fmt::Display] unless already constrained.
///
/// This approach allows struct/enum definitions deriving [crate::Display] to avoid hardcoding
/// a [core::fmt::Display] constraint into every type parameter.
///
/// If the type parameter isn't already constrained, we add a `where _: Display` clause to our
/// display implementation to expect to be able to format every enum case or struct member.
///
/// # TODO
/// This is possibly a little too coarse of a solution, since it's possible that a type
/// parameter may not be used in a way that affects whether the enum cases impl `Display`. It
/// would be nice to be able to ask rustc "does this type argument impl (e.g.)
/// `core::fmt::Display`?".
///
/// In fact, we would preferably only require `where _: Display` or `where _: Debug` where the
/// format string actually requires it. However, while [`std::fmt` defines a formal syntax for
/// `format!()`][format syntax], it *doesn't* expose the actual logic to parse the format string,
/// which appears to live in [`rustc_parse_format`]. While we use the [`syn`] crate to parse rust
/// syntax, it also doesn't currently provide any method to introspect a `format!()` string. It
/// would be nice to contribute this upstream in [`syn`].
///
/// [format syntax]: std::fmt#syntax
/// [`rustc_parse_format`]: https://doc.rust-lang.org/nightly/nightly-rustc/rustc_parse_format/index.html
fn generate_where_clause(generics: &Generics, where_clause: Option<&WhereClause>) -> WhereClause {
let mut where_clause = where_clause.cloned().unwrap_or_else(new_empty_where_clause);
let type_params: Vec<&TypeParam> = generics.type_params().collect();
ensure_where_clause_has_display_for_all_unconstrained_members(&mut where_clause, &type_params);
where_clause
}

fn impl_enum(input: &DeriveInput, data: &DataEnum) -> Result<TokenStream> {
let ty = &input.ident;
let (impl_generics, ty_generics, where_clause) = input.generics.split_for_impl();
let where_clause = generate_where_clause(&input.generics, where_clause);

let helper = AttrsHelper::new(&input.attrs);

Expand Down
Loading

0 comments on commit f6433f4

Please sign in to comment.