config.go

package youtube

import (
	"errors"
	"fmt"
	"log/slog"

	"github.com/rubpy/crawly"
	"github.com/rubpy/crawly/cclient"
	"google.golang.org/api/youtube/v3"
)

//////////////////////////////////////////////////

type config struct {
	logger  *slog.Logger
	client  cclient.Client
	service *youtube.Service

	settings struct {
		v  CrawlerSettings
		ok bool
	}
}

var (
	NilConfig  = errors.New("config is nil")
	NilClient  = errors.New("client is nil")
	NilService = errors.New("service is nil")
)

func validateConfig(cfg *config) error {
	if cfg == nil {
		return NilConfig
	}

	if cfg.service == nil {
		return NilService
	}

	return nil
}

func buildCrawlerFromConfig(cfg *config) (cr *Crawler, err error) {
	if cfg == nil {
		err = NilConfig
		return
	}

	cl := cfg.client
	if cl == nil {
		logger := cfg.logger
		if logger != nil {
			logger = logger.WithGroup("client")
		}

		cl, err = cclient.NewClient(cclient.WithLogger(logger))
		if err != nil {
			return nil, fmt.Errorf("cclient.NewClient: %w", err)
		}
	}

	svc := cfg.service
	if svc == nil {
		return nil, NilService
	}

	cr = &Crawler{
		client:  cl,
		service: svc,
	}

	cr.Crawler.SetLogger(cfg.logger)
	crawly.SetCrawlerHandlers(&cr.Crawler, crawly.CrawlerHandlers{
		Order:  cr.orderHandler,
		Entity: cr.entityHandler,
	})

	if cfg.settings.ok {
		cr.SetSettings(cfg.settings.v)
	} else {
		cr.SetSettings(DefaultSettings)
	}

	return cr, nil
}

type ConfigOption func(cfg *config)

//////////////////////////////////////////////////

func WithLogger(logger *slog.Logger) ConfigOption {
	return func(cfg *config) {
		cfg.logger = logger
	}
}

func WithClient(client cclient.Client) ConfigOption {
	return func(cfg *config) {
		cfg.client = client
	}
}

func WithService(service *youtube.Service) ConfigOption {
	return func(cfg *config) {
		cfg.service = service
	}
}

func WithSettings(settings CrawlerSettings) ConfigOption {
	return func(cfg *config) {
		cfg.settings.v = settings
		cfg.settings.ok = true
	}
}