feat: add notifications (#216)

* feat: initial notifications support

* chore: update deps
This commit is contained in:
Ludvig Lundgren 2022-04-04 19:13:09 +02:00 committed by GitHub
parent 3185832708
commit 431742fd94
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
20 changed files with 1632 additions and 36 deletions

View file

@ -4,6 +4,7 @@ import (
"context"
"github.com/autobrr/autobrr/internal/domain"
"github.com/autobrr/autobrr/internal/notification"
"github.com/autobrr/autobrr/internal/release"
"github.com/asaskevich/EventBus"
@ -11,12 +12,17 @@ import (
)
type Subscriber struct {
eventbus EventBus.Bus
releaseSvc release.Service
eventbus EventBus.Bus
notificationSvc notification.Service
releaseSvc release.Service
}
func NewSubscribers(eventbus EventBus.Bus, releaseSvc release.Service) Subscriber {
s := Subscriber{eventbus: eventbus, releaseSvc: releaseSvc}
func NewSubscribers(eventbus EventBus.Bus, notificationSvc notification.Service, releaseSvc release.Service) Subscriber {
s := Subscriber{
eventbus: eventbus,
notificationSvc: notificationSvc,
releaseSvc: releaseSvc,
}
s.Register()
@ -25,8 +31,8 @@ func NewSubscribers(eventbus EventBus.Bus, releaseSvc release.Service) Subscribe
func (s Subscriber) Register() {
s.eventbus.Subscribe("release:store-action-status", s.releaseActionStatus)
s.eventbus.Subscribe("release:push-rejected", s.releasePushRejected)
s.eventbus.Subscribe("release:push-approved", s.releasePushApproved)
s.eventbus.Subscribe("release:push", s.releasePushStatus)
s.eventbus.Subscribe("events:release:push", s.releasePush)
}
func (s Subscriber) releaseActionStatus(actionStatus *domain.ReleaseActionStatus) {
@ -38,20 +44,18 @@ func (s Subscriber) releaseActionStatus(actionStatus *domain.ReleaseActionStatus
}
}
func (s Subscriber) releasePushRejected(actionStatus *domain.ReleaseActionStatus) {
log.Trace().Msgf("events: 'release:push-rejected' '%+v'", actionStatus)
func (s Subscriber) releasePushStatus(actionStatus *domain.ReleaseActionStatus) {
log.Trace().Msgf("events: 'release:push' '%+v'", actionStatus)
err := s.releaseSvc.StoreReleaseActionStatus(context.Background(), actionStatus)
if err != nil {
log.Error().Err(err).Msgf("events: 'release:push-rejected' error")
if err := s.releaseSvc.StoreReleaseActionStatus(context.Background(), actionStatus); err != nil {
log.Error().Err(err).Msgf("events: 'release:push' error")
}
}
func (s Subscriber) releasePushApproved(actionStatus *domain.ReleaseActionStatus) {
log.Trace().Msgf("events: 'release:push-approved' '%+v'", actionStatus)
func (s Subscriber) releasePush(event *domain.EventsReleasePushed) {
log.Trace().Msgf("events: 'events:release:push' '%+v'", event)
err := s.releaseSvc.StoreReleaseActionStatus(context.Background(), actionStatus)
if err != nil {
log.Error().Err(err).Msgf("events: 'release:push-approved' error")
if err := s.notificationSvc.SendEvent(*event); err != nil {
log.Error().Err(err).Msgf("events: 'events:release:push' error sending notification")
}
}