mev-beta/pkg/bindings/bindings.go

// Package bindings provides utilities for working with Ethereum contract bindings
package bindings

import (
	"fmt"
	"strings"

	"github.com/ethereum/go-ethereum/accounts/abi"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
)

// FunctionSignature represents a parsed function signature
type FunctionSignature struct {
	Selector     common.Hash
	Name         string
	Signature    string
	Inputs       []abi.Argument
	Outputs      []abi.Argument
	IsEvent      bool
	TopicHash    common.Hash // For events
}

// ContractBinding represents a contract binding with its ABI
type ContractBinding struct {
	Name         string
	Address      common.Address
	ABI          abi.ABI
	Functions    map[string]*FunctionSignature
	Events       map[string]*FunctionSignature
	Constructor  *FunctionSignature
}

// BindingRegistry manages contract bindings
type BindingRegistry struct {
	bindings map[common.Address]*ContractBinding
	names    map[string]*ContractBinding
}

// NewBindingRegistry creates a new binding registry
func NewBindingRegistry() *BindingRegistry {
	return &BindingRegistry{
		bindings: make(map[common.Address]*ContractBinding),
		names:    make(map[string]*ContractBinding),
	}
}

// RegisterBinding registers a contract binding
func (br *BindingRegistry) RegisterBinding(binding *ContractBinding) error {
	if binding == nil {
		return fmt.Errorf("binding cannot be nil")
	}

	br.bindings[binding.Address] = binding
	br.names[binding.Name] = binding

	return nil
}

// GetBindingByAddress retrieves a binding by contract address
func (br *BindingRegistry) GetBindingByAddress(address common.Address) (*ContractBinding, bool) {
	binding, exists := br.bindings[address]
	return binding, exists
}

// GetBindingByName retrieves a binding by contract name
func (br *BindingRegistry) GetBindingByName(name string) (*ContractBinding, bool) {
	binding, exists := br.names[name]
	return binding, exists
}

// GetFunctionSignature derives a function signature from method name and ABI
func (br *BindingRegistry) GetFunctionSignature(contractName, methodName string) (*FunctionSignature, error) {
	binding, exists := br.GetBindingByName(contractName)
	if !exists {
		return nil, fmt.Errorf("contract binding not found: %s", contractName)
	}

	method, exists := binding.ABI.Methods[methodName]
	if !exists {
		return nil, fmt.Errorf("method not found in contract %s: %s", contractName, methodName)
	}

	signature := &FunctionSignature{
		Selector:  method.ID,
		Name:      method.Name,
		Signature: method.Sig,
		Inputs:    method.Inputs,
		Outputs:   method.Outputs,
		IsEvent:   false,
	}

	return signature, nil
}

// GetEventSignature derives an event signature from event name and ABI
func (br *BindingRegistry) GetEventSignature(contractName, eventName string) (*FunctionSignature, error) {
	binding, exists := br.GetBindingByName(contractName)
	if !exists {
		return nil, fmt.Errorf("contract binding not found: %s", contractName)
	}

	event, exists := binding.ABI.Events[eventName]
	if !exists {
		return nil, fmt.Errorf("event not found in contract %s: %s", contractName, eventName)
	}

	// Generate topic hash for the event
	topic := crypto.Keccak256Hash([]byte(event.Sig))

	signature := &FunctionSignature{
		Selector:  event.ID,
		Name:      event.Name,
		Signature: event.Sig,
		Inputs:    event.Inputs,
		Outputs:   nil, // Events don't have outputs
		IsEvent:   true,
		TopicHash: topic,
	}

	return signature, nil
}

// CreateFunctionSignature creates a function signature from method definition
func CreateFunctionSignature(methodName string, inputs []abi.Argument, outputs []abi.Argument) *FunctionSignature {
	// Create signature string
	inputTypes := make([]string, len(inputs))
	for i, input := range inputs {
		inputTypes[i] = input.Type.String()
	}

	outputTypes := make([]string, len(outputs))
	for i, output := range outputs {
		outputTypes[i] = output.Type.String()
	}

	signature := methodName + "(" + strings.Join(inputTypes, ",") + ")"
	if len(outputTypes) > 0 {
		signature += " returns (" + strings.Join(outputTypes, ",") + ")"
	}

	// Generate selector
	selector := crypto.Keccak256Hash([]byte(methodName + "(" + strings.Join(inputTypes, ",") + ")"))

	return &FunctionSignature{
		Selector:  selector,
		Name:      methodName,
		Signature: signature,
		Inputs:    inputs,
		Outputs:   outputs,
		IsEvent:   false,
	}
}

// CreateEventSignature creates an event signature from event definition
func CreateEventSignature(eventName string, inputs []abi.Argument) *FunctionSignature {
	// Create signature string
	inputTypes := make([]string, len(inputs))
	for i, input := range inputs {
		inputStr := input.Type.String()
		if input.Indexed {
			inputStr += " indexed"
		}
		inputTypes[i] = inputStr
	}

	signature := eventName + "(" + strings.Join(inputTypes, ",") + ")"

	// Generate selector and topic hash
	selector := crypto.Keccak256Hash([]byte(signature))
	topic := crypto.Keccak256Hash([]byte(signature))

	return &FunctionSignature{
		Selector:  selector,
		Name:      eventName,
		Signature: signature,
		Inputs:    inputs,
		Outputs:   nil,
		IsEvent:   true,
		TopicHash: topic,
	}
}

// GenerateCREATE2Address generates a CREATE2 address for a contract
func GenerateCREATE2Address(factory common.Address, salt [32]byte, initCode []byte) common.Address {
	return crypto.CreateAddress2(factory, salt, initCode)
}

// GetKnownDEXBindings returns known DEX contract bindings for Arbitrum
func GetKnownDEXBindings() map[string]*ContractBinding {
	bindings := make(map[string]*ContractBinding)

	// Add popular DEX contracts on Arbitrum
	bindings["UniswapV2Factory"] = &ContractBinding{
		Name:    "UniswapV2Factory",
		Address: common.HexToAddress("0xf1D7CC64Fb4452F05c498126312eBE29f30Fbcf9"),
		// ABI would be loaded from actual contract ABI files
		Functions: make(map[string]*FunctionSignature),
		Events:    make(map[string]*FunctionSignature),
	}

	bindings["UniswapV3Factory"] = &ContractBinding{
		Name:    "UniswapV3Factory",
		Address: common.HexToAddress("0x1F98431c8aD98523631AE4a59f267346ea31F984"),
		// ABI would be loaded from actual contract ABI files
		Functions: make(map[string]*FunctionSignature),
		Events:    make(map[string]*FunctionSignature),
	}

	bindings["SushiSwapFactory"] = &ContractBinding{
		Name:    "SushiSwapFactory",
		Address: common.HexToAddress("0xc35DADB65012eC5796536bD9864eD8773aBc74C4"),
		// ABI would be loaded from actual contract ABI files
		Functions: make(map[string]*FunctionSignature),
		Events:    make(map[string]*FunctionSignature),
	}

	return bindings
}