109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package server
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.ctrlz.es/mgdelacroix/birthdaybot/model"
|
|
"git.ctrlz.es/mgdelacroix/birthdaybot/notification"
|
|
"git.ctrlz.es/mgdelacroix/birthdaybot/parser"
|
|
"github.com/charmbracelet/log"
|
|
)
|
|
|
|
var ErrNoNotificationServices = errors.New("there are no notification services configured")
|
|
|
|
type Server struct {
|
|
Logger *log.Logger
|
|
config *model.Config
|
|
workers []*Worker
|
|
birthdays []*model.Birthday
|
|
notificationServices []notification.NotificationService
|
|
}
|
|
|
|
func createNotificationServices(logger *log.Logger, config *model.Config) ([]notification.NotificationService, error) {
|
|
notificationServices := []notification.NotificationService{}
|
|
|
|
if config.TelegramNotifications != nil {
|
|
telegramNotificationService, err := notification.NewTelegramNotificationService(logger, config.TelegramNotifications)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot create telegram notification service: %w", err)
|
|
}
|
|
notificationServices = append(notificationServices, telegramNotificationService)
|
|
}
|
|
|
|
if len(notificationServices) == 0 {
|
|
return nil, ErrNoNotificationServices
|
|
}
|
|
|
|
return notificationServices, nil
|
|
}
|
|
|
|
func New(options ...Option) (*Server, error) {
|
|
srv := &Server{}
|
|
for _, option := range options {
|
|
srv = option(srv)
|
|
}
|
|
|
|
srv.Logger.Debug("parsing CSV file", "birthdayFile", srv.config.BirthdayFile)
|
|
|
|
birthdays, err := parser.ParseCSV(srv.config.BirthdayFile)
|
|
if err != nil {
|
|
srv.Logger.Error("cannot parse CSV file", "birthdayFile", srv.config.BirthdayFile, "error", err)
|
|
return nil, fmt.Errorf("cannot parse CSV file %s: %w", srv.config.BirthdayFile, err)
|
|
}
|
|
|
|
srv.Logger.Debug("creating notification services from config")
|
|
|
|
notificationServices, err := createNotificationServices(srv.Logger, srv.config)
|
|
if err != nil {
|
|
srv.Logger.Error("error creating notification services", "error", err)
|
|
return nil, err
|
|
}
|
|
|
|
srv.Logger.Info("creating server")
|
|
|
|
server := &Server{
|
|
Logger: srv.Logger,
|
|
config: srv.config,
|
|
birthdays: birthdays,
|
|
notificationServices: notificationServices,
|
|
}
|
|
server.workers = []*Worker{NewWorker(server)}
|
|
|
|
return server, nil
|
|
}
|
|
|
|
func (s *Server) Start() {
|
|
s.Logger.Info("starting server")
|
|
for _, worker := range s.workers {
|
|
worker.Start()
|
|
}
|
|
s.Logger.Info("server started", "workers", len(s.workers))
|
|
}
|
|
|
|
func (s *Server) Stop() {
|
|
s.Logger.Info("stopping server")
|
|
for _, worker := range s.workers {
|
|
worker.Stop()
|
|
}
|
|
s.Logger.Info("server stopped", "workers", len(s.workers))
|
|
}
|
|
|
|
func (s *Server) Notify(birthday *model.Birthday) error {
|
|
errs := []error{}
|
|
for _, service := range s.notificationServices {
|
|
err := service.Notify(birthday)
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
}
|
|
|
|
if len(errs) == 0 {
|
|
return nil
|
|
}
|
|
return errors.Join(errs...)
|
|
}
|
|
|
|
func (s *Server) Birthdays() []*model.Birthday {
|
|
return s.birthdays
|
|
}
|