synapsecns/sanguine

package listener

import (
    "context"
    "strings"
    "time"

    "github.com/ethereum/go-ethereum/rpc"
)

// Option is a functional option for chainListener.
type Option func(*chainListener)

// NewBlockHandler is a function that is called when a new block is detected.
type NewBlockHandler func(ctx context.Context, block uint64) error

// WithNewBlockHandler sets the new block handler.
func WithNewBlockHandler(handler NewBlockHandler) Option {
    return func(c *chainListener) {
        c.newBlockHandler = handler
    }
}

// WithPollInterval sets the poll interval.
func WithPollInterval(interval time.Duration) Option {
    return func(c *chainListener) {
        c.pollIntervalSetting = interval
    }
}

// WithFinalityMode sets the finality mode.
func WithFinalityMode(mode string) Option {
    return func(c *chainListener) {
        switch strings.ToLower(mode) {
        case "latest":
            c.finalityMode = rpc.LatestBlockNumber
        case "safe":
            c.finalityMode = rpc.SafeBlockNumber
        case "finalized":
            c.finalityMode = rpc.FinalizedBlockNumber
        default:
            c.finalityMode = rpc.LatestBlockNumber
        }
    }
}

// WithBlockWait sets the block wait.
func WithBlockWait(wait uint64) Option {
    return func(c *chainListener) {
        c.blockWait = wait
    }
}

// WithName sets the listener name.
func WithName(name string) Option {
    return func(c *chainListener) {
        c.name = name
        c.store.SetListenerName(name)
    }
}