2018-01-12 11:49:53 -08:00
|
|
|
package bank
|
2018-01-03 17:20:21 -08:00
|
|
|
|
|
|
|
import (
|
2018-01-12 14:30:02 -08:00
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
2018-01-17 16:59:40 -08:00
|
|
|
"reflect"
|
2018-01-03 17:20:21 -08:00
|
|
|
)
|
|
|
|
|
2018-01-17 16:59:40 -08:00
|
|
|
// Handle all "bank" type messages.
|
2018-01-22 05:44:24 -08:00
|
|
|
func NewHandler(am sdk.AccountMapper) sdk.Handler {
|
2018-01-03 17:20:21 -08:00
|
|
|
|
2018-01-12 14:32:04 -08:00
|
|
|
return func(ctx sdk.Context, tx sdk.Tx) sdk.Result {
|
2018-01-22 05:44:24 -08:00
|
|
|
cm := CoinMapper{am}
|
2018-01-17 16:59:40 -08:00
|
|
|
msg := tx.(sdk.Msg)
|
|
|
|
switch msg := msg.(type) {
|
|
|
|
case SendMsg:
|
2018-01-22 05:44:24 -08:00
|
|
|
return handleSendMsg(ctx, cm, msg)
|
2018-01-17 16:59:40 -08:00
|
|
|
case IssueMsg:
|
2018-01-22 05:44:24 -08:00
|
|
|
return handleIssueMsg(ctx, cm, msg)
|
2018-01-17 16:59:40 -08:00
|
|
|
default:
|
2018-01-26 04:19:33 -08:00
|
|
|
errMsg := "Unrecognized bank Tx type: " + reflect.TypeOf(tx).Name()
|
|
|
|
return sdk.ErrUnknownRequest(errMsg).Result()
|
2018-01-03 17:20:21 -08:00
|
|
|
}
|
2018-01-17 16:59:40 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
2018-01-03 17:20:21 -08:00
|
|
|
|
2018-01-17 16:59:40 -08:00
|
|
|
// Handle SendMsg.
|
2018-01-22 05:44:24 -08:00
|
|
|
func handleSendMsg(ctx sdk.Context, cm CoinMapper, msg SendMsg) sdk.Result {
|
2018-01-17 16:59:40 -08:00
|
|
|
// NOTE: totalIn == totalOut should already have been checked
|
2018-01-03 17:20:21 -08:00
|
|
|
|
2018-01-17 16:59:40 -08:00
|
|
|
for _, in := range msg.Inputs {
|
2018-01-22 05:44:24 -08:00
|
|
|
_, err := cm.SubtractCoins(ctx, in.Address, in.Coins)
|
2018-01-17 16:59:40 -08:00
|
|
|
if err != nil {
|
2018-01-26 04:19:33 -08:00
|
|
|
return ErrInvalidInput("").TraceCause(err, "").Result()
|
2018-01-03 17:20:21 -08:00
|
|
|
}
|
2018-01-17 16:59:40 -08:00
|
|
|
}
|
2018-01-03 17:20:21 -08:00
|
|
|
|
2018-01-17 16:59:40 -08:00
|
|
|
for _, out := range msg.Outputs {
|
2018-01-22 05:44:24 -08:00
|
|
|
_, err := cm.AddCoins(ctx, out.Address, out.Coins)
|
2018-01-17 16:59:40 -08:00
|
|
|
if err != nil {
|
2018-01-26 04:19:33 -08:00
|
|
|
return ErrInvalidOutput("").TraceCause(err, "").Result()
|
2018-01-03 17:20:21 -08:00
|
|
|
}
|
|
|
|
}
|
2018-01-17 16:59:40 -08:00
|
|
|
|
|
|
|
return sdk.Result{} // TODO
|
|
|
|
}
|
|
|
|
|
|
|
|
// Handle IssueMsg.
|
2018-01-22 05:44:24 -08:00
|
|
|
func handleIssueMsg(ctx sdk.Context, cm CoinMapper, msg IssueMsg) sdk.Result {
|
2018-01-17 16:59:40 -08:00
|
|
|
panic("not implemented yet")
|
2018-01-03 17:20:21 -08:00
|
|
|
}
|