neo-go/pkg/rpc/wrappers/account_state.go

package wrappers

import (
	"bytes"
	"sort"

	"github.com/CityOfZion/neo-go/pkg/core"
	"github.com/CityOfZion/neo-go/pkg/crypto/keys"
	"github.com/CityOfZion/neo-go/pkg/util"
)

// AccountState wrapper used for the representation of
// core.AccountState on the RPC Server.
type AccountState struct {
	Version    uint8               `json:"version"`
	ScriptHash util.Uint160        `json:"script_hash"`
	IsFrozen   bool                `json:"frozen"`
	Votes      []*keys.PublicKey `json:"votes"`
	Balances   []Balance           `json:"balances"`
}

// Balances type for sorting balances in rpc response
type Balances []Balance

func (b Balances) Len() int           { return len(b) }
func (b Balances) Less(i, j int) bool { return bytes.Compare(b[i].Asset[:], b[j].Asset[:]) != -1 }
func (b Balances) Swap(i, j int)      { b[i], b[j] = b[j], b[i] }

// Balance response wrapper
type Balance struct {
	Asset util.Uint256 `json:"asset"`
	Value util.Fixed8  `json:"value"`
}

// NewAccountState creates a new AccountState wrapper.
func NewAccountState(a *core.AccountState) AccountState {
	balances := make(Balances, 0, len(a.Balances))
	for k, v := range a.Balances {
		balances = append(balances, Balance{
			Asset: k,
			Value: v,
		})
	}

	sort.Sort(balances)

	// reverse scriptHash to be consistent with other client
	scriptHash, err := util.Uint160DecodeBytes(a.ScriptHash.BytesReverse())
	if err != nil {
		scriptHash = a.ScriptHash
	}

	return AccountState{
		Version:    a.Version,
		ScriptHash: scriptHash,
		IsFrozen:   a.IsFrozen,
		Votes:      a.Votes,
		Balances:   balances,
	}
}