106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package rpc
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
ctypes "github.com/tendermint/tendermint/rpc/core/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/types/rest"
|
|
"github.com/cosmos/cosmos-sdk/version"
|
|
|
|
"github.com/tendermint/tendermint/p2p"
|
|
)
|
|
|
|
// StatusCommand returns the command to return the status of the network.
|
|
func StatusCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "status",
|
|
Short: "Query remote node for status",
|
|
RunE: printNodeStatus,
|
|
}
|
|
|
|
cmd.Flags().StringP(flags.FlagNode, "n", "tcp://localhost:26657", "Node to connect to")
|
|
viper.BindPFlag(flags.FlagNode, cmd.Flags().Lookup(flags.FlagNode))
|
|
cmd.Flags().Bool(flags.FlagIndentResponse, false, "Add indent to JSON response")
|
|
return cmd
|
|
}
|
|
|
|
func getNodeStatus(cliCtx context.CLIContext) (*ctypes.ResultStatus, error) {
|
|
node, err := cliCtx.GetNode()
|
|
if err != nil {
|
|
return &ctypes.ResultStatus{}, err
|
|
}
|
|
|
|
return node.Status()
|
|
}
|
|
|
|
func printNodeStatus(_ *cobra.Command, _ []string) error {
|
|
// No need to verify proof in getting node status
|
|
viper.Set(flags.FlagTrustNode, true)
|
|
cliCtx := context.NewCLIContext()
|
|
status, err := getNodeStatus(cliCtx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var output []byte
|
|
if cliCtx.Indent {
|
|
output, err = codec.Cdc.MarshalJSONIndent(status, "", " ")
|
|
} else {
|
|
output, err = codec.Cdc.MarshalJSON(status)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(string(output))
|
|
return nil
|
|
}
|
|
|
|
type nodeInfoResponse struct {
|
|
p2p.DefaultNodeInfo `json:"node_info"`
|
|
|
|
ApplicationVersion version.Info `json:"application_version"`
|
|
}
|
|
|
|
// REST handler for node info
|
|
func NodeInfoRequestHandlerFn(cliCtx context.CLIContext) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
status, err := getNodeStatus(cliCtx)
|
|
if err != nil {
|
|
rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
|
|
resp := nodeInfoResponse{
|
|
DefaultNodeInfo: status.NodeInfo,
|
|
ApplicationVersion: version.NewInfo(),
|
|
}
|
|
rest.PostProcessResponseBare(w, cliCtx, resp)
|
|
}
|
|
}
|
|
|
|
type syncingResponse struct {
|
|
Syncing bool `json:"syncing"`
|
|
}
|
|
|
|
// REST handler for node syncing
|
|
func NodeSyncingRequestHandlerFn(cliCtx context.CLIContext) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
status, err := getNodeStatus(cliCtx)
|
|
if err != nil {
|
|
rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
|
|
rest.PostProcessResponseBare(w, cliCtx, syncingResponse{Syncing: status.SyncInfo.CatchingUp})
|
|
}
|
|
}
|