nomino/options.go

package nomino

import "strings"

// Option sets configuration parameters for Config.
type Option func(c *Config)

// WithOriginal sets the original filename.
// This will be included in the generated name after the generated string and before the suffix.
func WithOriginal(o string) Option {
	return func(c *Config) {
		c.original = "_" + o
	}
}

// WithPrefix sets a prefix for the generated name.
func WithPrefix(p string) Option {
	return func(c *Config) {
		c.prefix = p + "_"
	}
}

// WithSuffix sets a suffix for the generated name. It will be included in the base name before the suffix.
func WithSuffix(s string) Option {
	return func(c *Config) {
		c.suffix = "_" + s
	}
}

// WithoutExtension sets no extension for the generated filename. By default, it will be txt
func WithoutExtension() Option {
	return func(c *Config) {
		c.extension = ""
	}
}

// WithExtension sets the extension for the generated filename.
func WithExtension(ext string) Option {
	return func(c *Config) {
		c.extension = "." + strings.TrimPrefix(ext, ".")
	}
}