Moved code to top-level.

config.go

@@ -0,0 +1,163 @@
+package honeybee
+
+import (
+	"git.wisehodl.dev/jay/go-honeybee/errors"
+	"time"
+)
+
+type CloseHandler func(code int, text string) error
+
+type Config struct {
+	CloseHandler CloseHandler
+	ReadTimeout  time.Duration
+	WriteTimeout time.Duration
+	Retry        *RetryConfig
+}
+
+type RetryConfig struct {
+	MaxRetries   int
+	InitialDelay time.Duration
+	MaxDelay     time.Duration
+	JitterFactor float64
+}
+
+type ConfigOption func(*Config) error
+
+func NewConfig(options ...ConfigOption) (*Config, error) {
+	conf := GetDefaultConfig()
+	if err := SetConfig(conf, options...); err != nil {
+		return nil, err
+	}
+	if err := ValidateConfig(conf); err != nil {
+		return nil, err
+	}
+	return conf, nil
+}
+
+func GetDefaultConfig() *Config {
+	return &Config{}
+}
+
+func GetDefaultRetryConfig() *RetryConfig {
+	return &RetryConfig{
+		MaxRetries:   0, // Infinite retries
+		InitialDelay: 1 * time.Second,
+		MaxDelay:     5 * time.Second,
+		JitterFactor: 0.5,
+	}
+}
+
+func SetConfig(config *Config, options ...ConfigOption) error {
+	for _, option := range options {
+		if err := option(config); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func ValidateConfig(config *Config) error {
+	if config.Retry != nil {
+		if config.Retry.InitialDelay > config.Retry.MaxDelay {
+			return errors.NewConfigError("initial delay may not exceed maximum delay")
+		}
+	}
+
+	return nil
+}
+
+// Configuration Options
+
+func WithCloseHandler(handler CloseHandler) ConfigOption {
+	return func(c *Config) error {
+		c.CloseHandler = handler
+		return nil
+	}
+}
+
+// When ReadTimeout is set to zero, read timeouts are disabled.
+func WithReadTimeout(value time.Duration) ConfigOption {
+	return func(c *Config) error {
+		if value < 0 {
+			return errors.InvalidReadTimeout
+		}
+		c.ReadTimeout = value
+		return nil
+	}
+}
+
+// When WriteTimeout is set to zero, read timeouts are disabled.
+func WithWriteTimeout(value time.Duration) ConfigOption {
+	return func(c *Config) error {
+		if value < 0 {
+			return errors.InvalidWriteTimeout
+		}
+		c.WriteTimeout = value
+		return nil
+	}
+}
+
+// WithRetry enables retry with default parameters (infinite retries,
+// 1s initial delay, 5s max delay, 0.5 jitter factor).
+//
+// If passed after granular retry options (WithRetryMaxRetries, etc.),
+// it will overwrite them. Use either WithRetry alone or the granular
+// options; not both.
+func WithRetry() ConfigOption {
+	return func(c *Config) error {
+		c.Retry = GetDefaultRetryConfig()
+		return nil
+	}
+}
+
+func WithRetryMaxRetries(value int) ConfigOption {
+	return func(c *Config) error {
+		if c.Retry == nil {
+			c.Retry = GetDefaultRetryConfig()
+		}
+		if value < 0 {
+			return errors.InvalidRetryMaxRetries
+		}
+		c.Retry.MaxRetries = value
+		return nil
+	}
+}
+
+func WithRetryInitialDelay(value time.Duration) ConfigOption {
+	return func(c *Config) error {
+		if c.Retry == nil {
+			c.Retry = GetDefaultRetryConfig()
+		}
+		if value <= 0 {
+			return errors.InvalidRetryInitialDelay
+		}
+		c.Retry.InitialDelay = value
+		return nil
+	}
+}
+
+func WithRetryMaxDelay(value time.Duration) ConfigOption {
+	return func(c *Config) error {
+		if c.Retry == nil {
+			c.Retry = GetDefaultRetryConfig()
+		}
+		if value <= 0 {
+			return errors.InvalidRetryMaxDelay
+		}
+		c.Retry.MaxDelay = value
+		return nil
+	}
+}
+
+func WithRetryJitterFactor(value float64) ConfigOption {
+	return func(c *Config) error {
+		if c.Retry == nil {
+			c.Retry = GetDefaultRetryConfig()
+		}
+		if value < 0.0 || value > 1.0 {
+			return errors.InvalidRetryJitterFactor
+		}
+		c.Retry.JitterFactor = value
+		return nil
+	}
+}