goholiday.go

package goholiday

import "time"

const dateFormat = "2006-01-02"

type Schedule interface {
	GetNationalHolidays() map[string]string
	GetWeekdayHolidays() map[time.Weekday]struct{}
}

type Goholiday struct {
	schedule       Schedule
	uniqueHolidays map[string]struct{}
}

func New(schedule Schedule) *Goholiday {
	return &Goholiday{
		schedule:       schedule,
		uniqueHolidays: map[string]struct{}{},
	}
}

func (g *Goholiday) IsNationalHoliday(t time.Time) bool {
	_, exist := g.schedule.GetNationalHolidays()[t.Format(dateFormat)]
	return exist
}

func (g *Goholiday) IsHoliday(t time.Time) bool {
	return g.isWeekdayHoliday(t) || g.IsNationalHoliday(t) || g.isUniqueHoliday(t)
}

func (g *Goholiday) isWeekdayHoliday(t time.Time) bool {
	_, exist := g.schedule.GetWeekdayHolidays()[t.Weekday()]
	return exist
}

func (g *Goholiday) SetUniqueHolidays(ts []time.Time) {
	for _, t := range ts {
		g.uniqueHolidays[t.Format(dateFormat)] = struct{}{}
	}
}

func (g *Goholiday) isUniqueHoliday(t time.Time) bool {
	_, exist := g.uniqueHolidays[t.Format(dateFormat)]
	return exist
}

func (g *Goholiday) IsBusinessDay(t time.Time) bool {
	return !g.IsHoliday(t)
}

func (g *Goholiday) BusinessDaysBefore(t time.Time, bds int) time.Time {
	return g.travelBusinessDays(t, bds, -1)
}

func (g *Goholiday) BusinessDaysAfter(t time.Time, bds int) time.Time {
	return g.travelBusinessDays(t, bds, 1)
}

func (g *Goholiday) travelBusinessDays(t time.Time, bds int, course int) time.Time {
	duration := time.Hour * 24 * time.Duration(course)
	for tbds := 0; tbds != bds; {
		if t = t.Add(duration); !g.IsHoliday(t) {
			tbds++
		}
	}
	return t
}