feat: support push notification clients

server/notify_client.go

@@ -0,0 +1,75 @@
+package server
+
+import (
+	"polaris/ent"
+	"polaris/log"
+	"polaris/pkg/notifier"
+	"strconv"
+
+	"github.com/gin-gonic/gin"
+	"github.com/pkg/errors"
+)
+
+func (s *Server) GetAllNotificationClients(c *gin.Context) (interface{}, error) {
+	return s.db.GetAllNotificationClients()
+}
+
+func (s *Server) GetNotificationClient(c *gin.Context) (interface{}, error) {
+	ids := c.Param("id")
+	id, err := strconv.Atoi(ids)
+	if err != nil {
+		return nil, errors.Wrap(err, "convert")
+	}
+	return s.db.GetNotificationClient(id)
+}
+
+func (s *Server) DeleteNotificationClient(c *gin.Context) (interface{}, error) {
+	ids := c.Param("id")
+	id, err := strconv.Atoi(ids)
+	if err != nil {
+		return nil, errors.Wrap(err, "convert")
+	}
+	return nil, s.db.DeleteNotificationClient(id)
+}
+
+func (s *Server) AddNotificationClient(c *gin.Context) (interface{}, error) {
+	var in ent.NotificationClient
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, errors.Wrap(err, "json")
+	}
+
+	err := s.db.AddNotificationClient(in.Name, in.Service, in.Settings, in.Enabled)
+	if err != nil {
+		return nil, errors.Wrap(err, "save db")
+	}
+	return nil, nil
+}
+
+func (s *Server) sendMsg(msg string)  {
+	clients, err := s.db.GetAllNotificationClients2()
+	if err != nil {
+		log.Errorf("query notification clients: %v", err)
+		return 
+	}
+	for _, cl := range clients {
+		if !cl.Enabled {
+			continue
+		}
+		handler, ok := notifier.Gethandler(cl.Service)
+		if !ok {
+			log.Errorf("no notification implementation of service %s", cl.Service)
+			continue
+		}
+		noCl, err := handler(cl.Settings)
+		if err != nil {
+			log.Errorf("handle setting for name %s error: %v", cl.Name, err)
+			continue
+		}
+		err = noCl.SendMsg(msg)
+		if err != nil {
+			log.Errorf("send message error: %v", err)
+			continue
+		}
+		log.Debugf("send message to %s success, msg is %s", cl.Name, msg)
+	}
+}

server/resources.go

