✨ Add options.go
This commit is contained in:
parent
511a81cab3
commit
bd8f9ae8a6
4 changed files with 118 additions and 0 deletions
52
options.go
Normal file
52
options.go
Normal file
|
|
@ -0,0 +1,52 @@
|
|||
package nomino
|
||||
|
||||
import "strings"
|
||||
|
||||
type generator func() string
|
||||
|
||||
type config struct {
|
||||
original string
|
||||
prefix string
|
||||
suffix string
|
||||
extension string
|
||||
generator generator
|
||||
}
|
||||
|
||||
func defaultConf() config {
|
||||
return config{
|
||||
extension: ".txt",
|
||||
generator: func() string { return "abc" },
|
||||
}
|
||||
}
|
||||
|
||||
type Option func(c *config)
|
||||
|
||||
func WithOriginal(o string) Option {
|
||||
return func(c *config) {
|
||||
c.original = "_" + o
|
||||
}
|
||||
}
|
||||
|
||||
func WithPrefix(p string) Option {
|
||||
return func(c *config) {
|
||||
c.prefix = p + "_"
|
||||
}
|
||||
}
|
||||
|
||||
func WithSuffix(s string) Option {
|
||||
return func(c *config) {
|
||||
c.suffix = "_" + s
|
||||
}
|
||||
}
|
||||
|
||||
func WithoutExtension() Option {
|
||||
return func(c *config) {
|
||||
c.extension = ""
|
||||
}
|
||||
}
|
||||
|
||||
func WithExtension(ext string) Option {
|
||||
return func(c *config) {
|
||||
c.extension = "." + strings.TrimPrefix(ext, ".")
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue