package gql import ( "context" "encoding/json" "fmt" "reflect" // #nosec G702 "strconv" auctiontypes "github.com/cerc-io/laconicd/x/auction/types" bondtypes "github.com/cerc-io/laconicd/x/bond/types" registrytypes "github.com/cerc-io/laconicd/x/registry/types" sdk "github.com/cosmos/cosmos-sdk/types" ) // OwnerAttributeName denotes the owner attribute name for a bond. const OwnerAttributeName = "owner" // BondIDAttributeName denotes the record bond ID. const BondIDAttributeName = "bondId" // ExpiryTimeAttributeName denotes the record expiry time. const ExpiryTimeAttributeName = "expiryTime" func getGQLCoin(coin sdk.Coin) *Coin { gqlCoin := Coin{ Type: coin.Denom, Quantity: coin.Amount.BigInt().String(), } return &gqlCoin } func getGQLCoins(coins sdk.Coins) []*Coin { gqlCoins := make([]*Coin, len(coins)) for index, coin := range coins { gqlCoins[index] = getGQLCoin(coin) } return gqlCoins } func GetGQLNameAuthorityRecord(record *registrytypes.NameAuthority) (*AuthorityRecord, error) { if record == nil { return nil, nil } return &AuthorityRecord{ OwnerAddress: record.OwnerAddress, OwnerPublicKey: record.OwnerPublicKey, Height: strconv.FormatUint(record.Height, 10), Status: record.Status, BondID: record.GetBondId(), ExpiryTime: record.GetExpiryTime().String(), }, nil } func getGQLRecord(ctx context.Context, resolver QueryResolver, record registrytypes.Record) (*Record, error) { // Nil record. if record.Deleted { return nil, nil } recordType := record.ToReadableRecord() attributes, err := getAttributes(&recordType) if err != nil { return nil, err } references, err := getReferences(ctx, resolver, &recordType) if err != nil { return nil, err } return &Record{ ID: record.Id, BondID: record.GetBondId(), CreateTime: record.GetCreateTime(), ExpiryTime: record.GetExpiryTime(), Owners: record.GetOwners(), Names: record.GetNames(), Attributes: attributes, References: references, }, nil } func getGQLNameRecord(record *registrytypes.NameRecord) (*NameRecord, error) { if record == nil { return nil, fmt.Errorf("got nil record") } records := make([]*NameRecordEntry, len(record.History)) for index, entry := range record.History { records[index] = getNameRecordEntry(entry) } return &NameRecord{ Latest: getNameRecordEntry(record.Latest), History: records, }, nil } func getNameRecordEntry(record *registrytypes.NameRecordEntry) *NameRecordEntry { return &NameRecordEntry{ ID: record.Id, Height: strconv.FormatUint(record.Height, 10), } } func getGQLBond(bondObj *bondtypes.Bond) (*Bond, error) { // Nil record. if bondObj == nil { return nil, nil } return &Bond{ ID: bondObj.Id, Owner: bondObj.Owner, Balance: getGQLCoins(bondObj.Balance), }, nil } func getAuctionBid(bid *auctiontypes.Bid) *AuctionBid { return &AuctionBid{ BidderAddress: bid.BidderAddress, Status: bid.Status, CommitHash: bid.CommitHash, CommitTime: bid.GetCommitTime(), RevealTime: bid.GetRevealTime(), CommitFee: getGQLCoin(bid.CommitFee), RevealFee: getGQLCoin(bid.RevealFee), BidAmount: getGQLCoin(bid.BidAmount), } } func GetGQLAuction(auction *auctiontypes.Auction, bids []*auctiontypes.Bid) (*Auction, error) { if auction == nil { return nil, nil } gqlAuction := Auction{ ID: auction.Id, Status: auction.Status, OwnerAddress: auction.OwnerAddress, CreateTime: auction.GetCreateTime(), CommitsEndTime: auction.GetCommitsEndTime(), RevealsEndTime: auction.GetRevealsEndTime(), CommitFee: getGQLCoin(auction.CommitFee), RevealFee: getGQLCoin(auction.RevealFee), MinimumBid: getGQLCoin(auction.MinimumBid), WinnerAddress: auction.WinnerAddress, WinnerBid: getGQLCoin(auction.WinningBid), WinnerPrice: getGQLCoin(auction.WinningPrice), } auctionBids := make([]*AuctionBid, len(bids)) for index, entry := range bids { auctionBids[index] = getAuctionBid(entry) } gqlAuction.Bids = auctionBids return &gqlAuction, nil } func getReferences(ctx context.Context, resolver QueryResolver, r *registrytypes.ReadableRecord) ([]*Record, error) { var ids []string // #nosec G705 for key := range r.Attributes { //nolint: all switch r.Attributes[key].(type) { case interface{}: if obj, ok := r.Attributes[key].(map[string]interface{}); ok { if _, ok := obj["/"]; ok && len(obj) == 1 { if _, ok := obj["/"].(string); ok { ids = append(ids, obj["/"].(string)) } } } } } return resolver.GetRecordsByIds(ctx, ids) } func getAttributes(r *registrytypes.ReadableRecord) ([]*KeyValue, error) { return mapToKeyValuePairs(r.Attributes) } func mapToKeyValuePairs(attrs map[string]interface{}) ([]*KeyValue, error) { kvPairs := []*KeyValue{} trueVal := true falseVal := false // #nosec G705 for key, value := range attrs { kvPair := &KeyValue{ Key: key, Value: &Value{}, } switch val := value.(type) { case nil: kvPair.Value.Null = &trueVal case int: kvPair.Value.Int = &val case float64: kvPair.Value.Float = &val case string: kvPair.Value.String = &val case bool: kvPair.Value.Boolean = &val case interface{}: if obj, ok := value.(map[string]interface{}); ok { if _, ok := obj["/"]; ok && len(obj) == 1 { if _, ok := obj["/"].(string); ok { kvPair.Value.Reference = &Reference{ ID: obj["/"].(string), } } } else { bytes, err := json.Marshal(obj) if err != nil { return nil, err } jsonStr := string(bytes) kvPair.Value.JSON = &jsonStr } } } if kvPair.Value.Null == nil { kvPair.Value.Null = &falseVal } valueType := reflect.ValueOf(value) if valueType.Kind() == reflect.Slice { bytes, err := json.Marshal(value) if err != nil { return nil, err } jsonStr := string(bytes) kvPair.Value.JSON = &jsonStr } kvPairs = append(kvPairs, kvPair) } return kvPairs, nil } func parseRequestAttributes(attrs []*KeyValueInput) []*registrytypes.QueryListRecordsRequest_KeyValueInput { kvPairs := []*registrytypes.QueryListRecordsRequest_KeyValueInput{} for _, value := range attrs { kvPair := ®istrytypes.QueryListRecordsRequest_KeyValueInput{ Key: value.Key, Value: ®istrytypes.QueryListRecordsRequest_ValueInput{}, } if value.Value.String != nil { kvPair.Value.String_ = *value.Value.String kvPair.Value.Type = "string" } if value.Value.Int != nil { kvPair.Value.Int = int64(*value.Value.Int) kvPair.Value.Type = "int" } if value.Value.Float != nil { kvPair.Value.Float = *value.Value.Float kvPair.Value.Type = "float" } if value.Value.Boolean != nil { kvPair.Value.Boolean = *value.Value.Boolean kvPair.Value.Type = "boolean" } if value.Value.Reference != nil { reference := ®istrytypes.QueryListRecordsRequest_ReferenceInput{ Id: value.Value.Reference.ID, } kvPair.Value.Reference = reference kvPair.Value.Type = "reference" } // TODO: Handle arrays. kvPairs = append(kvPairs, kvPair) } return kvPairs }