yanic/lib/duration/duration.go

package duration

import (
	"fmt"
	"strconv"
	"time"

	"github.com/pkg/errors"
)

// Duration is a TOML datatype
// A duration string is a possibly signed sequence of
// decimal numbers  and a unit suffix,
// such as "300s", "1.5h" or "5d".
// Valid time units are "s", "m", "h", "d", "w".
type Duration struct {
	time.Duration
}

// UnmarshalTOML parses a duration string.
func (d *Duration) UnmarshalTOML(dataInterface interface{}) error {
	var data string
	switch dataInterface.(type) {
	case string:
		data = dataInterface.(string)
	default:
		return fmt.Errorf("invalid duration: \"%s\"", dataInterface)
	}
	// " + int + unit + "
	if len(data) < 2 {
		return fmt.Errorf("invalid duration: \"%s\"", data)
	}

	unit := data[len(data)-1]
	value, err := strconv.Atoi(string(data[:len(data)-1]))
	if err != nil {
		return errors.Wrapf(err, "unable to parse duration \"%s\"", data)
	}

	switch unit {
	case 's':
		d.Duration = time.Duration(value) * time.Second
	case 'm':
		d.Duration = time.Duration(value) * time.Minute
	case 'h':
		d.Duration = time.Duration(value) * time.Hour
	case 'd':
		d.Duration = time.Duration(value) * time.Hour * 24
	case 'w':
		d.Duration = time.Duration(value) * time.Hour * 24 * 7
	case 'y':
		d.Duration = time.Duration(value) * time.Hour * 24 * 365
	default:
		return fmt.Errorf("invalid duration unit \"%s\"", string(unit))
	}

	return nil
}