plugin.go

package functions

import (
	"encoding/json"

	"github.com/ethereum/go-ethereum/common"
	"github.com/pkg/errors"

	"github.com/smartcontractkit/libocr/commontypes"
	ocr2types "github.com/smartcontractkit/libocr/offchainreporting2/types"

	"github.com/smartcontractkit/chainlink/v2/core/chains/evm"
	"github.com/smartcontractkit/chainlink/v2/core/gethwrappers/generated/ocr2dr_oracle"
	"github.com/smartcontractkit/chainlink/v2/core/logger"
	"github.com/smartcontractkit/chainlink/v2/core/services/functions"
	"github.com/smartcontractkit/chainlink/v2/core/services/job"
	"github.com/smartcontractkit/chainlink/v2/core/services/ocr2/plugins"
	"github.com/smartcontractkit/chainlink/v2/core/services/ocr2/plugins/functions/config"
	"github.com/smartcontractkit/chainlink/v2/core/services/pipeline"
	"github.com/smartcontractkit/chainlink/v2/core/utils"
)

type DROracle struct {
	jb             job.Job
	pipelineRunner pipeline.Runner
	jobORM         job.ORM
	pluginConfig   config.PluginConfig
	pluginORM      functions.ORM
	chain          evm.Chain
	lggr           logger.Logger
	ocrLogger      commontypes.Logger
	mailMon        *utils.MailboxMonitor
}

var _ plugins.OraclePlugin = &DROracle{}

func NewDROracle(jb job.Job, pipelineRunner pipeline.Runner, jobORM job.ORM, pluginORM functions.ORM, chain evm.Chain, lggr logger.Logger, ocrLogger commontypes.Logger, mailMon *utils.MailboxMonitor) (*DROracle, error) {
	var pluginConfig config.PluginConfig
	err := json.Unmarshal(jb.OCR2OracleSpec.PluginConfig.Bytes(), &pluginConfig)
	if err != nil {
		return &DROracle{}, err
	}
	err = config.ValidatePluginConfig(pluginConfig)
	if err != nil {
		return &DROracle{}, err
	}

	return &DROracle{
		jb:             jb,
		pipelineRunner: pipelineRunner,
		jobORM:         jobORM,
		pluginConfig:   pluginConfig,
		pluginORM:      pluginORM,
		chain:          chain,
		lggr:           lggr,
		ocrLogger:      ocrLogger,
		mailMon:        mailMon,
	}, nil
}

func (o *DROracle) GetPluginFactory() (ocr2types.ReportingPluginFactory, error) {
	return DirectRequestReportingPluginFactory{
		Logger:    o.ocrLogger,
		PluginORM: o.pluginORM,
		JobID:     o.jb.ExternalJobID,
	}, nil
}

func (o *DROracle) GetServices() ([]job.ServiceCtx, error) {
	contractAddress := common.HexToAddress(o.jb.OCR2OracleSpec.ContractID)
	oracle, err := ocr2dr_oracle.NewOCR2DROracle(contractAddress, o.chain.Client())
	if err != nil {
		return nil, errors.Wrapf(err, "OCR2DirectRequest: failed to create an OCR2DROracle wrapper for address: %v", contractAddress)
	}
	svcLogger := o.lggr.Named("FunctionsListener").
		With(
			"contract", contractAddress,
			"jobName", o.jb.PipelineSpec.JobName,
			"jobID", o.jb.PipelineSpec.JobID,
			"externalJobID", o.jb.ExternalJobID,
		)
	logListener := functions.NewFunctionsListener(oracle, o.jb, o.pipelineRunner, o.jobORM, o.pluginORM, o.pluginConfig, o.chain.LogBroadcaster(), svcLogger, o.mailMon)
	var services []job.ServiceCtx
	services = append(services, logListener)
	return services, nil
}