feat: email reports (resolve #124)

services/report.go

@@ -0,0 +1,96 @@
+package services
+
+import (
+	"github.com/emvi/logbuch"
+	"github.com/go-co-op/gocron"
+	"github.com/muety/wakapi/config"
+	"github.com/muety/wakapi/models"
+	"sync"
+	"time"
+)
+
+var reportLock = sync.Mutex{}
+
+type ReportService struct {
+	config           *config.Config
+	summaryService   ISummaryService
+	userService      IUserService
+	mailService      IMailService
+	schedulersWeekly map[string]*gocron.Scheduler // user id -> scheduler
+}
+
+func NewReportService(summaryService ISummaryService, userService IUserService, mailService IMailService) *ReportService {
+	return &ReportService{
+		config:           config.Get(),
+		summaryService:   summaryService,
+		userService:      userService,
+		mailService:      mailService,
+		schedulersWeekly: map[string]*gocron.Scheduler{},
+	}
+}
+
+func (srv *ReportService) Schedule() {
+	logbuch.Info("initializing report service")
+
+	users, err := srv.userService.GetAllByReports(true)
+	if err != nil {
+		config.Log().Fatal("%v", err)
+	}
+
+	logbuch.Info("scheduling reports for %d users", len(users))
+	for _, u := range users {
+		srv.UpdateUserSchedule(u)
+	}
+}
+
+func (srv *ReportService) UpdateUserSchedule(u *models.User) {
+	reportLock.Lock()
+	defer reportLock.Unlock()
+
+	// unschedule
+	if s, ok := srv.schedulersWeekly[u.ID]; ok && !u.ReportsWeekly {
+		s.Stop()
+		s.Clear()
+		delete(srv.schedulersWeekly, u.ID)
+		return
+	}
+
+	// schedule
+	if _, ok := srv.schedulersWeekly[u.ID]; !ok && u.ReportsWeekly {
+		s := gocron.NewScheduler(u.TZ())
+		s.
+			Every(1).
+			Week().
+			Weekday(srv.config.App.GetWeeklyReportDay()).
+			At(srv.config.App.GetWeeklyReportTime()).
+			Do(srv.Run, u, 7*24*time.Hour)
+		s.StartAsync()
+		srv.schedulersWeekly[u.ID] = s
+	}
+}
+
+func (srv *ReportService) Run(user *models.User, duration time.Duration) error {
+	end := time.Now().In(user.TZ())
+	start := time.Now().Add(-1 * duration)
+
+	summary, err := srv.summaryService.Aliased(start, end, user, srv.summaryService.Retrieve, false)
+	if err != nil {
+		config.Log().Error("failed to generate report for '%s' – %v", user.ID, err)
+		return err
+	}
+
+	report := &models.Report{
+		From:    start,
+		To:      end,
+		User:    user,
+		Summary: summary,
+	}
+
+	if err := srv.mailService.SendReport(user, report); err != nil {
+		config.Log().Error("failed to send report for '%s' – %v", user.ID, err)
+		return err
+	}
+
+	logbuch.Info("sent report to user '%s'", user.ID)
+	return nil
+}