52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
// ParseDurationString parses a string to a duration
|
|
// Duration notations are an integer followed by a unit
|
|
// Units are s = second, m = minute, d = day, w = week, M = month, y = year
|
|
// Example 1y is the same as 1 year.
|
|
func ParseDurationString(input string) (time.Duration, error) {
|
|
var duration time.Duration
|
|
|
|
matches := reDuration.FindStringSubmatch(input)
|
|
|
|
switch {
|
|
case len(matches) == 3 && matches[2] != "":
|
|
d, _ := strconv.Atoi(matches[1])
|
|
|
|
switch matches[2] {
|
|
case "y":
|
|
duration = time.Duration(d) * Year
|
|
case "M":
|
|
duration = time.Duration(d) * Month
|
|
case "w":
|
|
duration = time.Duration(d) * Week
|
|
case "d":
|
|
duration = time.Duration(d) * Day
|
|
case "h":
|
|
duration = time.Duration(d) * Hour
|
|
case "m":
|
|
duration = time.Duration(d) * time.Minute
|
|
case "s":
|
|
duration = time.Duration(d) * time.Second
|
|
}
|
|
case input == "0" || len(matches) == 3:
|
|
seconds, err := strconv.Atoi(input)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("could not parse '%s' as a duration: %w", input, err)
|
|
}
|
|
|
|
duration = time.Duration(seconds) * time.Second
|
|
case input != "":
|
|
// Throw this error if input is anything other than a blank string, blank string will default to a duration of nothing.
|
|
return 0, fmt.Errorf("could not parse '%s' as a duration", input)
|
|
}
|
|
|
|
return duration, nil
|
|
}
|