code.vegaprotocol.io/vega@v0.79.0/datanode/gateway/graphql/oracles_resolver.go

// Copyright (C) 2023 Gobalsky Labs Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package gql

import (
	"context"
	"errors"
	"time"

	"code.vegaprotocol.io/vega/libs/ptr"
	v2 "code.vegaprotocol.io/vega/protos/data-node/api/v2"
	vegapb "code.vegaprotocol.io/vega/protos/vega"
	v1 "code.vegaprotocol.io/vega/protos/vega/data/v1"
)

type oracleSpecResolver VegaResolverRoot

func (o *oracleSpecResolver) DataSourceSpec(_ context.Context, obj *vegapb.OracleSpec) (extDss *ExternalDataSourceSpec, _ error) {
	extDss = &ExternalDataSourceSpec{Spec: &DataSourceSpec{Data: &vegapb.DataSourceDefinition{}}}
	if obj.ExternalDataSourceSpec != nil {
		extDss.Spec = resolveDataSourceSpec(obj.ExternalDataSourceSpec.Spec)
	}
	return
}

func (o *oracleSpecResolver) DataConnection(ctx context.Context, obj *vegapb.OracleSpec, pagination *v2.Pagination) (*v2.OracleDataConnection, error) {
	var specID *string
	if ed := obj.ExternalDataSourceSpec; ed != nil && ed.Spec != nil && ed.Spec.Id != "" {
		specID = &ed.Spec.Id
	}

	req := v2.ListOracleDataRequest{
		OracleSpecId: specID,
		Pagination:   pagination,
	}

	resp, err := o.tradingDataClientV2.ListOracleData(ctx, &req)
	if err != nil {
		return nil, err
	}

	return resp.OracleData, nil
}

type oracleDataResolver VegaResolverRoot

func (o *oracleDataResolver) ExternalData(_ context.Context, obj *vegapb.OracleData) (ed *ExternalData, _ error) {
	ed = &ExternalData{
		Data: &Data{},
	}

	oed := obj.ExternalData
	if oed == nil || oed.Data == nil {
		return
	}

	ed.Data.Signers = resolveSigners(oed.Data.Signers)
	ed.Data.Data = oed.Data.Data
	ed.Data.MatchedSpecIds = oed.Data.MatchedSpecIds
	ed.Data.BroadcastAt = oed.Data.BroadcastAt

	return
}

func resolveTrigger(obj any) (trigger TriggerKind) {
	if obj != nil {
		switch trig := obj.(type) {
		case *vegapb.EthCallTrigger_TimeTrigger:
			if trig.TimeTrigger != nil {
				var (
					init  *int64
					until *int64
				)
				if trig.TimeTrigger.GetInitial() != 0 {
					init = ptr.From(time.Unix(int64(trig.TimeTrigger.GetInitial()), 0).UnixNano())
				}
				every := int(trig.TimeTrigger.GetEvery())
				if trig.TimeTrigger.GetUntil() != 0 {
					until = ptr.From(time.Unix(int64(trig.TimeTrigger.GetUntil()), 0).UnixNano())
				}
				trigger = &EthTimeTrigger{
					Initial: init,
					Every:   &every,
					Until:   until,
				}
			}
		}
	}

	return
}

func resolveSigners(obj []*v1.Signer) (signers []*Signer) {
	for i := range obj {
		signers = append(signers, &Signer{Signer: resolveSigner(obj[i].Signer)})
	}
	return
}

func resolveSigner(obj any) (signer SignerKind) {
	switch sig := obj.(type) {
	case *v1.Signer_PubKey:
		signer = &PubKey{Key: &sig.PubKey.Key}
	case *v1.Signer_EthAddress:
		signer = &ETHAddress{Address: &sig.EthAddress.Address}
	}
	return
}

func resolveFilters(obj []*v1.Filter) (filters []*Filter, e error) {
	filters = []*Filter{}
	if obj != nil {
		for _, f := range obj {
			if f != nil {
				filter, err := resolveFilter(f)
				if err != nil {
					e = err
					return
				}
				filters = append(filters, filter)
			}
		}
		return
	}

	return
}

func resolveFilter(obj *v1.Filter) (filter *Filter, e error) {
	filter = &Filter{
		Key:        &PropertyKey{},
		Conditions: []*Condition{},
	}

	if obj.Key != nil {
		filter.Key = &PropertyKey{
			Name: &obj.Key.Name,
			Type: obj.Key.Type,
		}

		if obj.Key.NumberDecimalPlaces != nil {
			indp := new(int)
			*indp = int(*obj.Key.NumberDecimalPlaces)
			filter.Key.NumberDecimalPlaces = indp
		}
	} else {
		e = errors.New("Property key is empty")
		return
	}

	if obj.Conditions != nil || len(obj.Conditions) > 0 {
		filter.Conditions = resolveConditions(obj.Conditions)
	} else {
		e = errors.New("Conditions list is empty")
		return
	}

	return
}

func resolveConditions(obj []*v1.Condition) (conditions []*Condition) {
	conditions = []*Condition{}
	for _, c := range obj {
		conditions = append(
			conditions,
			&Condition{
				Operator: c.Operator,
				Value:    &c.Value,
			},
		)
	}
	return
}

func resolveNormalisers(obj []*vegapb.Normaliser) (normalisers []*Normaliser) {
	if obj != nil {
		for _, n := range obj {
			normalisers = append(normalisers, resolveNormaliser(n))
		}
		return
	}

	normalisers = []*Normaliser{}
	return
}

func resolveNormaliser(obj any) (normaliser *Normaliser) {
	normaliser = &Normaliser{}

	// if obj != nil {
	switch norm := obj.(type) {
	case *vegapb.Normaliser:
		normaliser = &Normaliser{
			Name:       norm.Name,
			Expression: norm.Expression,
		}
		// return
	}
	// }

	return
}

func resolveDataSourceDefinition(d *vegapb.DataSourceDefinition) (ds *vegapb.DataSourceDefinition) {
	ds = &vegapb.DataSourceDefinition{}
	if d == nil || d.SourceType == nil {
		return ds
	}
	data := d.Content()
	if data != nil {
		switch tp := data.(type) {
		case *vegapb.DataSourceSpecConfiguration:
			ds.SourceType = &vegapb.DataSourceDefinition_External{
				External: &vegapb.DataSourceDefinitionExternal{
					SourceType: &vegapb.DataSourceDefinitionExternal_Oracle{
						Oracle: tp,
					},
				},
			}

		case *vegapb.EthCallSpec:
			ds.SourceType = &vegapb.DataSourceDefinition_External{
				External: &vegapb.DataSourceDefinitionExternal{
					SourceType: &vegapb.DataSourceDefinitionExternal_EthOracle{
						EthOracle: tp,
					},
				},
			}

		case *vegapb.DataSourceSpecConfigurationTime:
			ds.SourceType = &vegapb.DataSourceDefinition_Internal{
				Internal: &vegapb.DataSourceDefinitionInternal{
					SourceType: &vegapb.DataSourceDefinitionInternal_Time{
						Time: tp,
					},
				},
			}
		case *vegapb.DataSourceSpecConfigurationTimeTrigger:
			ds.SourceType = &vegapb.DataSourceDefinition_Internal{
				Internal: &vegapb.DataSourceDefinitionInternal{
					SourceType: &vegapb.DataSourceDefinitionInternal_TimeTrigger{
						TimeTrigger: tp,
					},
				},
			}
		}
	}

	return ds
}

func resolveDataSourceSpec(d *vegapb.DataSourceSpec) (ds *DataSourceSpec) {
	ds = &DataSourceSpec{
		Data: &vegapb.DataSourceDefinition{},
	}
	if d == nil {
		return
	}

	ds.ID = d.GetId()
	ds.CreatedAt = d.CreatedAt
	if d.UpdatedAt != 0 {
		ds.UpdatedAt = ptr.From(d.UpdatedAt)
	}

	switch d.Status {
	case vegapb.DataSourceSpec_STATUS_ACTIVE:
		ds.Status = DataSourceSpecStatusStatusActive
	case vegapb.DataSourceSpec_STATUS_DEACTIVATED:
		ds.Status = DataSourceSpecStatusStatusDeactivated
	}

	if d.Data != nil {
		ds.Data = resolveDataSourceDefinition(d.Data)
	}

	return
}