package rest import ( "fmt" "net/http" "github.com/gorilla/mux" "github.com/cosmos/cosmos-sdk/client" "github.com/cosmos/cosmos-sdk/client/tx" "github.com/cosmos/cosmos-sdk/types/rest" "github.com/kava-labs/kava/x/hard/types" ) func registerTxRoutes(cliCtx client.Context, r *mux.Router) { r.HandleFunc(fmt.Sprintf("/%s/deposit", types.ModuleName), postDepositHandlerFn(cliCtx)).Methods("POST") r.HandleFunc(fmt.Sprintf("/%s/withdraw", types.ModuleName), postWithdrawHandlerFn(cliCtx)).Methods("POST") r.HandleFunc(fmt.Sprintf("/%s/borrow", types.ModuleName), postBorrowHandlerFn(cliCtx)).Methods("POST") r.HandleFunc(fmt.Sprintf("/%s/repay", types.ModuleName), postRepayHandlerFn(cliCtx)).Methods("POST") r.HandleFunc(fmt.Sprintf("/%s/liquidate", types.ModuleName), postLiquidateHandlerFn(cliCtx)).Methods("POST") } func postDepositHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode POST request body var req PostCreateDepositReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } msg := types.NewMsgDeposit(req.From, req.Amount) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } } func postWithdrawHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode POST request body var req PostCreateWithdrawReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } msg := types.NewMsgWithdraw(req.From, req.Amount) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } } func postBorrowHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode POST request body var req PostBorrowReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } msg := types.NewMsgBorrow(req.From, req.Amount) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } } func postRepayHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode POST request body var req PostRepayReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } owner := req.Owner if owner.Empty() { owner = req.From } msg := types.NewMsgRepay(req.From, owner, req.Amount) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } } func postLiquidateHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode POST request body var req PostLiquidateReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } msg := types.NewMsgLiquidate(req.From, req.Borrower) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } }