package client

import (
	"math/rand"
	"time"

	"github.com/miekg/dns"
	"github.com/mr-karan/doggo/internal/models"
	"github.com/mr-karan/doggo/internal/resolvers"
	"github.com/mr-karan/logf"
)

// Client represents the structure for all app wide configuration.
type Client struct {
	Log         *logf.Logger
	Version     string
	QueryFlags  models.QueryFlags
	Questions   []dns.Question
	Resolvers   []resolvers.Resolver
	Nameservers []models.Nameserver
}

// New initializes an instance of App which holds app wide configuration.
func New(logger *logf.Logger, buildVersion string) Client {
	return Client{
		Log:     logger,
		Version: buildVersion,
		QueryFlags: models.QueryFlags{
			QNames:      []string{},
			QTypes:      []string{},
			QClasses:    []string{},
			Nameservers: []string{},
		},
		Nameservers: []models.Nameserver{},
	}
}

// LookupWithRetry attempts a DNS Lookup with retries for a given Question.
func (hub *Client) LookupWithRetry(attempts int, resolver resolvers.Resolver, ques dns.Question) (resolvers.Response, error) {
	resp, err := resolver.Lookup(ques)
	if err != nil {
		// Retry lookup.
		attempts--
		if attempts > 0 {
			// Add some random delay.
			time.Sleep(time.Millisecond*300 + (time.Duration(rand.Int63n(int64(time.Millisecond*100))))/2)
			hub.Log.Debug("retrying lookup")
			return hub.LookupWithRetry(attempts, resolver, ques)
		}
		return resolvers.Response{}, err
	}
	return resp, nil
}