feat(linting): customize severity per rule (#430)

Author: juleswritescode

Cargo.lock

@@ -24,6 +24,7 @@ macro_rules! declare_lint_rule {
     ( $( #[doc = $doc:literal] )+ $vis:vis $id:ident {
         version: $version:literal,
         name: $name:tt,
+        severity: $severity:expr_2021,
         $( $key:ident: $value:expr_2021, )*
     } ) => {
 
@@ -32,6 +33,7 @@ macro_rules! declare_lint_rule {
             $vis $id {
                 version: $version,
                 name: $name,
+                severity: $severity,
                 $( $key: $value, )*
             }
         );
@@ -53,6 +55,7 @@ macro_rules! declare_rule {
         ( $( #[doc = $doc:literal] )+ $vis:vis $id:ident {
         version: $version:literal,
         name: $name:tt,
+        severity: $severity:expr_2021,
         $( $key:ident: $value:expr_2021, )*
     } ) => {
         $( #[doc = $doc] )*
@@ -61,7 +64,7 @@ macro_rules! declare_rule {
         impl $crate::RuleMeta for $id {
             type Group = super::Group;
             const METADATA: $crate::RuleMetadata =
-                $crate::RuleMetadata::new($version, $name, concat!( $( $doc, "\n", )* )) $( .$key($value) )*;
+                $crate::RuleMetadata::new($version, $name, concat!( $( $doc, "\n", )* ), $severity) $( .$key($value) )*;
         }
     }
 }

crates/pgt_analyse/src/macros.rs

@@ -3,7 +3,7 @@ use pgt_console::{MarkupBuf, markup};
 use pgt_diagnostics::advice::CodeSuggestionAdvice;
 use pgt_diagnostics::{
     Advices, Category, Diagnostic, DiagnosticTags, Location, LogCategory, MessageAndDescription,
-    Visit,
+    Severity, Visit,
 };
 use pgt_text_size::TextRange;
 use std::cmp::Ordering;
@@ -31,17 +31,25 @@ pub struct RuleMetadata {
     pub recommended: bool,
     /// The source URL of the rule
     pub sources: &'static [RuleSource],
+    /// The default severity of the rule
+    pub severity: Severity,
 }
 
 impl RuleMetadata {
-    pub const fn new(version: &'static str, name: &'static str, docs: &'static str) -> Self {
+    pub const fn new(
+        version: &'static str,
+        name: &'static str,
+        docs: &'static str,
+        severity: Severity,
+    ) -> Self {
         Self {
             deprecated: None,
             version,
             name,
             docs,
             sources: &[],
             recommended: false,
+            severity,
         }
     }
 

crates/pgt_analyse/src/rule.rs

@@ -54,9 +54,11 @@ Let's say we want to create a new **lint** rule called `useMyRuleName`, follow t
 1. Run the command
 
    ```shell
-   just new-lintrule safety useMyRuleName
+   just new-lintrule safety useMyRuleName (<severity>)
    ```
 
+   Where severity is optional but can be "info", "warn", or "error" (default).
+
    The script will generate a bunch of files inside the `pgt_analyser` crate.
    Among the other files, you'll find a file called `use_my_new_rule_name.rs` inside the `pgt_analyser/lib/src/lint/safety` folder. You'll implement your rule in this file.
 
@@ -187,6 +189,7 @@ declare_lint_rule! {
     pub(crate) ExampleRule {
         version: "next",
         name: "myRuleName",
+        severity: Severity::Error,
         recommended: false,
     }
 }
@@ -206,6 +209,7 @@ declare_lint_rule! {
     pub(crate) ExampleRule {
         version: "next",
         name: "myRuleName",
+        severity: Severity::Error,
         recommended: false,
         sources: &[RuleSource::Squawk("ban-drop-column")],
     }
@@ -228,6 +232,7 @@ declare_lint_rule! {
     pub(crate) ExampleRule {
         version: "next",
         name: "myRuleName",
+        severity: Severity::Error,
         recommended: false,
     }
 }
@@ -280,6 +285,7 @@ declare_lint_rule! {
         version: "next",
         name: "banDropColumn",
         recommended: true,
+        severity: Severity::Error,
         sources: &[RuleSource::Squawk("ban-drop-column")],
     }
 }
@@ -351,6 +357,7 @@ declare_lint_rule! {
         version: "next",
         name: "banDropColumn",
         recommended: true,
+        severity: Severity::Error,
         deprecated: true,
         sources: &[RuleSource::Squawk("ban-drop-column")],
     }

crates/pgt_analyser/CONTRIBUTING.md

@@ -12,10 +12,11 @@ repository.workspace = true
 version              = "0.0.0"
 
 [dependencies]
-pgt_analyse   = { workspace = true }
-pgt_console   = { workspace = true }
-pgt_query_ext = { workspace = true }
-serde         = { workspace = true }
+pgt_analyse     = { workspace = true }
+pgt_console     = { workspace = true }
+pgt_diagnostics = { workspace = true }
+pgt_query_ext   = { workspace = true }
+serde           = { workspace = true }
 
 [dev-dependencies]
 insta           = { version = "1.42.1" }

crates/pgt_analyser/Cargo.toml

@@ -3,6 +3,8 @@
 use pgt_analyse::declare_lint_group;
 pub mod adding_required_field;
 pub mod ban_drop_column;
+pub mod ban_drop_database;
 pub mod ban_drop_not_null;
 pub mod ban_drop_table;
-declare_lint_group! { pub Safety { name : "safety" , rules : [self :: adding_required_field :: AddingRequiredField , self :: ban_drop_column :: BanDropColumn , self :: ban_drop_not_null :: BanDropNotNull , self :: ban_drop_table :: BanDropTable ,] } }
+pub mod ban_truncate_cascade;
+declare_lint_group! { pub Safety { name : "safety" , rules : [self :: adding_required_field :: AddingRequiredField , self :: ban_drop_column :: BanDropColumn , self :: ban_drop_database :: BanDropDatabase , self :: ban_drop_not_null :: BanDropNotNull , self :: ban_drop_table :: BanDropTable , self :: ban_truncate_cascade :: BanTruncateCascade ,] } }

crates/pgt_analyser/src/lint/safety.rs

@@ -1,5 +1,6 @@
 use pgt_analyse::{Rule, RuleDiagnostic, RuleSource, context::RuleContext, declare_lint_rule};
 use pgt_console::markup;
+use pgt_diagnostics::Severity;
 
 declare_lint_rule! {
     /// Adding a new column that is NOT NULL and has no default value to an existing table effectively makes it required.
@@ -17,6 +18,7 @@ declare_lint_rule! {
     pub AddingRequiredField {
         version: "next",
         name: "addingRequiredField",
+        severity: Severity::Error,
         recommended: false,
         sources: &[RuleSource::Squawk("adding-required-field")],
     }

crates/pgt_analyser/src/lint/safety/adding_required_field.rs

@@ -1,5 +1,6 @@
 use pgt_analyse::{Rule, RuleDiagnostic, RuleSource, context::RuleContext, declare_lint_rule};
 use pgt_console::markup;
+use pgt_diagnostics::Severity;
 
 declare_lint_rule! {
     /// Dropping a column may break existing clients.
@@ -19,6 +20,7 @@ declare_lint_rule! {
     pub BanDropColumn {
         version: "next",
         name: "banDropColumn",
+        severity: Severity::Warning,
         recommended: true,
         sources: &[RuleSource::Squawk("ban-drop-column")],
     }

crates/pgt_analyser/src/lint/safety/ban_drop_column.rs

@@ -0,0 +1,39 @@
+use pgt_analyse::{Rule, RuleDiagnostic, RuleSource, context::RuleContext, declare_lint_rule};
+use pgt_console::markup;
+use pgt_diagnostics::Severity;
+
+declare_lint_rule! {
+    /// Dropping a database may break existing clients (and everything else, really).
+    ///
+    /// Make sure that you really want to drop it.
+    pub BanDropDatabase {
+        version: "next",
+        name: "banDropDatabase",
+        severity: Severity::Warning,
+        recommended: false,
+        sources: &[RuleSource::Squawk("ban-drop-database")],
+    }
+}
+
+impl Rule for BanDropDatabase {
+    type Options = ();
+
+    fn run(ctx: &RuleContext<Self>) -> Vec<RuleDiagnostic> {
+        let mut diagnostics = vec![];
+
+        if let pgt_query_ext::NodeEnum::DropdbStmt(_) = &ctx.stmt() {
+            diagnostics.push(
+                RuleDiagnostic::new(
+                    rule_category!(),
+                    None,
+                    markup! {
+                        "Dropping a database may break existing clients."
+                    },
+                )
+                .detail(None, "You probably don't want to drop your database."),
+            );
+        }
+
+        diagnostics
+    }
+}

crates/pgt_analyser/src/lint/safety/ban_drop_database.rs

@@ -1,5 +1,6 @@
 use pgt_analyse::{Rule, RuleDiagnostic, RuleSource, context::RuleContext, declare_lint_rule};
 use pgt_console::markup;
+use pgt_diagnostics::Severity;
 
 declare_lint_rule! {
     /// Dropping a NOT NULL constraint may break existing clients.
@@ -18,6 +19,7 @@ declare_lint_rule! {
     pub BanDropNotNull {
         version: "next",
         name: "banDropNotNull",
+        severity: Severity::Warning,
         recommended: true,
         sources: &[RuleSource::Squawk("ban-drop-not-null")],