erigon-pulse/cmd/devnet/services/context.go

package services

import (
	"context"

	"github.com/ledgerwatch/erigon/cmd/devnet/devnet"
	"github.com/ledgerwatch/erigon/cmd/devnet/services/accounts"
	"github.com/ledgerwatch/erigon/cmd/devnet/services/bor"
)

type ctxKey int

const (
	ckFaucet ctxKey = iota
)

func Faucet(ctx context.Context) *accounts.Faucet {
	if network := devnet.CurrentNetwork(ctx); network != nil {
		for _, service := range network.Services {
			if faucet, ok := service.(*accounts.Faucet); ok {
				return faucet
			}
		}
	}

	return nil
}

func Heimdall(ctx context.Context) *bor.Heimdall {
	if network := devnet.CurrentNetwork(ctx); network != nil {
		for _, service := range network.Services {
			if heimdall, ok := service.(*bor.Heimdall); ok {
				return heimdall
			}
		}
	}

	return nil
}