package model

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

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)
	}

	customNhNftHolderModel struct {
		*defaultNhNftHolderModel
	}
)

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) 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))
}