hunter_options.go

package emailvalidator

import (
	"net/url"
)

// HunterAPIOption specifies options for Hunter.io API client
type HunterAPIOption struct {
	baseURL    *url.URL
	apiVersion string
	rate       HunterRate
	blocking   bool
}

// HunterAPIOptionFunc is a function type for setting Hunter.io API client options
type HunterAPIOptionFunc func(*HunterAPIOption) error

// WithHunterAPIRate is used to specific rate limit on http client
func WithHunterAPIRate(rate HunterRate) HunterAPIOptionFunc {
	return func(opts *HunterAPIOption) error {
		opts.rate = rate
		return nil
	}
}

// WithHunterAPIBlocking is used to direct rateLimiter to wait until rate limit interval ends, if rate limit is reached.
func WithHunterAPIBlocking() HunterAPIOptionFunc {
	return func(opts *HunterAPIOption) error {
		opts.blocking = true
		return nil
	}
}

// WithHunterAPIBaseURL is used to set base url of hunter api service
func WithHunterAPIBaseURL(url *url.URL) HunterAPIOptionFunc {
	return func(opts *HunterAPIOption) error {
		if url == nil || url.String() == "" {
			return ErrEmptyBaseURL
		}
		opts.baseURL = url
		return nil
	}
}

// WithHunterAPIVersion is used to set API version of abstract api
func WithHunterAPIVersion(version string) HunterAPIOptionFunc {
	return func(opts *HunterAPIOption) error {
		if version == "" {
			return ErrEmptyAPIVersion
		}
		opts.apiVersion = version
		return nil
	}
}

func parseHunterAPIOptions(options ...HunterAPIOptionFunc) (*HunterAPIOption, error) {
	defAPIURL, _ := url.Parse(defaultHunterAPIURL)

	opts := &HunterAPIOption{
		rate:       HunterAPIRate,
		blocking:   false,
		baseURL:    defAPIURL,
		apiVersion: defaultHunterAPIVersion,
	}

	for _, o := range options {
		if err := o(opts); err != nil {
			return nil, err
		}
	}

	return opts, nil
}