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) {
    n, err = WriteOnto(self.Type, w, n, err)
    if self.Type == ACCOUNT_TYPE_NUMBER || self.Type == ACCOUNT_TYPE_BOTH {
        n, err = WriteOnto(self.Number, w, n, err)
    }
    if self.Type == ACCOUNT_TYPE_PUBKEY || self.Type == ACCOUNT_TYPE_BOTH {
        n, err = WriteOnto(self.PubKey, w, n, err)
    }
    return
}

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