mirror of
https://github.com/wavetermdev/waveterm.git
synced 2024-12-22 16:48:23 +01:00
105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
// Copyright 2024, Command Line Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package daystr
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/wavetermdev/waveterm/pkg/util/utilfn"
|
|
)
|
|
|
|
var customDayStrRe = regexp.MustCompile(`^((?:\d{4}-\d{2}-\d{2})|today|yesterday|bom|bow)?((?:[+-]\d+[dwm])*)$`)
|
|
var daystrRe = regexp.MustCompile(`^(\d{4})-(\d{2})-(\d{2})$`)
|
|
|
|
func GetCurDayStr() string {
|
|
now := time.Now()
|
|
dayStr := now.Format("2006-01-02")
|
|
return dayStr
|
|
}
|
|
|
|
func GetRelDayStr(relDays int) string {
|
|
now := time.Now()
|
|
dayStr := now.AddDate(0, 0, relDays).Format("2006-01-02")
|
|
return dayStr
|
|
}
|
|
|
|
// accepts a custom format string to return a daystr
|
|
// can be either a prefix, a delta, or a prefix w/ a delta
|
|
// if no prefix is given, "today" is assumed
|
|
// examples: today-2d, bow, bom+1m-1d (that's end of the month), 2024-04-01+1w
|
|
//
|
|
// prefixes:
|
|
//
|
|
// yyyy-mm-dd
|
|
// today
|
|
// yesterday
|
|
// bom (beginning of month)
|
|
// bow (beginning of week -- sunday)
|
|
//
|
|
// deltas:
|
|
//
|
|
// +[n]d, -[n]d (e.g. +1d, -5d)
|
|
// +[n]w, -[n]w (e.g. +2w)
|
|
// +[n]m, -[n]m (e.g. -1m)
|
|
// deltas can be combined e.g. +1w-2d
|
|
func GetCustomDayStr(format string) (string, error) {
|
|
m := customDayStrRe.FindStringSubmatch(format)
|
|
if m == nil {
|
|
return "", fmt.Errorf("invalid daystr format")
|
|
}
|
|
prefix, deltas := m[1], m[2]
|
|
if prefix == "" {
|
|
prefix = "today"
|
|
}
|
|
var rtnTime time.Time
|
|
now := time.Now()
|
|
switch prefix {
|
|
case "today":
|
|
rtnTime = now
|
|
case "yesterday":
|
|
rtnTime = now.AddDate(0, 0, -1)
|
|
case "bom":
|
|
rtnTime = time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
|
|
case "bow":
|
|
weekday := now.Weekday()
|
|
if weekday == time.Sunday {
|
|
rtnTime = now
|
|
} else {
|
|
rtnTime = now.AddDate(0, 0, -int(weekday))
|
|
}
|
|
default:
|
|
m = daystrRe.FindStringSubmatch(prefix)
|
|
if m == nil {
|
|
return "", fmt.Errorf("invalid prefix format")
|
|
}
|
|
year, month, day := m[1], m[2], m[3]
|
|
yearInt, monthInt, dayInt := utilfn.AtoiNoErr(year), utilfn.AtoiNoErr(month), utilfn.AtoiNoErr(day)
|
|
if yearInt == 0 || monthInt == 0 || dayInt == 0 {
|
|
return "", fmt.Errorf("invalid prefix format")
|
|
}
|
|
rtnTime = time.Date(yearInt, time.Month(monthInt), dayInt, 0, 0, 0, 0, now.Location())
|
|
}
|
|
for _, delta := range regexp.MustCompile(`[+-]\d+[dwm]`).FindAllString(deltas, -1) {
|
|
deltaVal, err := strconv.Atoi(delta[1 : len(delta)-1])
|
|
if err != nil {
|
|
return "", fmt.Errorf("invalid delta format")
|
|
}
|
|
if delta[0] == '-' {
|
|
deltaVal = -deltaVal
|
|
}
|
|
switch delta[len(delta)-1] {
|
|
case 'd':
|
|
rtnTime = rtnTime.AddDate(0, 0, deltaVal)
|
|
case 'w':
|
|
rtnTime = rtnTime.AddDate(0, 0, deltaVal*7)
|
|
case 'm':
|
|
rtnTime = rtnTime.AddDate(0, deltaVal, 0)
|
|
}
|
|
}
|
|
return rtnTime.Format("2006-01-02"), nil
|
|
}
|