57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package nomino
|
|
|
|
import "strings"
|
|
|
|
type config struct {
|
|
original string
|
|
prefix string
|
|
suffix string
|
|
extension string
|
|
generator Generator
|
|
}
|
|
|
|
func defaultConf() config {
|
|
return config{
|
|
extension: ".txt",
|
|
generator: uuidGen,
|
|
}
|
|
}
|
|
|
|
// Option is an option for nomino
|
|
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, ".")
|
|
}
|
|
}
|