2018-05-14 16:43:51 -07:00
|
|
|
package rest
|
|
|
|
|
|
|
|
import (
|
2018-05-24 08:41:17 -07:00
|
|
|
"bytes"
|
2018-05-14 16:43:51 -07:00
|
|
|
"encoding/json"
|
2018-06-05 21:53:04 -07:00
|
|
|
"fmt"
|
2018-05-14 16:43:51 -07:00
|
|
|
"io/ioutil"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"github.com/gorilla/mux"
|
|
|
|
"github.com/tendermint/go-crypto/keys"
|
|
|
|
ctypes "github.com/tendermint/tendermint/rpc/core/types"
|
|
|
|
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/stake"
|
|
|
|
)
|
|
|
|
|
|
|
|
func registerTxRoutes(ctx context.CoreContext, r *mux.Router, cdc *wire.Codec, kb keys.Keybase) {
|
2018-05-24 08:41:17 -07:00
|
|
|
r.HandleFunc(
|
|
|
|
"/stake/delegations",
|
|
|
|
editDelegationsRequestHandlerFn(cdc, kb, ctx),
|
|
|
|
).Methods("POST")
|
2018-05-14 16:43:51 -07:00
|
|
|
}
|
|
|
|
|
2018-06-05 21:53:04 -07:00
|
|
|
type msgDelegateInput struct {
|
|
|
|
DelegatorAddr string `json:"delegator_addr"` // in bech32
|
|
|
|
ValidatorAddr string `json:"validator_addr"` // in bech32
|
|
|
|
Bond sdk.Coin `json:"bond"`
|
|
|
|
}
|
|
|
|
type msgUnbondInput struct {
|
|
|
|
DelegatorAddr string `json:"delegator_addr"` // in bech32
|
|
|
|
ValidatorAddr string `json:"validator_addr"` // in bech32
|
|
|
|
Shares string `json:"shares"`
|
|
|
|
}
|
|
|
|
|
2018-05-24 08:41:17 -07:00
|
|
|
type editDelegationsBody struct {
|
2018-06-05 21:53:04 -07:00
|
|
|
LocalAccountName string `json:"name"`
|
|
|
|
Password string `json:"password"`
|
|
|
|
ChainID string `json:"chain_id"`
|
|
|
|
Sequence int64 `json:"sequence"`
|
|
|
|
Delegate []msgDelegateInput `json:"delegate"`
|
|
|
|
Unbond []msgUnbondInput `json:"unbond"`
|
2018-05-14 16:43:51 -07:00
|
|
|
}
|
|
|
|
|
2018-05-24 08:41:17 -07:00
|
|
|
func editDelegationsRequestHandlerFn(cdc *wire.Codec, kb keys.Keybase, ctx context.CoreContext) http.HandlerFunc {
|
2018-05-14 16:43:51 -07:00
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
2018-05-24 08:41:17 -07:00
|
|
|
var m editDelegationsBody
|
2018-05-14 16:43:51 -07:00
|
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
|
|
w.Write([]byte(err.Error()))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
err = json.Unmarshal(body, &m)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
|
|
w.Write([]byte(err.Error()))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
info, err := kb.Get(m.LocalAccountName)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
|
|
w.Write([]byte(err.Error()))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// build messages
|
2018-05-30 09:00:00 -07:00
|
|
|
messages := make([]sdk.Msg, len(m.Delegate)+len(m.Unbond))
|
|
|
|
i := 0
|
2018-05-24 08:41:17 -07:00
|
|
|
for _, msg := range m.Delegate {
|
2018-06-05 21:53:04 -07:00
|
|
|
delegatorAddr, err := sdk.GetAccAddressBech32(msg.DelegatorAddr)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
w.Write([]byte(fmt.Sprintf("Couldn't decode delegator. Error: %s", err.Error())))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
validatorAddr, err := sdk.GetValAddressBech32(msg.ValidatorAddr)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
w.Write([]byte(fmt.Sprintf("Couldn't decode validator. Error: %s", err.Error())))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if !bytes.Equal(info.Address(), delegatorAddr) {
|
2018-05-24 08:41:17 -07:00
|
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
|
|
w.Write([]byte("Must use own delegator address"))
|
|
|
|
return
|
|
|
|
}
|
2018-06-05 21:53:04 -07:00
|
|
|
messages[i] = stake.MsgDelegate{
|
|
|
|
DelegatorAddr: delegatorAddr,
|
|
|
|
ValidatorAddr: validatorAddr,
|
|
|
|
Bond: msg.Bond,
|
|
|
|
}
|
2018-05-30 09:00:00 -07:00
|
|
|
i++
|
2018-05-14 16:43:51 -07:00
|
|
|
}
|
2018-05-24 08:41:17 -07:00
|
|
|
for _, msg := range m.Unbond {
|
2018-06-05 21:53:04 -07:00
|
|
|
delegatorAddr, err := sdk.GetAccAddressBech32(msg.DelegatorAddr)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
w.Write([]byte(fmt.Sprintf("Couldn't decode delegator. Error: %s", err.Error())))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
validatorAddr, err := sdk.GetValAddressBech32(msg.ValidatorAddr)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
w.Write([]byte(fmt.Sprintf("Couldn't decode validator. Error: %s", err.Error())))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if !bytes.Equal(info.Address(), delegatorAddr) {
|
2018-05-24 08:41:17 -07:00
|
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
|
|
w.Write([]byte("Must use own delegator address"))
|
|
|
|
return
|
|
|
|
}
|
2018-06-05 21:53:04 -07:00
|
|
|
messages[i] = stake.MsgUnbond{
|
|
|
|
DelegatorAddr: delegatorAddr,
|
|
|
|
ValidatorAddr: validatorAddr,
|
|
|
|
Shares: msg.Shares,
|
|
|
|
}
|
2018-05-30 09:00:00 -07:00
|
|
|
i++
|
2018-05-14 16:43:51 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// sign messages
|
2018-05-30 09:00:00 -07:00
|
|
|
signedTxs := make([][]byte, len(messages[:]))
|
|
|
|
for i, msg := range messages {
|
2018-05-14 16:43:51 -07:00
|
|
|
// increment sequence for each message
|
|
|
|
ctx = ctx.WithSequence(m.Sequence)
|
|
|
|
m.Sequence++
|
|
|
|
|
|
|
|
txBytes, err := ctx.SignAndBuild(m.LocalAccountName, m.Password, msg, cdc)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
|
|
w.Write([]byte(err.Error()))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-05-30 09:00:00 -07:00
|
|
|
signedTxs[i] = txBytes
|
2018-05-14 16:43:51 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// send
|
|
|
|
// XXX the operation might not be atomic if a tx fails
|
|
|
|
// should we have a sdk.MultiMsg type to make sending atomic?
|
2018-05-30 09:00:00 -07:00
|
|
|
results := make([]*ctypes.ResultBroadcastTxCommit, len(signedTxs[:]))
|
|
|
|
for i, txBytes := range signedTxs {
|
2018-05-14 16:43:51 -07:00
|
|
|
res, err := ctx.BroadcastTx(txBytes)
|
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
w.Write([]byte(err.Error()))
|
|
|
|
return
|
|
|
|
}
|
2018-05-30 09:00:00 -07:00
|
|
|
results[i] = res
|
2018-05-14 16:43:51 -07:00
|
|
|
}
|
|
|
|
|
2018-05-30 09:00:00 -07:00
|
|
|
output, err := json.MarshalIndent(results[:], "", " ")
|
2018-05-14 16:43:51 -07:00
|
|
|
if err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
w.Write([]byte(err.Error()))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
w.Write(output)
|
|
|
|
}
|
|
|
|
}
|