msg_server_attest.go

package keeper

import (
	"context"

	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
	"github.com/tendermint/tendermint/libs/rand"

	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/jackalLabs/canine-chain/v3/x/storage/types"
)

const ( // TODO: Figure out optimal values for these and replace them with chain params
	True = "true"
)

func (k Keeper) Attest(ctx sdk.Context, cid string, creator string) error {
	form, found := k.GetAttestationForm(ctx, cid)
	if !found {
		return sdkerrors.Wrapf(types.ErrAttestInvalid, "cannot find this attestation")
	}

	done := false

	var count int64

	attestations := form.Attestations
	for _, attestation := range attestations {
		if attestation.Provider == creator {
			attestation.Complete = true
			done = true
		}

		if attestation.Complete {
			count++
		}
	}

	if !done {
		return sdkerrors.Wrapf(types.ErrAttestInvalid, "you cannot attest to this deal")
	}

	if count < k.GetParams(ctx).AttestMinToPass {
		form.Attestations = attestations
		k.SetAttestationForm(ctx, form)
		return nil
	}

	deal, found := k.GetActiveDeals(ctx, cid)

	if !found {
		return sdkerrors.Wrapf(types.ErrDealNotFound, "cannot find active deal from form")
	}

	deal.LastProof = ctx.BlockHeight()
	k.SetActiveDeals(ctx, deal)
	k.RemoveAttestation(ctx, cid)

	return nil
}

func (k msgServer) Attest(goCtx context.Context, msg *types.MsgAttest) (*types.MsgAttestResponse, error) {
	ctx := sdk.UnwrapSDKContext(goCtx)

	err := k.Keeper.Attest(ctx, msg.Cid, msg.Creator)
	if err != nil {
		ctx.Logger().Debug(err.Error())
	}

	return &types.MsgAttestResponse{}, nil
}

func (k Keeper) RequestAttestation(ctx sdk.Context, cid string, creator string) ([]string, error) {
	deal, found := k.GetActiveDeals(ctx, cid)
	if !found {
		return nil, sdkerrors.Wrapf(types.ErrDealNotFound, "cannot find active deal for attestation form")
	}

	if deal.Provider != creator {
		return nil, sdkerrors.Wrapf(sdkerrors.ErrUnauthorized, "you do not own this deal")
	}

	_, found = k.GetAttestationForm(ctx, cid)
	if found {
		return nil, sdkerrors.Wrapf(types.ErrAttestAlreadyExists, "attestation form already exists")
	}

	dealProvider := deal.Provider
	provider, found := k.GetProviders(ctx, dealProvider)
	if !found {
		return nil, sdkerrors.Wrapf(types.ErrProviderNotFound, "cannot find provider matching deal")
	}

	providers := k.GetActiveProviders(ctx, provider.Ip) // get a random list of active providers
	params := k.GetParams(ctx)

	if len(providers) < int(params.AttestFormSize) {
		return nil, sdkerrors.Wrapf(types.ErrInvalidLengthQuery, "not enough providers online")
	}

	rand.Seed(ctx.BlockHeight())

	attestations := make([]*types.Attestation, params.AttestFormSize)

	providerAddresses := make([]string, params.AttestFormSize)

	for i := 0; i < int(params.AttestFormSize); i++ {
		p := providers[i]

		providerAddresses[i] = p.Address

		attestations[i] = &types.Attestation{
			Provider: p.Address,
			Complete: false,
		}
	}

	form := types.AttestationForm{
		Attestations: attestations,
		Cid:          cid,
	}

	k.SetAttestationForm(ctx, form)

	return providerAddresses, nil
}

func (k msgServer) RequestAttestationForm(goCtx context.Context, msg *types.MsgRequestAttestationForm) (*types.MsgRequestAttestationFormResponse, error) {
	ctx := sdk.UnwrapSDKContext(goCtx)
	cid := msg.Cid
	creator := msg.Creator

	providerAddresses, err := k.RequestAttestation(ctx, cid, creator)

	success := true

	errorString := ""

	if err != nil {
		success = false
		errorString = err.Error()
	}

	return &types.MsgRequestAttestationFormResponse{
		Providers: providerAddresses,
		Success:   success,
		Error:     errorString,
		Cid:       cid,
	}, nil
}