feat(task): disable notification support

indes · Feb 2, 2019 · b19eb18 · b19eb18
1 parent f4e498d
commit b19eb18
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/bot/service.go b/bot/service.go
@@ -44,6 +44,12 @@ func BroadNews(source *model.Source, subs []model.Subscribe, contents []model.Co
 	var u telebot.User
 	for _, content := range contents {
 		for _, sub := range subs {
+			var disableNotification bool
+			if sub.EnableNotification == 1 {
+				disableNotification = false
+			} else {
+				disableNotification = true
+			}
 			u.ID = int(sub.UserID)
 			message := `
 *%s*
@@ -54,7 +60,8 @@ func BroadNews(source *model.Source, subs []model.Subscribe, contents []model.Co
 			message = fmt.Sprintf(message, source.Title, content.Title, content.RawLink, content.TelegraphUrl)
 
 			_, err := B.Send(&u, message, &telebot.SendOptions{
-				ParseMode: telebot.ModeMarkdown,
+				ParseMode:           telebot.ModeMarkdown,
+				DisableNotification: disableNotification,
 			})
 			if err != nil {
 				log.Println(err)

diff --git a/model/subscribe.go b/model/subscribe.go
@@ -20,6 +20,8 @@ func RegistFeed(userID int64, sourceID uint) error {
 		if err.Error() == "record not found" {
 			subscribe.UserID = userID
 			subscribe.SourceID = sourceID
+			subscribe.EnableNotification = 1
+			subscribe.EnableTelegraph = 1
 			err := db.Create(&subscribe).Error
 			if err == nil {
 				return nil