handler.go

package comment

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

	"github.com/coinexchain/cet-sdk/modules/comment/internal/types"
	dex "github.com/coinexchain/cet-sdk/types"
)

func NewHandler(k Keeper) sdk.Handler {
	return func(ctx sdk.Context, msg sdk.Msg) sdk.Result {
		ctx = ctx.WithEventManager(sdk.NewEventManager())

		switch msg := msg.(type) {
		case types.MsgCommentToken:
			return handleMsgCommentToken(ctx, k, msg)
		default:
			return dex.ErrUnknownRequest(ModuleName, msg)
		}
	}
}

func handleMsgCommentToken(ctx sdk.Context, k Keeper, msg types.MsgCommentToken) sdk.Result {
	if !k.IsTokenExists(ctx, msg.Token) {
		return types.ErrNoSuchAsset().Result()
	}
	if msg.Donation > 0 {
		donatedCoin := sdk.Coins{sdk.NewCoin(dex.CET, sdk.NewInt(msg.Donation))}
		res := k.DonateToCommunityPool(ctx, msg.Sender, donatedCoin)
		if res != nil {
			return res.Result()
		}
	}

	for _, ref := range msg.References {
		if ref.RewardAmount <= 0 {
			continue
		}
		acc := k.GetAccount(ctx, ref.RewardTarget)
		if acc == nil {
			return types.ErrNoSuchAccount(ref.RewardTarget.String()).Result()
		}
		rewardCoin := sdk.NewCoin(ref.RewardToken, sdk.NewInt(ref.RewardAmount))
		res := k.SendCoins(ctx, msg.Sender, ref.RewardTarget, sdk.Coins{rewardCoin})
		if res != nil {
			return res.Result()
		}
	}

	lastCount := k.IncrCommentCount(ctx, msg.Token)

	if len(k.GetEventTypeMsgQueue()) != 0 {
		tokenComment := types.NewTokenComment(&msg, lastCount, ctx.BlockHeight())
		bytes := dex.SafeJSONMarshal(tokenComment)
		ctx.EventManager().EmitEvent(
			sdk.NewEvent(
				k.GetEventTypeMsgQueue(),
				sdk.NewAttribute(types.TokenCommentKey, string(bytes)),
			),
		)
	}

	ctx.EventManager().EmitEvent(
		sdk.NewEvent(
			sdk.EventTypeMessage,
			sdk.NewAttribute(sdk.AttributeKeyModule, types.ModuleName),
			sdk.NewAttribute(sdk.AttributeKeySender, msg.Sender.String()),
		),
	)

	return sdk.Result{
		Codespace: types.CodeSpaceComment,
		Events:    ctx.EventManager().Events(),
	}
}