s_cherry_e/sr/internal/extend/open_time/daily.go

package openTime

import (
	"fmt"
	"sort"
	"strings"
	"time"

	ctime "github.com/cherry-game/cherry/extend/time"
)

type (
	dailyParser struct{}

	daily struct {
		times []*durationRange
	}
)

func (p *dailyParser) Parse(params string) (OpenTime, error) {
	times := []*durationRange{}
	for _, v := range strings.Split(params, ";") {
		s := strings.Split(string(v), "-")
		if len(s) < 2 {
			return nil, fmt.Errorf("nonsupport params: %s", params)
		}

		startTime, err := parseDuration(s[0])
		if err != nil {
			return nil, err
		}

		endTime, err := parseDuration(s[1])
		if err != nil {
			return nil, err
		}

		if startTime > endTime {
			startTime, endTime = endTime, startTime
		}

		times = append(times, &durationRange{
			startTime: startTime,
			endTime:   endTime,
		})
	}

	sort.Slice(times, func(i, j int) bool { return times[i].startTime < times[j].startTime })

	return &daily{times: times}, nil
}

func (p *daily) Between(t *ctime.CherryTime) bool {
	duration := t.Sub(t.StartOfDay().Time).Seconds()
	for _, v := range p.times {
		if v.startTime.Seconds() <= duration && duration <= v.endTime.Seconds() {
			return true
		}
	}
	return false
}

func (p *daily) TimeRange(t *ctime.CherryTime) (ctime.CherryTime, ctime.CherryTime) {
	startOfDay := t.StartOfDay()
	duration := t.Sub(startOfDay.Time).Seconds()
	for _, v := range p.times {
		if duration <= v.startTime.Seconds() || duration <= v.endTime.Seconds() {
			startTime := ctime.NewTime(startOfDay.Add(v.startTime), true)
			endTime := ctime.NewTime(startOfDay.Add(v.endTime), true)
			return startTime, endTime
		}
	}

	// 下一天
	nextTime := ctime.NewTime(startOfDay.Add(24*time.Hour), true)
	return p.TimeRange(&nextTime)
}