package rest import ( "fmt" "net/http" "github.com/gorilla/mux" "github.com/cosmos/cosmos-sdk/client" "github.com/cosmos/cosmos-sdk/client/tx" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/rest" "github.com/kava-labs/kava/x/bep3/types" ) func registerTxRoutes(cliCtx client.Context, r *mux.Router) { r.HandleFunc(fmt.Sprintf("/%s/swap/create", types.ModuleName), postCreateHandlerFn(cliCtx)).Methods("POST") r.HandleFunc(fmt.Sprintf("/%s/swap/claim", types.ModuleName), postClaimHandlerFn(cliCtx)).Methods("POST") r.HandleFunc(fmt.Sprintf("/%s/swap/refund", types.ModuleName), postRefundHandlerFn(cliCtx)).Methods("POST") } func postCreateHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode PUT request body var req PostCreateSwapReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } // Create and return msg msg := types.NewMsgCreateAtomicSwap( req.From.String(), req.To.String(), req.RecipientOtherChain, req.SenderOtherChain, req.RandomNumberHash, req.Timestamp, req.Amount, req.HeightSpan, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } } func postClaimHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode PUT request body var req PostClaimSwapReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } // Create and return msg msg := types.NewMsgClaimAtomicSwap( req.From.String(), req.SwapID, req.RandomNumber, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } } func postRefundHandlerFn(cliCtx client.Context) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // Decode PUT request body var req PostRefundSwapReq if !rest.ReadRESTReq(w, r, cliCtx.LegacyAmino, &req) { return } req.BaseReq = req.BaseReq.Sanitize() if !req.BaseReq.ValidateBasic(w) { return } senderAddr, err := sdk.AccAddressFromBech32(req.BaseReq.From) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } // Create and return msg msg := types.NewMsgRefundAtomicSwap( senderAddr.String(), req.SwapID, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } tx.WriteGeneratedTxResponse(cliCtx, w, req.BaseReq, &msg) } }