♻️ Refactor Incremental Generator to single function with options
This commit is contained in:
parent
63d538d889
commit
586fe4f1de
4 changed files with 176 additions and 182 deletions
|
|
@ -7,7 +7,6 @@ import (
|
|||
"errors"
|
||||
"fmt"
|
||||
"hash"
|
||||
"strconv"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/gosimple/slug"
|
||||
|
|
@ -64,60 +63,6 @@ func UUID() Generator {
|
|||
return uuidGen
|
||||
}
|
||||
|
||||
// Incremental generates a name that is a series of integers starting at 0
|
||||
func Incremental() Generator {
|
||||
return IncrementalWithStartAndStep(0, 1)
|
||||
}
|
||||
|
||||
// IncrementalFormat generates a name that is a series of integers starting at 0, formatted with Printf
|
||||
// This is mostly likely useful with a format like "%02d"
|
||||
func IncrementalFormat(format string) Generator {
|
||||
return IncrementalFormatWithStartAndStep(0, 1, format)
|
||||
}
|
||||
|
||||
// IncrementalWithStart generates a name that is a series of integers starting at the specified number
|
||||
func IncrementalWithStart(start int) Generator {
|
||||
return IncrementalWithStartAndStep(start, 1)
|
||||
}
|
||||
|
||||
// IncrementalFormatWithStart generates a name that is a series of integers starting at the specified number, formatted with Printf
|
||||
func IncrementalFormatWithStart(start int, format string) Generator {
|
||||
return IncrementalFormatWithStartAndStep(start, 1, format)
|
||||
}
|
||||
|
||||
// IncrementalWithStep generates a name that is a series of integers, starting at 0, and increasing the specified number each time
|
||||
func IncrementalWithStep(step int) Generator {
|
||||
return IncrementalWithStartAndStep(0, step)
|
||||
}
|
||||
|
||||
// IncrementalFormatWithStep generates a name that is a series of integers, starting at 0, and increasing the specified number each time,
|
||||
// formatted with Printf
|
||||
func IncrementalFormatWithStep(step int, format string) Generator {
|
||||
return IncrementalFormatWithStartAndStep(0, step, format)
|
||||
}
|
||||
func incrementalHelper(start, step int, cb func(int) string) Generator {
|
||||
next := start
|
||||
return func(*Config) (string, error) {
|
||||
out := cb(next)
|
||||
next += step
|
||||
return out, nil
|
||||
}
|
||||
}
|
||||
|
||||
// InrementalWithStartAndStep generates a name that is a series of integers, starting at the specified number,
|
||||
// and increasing the specified step each time
|
||||
func IncrementalWithStartAndStep(start, step int) Generator {
|
||||
return incrementalHelper(start, step, strconv.Itoa)
|
||||
}
|
||||
|
||||
// IncrementalFormatWithStartAndStep generates a name that is a series of integers, starting at the specified number,
|
||||
// and increasing the specified step each time, formatted with Printf
|
||||
func IncrementalFormatWithStartAndStep(start, step int, format string) Generator {
|
||||
return incrementalHelper(start, step, func(i int) string {
|
||||
return fmt.Sprintf(format, i)
|
||||
})
|
||||
}
|
||||
|
||||
// ErrMissingOriginal is the error returned by Slug if there is no filename
|
||||
var ErrMissingOriginal = errors.New("missing original filename")
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue