110 lines
1.6 KiB
Go
110 lines
1.6 KiB
Go
package command
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
DateFormat = "2006-01-02"
|
|
TimeFormat = "15:04"
|
|
)
|
|
|
|
var (
|
|
ErrWrongCommand = errors.New("wrong command")
|
|
ErrInvalidArg = errors.New("invalid argument")
|
|
)
|
|
|
|
type Flag interface {
|
|
Set(val string) error
|
|
IsSet() bool
|
|
Get() any
|
|
}
|
|
|
|
type FlagString struct {
|
|
Name string
|
|
Value string
|
|
}
|
|
|
|
func (fs *FlagString) Set(val string) error {
|
|
fs.Value = val
|
|
return nil
|
|
}
|
|
|
|
func (fs *FlagString) IsSet() bool {
|
|
return fs.Value != ""
|
|
}
|
|
|
|
func (fs *FlagString) Get() any {
|
|
return fs.Value
|
|
}
|
|
|
|
type FlagDate struct {
|
|
Name string
|
|
Value time.Time
|
|
}
|
|
|
|
func (ft *FlagDate) Set(val string) error {
|
|
d, err := time.Parse(DateFormat, val)
|
|
if err != nil {
|
|
return fmt.Errorf("could not parse date: %v", d)
|
|
}
|
|
ft.Value = d
|
|
|
|
return nil
|
|
}
|
|
|
|
func (ft *FlagDate) IsSet() bool {
|
|
return ft.Value.IsZero()
|
|
}
|
|
|
|
func (fs *FlagDate) Get() any {
|
|
return fs.Value
|
|
}
|
|
|
|
type FlagTime struct {
|
|
Name string
|
|
Value time.Time
|
|
}
|
|
|
|
func (ft *FlagTime) Set(val string) error {
|
|
d, err := time.Parse(TimeFormat, val)
|
|
if err != nil {
|
|
return fmt.Errorf("could not parse date: %v", d)
|
|
}
|
|
ft.Value = d
|
|
|
|
return nil
|
|
}
|
|
|
|
func (fd *FlagTime) IsSet() bool {
|
|
return fd.Value.IsZero()
|
|
}
|
|
|
|
func (fs *FlagTime) Get() any {
|
|
return fs.Value
|
|
}
|
|
|
|
type FlagDuration struct {
|
|
Name string
|
|
Value time.Duration
|
|
}
|
|
|
|
func (fd *FlagDuration) Set(val string) error {
|
|
dur, err := time.ParseDuration(val)
|
|
if err != nil {
|
|
return fmt.Errorf("could not parse duration: %v", err)
|
|
}
|
|
fd.Value = dur
|
|
return nil
|
|
}
|
|
|
|
func (fd *FlagDuration) IsSet() bool {
|
|
return fd.Value.String() != "0s"
|
|
}
|
|
|
|
func (fs *FlagDuration) Get() any {
|
|
return fs.Value
|
|
}
|