120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
package rpc
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
)
|
|
|
|
// TODO these next two functions feel kinda hacky based on their placement
|
|
|
|
func validatorCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "validatorset [height]",
|
|
Short: "Get the full validator set at given height",
|
|
Args: cobra.MaximumNArgs(1),
|
|
RunE: printValidators,
|
|
}
|
|
cmd.Flags().StringP(client.FlagNode, "n", "tcp://localhost:46657", "Node to connect to")
|
|
// TODO: change this to false when we can
|
|
cmd.Flags().Bool(client.FlagTrustNode, true, "Don't verify proofs for responses")
|
|
return cmd
|
|
}
|
|
|
|
func getValidators(ctx context.CoreContext, height *int64) ([]byte, error) {
|
|
// get the node
|
|
node, err := ctx.GetNode()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
res, err := node.Validators(height)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
output, err := cdc.MarshalJSON(res)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return output, nil
|
|
}
|
|
|
|
// CMD
|
|
|
|
func printValidators(cmd *cobra.Command, args []string) error {
|
|
var height *int64
|
|
// optional height
|
|
if len(args) > 0 {
|
|
h, err := strconv.Atoi(args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if h > 0 {
|
|
tmp := int64(h)
|
|
height = &tmp
|
|
}
|
|
}
|
|
|
|
output, err := getValidators(context.NewCoreContextFromViper(), height)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(string(output))
|
|
return nil
|
|
}
|
|
|
|
// REST
|
|
|
|
// Validator Set at a height REST handler
|
|
func ValidatorSetRequestHandlerFn(ctx context.CoreContext) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
height, err := strconv.ParseInt(vars["height"], 10, 64)
|
|
if err != nil {
|
|
w.WriteHeader(400)
|
|
w.Write([]byte("ERROR: Couldn't parse block height. Assumed format is '/validatorsets/{height}'."))
|
|
return
|
|
}
|
|
chainHeight, err := GetChainHeight(ctx)
|
|
if height > chainHeight {
|
|
w.WriteHeader(404)
|
|
w.Write([]byte("ERROR: Requested block height is bigger then the chain length."))
|
|
return
|
|
}
|
|
output, err := getValidators(ctx, &height)
|
|
if err != nil {
|
|
w.WriteHeader(500)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
w.Write(output)
|
|
}
|
|
}
|
|
|
|
// Latest Validator Set REST handler
|
|
func LatestValidatorSetRequestHandlerFn(ctx context.CoreContext) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
height, err := GetChainHeight(ctx)
|
|
if err != nil {
|
|
w.WriteHeader(500)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
output, err := getValidators(ctx, &height)
|
|
if err != nil {
|
|
w.WriteHeader(500)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
w.Write(output)
|
|
}
|
|
}
|