Add query REST endpoint for buyback module balance

x/buyback/client/rest/query.go

@@ -0,0 +1,37 @@
+package rest
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/gorilla/mux"
+
+	"github.com/cosmos/cosmos-sdk/client/context"
+	"github.com/cosmos/cosmos-sdk/types/rest"
+
+	"github.com/e-money/em-ledger/x/buyback/internal/types"
+)
+
+func RegisterQueryRoutes(cliCtx context.CLIContext, r *mux.Router) {
+	r.HandleFunc("/buyback/balance", queryBalanceHandlerFn(cliCtx)).Methods("GET")
+}
+
+func queryBalanceHandlerFn(cliCtx context.CLIContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		route := fmt.Sprintf("custom/%s/%s", types.QuerierRoute, types.QueryBalance)
+
+		cliCtx, ok := rest.ParseQueryHeightOrReturnBadRequest(w, cliCtx, r)
+		if !ok {
+			return
+		}
+
+		res, height, err := cliCtx.QueryWithData(route, nil)
+		if err != nil {
+			rest.WriteErrorResponse(w, http.StatusInternalServerError, err.Error())
+			return
+		}
+
+		cliCtx = cliCtx.WithHeight(height)
+		rest.PostProcessResponse(w, cliCtx, res)
+	}
+}

x/buyback/module.go

@@ -7,6 +7,7 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
 	"github.com/e-money/em-ledger/x/buyback/client/cli"
+	"github.com/e-money/em-ledger/x/buyback/client/rest"
 	"github.com/e-money/em-ledger/x/buyback/internal/keeper"
 	"github.com/gorilla/mux"
 	"github.com/spf13/cobra"
@@ -49,7 +50,9 @@ func (amb AppModuleBasic) ValidateGenesis(json.RawMessage) error {
 	return nil
 }
 
-func (amb AppModuleBasic) RegisterRESTRoutes(context.CLIContext, *mux.Router) {}
+func (amb AppModuleBasic) RegisterRESTRoutes(ctx context.CLIContext, r *mux.Router) {
+	rest.RegisterQueryRoutes(ctx, r)
+}
 
 func (amb AppModuleBasic) GetTxCmd(*codec.Codec) *cobra.Command {
 	return nil