69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type FlagStringArray []string
|
|
|
|
func (array *FlagStringArray) String() string {
|
|
return strings.Join(*array, ", ")
|
|
}
|
|
|
|
func (array *FlagStringArray) Set(s string) error {
|
|
*array = append(*array, s)
|
|
return nil
|
|
}
|
|
|
|
type FileList FlagStringArray
|
|
|
|
func (fileList FileList) Strings() []string {
|
|
return ([]string)(fileList)
|
|
}
|
|
|
|
func (fileList *FileList) String() string {
|
|
return (*FlagStringArray)(fileList).String()
|
|
}
|
|
|
|
func (fileList *FileList) Set(s string) error {
|
|
if _, err := os.Stat(s); os.IsNotExist(err) {
|
|
return fmt.Errorf("file does not exist: %s", s)
|
|
}
|
|
return (*FlagStringArray)(fileList).Set(s)
|
|
}
|
|
|
|
type Period struct {
|
|
Start time.Time
|
|
Stop time.Time
|
|
}
|
|
|
|
func (period Period) Empty() bool {
|
|
return period.Stop.Sub(period.Start) == 0
|
|
}
|
|
|
|
func (period *Period) String() string {
|
|
return fmt.Sprintf("%s..%s", period.Start, period.Stop)
|
|
}
|
|
|
|
func (period *Period) Set(s string) error {
|
|
if result, err := time.Parse("2006", s); err == nil {
|
|
period.Start = result
|
|
period.Stop = result.AddDate(1, 0, 0).Add(-1 * time.Minute)
|
|
return nil
|
|
}
|
|
if result, err := time.Parse("2006-01", s); err == nil {
|
|
period.Start = result
|
|
period.Stop = result.AddDate(0, 1, 0).Add(-1 * time.Minute)
|
|
return nil
|
|
}
|
|
if result, err := time.Parse("2006-01-02", s); err == nil {
|
|
period.Start = result
|
|
period.Stop = result.AddDate(0, 0, 1).Add(-1 * time.Minute)
|
|
return nil
|
|
}
|
|
return fmt.Errorf("unimplemented format: %s", s)
|
|
}
|