actions.go

package eos

import (
	"bytes"
	"crypto/sha256"
	"encoding/json"
	"fmt"
	"reflect"
)

// See: libraries/chain/include/eosio/chain/contracts/types.hpp:203
// See: build/contracts/eosio.system/eosio.system.abi

// SetCode represents the hard-coded `setcode` action.
type SetCode struct {
	Account   AccountName `json:"account"`
	VMType    byte        `json:"vmtype"`
	VMVersion byte        `json:"vmversion"`
	Code      HexBytes    `json:"code"`
}

// SetABI represents the hard-coded `setabi` action.
type SetABI struct {
	Account AccountName `json:"account"`
	ABI     ABI         `json:"abi"`
}

// Action
type Action struct {
	Account       AccountName       `json:"account"`
	Name          ActionName        `json:"name"`
	Authorization []PermissionLevel `json:"authorization,omitempty"`
	ActionData
}

func (a Action) Digest() Checksum256 {
	toEat := jsonActionToServer{
		Account:       a.Account,
		Name:          a.Name,
		Authorization: a.Authorization,
		Data:          a.ActionData.HexData,
	}
	bin, err := MarshalBinary(toEat)
	if err != nil {
		panic("this should never panic, we know it marshals properly all the time")
	}

	h := sha256.New()
	_, _ = h.Write(bin)
	return h.Sum(nil)
}

type ActionData struct {
	HexData  HexBytes    `json:"hex_data,omitempty"`
	Data     interface{} `json:"data,omitempty" eos:"-"`
	abi      []byte      // TBD: we could use the ABI to decode in obj
	toServer bool
}

func NewActionData(obj interface{}) ActionData {
	return ActionData{
		HexData:  []byte{},
		Data:     obj,
		toServer: true,
	}
}

func NewActionDataFromHexData(data []byte) ActionData {
	return ActionData{
		HexData:  data,
		Data:     nil,
		toServer: true,
	}
}

func (a *ActionData) SetToServer(toServer bool) {
	// FIXME: let's clarify this design, make it simpler..
	// toServer doesn't speak of the intent..
	a.toServer = toServer
}

// jsonActionToServer represents what /v1/chain/push_transaction
// expects, which isn't allllways the same everywhere.
type jsonActionToServer struct {
	Account       AccountName       `json:"account"`
	Name          ActionName        `json:"name"`
	Authorization []PermissionLevel `json:"authorization"`
	Data          HexBytes          `json:"data,omitempty"`
}

type jsonActionFromServer struct {
	Account       AccountName       `json:"account"`
	Name          ActionName        `json:"name"`
	Authorization []PermissionLevel `json:"authorization"`
	Data          interface{}       `json:"data,omitempty"`
	HexData       HexBytes          `json:"hex_data,omitempty"`
}

func (a *Action) MarshalJSON() ([]byte, error) {
	auths := a.Authorization
	if auths == nil {
		auths = make([]PermissionLevel, 0)
	}
	if a.toServer {
		data, err := a.ActionData.EncodeActionData()
		if err != nil {
			return nil, err
		}

		return json.Marshal(&jsonActionToServer{
			Account:       a.Account,
			Name:          a.Name,
			Authorization: auths,
			Data:          data,
		})
	}

	return json.Marshal(&jsonActionFromServer{
		Account:       a.Account,
		Name:          a.Name,
		Authorization: auths,
		HexData:       a.HexData,
		Data:          a.Data,
	})
}

func (data *ActionData) EncodeActionData() ([]byte, error) {
	if data.Data == nil {
		return data.HexData, nil
	}

	buf := new(bytes.Buffer)
	encoder := NewEncoder(buf)

	if err := encoder.Encode(data.Data); err != nil {
		return nil, err
	}

	return buf.Bytes(), nil
}

func (a *Action) MapToRegisteredAction() error {
	src, ok := a.ActionData.Data.(map[string]interface{})
	if !ok {
		return nil
	}

	actionMap := RegisteredActions[a.Account]

	var decodeInto reflect.Type
	if actionMap != nil {
		objType := actionMap[a.Name]
		if objType != nil {
			decodeInto = objType
		}
	}
	if decodeInto == nil {
		return nil
	}

	obj := reflect.New(decodeInto)
	objIface := obj.Interface()

	cnt, err := json.Marshal(src)
	if err != nil {
		return fmt.Errorf("marshaling data: %w", err)
	}
	err = json.Unmarshal(cnt, objIface)
	if err != nil {
		return fmt.Errorf("json unmarshal into registered actions: %w", err)
	}

	a.ActionData.Data = objIface

	return nil
}