package resolver

import (
	"context"

	"github.com/miekg/dns"
)

// ExtractRecordValue exports extractRecordValue for testing.
func ExtractRecordValue(rr dns.RR) string {
	return extractRecordValue(rr)
}

// UsableReply exports usableReply for testing.
func UsableReply(resp *dns.Msg, zone string, name string) bool {
	return usableReply(resp, zone, name)
}

// NSSetFrom exports nsSetFrom for testing.
func NSSetFrom(resp *dns.Msg, domain string) []string {
	return nsSetFrom(resp, domain)
}

// CollectIPs exports collectIPs for testing.
func CollectIPs(
	results map[string]*NameserverResponse,
) ([]string, string, error) {
	return collectIPs(results)
}

// QueryServers exports queryServers for testing.
func (r *Resolver) QueryServers(
	ctx context.Context,
	servers []string,
	zone string,
	name string,
	qtype uint16,
) (*dns.Msg, error) {
	return r.queryServers(ctx, servers, zone, name, qtype)
}

// QueryEachNS exports queryEachNS for testing.
func (r *Resolver) QueryEachNS(
	ctx context.Context,
	nameservers []string,
	hostname string,
) (map[string]*NameserverResponse, error) {
	return r.queryEachNS(ctx, nameservers, hostname)
}

// ResolveNSIPs exports resolveNSIPs for testing.
func (r *Resolver) ResolveNSIPs(
	ctx context.Context,
	nsNames []string,
) []string {
	return r.resolveNSIPs(ctx, nsNames)
}

// RootServerList exports rootServerList for testing.
func RootServerList() []string {
	return rootServerList()
}

// Shuffled exports shuffled for testing.
func Shuffled(
	servers []string,
	shuffle func(n int, swap func(i, j int)),
) []string {
	return shuffled(servers, shuffle)
}