From 97794855b401dab082c9df9473fcada65db83afb Mon Sep 17 00:00:00 2001
From: Delweng <delweng@gmail.com>
Date: Fri, 27 Jun 2025 16:02:53 +0800
Subject: [PATCH] Revert "cmd: disable preimage in pbss archive"

This reverts commit 11ea6dde2eee3d6536fef3fa26a75b92bc51d7fa.

Signed-off-by: Delweng <delweng@gmail.com>
---
 cmd/utils/flags.go | 30 ++++++++++++++++--------------
 1 file changed, 16 insertions(+), 14 deletions(-)

diff --git a/cmd/utils/flags.go b/cmd/utils/flags.go
index a71c17d281..71941cdf9d 100644
--- a/cmd/utils/flags.go
+++ b/cmd/utils/flags.go
@@ -1632,15 +1632,6 @@ func SetEthConfig(ctx *cli.Context, stack *node.Node, cfg *ethconfig.Config) {
 	if ctx.IsSet(CacheNoPrefetchFlag.Name) {
 		cfg.NoPrefetch = ctx.Bool(CacheNoPrefetchFlag.Name)
 	}
-	if ctx.IsSet(DataDirFlag.Name) {
-		chaindb := tryMakeReadOnlyDatabase(ctx, stack)
-		scheme, err := rawdb.ParseStateScheme(ctx.String(StateSchemeFlag.Name), chaindb)
-		if err != nil {
-			Fatalf("%v", err)
-		}
-		cfg.StateScheme = scheme
-		chaindb.Close()
-	}
 	// Read the value from the flag no matter if it's set or not.
 	cfg.Preimages = ctx.Bool(CachePreimagesFlag.Name)
 	if cfg.NoPruning && !cfg.Preimages {
@@ -1650,6 +1641,9 @@ func SetEthConfig(ctx *cli.Context, stack *node.Node, cfg *ethconfig.Config) {
 	if ctx.IsSet(StateHistoryFlag.Name) {
 		cfg.StateHistory = ctx.Uint64(StateHistoryFlag.Name)
 	}
+	if ctx.IsSet(StateSchemeFlag.Name) {
+		cfg.StateScheme = ctx.String(StateSchemeFlag.Name)
+	}
 	// Parse transaction history flag, if user is still using legacy config
 	// file with 'TxLookupLimit' configured, copy the value to 'TransactionHistory'.
 	if cfg.TransactionHistory == ethconfig.Defaults.TransactionHistory && cfg.TxLookupLimit != ethconfig.Defaults.TxLookupLimit {
@@ -1663,9 +1657,17 @@ func SetEthConfig(ctx *cli.Context, stack *node.Node, cfg *ethconfig.Config) {
 		cfg.TransactionHistory = ctx.Uint64(TxLookupLimitFlag.Name)
 	}
 
-	if ctx.String(GCModeFlag.Name) == "archive" && cfg.TransactionHistory != 0 && cfg.StateScheme == rawdb.HashScheme {
-		cfg.TransactionHistory = 0
-		log.Warn("Disabled transaction unindexing for archive node with hash state scheme")
+	if ctx.String(GCModeFlag.Name) == "archive" && cfg.TransactionHistory != 0 && ctx.IsSet(DataDirFlag.Name) {
+		chaindb := tryMakeReadOnlyDatabase(ctx, stack)
+		scheme, err := rawdb.ParseStateScheme(cfg.StateScheme, chaindb)
+		if err != nil {
+			Fatalf("%v", err)
+		}
+		if scheme == rawdb.HashScheme {
+			cfg.TransactionHistory = 0
+			log.Warn("Disabled transaction unindexing for archive node with hash state scheme")
+		}
+		chaindb.Close()
 	}
 	if ctx.IsSet(LogHistoryFlag.Name) {
 		cfg.LogHistory = ctx.Uint64(LogHistoryFlag.Name)
@@ -2204,9 +2206,9 @@ func MakeChain(ctx *cli.Context, stack *node.Node, readonly bool) (*core.BlockCh
 		// Disable transaction indexing/unindexing.
 		TxLookupLimit: -1,
 	}
-	if options.ArchiveMode && !options.Preimages && scheme == rawdb.HashScheme {
+	if options.ArchiveMode && !options.Preimages {
 		options.Preimages = true
-		log.Info("Enabling recording of key preimages since archive mode is used in hash state scheme")
+		log.Info("Enabling recording of key preimages since archive mode is used")
 	}
 	if !ctx.Bool(SnapshotFlag.Name) {
 		options.SnapshotLimit = 0 // Disabled