blockresponse.go

package clienttypes

import (
	"encoding/json"
	"errors"
	"math/big"

	handlertypes "github.com/opennetsys/golkadot/client/p2p/handler/types"
	"github.com/opennetsys/golkadot/common/bnutil"
	"github.com/opennetsys/golkadot/common/codec"
	"github.com/opennetsys/golkadot/common/u8util"
	"github.com/opennetsys/golkadot/logger"
)

// Kind ...
func (b *BlockResponse) Kind() handlertypes.FuncEnum {
	return handlertypes.BlockResponse
}

// Encode serializes the message into a bytes array
// TODO: fix...
func (b *BlockResponse) Encode() ([]byte, error) {
	jsn, err := b.MarshalJSON()
	if err != nil {
		return nil, err
	}

	msgBytes, err := codec.Encode(jsn)
	if err != nil {
		return nil, err
	}

	return u8util.Concat(bnutil.ToUint8Slice(big.NewInt(int64(handlertypes.BlockResponse)), 8, true, false), msgBytes), nil
}

// Decode deserializes a bytes array into a message
// TODO: fix...
func (b *BlockResponse) Decode(data []byte) error {
	if data == nil || len(data) == 0 {
		return errors.New("nil data")
	}

	bn := bnutil.ToBN(data[0], true)
	if bn == nil {
		return errors.New("nil kind")
	}

	if bn.Int64() != int64(handlertypes.BlockResponse) {
		logger.Errorf("[block request] expected Block Response, but received %v", bn.Int64())
		return errors.New("wrong kind")
	}

	return b.UnmarshalJSON(data[1:])
}

// MarshalJSON returns json
func (b *BlockResponse) MarshalJSON() ([]byte, error) {
	return json.Marshal(b.Message)
}

// UnmarshalJSON converts json to a message
func (b *BlockResponse) UnmarshalJSON(data []byte) error {
	msg := new(BlockResponseMessage)
	if err := json.Unmarshal(data, msg); err != nil {
		return err
	}

	b.Message = msg
	return nil
}

// GetHeader ...
func (b *BlockResponse) GetHeader() *Header {
	return nil
}