config.go

package moonshot

import (
	"errors"
	"strings"
)

type Config struct {
	Host   string
	APIKey string
	Debug  bool
}

const DefaultHost = "https://api.moonshot.cn"

func newConfigDefault() *Config {
	return &Config{
		Host: DefaultHost,
	}
}

// NewConfig creates a new config
func NewConfig(opts ...Option) *Config {
	cfg := newConfigDefault()
	for _, opt := range opts {
		opt(cfg)
	}
	return cfg
}

func (c *Config) preCheck() error {
	if len(c.APIKey) == 0 {
		return errors.New("API key is required")
	}
	return nil
}

type Option func(*Config)

// WithHost sets the host
func WithHost(host string) Option {
	return func(c *Config) {
		c.Host = strings.TrimSuffix(host, "/")
	}
}

// WithAPIKey sets the API key
func WithAPIKey(key string) Option {
	return func(c *Config) {
		c.APIKey = key
	}
}