package model

import (
	"context"
	"errors"
	"fmt"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlc"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
	"strings"
)

var _ NhNftHolderModel = (*customNhNftHolderModel)(nil)

type (
	// NhNftHolderModel is an interface to be customized, add more methods here,
	// and implement the added methods in customNhNftHolderModel.
	NhNftHolderModel interface {
		nhNftHolderModel
		withSession(session sqlx.Session) NhNftHolderModel
		FindOtherUpdateSeq(ctx context.Context, updateSeq int) ([]*NhNftHolder, error)
		DeleteOtherUpdateSeq(ctx context.Context, updateSeq int) error
		FindTokensByAddress(ctx context.Context, address string) ([]string, error)
		HoldNft(ctx context.Context, uid uint) bool
		FindTokensByAddresses(ctx context.Context, addresses []string) ([]*NhNftHolder, error)
	}

	customNhNftHolderModel struct {
		*defaultNhNftHolderModel
	}
)

func (m *customNhNftHolderModel) HoldNft(ctx context.Context, uid uint) bool {
	query := fmt.Sprintf("select count(1) from %s h join nh_wallet w on h.address = w.address where w.`uid` = ?", m.table)
	var count int
	err := m.conn.QueryRowCtx(ctx, &count, query, uid)
	if err != nil {
		if !errors.Is(err, sqlc.ErrNotFound) {
			logx.Errorw("HoldNft query failed", logx.Field("query", query), logx.Field("args", uid), logx.Field("err", err))
		}
		return false
	}
	return count > 0
}

func (m *customNhNftHolderModel) FindTokensByAddress(ctx context.Context, address string) ([]string, error) {
	query := fmt.Sprintf("select `token_id` from %s where `address` = ?", m.table)
	var tokens []string
	err := m.conn.QueryRowsCtx(ctx, &tokens, query, address)
	if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
		return nil, err
	}
	return tokens, nil
}

func (m *customNhNftHolderModel) FindTokensByAddresses(ctx context.Context, addresses []string) ([]*NhNftHolder, error) {
	// 生成占位符
	placeholders := make([]string, len(addresses))
	args := make([]interface{}, len(addresses))

	for i, val := range addresses {
		placeholders[i] = "?"
		args[i] = val
	}
	query := fmt.Sprintf("select %s from %s where `address` in (%s)", nhNftHolderRows, m.table, strings.Join(placeholders, ","))
	var tokens []*NhNftHolder
	err := m.conn.QueryRowsCtx(ctx, &tokens, query, args...)
	if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
		return nil, err
	}
	return tokens, nil
}

func (m *customNhNftHolderModel) DeleteOtherUpdateSeq(ctx context.Context, updateSeq int) error {
	delSql := fmt.Sprintf("delete from %s where `update_seq` != ?", m.table)
	_, err := m.conn.ExecCtx(ctx, delSql, updateSeq)
	return err
}

func (m *customNhNftHolderModel) FindOtherUpdateSeq(ctx context.Context, updateSeq int) ([]*NhNftHolder, error) {
	query := fmt.Sprintf("select %s from %s where `update_seq` != ?", nhNftHolderRows, m.table)
	var resp []*NhNftHolder
	err := m.conn.QueryRowsCtx(ctx, &resp, query, updateSeq)
	if err != nil && !errors.Is(err, sqlx.ErrNotFound) {
		return nil, err
	}
	return resp, nil
}

// NewNhNftHolderModel returns a model for the database table.
func NewNhNftHolderModel(conn sqlx.SqlConn) NhNftHolderModel {
	return &customNhNftHolderModel{
		defaultNhNftHolderModel: newNhNftHolderModel(conn),
	}
}

func (m *customNhNftHolderModel) withSession(session sqlx.Session) NhNftHolderModel {
	return NewNhNftHolderModel(sqlx.NewSqlConnFromSession(session))
}