forked from TrueCloudLab/neoneo-go
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package result
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/blockchainer"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
)
|
|
|
|
type (
|
|
// Header wrapper used for the representation of
|
|
// block header on the RPC Server.
|
|
Header struct {
|
|
block.Header
|
|
BlockMetadata
|
|
}
|
|
)
|
|
|
|
// NewHeader creates a new Header wrapper.
|
|
func NewHeader(h *block.Header, chain blockchainer.Blockchainer) Header {
|
|
res := Header{
|
|
Header: *h,
|
|
BlockMetadata: BlockMetadata{
|
|
Size: io.GetVarSize(h),
|
|
Confirmations: chain.BlockHeight() - h.Index + 1,
|
|
},
|
|
}
|
|
|
|
hash := chain.GetHeaderHash(int(h.Index) + 1)
|
|
if !hash.Equals(util.Uint256{}) {
|
|
res.NextBlockHash = &hash
|
|
}
|
|
return res
|
|
}
|
|
|
|
// MarshalJSON implements json.Marshaler interface.
|
|
func (h Header) MarshalJSON() ([]byte, error) {
|
|
output, err := json.Marshal(h.BlockMetadata)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
baseBytes, err := json.Marshal(h.Header)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// We have to keep both "fields" at the same level in json in order to
|
|
// match C# API, so there's no way to marshall Block correctly with
|
|
// standard json.Marshaller tool.
|
|
if output[len(output)-1] != '}' || baseBytes[0] != '{' {
|
|
return nil, errors.New("can't merge internal jsons")
|
|
}
|
|
output[len(output)-1] = ','
|
|
output = append(output, baseBytes[1:]...)
|
|
return output, nil
|
|
}
|
|
|
|
// UnmarshalJSON implements json.Unmarshaler interface.
|
|
func (h *Header) UnmarshalJSON(data []byte) error {
|
|
// As block.Block and BlockMetadata are at the same level in json,
|
|
// do unmarshalling separately for both structs.
|
|
meta := new(BlockMetadata)
|
|
err := json.Unmarshal(data, meta)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = json.Unmarshal(data, &h.Header)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
h.BlockMetadata = *meta
|
|
return nil
|
|
}
|