frostfs-node/cmd/neofs-node/modules/morph/common.go

package morph

import (
	"github.com/nspcc-dev/neo-go/pkg/util"
	"github.com/nspcc-dev/neofs-node/pkg/core/netmap"
	"github.com/nspcc-dev/neofs-node/pkg/morph/client"
	"github.com/nspcc-dev/neofs-node/pkg/morph/event"
	"github.com/spf13/viper"
	"go.uber.org/dig"
	"go.uber.org/zap"
)

// SmartContracts maps smart contract name to contract client.
type SmartContracts map[string]*client.StaticClient

// EventHandlers maps notification event name to handler information.
type EventHandlers map[string]event.HandlerInfo

type morphContractsParams struct {
	dig.In

	Viper *viper.Viper

	Client *client.Client

	Listener event.Listener
}

type contractParams struct {
	dig.In

	Viper *viper.Viper

	Logger *zap.Logger

	MorphContracts SmartContracts

	NodeInfo netmap.Info
}

func newMorphContracts(p morphContractsParams) (SmartContracts, EventHandlers, error) {
	mContracts := make(map[string]*client.StaticClient, len(ContractNames))
	mHandlers := make(map[string]event.HandlerInfo)

	for _, contractName := range ContractNames {
		scHash, err := util.Uint160DecodeStringLE(
			p.Viper.GetString(
				ScriptHashOptPath(contractName),
			),
		)
		if err != nil {
			return nil, nil, err
		}

		fee := util.Fixed8FromInt64(
			p.Viper.GetInt64(
				InvocationFeeOptPath(contractName),
			),
		)

		mContracts[contractName], err = client.NewStatic(p.Client, scHash, fee)
		if err != nil {
			return nil, nil, err
		}

		// set event parsers
		parserInfo := event.ParserInfo{}
		parserInfo.SetScriptHash(scHash)

		handlerInfo := event.HandlerInfo{}
		handlerInfo.SetScriptHash(scHash)

		for _, item := range mParsers[contractName] {
			parserInfo.SetParser(item.parser)

			optPath := ContractEventOptPath(contractName, item.typ)

			typEvent := event.TypeFromString(
				p.Viper.GetString(optPath),
			)

			parserInfo.SetType(typEvent)
			handlerInfo.SetType(typEvent)

			p.Listener.SetParser(parserInfo)

			mHandlers[optPath] = handlerInfo
		}
	}

	return mContracts, mHandlers, nil
}

const prefix = "morph"

const (
	endpointOpt = "endpoint"

	dialTimeoutOpt = "dial_timeout"

	magicNumberOpt = "magic_number"

	scriptHashOpt = "script_hash"

	invocationFeeOpt = "invocation_fee"
)

// ContractNames is a list of smart contract names.
var ContractNames = []string{
	containerContractName,
	NetmapContractName,
	BalanceContractName,
}

// EndpointOptPath returns the config path to goclient endpoint.
func EndpointOptPath() string {
	return optPath(prefix, endpointOpt)
}

// MagicNumberOptPath returns the config path to goclient magic number.
func MagicNumberOptPath() string {
	return optPath(prefix, magicNumberOpt)
}

// DialTimeoutOptPath returns the config path to goclient dial timeout.
func DialTimeoutOptPath() string {
	return optPath(prefix, dialTimeoutOpt)
}

// ScriptHashOptPath calculates the config path to script hash config of particular contract.
func ScriptHashOptPath(name string) string {
	return optPath(prefix, name, scriptHashOpt)
}

// InvocationFeeOptPath calculates the config path to invocation fee config of particular contract.
func InvocationFeeOptPath(name string) string {
	return optPath(prefix, name, invocationFeeOpt)
}