cleanup conditional

stackup-wallet · Apr 8, 2024 · 04e99a5 · 04e99a5
1 parent 429a8d2
commit 04e99a5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/internal/config/constants.go b/internal/config/constants.go
@@ -34,4 +34,10 @@ var (
 		LyraSepoliaChainID.Uint64(),
 		Ancient8SepoliaChainID.Uint64(),
 	)
+
+	ArbStackChains = mapset.NewSet(
+		ArbitrumOneChainID.Uint64(),
+		ArbitrumGoerliChainID.Uint64(),
+		ArbitrumSepoliaChainID.Uint64(),
+	)
 )
diff --git a/internal/start/private.go b/internal/start/private.go
@@ -84,9 +84,7 @@ func PrivateMode() {
 	}
 
 	ov := gas.NewDefaultOverhead()
-	if conf.IsArbStackNetwork || chain.Cmp(config.ArbitrumOneChainID) == 0 ||
-		chain.Cmp(config.ArbitrumGoerliChainID) == 0 ||
-		chain.Cmp(config.ArbitrumSepoliaChainID) == 0 {
+	if conf.IsArbStackNetwork || config.ArbStackChains.Contains(chain.Uint64()) {
 		ov.SetCalcPreVerificationGasFunc(gas.CalcArbitrumPVGWithEthClient(rpc, conf.SupportedEntryPoints[0]))
 		ov.SetPreVerificationGasBufferFactor(16)
 	}