option.go

package tiktok

import (
	"net/http"
	"time"
)

type HTTPClient interface {
	Do(*http.Request) (*http.Response, error)
}

type Logger interface {
	Print(...interface{})
	Printf(string, ...interface{})
}

type option struct {
	client   HTTPClient
	logger   Logger
	endpoint string
}

type Option func(o *option)

// WithHTTPClient setup HTTPClient with given client.
func WithHTTPClient(c HTTPClient) Option {
	return func(o *option) {
		o.client = c
	}
}

// WithLogger setup logger for debug perporse.
// Default Logger will be discard.
func WithLogger(l Logger) Option {
	return func(o *option) {
		o.logger = l
	}
}

// WithEndpoint setup endpoint for request.
// TikTok is not support sandbox mode yet, but it will be lauching soon.
// You can also use this for mocking purpose.
func WithEndpoint(e string) Option {
	return func(o *option) {
		o.endpoint = e
	}
}

type nopLogger struct{}

func (n *nopLogger) Print(...interface{}) {
}

func (n *nopLogger) Printf(string, ...interface{}) {
}

func defaultOpt() *option {
	return &option{
		client: &http.Client{
			Timeout: 30 * time.Second,
		},
		logger:   &nopLogger{},
		endpoint: APIBaseURL,
	}
}