95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package types
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
"github.com/cosmos/ethermint/x/bank"
|
|
|
|
ethcmn "github.com/ethereum/go-ethereum/common"
|
|
)
|
|
|
|
var _ auth.Account = (*Account)(nil)
|
|
|
|
// BaseAccount implements the auth.Account interface and embeds an
|
|
// auth.BaseAccount type. It is compatible with the auth.AccountMapper.
|
|
type Account struct {
|
|
*auth.BaseAccount
|
|
|
|
Root ethcmn.Hash // merkle root of the storage trie
|
|
CodeHash []byte
|
|
}
|
|
|
|
// ProtoBaseAccount defines the prototype function for BaseAccount used for an
|
|
// account mapper.
|
|
func ProtoBaseAccount() auth.Account {
|
|
return &Account{BaseAccount: &auth.BaseAccount{}}
|
|
}
|
|
|
|
// Balance returns the balance of an account.
|
|
func (acc Account) Balance() sdk.Int {
|
|
return acc.GetCoins().AmountOf(bank.DenomEthereum)
|
|
}
|
|
|
|
// SetBalance sets an account's balance.
|
|
func (acc Account) SetBalance(amt sdk.Int) {
|
|
acc.SetCoins(sdk.Coins{sdk.NewCoin(bank.DenomEthereum, amt)})
|
|
}
|
|
|
|
// // NewAccount returns a reference to a new initialized account.
|
|
// func NewAccount(base auth.BaseAccount, code []byte) *Account {
|
|
// return &Account{
|
|
// BaseAccount: base,
|
|
// Code: code,
|
|
// Storage: storage,
|
|
// }
|
|
// }
|
|
|
|
// GetAccountDecoder returns the auth.AccountDecoder function for the custom
|
|
// Account type.
|
|
// func GetAccountDecoder(cdc *wire.Codec) auth.AccountDecoder {
|
|
// return func(accBytes []byte) (auth.Account, error) {
|
|
// if len(accBytes) == 0 {
|
|
// return nil, sdk.ErrTxDecode("account bytes are empty")
|
|
// }
|
|
|
|
// acc := new(Account)
|
|
|
|
// err := cdc.UnmarshalBinaryBare(accBytes, &acc)
|
|
// if err != nil {
|
|
// return nil, sdk.ErrTxDecode("failed to decode account bytes")
|
|
// }
|
|
|
|
// return acc, err
|
|
// }
|
|
// }
|
|
|
|
// Account code and storage type aliases.
|
|
type (
|
|
Code []byte
|
|
Storage map[ethcmn.Hash]ethcmn.Hash
|
|
)
|
|
|
|
func (c Code) String() string {
|
|
return string(c)
|
|
}
|
|
|
|
func (c Storage) String() (str string) {
|
|
for key, value := range c {
|
|
str += fmt.Sprintf("%X : %X\n", key, value)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Copy returns a copy of storage.
|
|
func (c Storage) Copy() Storage {
|
|
cpy := make(Storage)
|
|
for key, value := range c {
|
|
cpy[key] = value
|
|
}
|
|
|
|
return cpy
|
|
}
|