mercury

type HttpClient interface {
	Do(req *http.Request) (*http.Response, error)
}

type MercuryClient interface {
	// DoRequest makes a data stream request, it manages retries and returns the following:
	// state: the state of the pipeline execution. This field is coupled with err. If state is NoPipelineError then err is nil, otherwise err is non nil and appropriate state is returned
	// data: the data returned from the data stream if there's NoPipelineError
	// errCode: the errorCode of streams request if data is nil and there's NoPipelineError
	// retryable: whether the request is retryable. Only applicable for errored states.
	// retryInterval: the interval to wait before retrying the request. Only applicable for errored states.
	// err: non nil if some internal error occurs in pipeline
	DoRequest(ctx context.Context, streamsLookup *StreamsLookup, upkeepType automationTypes.UpkeepType, pluginRetryKey string) (encoding.PipelineExecutionState, [][]byte, encoding.ErrCode, bool, time.Duration, error)
}

type MercuryConfigProvider interface {
	Credentials() *types.MercuryCredentials
	IsUpkeepAllowed(string) (any, bool)
	SetUpkeepAllowed(string, any, time.Duration)
	GetPluginRetry(string) (any, bool)
	SetPluginRetry(string, any, time.Duration)
}

type MercuryData struct {
	Index     int
	Bytes     [][]byte                        // Mercury values if request is successful
	ErrCode   encoding.ErrCode                // Error code if mercury gives an error
	State     encoding.PipelineExecutionState // NoPipelineError if no error during execution, otherwise appropriate error
	Retryable bool                            // Applicable if State != NoPipelineError
	Error     error                           // non nil if State != NoPipelineError
}

type MercuryEndpointMock interface {
	URL() string
	Username() string
	Password() string
	CallCount() int
	RegisterHandler(http.HandlerFunc)
}

type Packer interface {
	UnpackCheckCallbackResult(callbackResp []byte) (encoding.PipelineExecutionState, bool, []byte, encoding.UpkeepFailureReason, *big.Int, error)
	PackGetUpkeepPrivilegeConfig(upkeepId *big.Int) ([]byte, error)
	UnpackGetUpkeepPrivilegeConfig(resp []byte) ([]byte, error)
	DecodeStreamsLookupRequest(data []byte) (*StreamsLookupError, error)
	PackUserCheckErrorHandler(errCode encoding.ErrCode, extraData []byte) ([]byte, error)
}

type SimulatedMercuryServer struct {
	// contains filtered or unexported fields
}

type StreamsLookup struct {
	*StreamsLookupError
	UpkeepId *big.Int
	Block    uint64
}

type StreamsLookupError struct {
	FeedParamKey string
	Feeds        []string
	TimeParamKey string
	Time         *big.Int
	ExtraData    []byte
}