kroma.go

package blockchain

import (
	"context"

	"github.com/ethereum/go-ethereum"
	"github.com/ethereum/go-ethereum/accounts/abi/bind"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/core/types"
	"github.com/rs/zerolog/log"

	ethcontracts "github.com/goplugin/plugin-testing-framework/lib/contracts/ethereum"
)

// Kroma Gas Oracle Address https://docs.kroma.network/testnet/contract-addresses
const kromaGasOracleAddress string = "0x4200000000000000000000000000000000000005"

// KromaMultinodeClient represents a multi-node, EVM compatible client for the Kroma network
type KromaMultinodeClient struct {
	*EthereumMultinodeClient
}

// KromaClient represents a single node, EVM compatible client for the Kroma network
type KromaClient struct {
	*EthereumClient
}

func (o *KromaClient) EstimateGas(callData ethereum.CallMsg) (GasEstimations, error) {
	// Kroma is based on Optimism and uses the same gas mechanism. The L1 fee needs to be added on top of the regular gas costs.
	gasOracle, err := ethcontracts.NewOptimismGas(common.HexToAddress(kromaGasOracleAddress), o.Client)
	if err != nil {
		return GasEstimations{}, err
	}
	opts := &bind.CallOpts{
		From:    common.HexToAddress(o.GetDefaultWallet().Address()),
		Context: context.Background(),
	}
	l1Fee, err := gasOracle.GetL1Fee(opts, types.DynamicFeeTx{}.Data)
	if err != nil {
		return GasEstimations{}, err
	}
	gasEstimations, err := o.EthereumClient.EstimateGas(callData)
	if err != nil {
		return GasEstimations{}, err
	}
	initialEstimate := gasEstimations.TotalGasCost
	gasEstimations.TotalGasCost.Add(initialEstimate, l1Fee)
	log.Debug().
		Uint64("New Total Cost", gasEstimations.TotalGasCost.Uint64()).
		Uint64("Initial Estimate", initialEstimate.Uint64()).
		Uint64("L1 Fee", l1Fee.Uint64()).
		Msg("Adding Kroma L1 Fee")
	return gasEstimations, err
}