Remove old define_violation! (in favor of #[violation]) (#3310)

crates/ruff_macros/src/violation.rs

@@ -0,0 +1,63 @@
+use proc_macro2::TokenStream;
+use quote::quote;
+use syn::{Attribute, Error, ItemStruct, Lit, LitStr, Meta, Result};
+
+fn parse_attr<const LEN: usize>(path: [&'static str; LEN], attr: &Attribute) -> Option<LitStr> {
+    if let Meta::NameValue(name_value) = attr.parse_meta().ok()? {
+        let path_idents = name_value
+            .path
+            .segments
+            .iter()
+            .map(|segment| &segment.ident);
+
+        if itertools::equal(path_idents, path) {
+            if let Lit::Str(lit) = name_value.lit {
+                return Some(lit);
+            }
+        }
+    }
+
+    None
+}
+
+/// Collect all doc comment attributes into a string
+fn get_docs(attrs: &[Attribute]) -> Result<String> {
+    let mut explanation = String::new();
+    for attr in attrs {
+        if attr.path.is_ident("doc") {
+            if let Some(lit) = parse_attr(["doc"], attr) {
+                let value = lit.value();
+                // `/// ` adds
+                let line = value.strip_prefix(' ').unwrap_or(&value);
+                explanation.push_str(line);
+                explanation.push('\n');
+            } else {
+                return Err(Error::new_spanned(attr, "unimplemented doc comment style"));
+            }
+        }
+    }
+    Ok(explanation)
+}
+
+pub fn violation(violation: &ItemStruct) -> Result<TokenStream> {
+    let ident = &violation.ident;
+    let explanation = get_docs(&violation.attrs)?;
+    let violation = if explanation.trim().is_empty() {
+        quote! {
+            #[derive(Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
+            #violation
+        }
+    } else {
+        quote! {
+            #[derive(Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
+            #violation
+
+            impl #ident {
+                pub fn explanation() -> Option<&'static str> {
+                    Some(#explanation)
+                }
+            }
+        }
+    };
+    Ok(violation)
+}