xevion/go-ha
1
0
Fork 0
mirror of https://github.com/Xevion/go-ha.git synced 2025-12-06 01:15:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

add OnlyOnDates to schedule, change exceptionday to exceptiondates everywhere

Browse Source
This commit is contained in:
Sam Lewis
2022-11-26 23:36:19 -05:00
parent 555bafa2e5
commit 60349852e6
6 changed files with 70 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
schedule.go
View File

@@ -23,8 +23,8 @@ type DailySchedule struct {
isSunset bool
sunOffset DurationString
exceptionDays []time.Time
exceptionRanges []timeRange
exceptionDates []time.Time
allowlistDates []time.Time
}
func (s DailySchedule) Hash() string {
@@ -93,13 +93,13 @@ func (sb scheduleBuilderCall) Sunset(offset ...DurationString) scheduleBuilderEn
return scheduleBuilderEnd(sb)
}
func (sb scheduleBuilderEnd) ExceptionDay(t time.Time) scheduleBuilderEnd {
sb.schedule.exceptionDays = append(sb.schedule.exceptionDays, t)
func (sb scheduleBuilderEnd) ExceptionDates(t time.Time, tl ...time.Time) scheduleBuilderEnd {
sb.schedule.exceptionDates = append(tl, t)
return sb
}
func (sb scheduleBuilderEnd) ExceptionRange(start, end time.Time) scheduleBuilderEnd {
sb.schedule.exceptionRanges = append(sb.schedule.exceptionRanges, timeRange{start, end})
func (sb scheduleBuilderEnd) OnlyOnDates(t time.Time, tl ...time.Time) scheduleBuilderEnd {
sb.schedule.allowlistDates = append(tl, t)
return sb
}
@@ -132,10 +132,10 @@ func runSchedules(a *App) {
}
func (s DailySchedule) maybeRunCallback(a *App) {
if c := checkExceptionDays(s.exceptionDays); c.fail {
if c := checkExceptionDates(s.exceptionDates); c.fail {
return
}
if c := checkExceptionRanges(s.exceptionRanges); c.fail {
if c := checkAllowlistDates(s.allowlistDates); c.fail {
return
}
go s.callback(a.service, a.state)