monza/encoding.go

package main

import (
	"encoding/hex"
	"fmt"
	"time"

	"github.com/nspcc-dev/neo-go/pkg/core/block"
	"github.com/nspcc-dev/neo-go/pkg/core/state"
	"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
)

const nonCompatibleMsg = "not compatible with FrostFS"

func PrintEvent(b *block.Block, n state.NotificationEvent, extra string) {
	d := time.Unix(int64(b.Timestamp/1e3), 0)
	s := fmt.Sprintf("block:%d at:%s name:%s",
		b.Index, d.Format(time.RFC3339), n.Name,
	)

	if len(extra) != 0 {
		s += fmt.Sprintf(" [%s]", extra)
	}

	fmt.Println(s)
}

func PrintTransfer(b *block.Block, n state.NotificationEvent) {
	const nonCompatibleMsg = "not NEP-17 compatible"

	items, ok := n.Item.Value().([]stackitem.Item)
	if !ok {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	if len(items) != 3 {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	snd, err := items[0].TryBytes()
	if err != nil {
		snd = nil
	}

	rcv, err := items[1].TryBytes()
	if err != nil {
		rcv = nil
	}

	bigAmount, err := items[2].TryInteger()
	if err != nil {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	var sndStr, rcvStr = "nil", "nil"
	if snd != nil {
		sndStr = hex.EncodeToString(revertBytes(snd))
	}

	if rcv != nil {
		rcvStr = hex.EncodeToString(revertBytes(rcv))
	}

	d := time.Unix(int64(b.Timestamp/1e3), 0)

	s := fmt.Sprintf("block:%d at:%s name:%s from:%s to:%s amount:%d",
		b.Index, d.Format(time.RFC3339), n.Name,
		sndStr, rcvStr, bigAmount.Int64(),
	)

	fmt.Println(s)
}

func PrintNewEpoch(b *block.Block, n state.NotificationEvent) {
	items, ok := n.Item.Value().([]stackitem.Item)
	if !ok {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	if len(items) != 1 {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	epoch, err := items[0].TryInteger()
	if err != nil {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	d := time.Unix(int64(b.Timestamp/1e3), 0)

	s := fmt.Sprintf("block:%d at:%s name:%s epoch:%d",
		b.Index, d.Format(time.RFC3339), n.Name, epoch,
	)

	fmt.Println(s)
}

func PrintAddPeerSuccess(b *block.Block, n state.NotificationEvent) {
	items, ok := n.Item.Value().([]stackitem.Item)
	if !ok {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	if len(items) != 1 {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	data, err := items[0].TryBytes()
	if err != nil {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	d := time.Unix(int64(b.Timestamp/1e3), 0)

	s := fmt.Sprintf("block:%d at:%s name:%s pubkey:[..%s]",
		b.Index, d.Format(time.RFC3339), n.Name,
		hex.EncodeToString(data[len(data)-3:]),
	)

	fmt.Println(s)
}

func PrintUpdateState(b *block.Block, n state.NotificationEvent) {
	items, ok := n.Item.Value().([]stackitem.Item)
	if !ok {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	if len(items) != 2 {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	st, err := items[0].TryInteger()
	if err != nil {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	var stateStr string
	switch v := st.Uint64(); v {
	case 1:
		stateStr = "online"
	case 2:
		stateStr = "offline"
	default:
		stateStr = fmt.Sprintf("%d(unknown)", v)
	}

	pubkey, err := items[1].TryBytes()
	if err != nil {
		PrintEvent(b, n, nonCompatibleMsg)
		return
	}

	d := time.Unix(int64(b.Timestamp/1e3), 0)

	s := fmt.Sprintf("block:%d at:%s name:%s pubkey:[..%s] state:%s",
		b.Index, d.Format(time.RFC3339), n.Name,
		hex.EncodeToString(pubkey[len(pubkey)-3:]),
		stateStr,
	)

	fmt.Println(s)
}

func PrintBlock(b *block.Block, extra string) {
	d := time.Unix(int64(b.Timestamp/1e3), 0)
	s := fmt.Sprintf("block:%d at:%s", b.Index, d.Format(time.RFC3339))

	if len(extra) != 0 {
		s += fmt.Sprintf(" [%s]", extra)
	}

	fmt.Println(s)
}

func revertBytes(data []byte) []byte {
	ln := len(data)
	for i := 0; i < ln/2; i++ {
		data[i], data[ln-1-i] = data[ln-1-i], data[i]
	}
	return data
}