package admin

import (
	"context"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/threading"
	"nova_task/internal/consts"
	"nova_task/internal/pkg/errs"
	"nova_task/internal/svc"
)

type SendEmailRewardLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// 执行发放奖励操作
func NewSendEmailRewardLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendEmailRewardLogic {
	return &SendEmailRewardLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *SendEmailRewardLogic) SendEmailReward() error {
	rewards, err := l.svcCtx.EmailRewardModel.AllRequireSend(l.ctx)
	if err != nil {
		l.Errorw("get email reward failed", logx.Field("err", err))
		return errs.New(errs.ErrDatabaseOperate, err)
	}

	if len(rewards) <= 0 {
		l.Infow("no email reward to send")
		return errs.Success()
	}

	threading.GoSafe(func() {
		ctx := context.Background()
		for _, rw := range rewards {
			u, err := l.svcCtx.UserModel.FindOneByEmail(ctx, rw.Email)
			if err != nil {
				l.Errorw("find user failed", logx.Field("err", err), logx.Field("email", rw.Email))
				continue
			}

			err = l.svcCtx.EmailRewardModel.SetSent(ctx, rw.Id, u.Id)
			if err != nil {
				l.Errorw("set sent failed", logx.Field("err", err), logx.Field("id", rw.Id), logx.Field("uid", u.Id))
				continue
			}

			err = l.svcCtx.AddUserAsset(ctx, u.Id, consts.AssetType(rw.RewardType), rw.Value, rw.Remark, 0, 0, false)
			if err != nil {
				l.Errorw("insert task asset record failed", logx.Field("err", err), logx.Field("uid", u.Id), logx.Field("rewardType", rw.RewardType), logx.Field("value", rw.Value))
			}
		}
	})

	return errs.Success()
}