Implement PartialEq for proc_macro::Ident == strings

Author: dtolnay

compiler/rustc_expand/src/proc_macro_server.rs

@@ -513,6 +513,9 @@ impl server::Ident for Rustc<'_> {
     fn new(&mut self, string: &str, span: Self::Span, is_raw: bool) -> Self::Ident {
         Ident::new(self.sess, Symbol::intern(string), is_raw, span)
     }
+    fn eq(&mut self, ident: Self::Ident, rhs: &str) -> bool {
+        ident.sym.as_str() == rhs
+    }
     fn span(&mut self, ident: Self::Ident) -> Self::Span {
         ident.span
     }

library/proc_macro/src/bridge/mod.rs

@@ -101,6 +101,7 @@ macro_rules! with_api {
             },
             Ident {
                 fn new(string: &str, span: $S::Span, is_raw: bool) -> $S::Ident;
+                fn eq($self: $S::Ident, rhs: &str) -> bool;
                 fn span($self: $S::Ident) -> $S::Span;
                 fn with_span($self: $S::Ident, span: $S::Span) -> $S::Ident;
             },

library/proc_macro/src/lib.rs

@@ -923,6 +923,31 @@ impl fmt::Debug for Ident {
     }
 }
 
+#[stable(feature = "proc_macro_ident_eq", since = "1.49.0")]
+impl PartialEq<str> for Ident {
+    fn eq(&self, rhs: &str) -> bool {
+        self.0.eq(rhs)
+    }
+}
+
+#[stable(feature = "proc_macro_ident_eq", since = "1.49.0")]
+impl PartialEq<String> for Ident {
+    fn eq(&self, rhs: &String) -> bool {
+        <Self as PartialEq<str>>::eq(self, rhs.as_str())
+    }
+}
+
+#[stable(feature = "proc_macro_ident_eq", since = "1.49.0")]
+impl<'a, T> PartialEq<&'a T> for Ident
+where
+    T: ?Sized,
+    Ident: PartialEq<T>,
+{
+    fn eq(&self, rhs: &&T) -> bool {
+        <Self as PartialEq<T>>::eq(self, *rhs)
+    }
+}
+
 /// A literal string (`"hello"`), byte string (`b"hello"`),
 /// character (`'a'`), byte character (`b'a'`), an integer or floating point number
 /// with or without a suffix (`1`, `1u8`, `2.3`, `2.3f32`).