chain_events.go

package vega

import (
	"errors"
	"fmt"
)

var (
	ErrUnsupportedChainEvent = errors.New("unsupported chain event")
)

func (c *BuiltinAssetEvent) PrepareToSign() ([]byte, error) {
	switch act := c.Action.(type) {
	case *BuiltinAssetEvent_Deposit:
		return act.Deposit.PrepareToSign()
	case *BuiltinAssetEvent_Withdrawal:
		return act.Withdrawal.PrepareToSign()
	default:
		return nil, ErrUnsupportedChainEvent
	}
}

func (c *BuiltinAssetDeposit) PrepareToSign() ([]byte, error) {
	out := []byte{}
	out = append(out, []byte(c.VegaAssetId)...)
	out = append(out, []byte(c.PartyId)...)
	out = append(out, []byte(fmt.Sprintf("%v", c.Amount))...)
	return out, nil
}

func (c *BuiltinAssetWithdrawal) PrepareToSign() ([]byte, error) {
	out := []byte{}
	out = append(out, []byte(c.VegaAssetId)...)
	out = append(out, []byte(c.PartyId)...)
	out = append(out, []byte(fmt.Sprintf("%v", c.Amount))...)
	return out, nil
}

func (c *ERC20Event) PrepareToSign() ([]byte, error) {
	out := []byte{}
	out = append(out, []byte(fmt.Sprintf("%v", c.Block))...)
	out = append(out, []byte(fmt.Sprintf("%v", c.Index))...)

	var (
		next []byte
		err  error
	)
	switch act := c.Action.(type) {
	case *ERC20Event_AssetList:
		next, err = act.AssetList.PrepareToSign()
	case *ERC20Event_AssetDelist:
		next, err = act.AssetDelist.PrepareToSign()
	case *ERC20Event_Deposit:
		next, err = act.Deposit.PrepareToSign()
	case *ERC20Event_Withdrawal:
		next, err = act.Withdrawal.PrepareToSign()
	default:
		err = ErrUnsupportedChainEvent
	}

	if err != nil {
		return nil, err
	}

	out = append(out, next...)
	return out, nil
}

func (c *ERC20AssetList) PrepareToSign() ([]byte, error) {
	return []byte(c.VegaAssetId), nil
}

func (c *ERC20AssetDelist) PrepareToSign() ([]byte, error) {
	return []byte(c.VegaAssetId), nil
}

func (c *ERC20Deposit) PrepareToSign() ([]byte, error) {
	out := []byte{}
	out = append(out, []byte(c.VegaAssetId)...)
	out = append(out, []byte(c.SourceEthereumAddress)...)
	out = append(out, []byte(c.TargetPartyId)...)
	out = append(out, []byte(c.Amount)...)
	return out, nil
}

func (c *ERC20Withdrawal) PrepareToSign() ([]byte, error) {
	out := []byte{}
	out = append(out, []byte(c.VegaAssetId)...)
	out = append(out, []byte(c.TargetEthereumAddress)...)
	out = append(out, []byte(c.ReferenceNonce)...)
	return out, nil
}