s_cherry_e/sr/internal/extend/open_time/monthly_continual.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 (
	monthlyContinualParser struct{}

	monthlyContinual struct {
		times []*durationRange
	}
)

// 1/12:00-10/24:00|15/00:00-30/24:00
func (p *monthlyContinualParser) 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)
		}

		startTime, err := p.toDateTime(wt[0])
		if err != nil {
			return nil, err
		}

		endTime, err := p.toDateTime(wt[1])
		if err != nil {
			return nil, err
		}

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

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

	return &monthlyContinual{times: times}, nil
}

func (p *monthlyContinualParser) toDateTime(str string) (time.Duration, error) {
	s := strings.Split(str, "/")
	if len(s) < 2 {
		return 0, fmt.Errorf("nonsupport params: %s", str)
	}

	d, ok := cherryString.ToInt64(s[0])
	if !ok {
		return 0, fmt.Errorf("nonsupport params: %s", str)
	}

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

	duration := time.Duration((d-1)*ctime.SecondsPerDay)*time.Second + t
	return duration, nil
}

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

func (p *monthlyContinual) TimeRange(t *ctime.CherryTime) (ctime.CherryTime, ctime.CherryTime) {
	monthStart := t.StartOfMonth()
	monthEnd := t.EndOfMonth()
	duration := t.Sub(monthStart.Time).Seconds()
	for _, v := range p.times {
		if duration <= v.startTime.Seconds() || duration <= v.endTime.Seconds() {
			st := monthStart.Add(v.startTime)
			if st.After(monthEnd.Time) { // 开始时间比月结日大，下个月
				break
			}

			et := monthStart.Add(v.endTime)
			if et.After(monthEnd.Time) {
				et = monthEnd.Time
			}

			startTime := ctime.NewTime(st, true)
			endTime := ctime.NewTime(et, true)
			return startTime, endTime
		}
	}

	// 下个月
	t.AddMonth()
	nextTime := ctime.NewTime(t.StartOfMonth().Time, true)
	return p.TimeRange(&nextTime)
}