Disable clippy on generated modules

golemcloud · Mar 20, 2024 · 7894809 · 7894809
1 parent c6d8d2e
commit 7894809
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/src/rust/lib_gen.rs b/src/rust/lib_gen.rs
@@ -32,28 +32,51 @@ impl Verbosity {
     }
 }
 
+#[derive(Debug, Clone, PartialEq, Eq, Ord, PartialOrd)]
+pub enum Attribute {
+    DenyClippy,
+}
+
+impl Attribute {
+    fn code(&self) -> RustPrinter {
+        match self {
+            Attribute::DenyClippy => line(unit() + "#![deny(clippy)]"),
+        }
+    }
+}
+
 #[derive(Debug, Clone, PartialEq, Eq, Ord, PartialOrd)]
 pub struct ModuleName {
     name: String,
     verbosity: Verbosity,
+    attributes: Vec<Attribute>,
 }
 
 impl ModuleName {
     fn code(&self) -> RustPrinter {
-        line(unit() + self.verbosity.render() + "mod " + escape_keywords(&self.name) + ";")
+        unit()
+            + self
+                .attributes
+                .iter()
+                .map(|a| a.code())
+                .reduce(|acc, e| acc + line(e))
+                .unwrap_or_else(unit)
+            + line(unit() + self.verbosity.render() + "mod " + escape_keywords(&self.name) + ";")
     }
 
     pub fn new<S: Into<String>>(s: S) -> ModuleName {
         ModuleName {
             name: s.into(),
             verbosity: Verbosity::Default,
+            attributes: vec![Attribute::DenyClippy],
         }
     }
 
     pub fn new_pub<S: Into<String>>(s: S) -> ModuleName {
         ModuleName {
             name: s.into(),
             verbosity: Verbosity::Pub,
+            attributes: vec![Attribute::DenyClippy],
         }
     }
 
@@ -136,7 +159,9 @@ mod tests {
         );
 
         let expected = indoc! { r#"
+            #![deny(clippy)]
             mod abc;
+            #![deny(clippy)]
             mod xyz;
 
             pub use lib::abc::B;