package task

import (
	"context"
	"errors"
	"nova_task/internal/model"
	"nova_task/internal/pkg/errs"
	"nova_task/internal/pkg/utils"
	"nova_task/internal/svc"
	"nova_task/internal/types"
	"time"

	"github.com/zeromicro/go-zero/core/logx"
)

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

// 拉取玩家持有的nft列表
func NewGetNftListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetNftListLogic {
	return &GetNftListLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *GetNftListLogic) GetNftList() (*types.UserNftList, error) {
	uid := utils.GetUidUint(l.ctx)
	address, err := l.svcCtx.WalletModel.FindAddressByUid(l.ctx, uid)
	if err != nil {
		if errors.Is(err, model.ErrNotFound) {
			return nil, errs.New(errs.ErrNotBindWallet, "")
		}
		l.Errorw("find address by uid failed", logx.Field("err", err), logx.Field("uid", uid))
		return nil, errs.New(errs.ErrDatabaseOperate, err)
	}

	tokens, err := l.svcCtx.NftHolderModel.FindTokensByAddress(l.ctx, address)
	if err != nil {
		l.Errorw("find tokens by address failed", logx.Field("err", err), logx.Field("address", address), logx.Field("uid", uid))
		return nil, errs.New(errs.ErrDatabaseOperate, err)
	}

	var nftList []types.UserNft
	for _, token := range tokens {
		var hasStake bool
		var stakeAt string
		nft, err := l.svcCtx.StakeNftModel.FindOneByUidTokenId(l.ctx, uid, token)
		if err == nil {
			hasStake = nft.State == 1
			if hasStake {
				stakeAt = nft.UpdatedAt.Format(time.DateOnly)
			}
		}
		nftList = append(nftList, types.UserNft{
			TokenId:  token,
			HasStake: hasStake,
			StakeAt:  stakeAt,
		})
	}

	return &types.UserNftList{
		NftList: nftList,
	}, nil
}