mirror of https://github.com/muety/wakapi.git
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package helpers
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/muety/wakapi/models"
|
|
"github.com/muety/wakapi/utils"
|
|
"time"
|
|
)
|
|
|
|
func ParseInterval(interval string) (*models.IntervalKey, error) {
|
|
for _, i := range models.AllIntervals {
|
|
if i.HasAlias(interval) {
|
|
return i, nil
|
|
}
|
|
}
|
|
return nil, errors.New("not a valid interval")
|
|
}
|
|
|
|
func MustParseInterval(interval string) *models.IntervalKey {
|
|
key, _ := ParseInterval(interval)
|
|
return key
|
|
}
|
|
|
|
func MustResolveIntervalRawTZ(interval string, tz *time.Location) (from, to time.Time) {
|
|
_, from, to = ResolveIntervalRawTZ(interval, tz)
|
|
return from, to
|
|
}
|
|
|
|
func ResolveIntervalRawTZ(interval string, tz *time.Location) (err error, from, to time.Time) {
|
|
parsed, err := ParseInterval(interval)
|
|
if err != nil {
|
|
return err, time.Time{}, time.Time{}
|
|
}
|
|
return ResolveIntervalTZ(parsed, tz)
|
|
}
|
|
|
|
func ResolveIntervalTZ(interval *models.IntervalKey, tz *time.Location) (err error, from, to time.Time) {
|
|
now := time.Now().In(tz)
|
|
to = now
|
|
|
|
switch interval {
|
|
case models.IntervalToday:
|
|
from = utils.BeginOfToday(tz)
|
|
case models.IntervalYesterday:
|
|
from = utils.BeginOfToday(tz).Add(-24 * time.Hour)
|
|
to = utils.BeginOfToday(tz)
|
|
case models.IntervalPastDay:
|
|
from = now.Add(-24 * time.Hour)
|
|
case models.IntervalThisWeek:
|
|
from = utils.BeginOfThisWeek(tz)
|
|
case models.IntervalLastWeek:
|
|
from = utils.BeginOfThisWeek(tz).AddDate(0, 0, -7)
|
|
to = utils.BeginOfThisWeek(tz)
|
|
case models.IntervalThisMonth:
|
|
from = utils.BeginOfThisMonth(tz)
|
|
case models.IntervalLastMonth:
|
|
from = utils.BeginOfThisMonth(tz).AddDate(0, -1, 0)
|
|
to = utils.BeginOfThisMonth(tz)
|
|
case models.IntervalThisYear:
|
|
from = utils.BeginOfThisYear(tz)
|
|
case models.IntervalPast7Days:
|
|
from = now.AddDate(0, 0, -7)
|
|
case models.IntervalPast7DaysYesterday:
|
|
from = utils.BeginOfToday(tz).AddDate(0, 0, -1).AddDate(0, 0, -7)
|
|
to = utils.BeginOfToday(tz).AddDate(0, 0, -1)
|
|
case models.IntervalPast14Days:
|
|
from = now.AddDate(0, 0, -14)
|
|
case models.IntervalPast30Days:
|
|
from = now.AddDate(0, 0, -30)
|
|
case models.IntervalPast6Months:
|
|
from = now.AddDate(0, -6, 0)
|
|
case models.IntervalPast12Months:
|
|
from = now.AddDate(0, -12, 0)
|
|
case models.IntervalAny:
|
|
from = time.Time{}
|
|
default:
|
|
err = errors.New("invalid interval")
|
|
}
|
|
|
|
return err, from, to
|
|
}
|
|
|
|
// ResolveClosestRange returns the interval label (e.g. "last_7_days") of the maximum allowed range when having opted to share this many days or an error for days == 0.
|
|
func ResolveMaximumRange(days int) (error, *models.IntervalKey) {
|
|
if days == 0 {
|
|
return errors.New("no matching interval"), nil
|
|
}
|
|
if days < 0 {
|
|
return nil, models.IntervalAny
|
|
}
|
|
if days < 7 {
|
|
return nil, models.IntervalPastDay
|
|
}
|
|
if days < 14 {
|
|
return nil, models.IntervalPast7Days
|
|
}
|
|
if days < 30 {
|
|
return nil, models.IntervalPast14Days
|
|
}
|
|
if days < 181 { // 3*31 + 2*30 + 1*28
|
|
return nil, models.IntervalPast30Days
|
|
}
|
|
if days < 365 { // 7*31 + 4*30 + 1*28
|
|
return nil, models.IntervalPast6Months
|
|
}
|
|
return nil, models.IntervalPast12Months
|
|
}
|