neo-go/pkg/core/asset_state.go

package core

import (
	"bytes"
	"encoding/binary"
	"io"

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

const feeMode = 0x0

// Assets is mapping between AssetID and the AssetState.
type Assets map[util.Uint256]*AssetState

func (a Assets) commit(b storage.Batch) error {
	buf := new(bytes.Buffer)
	for hash, state := range a {
		if err := state.EncodeBinary(buf); err != nil {
			return err
		}
		key := storage.AppendPrefix(storage.STAsset, hash.Bytes())
		b.Put(key, buf.Bytes())
		buf.Reset()
	}
	return nil
}

// AssetState represents the state of an NEO registered Asset.
type AssetState struct {
	ID         util.Uint256
	AssetType  transaction.AssetType
	Name       string
	Amount     util.Fixed8
	Available  util.Fixed8
	Precision  uint8
	FeeMode    uint8
	FeeAddress util.Uint160
	Owner      *crypto.PublicKey
	Admin      util.Uint160
	Issuer     util.Uint160
	Expiration uint32
	IsFrozen   bool
}

// DecodeBinary implements the Payload interface.
func (a *AssetState) DecodeBinary(r io.Reader) error {
	if err := binary.Read(r, binary.LittleEndian, &a.ID); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.AssetType); err != nil {
		return err
	}

	var err error
	a.Name, err = util.ReadVarString(r)
	if err != nil {
		return err
	}

	if err := binary.Read(r, binary.LittleEndian, &a.Amount); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.Available); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.Precision); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.FeeMode); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.FeeAddress); err != nil {
		return err
	}

	a.Owner = &crypto.PublicKey{}
	if err := a.Owner.DecodeBinary(r); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.Admin); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.Issuer); err != nil {
		return err
	}
	if err := binary.Read(r, binary.LittleEndian, &a.Expiration); err != nil {
		return err
	}
	return binary.Read(r, binary.LittleEndian, &a.IsFrozen)
}

// EncodeBinary implements the Payload interface.
func (a *AssetState) EncodeBinary(w io.Writer) error {
	if err := binary.Write(w, binary.LittleEndian, a.ID); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.AssetType); err != nil {
		return err
	}
	if err := util.WriteVarUint(w, uint64(len(a.Name))); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, []byte(a.Name)); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.Amount); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.Available); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.Precision); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.FeeMode); err != nil {
		return err
	}

	if err := binary.Write(w, binary.LittleEndian, a.FeeAddress); err != nil {
		return err
	}

	if err := a.Owner.EncodeBinary(w); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.Admin); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.Issuer); err != nil {
		return err
	}
	if err := binary.Write(w, binary.LittleEndian, a.Expiration); err != nil {
		return err
	}
	return binary.Write(w, binary.LittleEndian, a.IsFrozen)
}

// GetName returns the asset name based on its type.
func (a *AssetState) GetName() string {

	if a.AssetType == transaction.GoverningToken {
		return "NEO"
	} else if a.AssetType == transaction.UtilityToken {
		return "NEOGas"
	}

	return a.Name
}