Flag to enable stack specific gas calculation (#368)

Author: NickSolante

internal/config/values.go

@@ -41,6 +41,9 @@ type Values struct {
 	AltMempoolIPFSGateway string
 	AltMempoolIds         []string
 
+	// Stack Specific gas calculation variables.
+	IsOpStackNetwork bool
+
 	// Undocumented variables.
 	DebugMode bool
 	GinMode   string
@@ -91,6 +94,7 @@ func GetValues() *Values {
 	viper.SetDefault("erc4337_bundler_max_op_ttl_seconds", 180)
 	viper.SetDefault("erc4337_bundler_blocks_in_the_future", 6)
 	viper.SetDefault("erc4337_bundler_otel_insecure_mode", false)
+	viper.SetDefault("erc4337_is_op_stack_network", false)
 	viper.SetDefault("erc4337_bundler_debug_mode", false)
 	viper.SetDefault("erc4337_bundler_gin_mode", gin.ReleaseMode)
 
@@ -126,6 +130,7 @@ func GetValues() *Values {
 	_ = viper.BindEnv("erc4337_bundler_otel_insecure_mode")
 	_ = viper.BindEnv("erc4337_bundler_alt_mempool_ipfs_gateway")
 	_ = viper.BindEnv("erc4337_bundler_alt_mempool_ids")
+	_ = viper.BindEnv("erc4337_is_op_stack_network")
 	_ = viper.BindEnv("erc4337_bundler_debug_mode")
 	_ = viper.BindEnv("erc4337_bundler_gin_mode")
 
@@ -184,6 +189,7 @@ func GetValues() *Values {
 	otelInsecureMode := viper.GetBool("erc4337_bundler_otel_insecure_mode")
 	altMempoolIPFSGateway := viper.GetString("erc4337_bundler_alt_mempool_ipfs_gateway")
 	altMempoolIds := envArrayToStringSlice(viper.GetString("erc4337_bundler_alt_mempool_ids"))
+	isOpStackNetwork := viper.GetBool("erc4337_is_op_stack_network")
 	debugMode := viper.GetBool("erc4337_bundler_debug_mode")
 	ginMode := viper.GetString("erc4337_bundler_gin_mode")
 	return &Values{
@@ -206,6 +212,7 @@ func GetValues() *Values {
 		OTELInsecureMode:             otelInsecureMode,
 		AltMempoolIPFSGateway:        altMempoolIPFSGateway,
 		AltMempoolIds:                altMempoolIds,
+		IsOpStackNetwork:             isOpStackNetwork,
 		DebugMode:                    debugMode,
 		GinMode:                      ginMode,
 	}

internal/start/private.go

@@ -91,7 +91,7 @@ func PrivateMode() {
 		ov.SetPreVerificationGasBufferFactor(16)
 	}
 
-	if config.OpStackChains.Contains(chain) {
+	if conf.IsOpStackNetwork || config.OpStackChains.Contains(chain) {
 		ov.SetCalcPreVerificationGasFunc(
 			gas.CalcOptimismPVGWithEthClient(rpc, chain, conf.SupportedEntryPoints[0]),
 		)