options.go

package graceful

import "context"

// Option interface for configuration.
type Option interface {
	Apply(*Options)
}

// OptionFunc is a function that configures a graceful shutdown.
type OptionFunc func(*Options)

// Apply calls f(option)
func (f OptionFunc) Apply(option *Options) {
	f(option)
}

// Options for graceful shutdown
type Options struct {
	ctx    context.Context
	logger Logger
}

// WithContext custom context
func WithContext(ctx context.Context) Option {
	return OptionFunc(func(o *Options) {
		o.ctx = ctx
	})
}

// WithLogger custom logger
func WithLogger(logger Logger) Option {
	return OptionFunc(func(o *Options) {
		o.logger = logger
	})
}

// newOptions creates a new Options instance with default settings and applies any provided Option modifiers.
// It initializes the Options struct with a default background context and a new logger,
// then iterates over each given Option to adjust the configuration accordingly.
//
// Parameters:
//   - opts: A variadic list of Option functions that modify the default Options.
//
// Returns:
//   - Options: The customized Options struct after all modifications have been applied.
func newOptions(opts ...Option) Options {
	defaultOpts := Options{
		ctx:    context.Background(),
		logger: NewLogger(),
	}

	// Loop through each option
	for _, opt := range opts {
		// Call the option giving the instantiated
		opt.Apply(&defaultOpts)
	}

	return defaultOpts
}