package rest import ( "bytes" "fmt" "net/http" "github.com/cosmos/cosmos-sdk/client/context" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/rest" "github.com/cosmos/cosmos-sdk/x/auth/client/utils" "github.com/gorilla/mux" "github.com/kava-labs/kava/x/cdp/types" ) func registerTxRoutes(cliCtx context.CLIContext, r *mux.Router) { r.HandleFunc("/cdp", postCdpHandlerFn(cliCtx)).Methods("POST") r.HandleFunc("/cdp/{owner}/{collateralType}/deposits", postDepositHandlerFn(cliCtx)).Methods("POST") r.HandleFunc("/cdp/{owner}/{collateralType}/withdraw", postWithdrawHandlerFn(cliCtx)).Methods("POST") r.HandleFunc("/cdp/{owner}/{collateralType}/draw", postDrawHandlerFn(cliCtx)).Methods("POST") r.HandleFunc("/cdp/{owner}/{collateralType}/repay", postRepayHandlerFn(cliCtx)).Methods("POST") } func postCdpHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { var requestBody PostCdpReq if !rest.ReadRESTReq(w, r, cliCtx.Codec, &requestBody) { return } requestBody.BaseReq = requestBody.BaseReq.Sanitize() if !requestBody.BaseReq.ValidateBasic(w) { return } fromAddr, err := sdk.AccAddressFromBech32(requestBody.BaseReq.From) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } if !bytes.Equal(fromAddr, requestBody.Sender) { rest.WriteErrorResponse(w, http.StatusUnauthorized, fmt.Sprintf("expected: %s, got: %s", fromAddr, requestBody.Sender)) return } msg := types.NewMsgCreateCDP( requestBody.Sender, requestBody.Collateral, requestBody.Principal, requestBody.CollateralType, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } utils.WriteGenerateStdTxResponse(w, cliCtx, requestBody.BaseReq, []sdk.Msg{msg}) } } func postDepositHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { var requestBody PostDepositReq if !rest.ReadRESTReq(w, r, cliCtx.Codec, &requestBody) { return } requestBody.BaseReq = requestBody.BaseReq.Sanitize() if !requestBody.BaseReq.ValidateBasic(w) { return } msg := types.NewMsgDeposit( requestBody.Owner, requestBody.Depositor, requestBody.Collateral, requestBody.CollateralType, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } utils.WriteGenerateStdTxResponse(w, cliCtx, requestBody.BaseReq, []sdk.Msg{msg}) } } func postWithdrawHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { var requestBody PostWithdrawalReq if !rest.ReadRESTReq(w, r, cliCtx.Codec, &requestBody) { return } requestBody.BaseReq = requestBody.BaseReq.Sanitize() if !requestBody.BaseReq.ValidateBasic(w) { return } msg := types.NewMsgWithdraw( requestBody.Owner, requestBody.Depositor, requestBody.Collateral, requestBody.CollateralType, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } utils.WriteGenerateStdTxResponse(w, cliCtx, requestBody.BaseReq, []sdk.Msg{msg}) } } func postDrawHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { var requestBody PostDrawReq if !rest.ReadRESTReq(w, r, cliCtx.Codec, &requestBody) { return } requestBody.BaseReq = requestBody.BaseReq.Sanitize() if !requestBody.BaseReq.ValidateBasic(w) { return } fromAddr, err := sdk.AccAddressFromBech32(requestBody.BaseReq.From) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } if !bytes.Equal(fromAddr, requestBody.Owner) { rest.WriteErrorResponse(w, http.StatusUnauthorized, fmt.Sprintf("expected: %s, got: %s", fromAddr, requestBody.Owner)) return } msg := types.NewMsgDrawDebt( requestBody.Owner, requestBody.CollateralType, requestBody.Principal, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } utils.WriteGenerateStdTxResponse(w, cliCtx, requestBody.BaseReq, []sdk.Msg{msg}) } } func postRepayHandlerFn(cliCtx context.CLIContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { var requestBody PostRepayReq if !rest.ReadRESTReq(w, r, cliCtx.Codec, &requestBody) { return } requestBody.BaseReq = requestBody.BaseReq.Sanitize() if !requestBody.BaseReq.ValidateBasic(w) { return } fromAddr, err := sdk.AccAddressFromBech32(requestBody.BaseReq.From) if err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } if !bytes.Equal(fromAddr, requestBody.Owner) { rest.WriteErrorResponse(w, http.StatusUnauthorized, fmt.Sprintf("expected: %s, got: %s", fromAddr, requestBody.Owner)) return } msg := types.NewMsgRepayDebt( requestBody.Owner, requestBody.CollateralType, requestBody.Payment, ) if err := msg.ValidateBasic(); err != nil { rest.WriteErrorResponse(w, http.StatusBadRequest, err.Error()) return } utils.WriteGenerateStdTxResponse(w, cliCtx, requestBody.BaseReq, []sdk.Msg{msg}) } }