woop/rosetta/common/operations.go

package common

import (
	"encoding/json"
	"fmt"

	"github.com/harmony-one/harmony/internal/common"

	"github.com/coinbase/rosetta-sdk-go/types"

	rpcV2 "github.com/harmony-one/harmony/rpc/v2"
	staking "github.com/harmony-one/harmony/staking/types"
)

const (
	// ExpendGasOperation is an operation that only affects the native currency.
	ExpendGasOperation = "Gas"

	// ContractCreationOperation is an operation that only affects the native currency.
	ContractCreationOperation = "ContractCreation"

	// NativeTransferOperation is an operation that only affects the native currency.
	NativeTransferOperation = "NativeTransfer"

	// NativeCrossShardTransferOperation is an operation that only affects the native currency.
	NativeCrossShardTransferOperation = "NativeCrossShardTransfer"

	// CreateValidatorOperation is an operation that only affects the native currency.
	CreateValidatorOperation = "CreateValidator"

	// EditValidatorOperation is an operation that only affects the native currency.
	EditValidatorOperation = "EditValidator"

	// DelegateOperation is an operation that only affects the native currency.
	DelegateOperation = "Delegate"

	// UndelegateOperation is an operation that only affects the native currency.
	UndelegateOperation = "Undelegate"

	// CollectRewardsOperation is an operation that only affects the native currency.
	CollectRewardsOperation = "CollectRewards"

	// GenesisFundsOperation is a side effect operation for genesis block only.
	// Note that no transaction can be constructed with this operation.
	GenesisFundsOperation = "Genesis"

	// PreStakingBlockRewardOperation is a side effect operation for pre-staking era only.
	// Note that no transaction can be constructed with this operation.
	PreStakingBlockRewardOperation = "PreStakingBlockReward"

	// UndelegationPayoutOperation is a side effect operation for committee election block only.
	// Note that no transaction can be constructed with this operation.
	UndelegationPayoutOperation = "UndelegationPayout"
)

var (
	// PlainOperationTypes ..
	PlainOperationTypes = []string{
		ExpendGasOperation,
		NativeTransferOperation,
		NativeCrossShardTransferOperation,
		ContractCreationOperation,
		GenesisFundsOperation,
		PreStakingBlockRewardOperation,
		UndelegationPayoutOperation,
	}

	// StakingOperationTypes ..
	StakingOperationTypes = []string{
		staking.DirectiveCreateValidator.String(),
		staking.DirectiveEditValidator.String(),
		staking.DirectiveDelegate.String(),
		staking.DirectiveUndelegate.String(),
		staking.DirectiveCollectRewards.String(),
	}

	// MutuallyExclusiveOperations for invariant: A transaction can only contain 1 type of 'native' operation.
	MutuallyExclusiveOperations = map[string]interface{}{
		NativeTransferOperation:           struct{}{},
		NativeCrossShardTransferOperation: struct{}{},
	}
)

var (
	// SuccessOperationStatus for tx operations who's amount affects the account
	SuccessOperationStatus = &types.OperationStatus{
		Status:     "success",
		Successful: true,
	}

	// ContractFailureOperationStatus for tx operations who's amount does not affect the account
	// due to a contract call failure (but still incurs gas).
	ContractFailureOperationStatus = &types.OperationStatus{
		Status:     "contract_failure",
		Successful: false,
	}

	// FailureOperationStatus ..
	FailureOperationStatus = &types.OperationStatus{
		Status:     "failure",
		Successful: false,
	}
)

// CreateValidatorOperationMetadata ..
type CreateValidatorOperationMetadata rpcV2.CreateValidatorMsg

// EditValidatorOperationMetadata ..
type EditValidatorOperationMetadata rpcV2.EditValidatorMsg

// DelegateOperationMetadata ..
type DelegateOperationMetadata rpcV2.DelegateMsg

// UndelegateOperationMetadata ..
type UndelegateOperationMetadata rpcV2.UndelegateMsg

// CollectRewardsMetadata ..
type CollectRewardsMetadata rpcV2.CollectRewardsMsg

// CrossShardTransactionOperationMetadata ..
type CrossShardTransactionOperationMetadata struct {
	From *types.AccountIdentifier `json:"from"`
	To   *types.AccountIdentifier `json:"to"`
}

// UnmarshalFromInterface ..
func (s *CrossShardTransactionOperationMetadata) UnmarshalFromInterface(data interface{}) error {
	var T CrossShardTransactionOperationMetadata
	dat, err := json.Marshal(data)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(dat, &T); err != nil {
		return err
	}
	if T.To == nil || T.From == nil {
		return fmt.Errorf("expected to & from to be present for CrossShardTransactionOperationMetadata")
	}
	*s = T
	return nil
}

func (s *CreateValidatorOperationMetadata) UnmarshalFromInterface(data interface{}) error {
	var T CreateValidatorOperationMetadata
	dat, err := json.Marshal(data)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(dat, &T); err != nil {
		return err
	}
	*s = T
	return nil
}

func (s *EditValidatorOperationMetadata) UnmarshalFromInterface(data interface{}) error {
	var T EditValidatorOperationMetadata
	dat, err := json.Marshal(data)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(dat, &T); err != nil {
		return err
	}
	*s = T
	return nil
}

func (s *DelegateOperationMetadata) UnmarshalFromInterface(data interface{}) error {
	var T DelegateOperationMetadata
	dat, err := json.Marshal(data)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(dat, &T); err != nil {
		return err
	}
	if T.Amount == nil || T.ValidatorAddress == "" || T.DelegatorAddress == "" {
		return fmt.Errorf("expected validator address & delegator address & amount be present for DelegateOperationMetadata")
	}

	if !common.IsBech32Address(T.ValidatorAddress) || !common.IsBech32Address(T.DelegatorAddress) {
		return fmt.Errorf("expected validator address & delegator address to be bech32 format for DelegateOperationMetadata")
	}

	*s = T
	return nil
}

func (s *UndelegateOperationMetadata) UnmarshalFromInterface(data interface{}) error {
	var T UndelegateOperationMetadata
	dat, err := json.Marshal(data)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(dat, &T); err != nil {
		return err
	}

	if T.Amount == nil || T.ValidatorAddress == "" || T.DelegatorAddress == "" {
		return fmt.Errorf("expected validator address & delegator address & amount be present for UndelegateOperationMetadata")
	}

	if !common.IsBech32Address(T.ValidatorAddress) || !common.IsBech32Address(T.DelegatorAddress) {
		return fmt.Errorf("expected validator address & delegator address to be bech32 format for UndelegateOperationMetadata")
	}

	*s = T
	return nil
}

func (s *CollectRewardsMetadata) UnmarshalFromInterface(data interface{}) error {
	var T CollectRewardsMetadata
	dat, err := json.Marshal(data)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(dat, &T); err != nil {
		return err
	}
	if T.DelegatorAddress == "" {
		return fmt.Errorf("expected delegator address be present for CollectRewardsMetadata")
	}
	if !common.IsBech32Address(T.DelegatorAddress) {
		return fmt.Errorf("expected delegator address to be bech32 format for CollectRewardsMetadata")
	}
	*s = T
	return nil
}