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

package openTime

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

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

type (
	weeklyParser struct{}

	weekly struct {
		times []*durationRange
	}
)

// 1,3,5/12:30-13:30|2,4,6/12:00-13:00
func (p *weeklyParser) Parse(params string) (OpenTime, error) {
	times := []*durationRange{}
	for _, v := range strings.Split(params, "|") {
		wt := strings.Split(v, "/")
		if len(wt) < 2 {
			return nil, fmt.Errorf("nonsupport params: %s", params)
		}

		ts := strings.Split(wt[1], "-")
		if len(ts) < 2 {
			return nil, fmt.Errorf("nonsupport params: %s", params)
		}

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

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

		for _, wd := range strings.Split(wt[0], ",") {
			w, ok := cherryString.ToInt64(wd)
			if !ok {
				return nil, fmt.Errorf("nonsupport params: %s", params)
			}

			newStartTime := time.Duration((w-1)*ctime.SecondsPerDay)*time.Second + startTime
			newEndTime := time.Duration((w-1)*ctime.SecondsPerDay)*time.Second + endTime
			times = append(times, &durationRange{
				startTime: newStartTime,
				endTime:   newEndTime,
			})
		}
	}

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

	return &weekly{times: times}, nil
}

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

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

	// 下周
	nextTime := ctime.NewTime(weekStart.Add(7*24*time.Hour), true)
	return p.TimeRange(&nextTime)
}