feat(rules): add ban-drop-table

Author: juleswritescode

crates/pglt_analyser/src/lint.rs

@@ -1,4 +1,4 @@
 //! Generated file, do not edit by hand, see `xtask/codegen`
 
 pub mod safety;
-::pglt_analyse::declare_category! { pub Lint { kind : Lint , groups : [self :: safety :: Safety ,] } }
+::pglt_analyse::declare_category! { pub Lint { kind : Lint , groups : [self :: safety :: Safety , ] } }

crates/pglt_analyser/src/lint/safety.rs

@@ -3,4 +3,5 @@
 use pglt_analyse::declare_lint_group;
 pub mod ban_drop_column;
 pub mod ban_drop_not_null;
-declare_lint_group! { pub Safety { name : "safety" , rules : [self :: ban_drop_column :: BanDropColumn , self :: ban_drop_not_null :: BanDropNotNull ,] } }
+pub mod ban_drop_table;
+declare_lint_group! { pub Safety { name : "safety" , rules : [self :: ban_drop_column :: BanDropColumn , self :: ban_drop_not_null :: BanDropNotNull , self :: ban_drop_table :: BanDropTable ,] } }

crates/pglt_analyser/src/lint/safety/ban_drop_table.rs

@@ -0,0 +1,52 @@
+use pglt_analyse::{context::RuleContext, declare_lint_rule, Rule, RuleDiagnostic, RuleSource};
+use pglt_console::markup;
+
+declare_lint_rule! {
+    /// Dropping a table may break existing clients.
+    ///
+    /// Update your application code to no longer read or write the table.
+    ///
+    /// Once the table is no longer needed, you can delete it by running the command "DROP TABLE mytable;".
+    ///
+    /// This command will permanently remove the table from the database and all its contents.
+    /// Be sure to back up the table before deleting it, just in case you need to restore it in the future.
+    ///
+    /// ## Examples
+    /// ```sql,expect_diagnostic
+    /// drop table some_table;
+    /// ```
+    pub BanDropTable {
+        version: "next",
+        name: "banDropTable",
+        recommended: false,
+        sources: &[RuleSource::Squawk("ban-drop-table")],
+    }
+}
+
+impl Rule for BanDropTable {
+    type Options = ();
+
+    fn run(ctx: &RuleContext<Self>) -> Vec<RuleDiagnostic> {
+        let mut diagnostics = vec![];
+
+        if let pglt_query_ext::NodeEnum::DropStmt(stmt) = &ctx.stmt() {
+            if stmt.remove_type() == pglt_query_ext::protobuf::ObjectType::ObjectTable {
+                diagnostics.push(
+                    RuleDiagnostic::new(
+                        rule_category!(),
+                        None,
+                        markup! {
+                            "Dropping a table may break existing clients."
+                        },
+                    )
+                    .detail(
+                        None,
+                        "Update your application code to no longer read or write the table, and only then delete the table. Be sure to create a backup.",
+                    ),
+                );
+            }
+        }
+
+        diagnostics
+    }
+}

crates/pglt_analyser/src/options.rs

@@ -5,3 +5,4 @@ pub type BanDropColumn =
     <lint::safety::ban_drop_column::BanDropColumn as pglt_analyse::Rule>::Options;
 pub type BanDropNotNull =
     <lint::safety::ban_drop_not_null::BanDropNotNull as pglt_analyse::Rule>::Options;
+pub type BanDropTable = <lint::safety::ban_drop_table::BanDropTable as pglt_analyse::Rule>::Options;

crates/pglt_analyser/tests/specs/safety/banDropTable/basic.sql

@@ -0,0 +1,2 @@
+-- expect_only_lint/safety/banDropTable
+drop table test;

crates/pglt_analyser/tests/specs/safety/banDropTable/basic.sql.snap

@@ -0,0 +1,16 @@
+---
+source: crates/pglt_analyser/tests/rules_tests.rs
+expression: snapshot
+---
+# Input
+```
+-- expect_only_lint/safety/banDropTable
+drop table test;
+```
+
+# Diagnostics
+lint/safety/banDropTable ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+
+  × Dropping a table may break existing clients.
+  
+  i Update your application code to no longer read or write the table, and only then delete the table. Be sure to create a backup.

crates/pglt_configuration/src/analyser/linter/rules.rs

@@ -54,6 +54,7 @@ pub struct Rules {
     pub all: Option<bool>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub safety: Option<Safety>,
+    #[serde(skip_serializing_if = "Option::is_none")]
 }
 impl Rules {
     #[doc = r" Checks if the code coming from [pglt_diagnostics::Diagnostic] corresponds to a rule."]

crates/pglt_diagnostics_categories/src/categories.rs

@@ -15,6 +15,7 @@
 define_categories! {
     "lint/safety/banDropColumn": "https://pglt.dev/linter/rules/ban-drop-column",
     "lint/safety/banDropNotNull": "https://pglt.dev/linter/rules/ban-drop-not-null",
+    "lint/safety/banDropTable": "https://pglt.dev/linter/rules/ban-drop-table",
     // end lint rules
     ;
     // General categories