package nft

import (
	"context"
	"github.com/shopspring/decimal"
	"github.com/spf13/cast"
	"github.com/zeromicro/go-zero/core/logx"
	"nova_task/internal/model"
	"nova_task/internal/pkg/utils"
	"nova_task/internal/svc"
	"time"
)

type StakeSettleLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewStakeSettleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *StakeSettleLogic {
	return &StakeSettleLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *StakeSettleLogic) StakeSettle() {
	start, end, err := l.svcCtx.ConfigModel.GetNftStakeTaskOpenDate(l.ctx)
	if err != nil {
		logx.Errorw("get nft stake task open date failed", logx.Field("err", err))
		return
	}
	end = end.AddDate(0, 0, 1).Add(-time.Second)
	now := time.Now()
	if now.Before(start) || now.After(end) {
		logx.Infow("now is not in the date range", logx.Field("now", now), logx.Field("start", start), logx.Field("end", end))
		return
	}

	taskConf, err := l.svcCtx.ConfigModel.GetNftStakeTaskConf(l.ctx)
	if err != nil {
		logx.Errorw("get nft stake task conf failed", logx.Field("err", err))
		return
	}

	stakes, err := l.svcCtx.StakeNftModel.AllStakeNft(l.ctx)
	if err != nil {
		logx.Errorw("get all stake nft failed", logx.Field("err", err))
		return
	}

	uid2tokens := map[uint]float64{}
	for _, s := range stakes {
		if s.State != 1 {
			continue
		}
		nftHolder, err := l.svcCtx.NftHolderModel.FindOneByTokenId(l.ctx, s.TokenId)
		if err != nil {
			logx.Errorw("find nft holder failed", logx.Field("err", err), logx.Field("tokenId", s.TokenId))
			continue
		}

		uid, err := l.svcCtx.WalletModel.FindUidByAddress(l.ctx, nftHolder.Address)
		if err != nil {
			logx.Errorw("find uid by address failed", logx.Field("err", err), logx.Field("address", nftHolder.Address))
			continue
		}

		if uid != s.Uid {
			logx.Errorw("uid not match", logx.Field("uid", s.Uid), logx.Field("nftHolderUid", uid))
			continue
		}

		if utils.IsBigTarot(s.TokenId) {
			uid2tokens[s.Uid] += float64(taskConf.GreatTarot)
		} else {
			uid2tokens[s.Uid] += float64(taskConf.LittleTarot)
		}
	}
	awardSeq := cast.ToInt(time.Now().AddDate(0, 0, -1).Format("20060102"))
	for uid, tokens := range uid2tokens {
		coefficient, err := l.svcCtx.StakeRewardModel.GetRandomCoefficientByUid(l.ctx, uid, awardSeq, taskConf.MinCoefficient, taskConf.MaxCoefficient)
		if err != nil {
			logx.Errorw("get random coefficient failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq))
			continue
		}
		tokensDecimal := decimal.NewFromFloat(tokens)
		reward := tokensDecimal.Mul(decimal.NewFromFloat(coefficient))
		err = l.svcCtx.StakeRewardModel.SetReward(l.ctx, uid, awardSeq, taskConf.OccupyPercent, tokensDecimal, reward)
		if err != nil {
			logx.Errorw("set reward failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq), logx.Field("coefficient", coefficient), logx.Field("tokens", tokens), logx.Field("reward", reward))
			continue
		}

		// 加资产
		err = l.svcCtx.TaskAssetModel.AddCastile(l.ctx, uid, reward)
		if err != nil {
			logx.Errorw("add castile failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq), logx.Field("reward", reward))
			continue
		}
		// 加资产记录表
		_, err = l.svcCtx.TaskAssetRecordModel.Insert(l.ctx, &model.NhTaskAssetRecord{
			Uid:        int(uid),
			EventId:    0,
			AssetField: "castile",
			Count:      reward.InexactFloat64(),
			Remark:     "nft软质押奖励",
			ProvideUid: 0,
			CreateTime: int(now.Unix()),
		})
		if err != nil {
			logx.Errorw("insert task asset record failed", logx.Field("err", err), logx.Field("uid", uid), logx.Field("awardSeq", awardSeq), logx.Field("reward", reward))
		}
	}
}