From 21569180e10cc00ba2be73db630526ed8d9049c8 Mon Sep 17 00:00:00 2001
From: Roman Krasiuk <rokrassyuk@gmail.com>
Date: Tue, 25 Apr 2023 00:31:24 +0300
Subject: [PATCH] feat(cli): drop history stages (#2369)

---
 bin/reth/src/args/stage_args.rs |  1 +
 bin/reth/src/drop_stage.rs      | 12 +++++++++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/bin/reth/src/args/stage_args.rs b/bin/reth/src/args/stage_args.rs
index c2a9da6973..6bf71d824c 100644
--- a/bin/reth/src/args/stage_args.rs
+++ b/bin/reth/src/args/stage_args.rs
@@ -11,4 +11,5 @@ pub enum StageEnum {
     Hashing,
     Merkle,
     TxLookup,
+    History,
 }
diff --git a/bin/reth/src/drop_stage.rs b/bin/reth/src/drop_stage.rs
index 9ef004a64d..8ce5d6bd05 100644
--- a/bin/reth/src/drop_stage.rs
+++ b/bin/reth/src/drop_stage.rs
@@ -14,7 +14,8 @@ use reth_db::{
 use reth_primitives::ChainSpec;
 use reth_staged_sync::utils::{chainspec::genesis_value_parser, init::insert_genesis_state};
 use reth_stages::stages::{
-    ACCOUNT_HASHING, EXECUTION, MERKLE_EXECUTION, MERKLE_UNWIND, STORAGE_HASHING,
+    ACCOUNT_HASHING, EXECUTION, INDEX_ACCOUNT_HISTORY, INDEX_STORAGE_HISTORY, MERKLE_EXECUTION,
+    MERKLE_UNWIND, STORAGE_HASHING,
 };
 use std::sync::Arc;
 use tracing::info;
@@ -102,6 +103,15 @@ impl Command {
                     Ok::<_, eyre::Error>(())
                 })??;
             }
+            StageEnum::History => {
+                tool.db.update(|tx| {
+                    tx.clear::<tables::AccountHistory>()?;
+                    tx.clear::<tables::StorageHistory>()?;
+                    tx.put::<tables::SyncStage>(INDEX_ACCOUNT_HISTORY.0.to_string(), 0)?;
+                    tx.put::<tables::SyncStage>(INDEX_STORAGE_HISTORY.0.to_string(), 0)?;
+                    Ok::<_, eyre::Error>(())
+                })??;
+            }
             _ => {
                 info!("Nothing to do for stage {:?}", self.stage);
             }