package rest import ( "encoding/json" "fmt" "net/http" "github.com/cosmos/cosmos-sdk/client/context" "github.com/cosmos/cosmos-sdk/types/rest" "github.com/gorilla/mux" "github.com/kava-labs/kava/x/validator-vesting/internal/types" ) // define routes that get registered by the main application func registerQueryRoutes(cliCtx context.CLIContext, r *mux.Router) { r.HandleFunc("/vesting/circulatingsupply", getCirculatingSupplyHandlerFn(cliCtx)).Methods("GET") r.HandleFunc("/vesting/totalsupply", getTotalSupplyHandlerFn(cliCtx)).Methods("GET") } func getTotalSupplyHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { _, page, limit, err := rest.ParseHTTPArgsWithLimit(r, 0) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } cliCtx, ok := rest.ParseQueryHeightOrReturnBadRequest(w, cliCtx, r) if !ok { return } params := types.NewBaseQueryParams(page, limit) bz, err := cliCtx.Codec.MarshalJSON(params) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, fmt.Sprintf("failed to marshal query params: %s", err)) return } route := fmt.Sprintf("custom/%s/%s", types.QuerierRoute, types.QueryTotalSupply) res, _, err := cliCtx.QueryWithData(route, bz) if err != nil { rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error()) return } var totalSupply int64 err = cliCtx.Codec.UnmarshalJSON(res, &totalSupply) if err != nil { rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error()) return } resBytes, err := json.Marshal(totalSupply) if err != nil { rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error()) return } w.Write(resBytes) } } func getCirculatingSupplyHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { _, page, limit, err := rest.ParseHTTPArgsWithLimit(r, 0) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } cliCtx, ok := rest.ParseQueryHeightOrReturnBadRequest(w, cliCtx, r) if !ok { return } params := types.NewBaseQueryParams(page, limit) bz, err := cliCtx.Codec.MarshalJSON(params) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, fmt.Sprintf("failed to marshal query params: %s", err)) return } route := fmt.Sprintf("custom/%s/%s", types.QuerierRoute, types.QueryCirculatingSupply) res, _, err := cliCtx.QueryWithData(route, bz) if err != nil { rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error()) return } var circSupply int64 err = cliCtx.Codec.UnmarshalJSON(res, &circSupply) if err != nil { rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error()) return } resBytes, err := json.Marshal(circSupply) if err != nil { rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error()) return } w.Write(resBytes) } }