96 lines
2.0 KiB
Go
96 lines
2.0 KiB
Go
package coin
|
|
|
|
import (
|
|
"github.com/tendermint/basecoin"
|
|
"github.com/tendermint/basecoin/errors"
|
|
"github.com/tendermint/basecoin/types"
|
|
)
|
|
|
|
const (
|
|
NameCoin = "coin"
|
|
)
|
|
|
|
// Handler writes
|
|
type Handler struct {
|
|
Accountant
|
|
}
|
|
|
|
var _ basecoin.Handler = Handler{}
|
|
|
|
func NewHandler() Handler {
|
|
return Handler{
|
|
Accountant: Accountant{Prefix: []byte(NameCoin + "/")},
|
|
}
|
|
}
|
|
|
|
func (_ Handler) Name() string {
|
|
return NameCoin
|
|
}
|
|
|
|
// CheckTx checks if there is enough money in the account
|
|
func (h Handler) CheckTx(ctx basecoin.Context, store types.KVStore, tx basecoin.Tx) (res basecoin.Result, err error) {
|
|
send, err := checkTx(ctx, tx)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
|
|
// now make sure there is money
|
|
for _, in := range send.Inputs {
|
|
_, err = h.CheckCoins(store, in.Address, in.Coins, in.Sequence)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
}
|
|
|
|
// otherwise, we are good
|
|
return res, nil
|
|
}
|
|
|
|
// DeliverTx moves the money
|
|
func (h Handler) DeliverTx(ctx basecoin.Context, store types.KVStore, tx basecoin.Tx) (res basecoin.Result, err error) {
|
|
send, err := checkTx(ctx, tx)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
|
|
// deduct from all input accounts
|
|
for _, in := range send.Inputs {
|
|
_, err = h.ChangeCoins(store, in.Address, in.Coins.Negative(), in.Sequence)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
}
|
|
|
|
// add to all output accounts
|
|
for _, out := range send.Outputs {
|
|
// note: sequence number is ignored when adding coins, only checked for subtracting
|
|
_, err = h.ChangeCoins(store, out.Address, out.Coins, 0)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
}
|
|
|
|
// a-ok!
|
|
return basecoin.Result{}, nil
|
|
}
|
|
|
|
func checkTx(ctx basecoin.Context, tx basecoin.Tx) (send SendTx, err error) {
|
|
// check if the tx is proper type and valid
|
|
send, ok := tx.Unwrap().(SendTx)
|
|
if !ok {
|
|
return send, errors.ErrInvalidFormat(tx)
|
|
}
|
|
err = send.ValidateBasic()
|
|
if err != nil {
|
|
return send, err
|
|
}
|
|
|
|
// check if all inputs have permission
|
|
for _, in := range send.Inputs {
|
|
if !ctx.HasPermission(in.Address) {
|
|
return send, errors.ErrUnauthorized()
|
|
}
|
|
}
|
|
return send, nil
|
|
}
|