mukan-ignite/ignite/pkg/cosmosfaucet/http_faucet.go

package cosmosfaucet

import (
	"context"
	"encoding/json"
	"net/http"

	sdk "github.com/cosmos/cosmos-sdk/types"

	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/errors"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/xhttp"
)

type TransferRequest struct {
	// AccountAddress to request for coins.
	AccountAddress string `json:"address"`

	// Coins that are requested.
	// default ones used when this one isn't provided.
	Coins []string `json:"coins,omitempty"`
}

func NewTransferRequest(accountAddress string, coins []string) TransferRequest {
	return TransferRequest{
		AccountAddress: accountAddress,
		Coins:          coins,
	}
}

type TransferResponse struct {
	Hash  string `json:"hash,omitempty"`
	Error string `json:"error,omitempty"`
}

func (f Faucet) faucetHandler(w http.ResponseWriter, r *http.Request) {
	var req TransferRequest

	// decode request into req.
	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
		responseError(w, http.StatusBadRequest, err)
		return
	}

	// determine coins to transfer.
	coins, err := f.coinsFromRequest(req)
	if err != nil {
		responseError(w, http.StatusBadRequest, err)
		return
	}

	// try performing the transfer
	hash, err := f.Transfer(r.Context(), req.AccountAddress, coins)
	if err != nil {
		if errors.Is(err, context.Canceled) {
			return
		}
		responseError(w, http.StatusInternalServerError, err)
		return
	}
	responseSuccess(w, hash)
}

// FaucetInfoResponse is the faucet info payload.
type FaucetInfoResponse struct {
	// IsAFaucet indicates that this is a faucet endpoint.
	// useful for auto discoveries.
	IsAFaucet bool `json:"is_a_faucet"`

	// ChainID is chain id of the chain that faucet is running for.
	ChainID string `json:"chain_id"`
}

func (f Faucet) faucetInfoHandler(w http.ResponseWriter, _ *http.Request) {
	_ = xhttp.ResponseJSON(w, http.StatusOK, FaucetInfoResponse{
		IsAFaucet: true,
		ChainID:   f.chainID,
	})
}

// coinsFromRequest determines tokens to transfer from transfer request.
func (f Faucet) coinsFromRequest(req TransferRequest) (sdk.Coins, error) {
	if len(req.Coins) == 0 {
		return f.coins.Sort(), nil
	}

	coins := sdk.NewCoins()
	for _, c := range req.Coins {
		coin, err := sdk.ParseCoinNormalized(c)
		if err != nil {
			return nil, err
		}
		coins = coins.Add(coin)
	}

	return coins, nil
}

func responseSuccess(w http.ResponseWriter, hash string) {
	_ = xhttp.ResponseJSON(w, http.StatusOK, TransferResponse{
		Hash: hash,
	})
}

func responseError(w http.ResponseWriter, code int, err error) {
	_ = xhttp.ResponseJSON(w, code, TransferResponse{
		Error: err.Error(),
	})
}