feat(bin): delete snapshot jars in db clear (#6311)

Author: shekhirin

bin/reth/src/commands/db/clear.rs

@@ -1,27 +1,53 @@
-use clap::Parser;
+use clap::{Parser, Subcommand};
 use reth_db::{
     database::Database,
+    snapshot::iter_snapshots,
     table::Table,
     transaction::{DbTx, DbTxMut},
     TableViewer, Tables,
 };
+use reth_primitives::{snapshot::find_fixed_range, SnapshotSegment};
+use reth_provider::ProviderFactory;
 
 /// The arguments for the `reth db clear` command
 #[derive(Parser, Debug)]
 pub struct Command {
-    /// Table name
-    pub table: Tables,
+    #[clap(subcommand)]
+    subcommand: Subcommands,
 }
 
 impl Command {
     /// Execute `db clear` command
-    pub fn execute<DB: Database>(self, db: &DB) -> eyre::Result<()> {
-        self.table.view(&ClearViewer { db })?;
+    pub fn execute<DB: Database>(self, provider_factory: ProviderFactory<DB>) -> eyre::Result<()> {
+        match self.subcommand {
+            Subcommands::Mdbx { table } => {
+                table.view(&ClearViewer { db: provider_factory.db_ref() })?
+            }
+            Subcommands::Snapshot { segment } => {
+                let snapshot_provider = provider_factory
+                    .snapshot_provider()
+                    .expect("snapshot provider initialized via provider factory");
+                let snapshots = iter_snapshots(snapshot_provider.directory())?;
+
+                if let Some(segment_snapshots) = snapshots.get(&segment) {
+                    for (block_range, _) in segment_snapshots {
+                        snapshot_provider
+                            .delete_jar(segment, find_fixed_range(*block_range.start()))?;
+                    }
+                }
+            }
+        }
 
         Ok(())
     }
 }
 
+#[derive(Subcommand, Debug)]
+enum Subcommands {
+    Mdbx { table: Tables },
+    Snapshot { segment: SnapshotSegment },
+}
+
 struct ClearViewer<'a, DB: Database> {
     db: &'a DB,
 }

bin/reth/src/commands/db/mod.rs

@@ -249,7 +249,9 @@ impl Command {
             Subcommands::Clear(command) => {
                 let db =
                     open_db(&db_path, DatabaseArguments::default().log_level(self.db.log_level))?;
-                command.execute(&db)?;
+                let provider_factory = ProviderFactory::new(db, self.chain.clone())
+                    .with_snapshots(data_dir.snapshots_path())?;
+                command.execute(provider_factory)?;
             }
             Subcommands::Snapshot(command) => {
                 command.execute(&db_path, self.db.log_level, self.chain.clone())?;