Implied bounds for the remaining std::fmt traits

dtolnay · Sep 5, 2021 · cc65053 · cc65053
1 parent 0a1c5bd
commit cc65053
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 5 deletions.
diff --git a/impl/src/attr.rs b/impl/src/attr.rs
@@ -31,10 +31,17 @@ pub struct Transparent<'a> {
     pub span: Span,
 }
 
-#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd)]
+#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Debug)]
 pub enum Trait {
     Debug,
     Display,
+    Octal,
+    LowerHex,
+    UpperHex,
+    Pointer,
+    Binary,
+    LowerExp,
+    UpperExp,
 }
 
 pub fn get(input: &[Attribute]) -> Result<Attrs> {
@@ -200,9 +207,7 @@ impl ToTokens for Display<'_> {
 
 impl ToTokens for Trait {
     fn to_tokens(&self, tokens: &mut TokenStream) {
-        tokens.extend(match self {
-            Trait::Debug => quote!(std::fmt::Debug),
-            Trait::Display => quote!(std::fmt::Display),
-        });
+        let trait_name = format_ident!("{}", format!("{:?}", self));
+        tokens.extend(quote!(std::fmt::#trait_name));
     }
 }
diff --git a/impl/src/fmt.rs b/impl/src/fmt.rs
@@ -93,6 +93,13 @@ impl Display<'_> {
                 };
                 let bound = match read[..end_spec].chars().next_back() {
                     Some('?') => Trait::Debug,
+                    Some('o') => Trait::Octal,
+                    Some('x') => Trait::LowerHex,
+                    Some('X') => Trait::UpperHex,
+                    Some('p') => Trait::Pointer,
+                    Some('b') => Trait::Binary,
+                    Some('e') => Trait::LowerExp,
+                    Some('E') => Trait::UpperExp,
                     Some(_) => Trait::Display,
                     None => {
                         has_bonus_display = true;