domogeek/pkg/calendar/calendar.go

package calendar

import (
	"fmt"
	"github.com/avast/retry-go"
	"github.com/dolanor/caldav-go/caldav"
	"github.com/dolanor/caldav-go/caldav/entities"
	"github.com/dolanor/caldav-go/icalendar/components"
	"go.uber.org/zap"
	"math"
	"net/http"
	"strings"
	"time"
)

type Caldav interface {
	QueryEvents(path string, query *entities.CalendarQuery) (events []*components.Event, oerr error)
}

type Calendar struct {
	Location             *time.Location
	cdav                 Caldav
	caldavPath           string
	caldavSummaryPattern string
}

func NewCaldav(caldavUrl, caldavPath string) (Caldav, error) {
	// create a reference to your CalDAV-compliant server
	server, _ := caldav.NewServer(caldavUrl)
	// create a CalDAV client to speak to the server
	var client = caldav.NewClient(server, http.DefaultClient)
	err := retry.Do(
		func() error {
			// start executing requests!
			err := client.ValidateServer(caldavPath)
			if err != nil {
				return fmt.Errorf("bad caldav configuration, unable to validate connexion: %w", err)
			}
			return nil
		},
		retry.OnRetry(
			func(n uint, err error) {
				zap.S().Errorf("unable to validate caldav connection on retry %d: %v", n, err)
			},
		),
		retry.Attempts(0), // Infinite attempts
		retry.DelayType(retry.BackOffDelay),
	)
	if err != nil {
		return nil, fmt.Errorf("unable to validate caldav connection: %w", err)
	}
	return client, nil
}

type Option func(calendar *Calendar)

func WithCaldav(cdav Caldav) Option {
	return func(calendar *Calendar) {
		calendar.cdav = cdav
	}
}

func WithCaldavSummaryPattern(caldavSummaryPattern string) Option {
	return func(calendar *Calendar) {
		calendar.caldavSummaryPattern = caldavSummaryPattern
	}
}

func WithCaldavPath(caldavPath string) Option {
	return func(calendar *Calendar) {
		calendar.caldavPath = caldavPath
	}
}

func New(location *time.Location, opts ...Option) *Calendar {
	c := &Calendar{
		location,
		nil,
		"",
		"",
	}

	for _, opt := range opts {
		opt(c)
	}
	return c
}

func (cal *Calendar) GetEasterDay(year int) time.Time {
	g := float64(year % 19.0)
	c := math.Floor(float64(year) / 100.0)
	c4 := math.Floor(c / 4.0)
	h := float64(int(19.0*g+c-c4-math.Floor((8.0*c+13)/25)+15) % 30.0)
	k := math.Floor(h / 28.0)
	i := (k*math.Floor(29./(h+1.))*math.Floor((21.-g)/11.)-1.)*k + h

	// jour de Pâques (0=dimanche, 1=lundi....)
	dayWeek := int(math.Floor(float64(year)/4.)+float64(year)+i+2+c4-c) % 7

	// Jour de Pâques en jours enpartant de 1 = 1er mars
	presJour := int(28 + int(i) - dayWeek)

	// mois (0 = janvier, ... 2 = mars, 3 = avril)
	month := 2
	if presJour > 31 {
		month = 3
	}

	// Mois dans l'année
	month += 1

	// jour du mois
	day := presJour - 31
	if month == 2 {
		day = presJour
	}

	return time.Date(year, 3, 31, 0, 0, 0, 0, cal.Location).AddDate(0, 0, day)
}

func (cal *Calendar) GetHolidays(year int) *[]time.Time {

	// Calcul du jour de pâques
	paques := cal.GetEasterDay(year)

	joursFeries := []time.Time{
		// Jour de l'an
		time.Date(year, time.January, 1, 0, 0, 0, 0, cal.Location),
		// Easter
		paques.AddDate(0, 0, 1),
		// 1 mai
		time.Date(year, time.May, 1, 0, 0, 0, 0, cal.Location),
		// 8 mai
		time.Date(year, time.May, 8, 0, 0, 0, 0, cal.Location),
		// Ascension
		paques.AddDate(0, 0, 39),
		// 14 juillet
		time.Date(year, time.July, 14, 0, 0, 0, 0, cal.Location),
		// 15 aout
		time.Date(year, time.August, 15, 0, 0, 0, 0, cal.Location),
		// Toussaint
		time.Date(year, time.November, 1, 0, 0, 0, 0, cal.Location),
		// 11 novembre
		time.Date(year, time.November, 11, 0, 0, 0, 0, cal.Location),
		// noël
		time.Date(year, time.December, 25, 0, 0, 0, 0, cal.Location),
	}

	return &joursFeries
}

func (cal *Calendar) GetHolidaysSet(year int) map[time.Time]bool {
	holidays := cal.GetHolidays(year)
	result := make(map[time.Time]bool, len(*holidays))
	for _, h := range *holidays {
		result[h] = true
	}
	return result
}

func (cal *Calendar) IsHoliday(date time.Time) bool {
	h := cal.GetHolidaysSet(date.Year())
	d := date.In(cal.Location)
	day := time.Date(d.Year(), d.Month(), d.Day(), 0, 0, 0, 0, cal.Location)
	caldavHolidays, err := cal.IsHolidaysFromCaldav(day)
	if err != nil {
		zap.S().Errorf("unable to check holidays from caldav: %v", err)
	}
	return h[day] || caldavHolidays
}

func (cal *Calendar) IsWorkingDay(date time.Time) bool {
	return !cal.IsHoliday(date) && date.Weekday() >= time.Monday && date.Weekday() <= time.Friday
}

func (cal *Calendar) IsWorkingDayToday() bool {
	return cal.IsWorkingDay(time.Now())
}

func (cal *Calendar) IsWeekDay(day time.Time) bool {
	return day.Weekday() >= time.Monday && day.Weekday() <= time.Friday
}

func (cal *Calendar) IsHolidaysFromCaldav(day time.Time) (bool, error) {
	if cal.cdav == nil {
		return false, nil
	}
	query, err := entities.NewEventRangeQuery(day.UTC(), day.UTC().Add(23*time.Hour+59*time.Minute))
	if err != nil {
		return false, fmt.Errorf("unable to build events range query: %v", err)
	}
	events, err := cal.cdav.QueryEvents(cal.caldavPath, query)
	if err != nil {
		return false, fmt.Errorf("unable list events from caldav: %v", err)
	}

	for _, evt := range events {
		if strings.Contains(evt.Summary, cal.caldavSummaryPattern) {
			return true, nil
		}
	}
	return false, nil
}