tendermint/blocks/account.go

package blocks

import (
    . "github.com/tendermint/tendermint/binary"
    "io"
)

type AccountId struct {
    Type            Byte
    Number          UInt64
    PubKey          ByteSlice
}

const (
    ACCOUNT_TYPE_NUMBER =   Byte(0x01)
    ACCOUNT_TYPE_PUBKEY =   Byte(0x02)
    ACCOUNT_TYPE_BOTH =     Byte(0x03)
)

func ReadAccountId(r io.Reader) AccountId {
    switch t := ReadByte(r); t {
    case ACCOUNT_TYPE_NUMBER:
        return AccountId{t, ReadUInt64(r), nil}
    case ACCOUNT_TYPE_PUBKEY:
        return AccountId{t, 0, ReadByteSlice(r)}
    case ACCOUNT_TYPE_BOTH:
        return AccountId{t, ReadUInt64(r), ReadByteSlice(r)}
    default:
        panicf("Unknown AccountId type %x", t)
        return AccountId{}
    }
}

func (self *AccountId) WriteTo(w io.Writer) (n int64, err error) {
    var n_ int64
    n_, err = self.Type.WriteTo(w)
    n += n_; if err != nil { return n, err }
    if self.Type == ACCOUNT_TYPE_NUMBER ||
       self.Type == ACCOUNT_TYPE_BOTH {
        n_, err = self.Number.WriteTo(w)
        n += n_; if err != nil { return n, err }
    }
    if self.Type == ACCOUNT_TYPE_PUBKEY ||
       self.Type == ACCOUNT_TYPE_BOTH {
        n_, err = self.PubKey.WriteTo(w)
        n += n_; if err != nil { return n, err }
    }
    return
}

func AccountNumber(n UInt64) AccountId {
    return AccountId{ACCOUNT_TYPE_NUMBER, n, nil}
}