@@ -7,6 +7,7 @@ import (
 	"polaris/ent/history"
 	"polaris/ent/media"
 	"polaris/log"
+	"polaris/pkg/notifier/message"
 	"polaris/pkg/torznab"
 	"polaris/pkg/utils"
 	"polaris/server/core"
@@ -67,6 +68,8 @@ func (s *Server) downloadSeasonPackage(r1 torznab.Result, seriesId, seasonNum in
 	s.db.SetSeasonAllEpisodeStatus(seriesId, seasonNum, episode.StatusDownloading)
 
 	s.tasks[history.ID] = &Task{Torrent: torrent}
+
+	s.sendMsg(fmt.Sprintf(message.BeginDownload, r1.Name))
 	return &r1.Name, nil
 
 }
@@ -112,6 +115,7 @@ func (s *Server) downloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,
 	s.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
 
 	s.tasks[history.ID] = &Task{Torrent: torrent}
+	s.sendMsg(fmt.Sprintf(message.BeginDownload, r1.Name))
 
 	log.Infof("success add %s to download task", r1.Name)
 	return &r1.Name, nil
@@ -293,7 +297,8 @@ func (s *Server) DownloadTorrent(c *gin.Context) (interface{}, error) {
 
 		s.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
 	}()
-
+	
+	s.sendMsg(fmt.Sprintf(message.BeginDownload, in.Name))
 	log.Infof("success add %s to download task", in.Name)
 	return in.Name, nil
 

server/scheduler.go

@@ -1,6 +1,7 @@
 package server
 
 import (
+	"fmt"
 	"path/filepath"
 	"polaris/ent"
 	"polaris/ent/episode"
@@ -9,6 +10,7 @@ import (
 	storage1 "polaris/ent/storage"
 	"polaris/log"
 	"polaris/pkg"
+	"polaris/pkg/notifier/message"
 	"polaris/pkg/storage"
 	"polaris/pkg/utils"
 	"polaris/server/core"
@@ -44,6 +46,7 @@ func (s *Server) checkTasks() {
 		log.Infof("task (%s) percentage done: %d%%", t.Name(), t.Progress())
 		if t.Progress() == 100 {
 			log.Infof("task is done: %v", t.Name())
+			s.sendMsg(fmt.Sprintf(message.DownloadComplete, t.Name()))
 			go func() {
 				if err := s.moveCompletedTask(id); err != nil {
 					log.Infof("post tasks for id %v fail: %v", id, err)
@@ -76,7 +79,7 @@ func (s *Server) moveCompletedTask(id int) (err1 error) {
 			} else {
 				s.db.SetSeasonAllEpisodeStatus(r.MediaID, seasonNum, episode.StatusMissing)
 			}
-
+			s.sendMsg(fmt.Sprintf(message.ProcessingFailed, err))
 		} else {
 			delete(s.tasks, r.ID)
 			s.db.SetHistoryStatus(r.ID, history.StatusSuccess)
@@ -85,6 +88,7 @@ func (s *Server) moveCompletedTask(id int) (err1 error) {
 			} else {
 				s.db.SetSeasonAllEpisodeStatus(r.MediaID, seasonNum, episode.StatusDownloaded)
 			}
+			s.sendMsg(fmt.Sprintf(message.ProcessingComplete, torrent.Name()))
 
 			torrent.Remove()
 		}
@@ -229,9 +233,9 @@ func (s *Server) downloadTvSeries() {
 				}
 				if series.CreatedAt.Sub(t) > 24*time.Hour { //剧集在加入watchlist之前，不去下载
 					continue
-				}	
+				}
 			}
-			
+
 			if ep.Status != episode.StatusMissing { //已经下载的不去下载
 				continue
 			}
@@ -252,7 +256,7 @@ func (s *Server) downloadMovie() {
 	allSeries := s.db.GetMediaWatchlist(media.MediaTypeMovie)
 
 	for _, series := range allSeries {
-		detail := s.db.GetMediaDetails(series.ID)		
+		detail := s.db.GetMediaDetails(series.ID)
 		if len(detail.Episodes) == 0 {
 			log.Errorf("no related dummy episode: %v", detail.NameEn)
 			continue
@@ -317,7 +321,7 @@ func (s *Server) checkAllSeriesNewSeason() {
 	}
 }
 
-func (s *Server) checkSeiesNewSeason(media *ent.Media) error{
+func (s *Server) checkSeiesNewSeason(media *ent.Media) error {
 	d, err := s.MustTMDB().GetTvDetails(media.TmdbID, s.language)
 	if err != nil {
 		return errors.Wrap(err, "tmdb")
@@ -327,29 +331,29 @@ func (s *Server) checkSeiesNewSeason(media *ent.Media) error{
 	if err != nil {
 		return errors.Wrap(err, "tmdb season")
 	}
-	
+
 	for _, ep := range seasonDetail.Episodes {
 		epDb, err := s.db.GetEpisode(media.ID, ep.SeasonNumber, ep.EpisodeNumber)
 		if err != nil {
 			if ent.IsNotFound(err) {
 				log.Infof("add new episode: %+v", ep)
 				episode := &ent.Episode{
-					MediaID: media.ID,
-					SeasonNumber: ep.SeasonNumber,
+					MediaID:       media.ID,
+					SeasonNumber:  ep.SeasonNumber,
 					EpisodeNumber: ep.EpisodeNumber,
-					Title: ep.Name,
-					Overview: ep.Overview,
-					AirDate: ep.AirDate,
-					Status: episode.StatusMissing,
+					Title:         ep.Name,
+					Overview:      ep.Overview,
+					AirDate:       ep.AirDate,
+					Status:        episode.StatusMissing,
 				}
 				s.db.SaveEposideDetail2(episode)
 			}
-		} else {//update episode
+		} else { //update episode
 			if ep.Name != epDb.Title || ep.Overview != epDb.Overview || ep.AirDate != epDb.AirDate {
 				log.Infof("update new episode: %+v", ep)
-				s.db.UpdateEpiode2(epDb.ID, ep.Name, ep.Overview, ep.AirDate)	
+				s.db.UpdateEpiode2(epDb.ID, ep.Name, ep.Overview, ep.AirDate)
 			}
 		}
 	}
 	return nil
-}
+}

server/server.go

@@ -115,6 +115,13 @@ func (s *Server) Serve() error {
 		storage.POST("/", HttpHandler(s.AddStorage))
 		storage.DELETE("/:id", HttpHandler(s.DeleteStorage))
 	}
+	notifier := api.Group("/notifier")
+	{
+		notifier.GET("/all", HttpHandler(s.GetAllNotificationClients))
+		notifier.GET("/id/:id", HttpHandler(s.GetNotificationClient))
+		notifier.DELETE("/id/:id", HttpHandler(s.DeleteNotificationClient))
+		notifier.POST("/add", HttpHandler(s.AddNotificationClient))
+	}
 
 	s.language = s.db.GetLanguage()
 	return s.r.Run(":8080")