gecko/network/builder.go

// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package network

import (
	"github.com/ava-labs/gecko/ids"
	"github.com/ava-labs/gecko/utils"
)

// Builder extends a Codec to build messages safely
type Builder struct{ Codec }

// GetVersion message
func (m Builder) GetVersion() (Msg, error) { return m.Pack(GetVersion, nil) }

// Version message
func (m Builder) Version(networkID, nodeID uint32, myTime uint64, ip utils.IPDesc, myVersion string) (Msg, error) {
	return m.Pack(Version, map[Field]interface{}{
		NetworkID:  networkID,
		NodeID:     nodeID,
		MyTime:     myTime,
		IP:         ip,
		VersionStr: myVersion,
	})
}

// GetPeerList message
func (m Builder) GetPeerList() (Msg, error) { return m.Pack(GetPeerList, nil) }

// PeerList message
func (m Builder) PeerList(ipDescs []utils.IPDesc) (Msg, error) {
	return m.Pack(PeerList, map[Field]interface{}{Peers: ipDescs})
}

// Ping message
func (m Builder) Ping() (Msg, error) { return m.Pack(Ping, nil) }

// Pong message
func (m Builder) Pong() (Msg, error) { return m.Pack(Pong, nil) }

// GetAcceptedFrontier message
func (m Builder) GetAcceptedFrontier(chainID ids.ID, requestID uint32) (Msg, error) {
	return m.Pack(GetAcceptedFrontier, map[Field]interface{}{
		ChainID:   chainID.Bytes(),
		RequestID: requestID,
	})
}

// AcceptedFrontier message
func (m Builder) AcceptedFrontier(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
	containerIDBytes := make([][]byte, containerIDs.Len())
	for i, containerID := range containerIDs.List() {
		containerIDBytes[i] = containerID.Bytes()
	}
	return m.Pack(AcceptedFrontier, map[Field]interface{}{
		ChainID:      chainID.Bytes(),
		RequestID:    requestID,
		ContainerIDs: containerIDBytes,
	})
}

// GetAccepted message
func (m Builder) GetAccepted(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
	containerIDBytes := make([][]byte, containerIDs.Len())
	for i, containerID := range containerIDs.List() {
		containerIDBytes[i] = containerID.Bytes()
	}
	return m.Pack(GetAccepted, map[Field]interface{}{
		ChainID:      chainID.Bytes(),
		RequestID:    requestID,
		ContainerIDs: containerIDBytes,
	})
}

// Accepted message
func (m Builder) Accepted(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
	containerIDBytes := make([][]byte, containerIDs.Len())
	for i, containerID := range containerIDs.List() {
		containerIDBytes[i] = containerID.Bytes()
	}
	return m.Pack(Accepted, map[Field]interface{}{
		ChainID:      chainID.Bytes(),
		RequestID:    requestID,
		ContainerIDs: containerIDBytes,
	})
}

// Get message
func (m Builder) Get(chainID ids.ID, requestID uint32, containerID ids.ID) (Msg, error) {
	return m.Pack(Get, map[Field]interface{}{
		ChainID:     chainID.Bytes(),
		RequestID:   requestID,
		ContainerID: containerID.Bytes(),
	})
}

// GetAncestors message
func (m Builder) GetAncestors(chainID ids.ID, requestID uint32, containerID ids.ID) (Msg, error) {
	return m.Pack(GetAncestors, map[Field]interface{}{
		ChainID:     chainID.Bytes(),
		RequestID:   requestID,
		ContainerID: containerID.Bytes(),
	})
}

// Put message
func (m Builder) Put(chainID ids.ID, requestID uint32, containerID ids.ID, container []byte) (Msg, error) {
	return m.Pack(Put, map[Field]interface{}{
		ChainID:        chainID.Bytes(),
		RequestID:      requestID,
		ContainerID:    containerID.Bytes(),
		ContainerBytes: container,
	})
}

// MultiPut message
func (m Builder) MultiPut(chainID ids.ID, requestID uint32, containers [][]byte) (Msg, error) {
	return m.Pack(MultiPut, map[Field]interface{}{
		ChainID:             chainID.Bytes(),
		RequestID:           requestID,
		MultiContainerBytes: containers,
	})
}

// PushQuery message
func (m Builder) PushQuery(chainID ids.ID, requestID uint32, containerID ids.ID, container []byte) (Msg, error) {
	return m.Pack(PushQuery, map[Field]interface{}{
		ChainID:        chainID.Bytes(),
		RequestID:      requestID,
		ContainerID:    containerID.Bytes(),
		ContainerBytes: container,
	})
}

// PullQuery message
func (m Builder) PullQuery(chainID ids.ID, requestID uint32, containerID ids.ID) (Msg, error) {
	return m.Pack(PullQuery, map[Field]interface{}{
		ChainID:     chainID.Bytes(),
		RequestID:   requestID,
		ContainerID: containerID.Bytes(),
	})
}

// Chits message
func (m Builder) Chits(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
	containerIDBytes := make([][]byte, containerIDs.Len())
	for i, containerID := range containerIDs.List() {
		containerIDBytes[i] = containerID.Bytes()
	}
	return m.Pack(Chits, map[Field]interface{}{
		ChainID:      chainID.Bytes(),
		RequestID:    requestID,
		ContainerIDs: containerIDBytes,
	})
}