Enable replies on Twitter and Telegram

By default, when a product is available, a notification is sent. When that same
product is not available, a reply is sent to the original message. With tons of
notifications, replies might be seen as flooding. This commit adds an option to
explicitly enable replies on Twitter and Telegram notifiers. By default, reply
messages are disabled.

Signed-off-by: Julien Riou <julien@riou.xyz>

README.md

@@ -131,10 +131,12 @@ Options:
     * `access_token`: authentication token generated for your Twitter account
     * `access_token_secret`: authentication token secret generated for your Twitter account
     * `hashtags`: list of key/value used to append hashtags to each tweet. Key is the pattern to match in the product name, value is the string to append to the tweet. For example, `{"twitter": {"hashtags": [{"rtx 3090": "#nvidia #rtx3090"}]}}` will detect `rtx 3090` to append `#nvidia #rtx3090` at the end of the tweet.
+    * `enable_replies`: reply to original message when product is not available anymore
 * `telegram` (optional):
     * `channel_name`: send message to a channel (ex: `@channel`)
     * `chat_id`: send message to a chat (ex: `1234`)
     * `token`: key returned by BotFather
+    * `enable_replies`: reply to original message when product is not available anymore
 * `include_regex` (optional): include products with a name matching this regexp
 * `exclude_regex` (optional): exclude products with a name matching this regexp
 * `browser_address` (optional): set headless browser address (ex: `http://127.0.0.1:9222`)

config.go

@@ -25,6 +25,7 @@ type TwitterConfig struct {
 	AccessToken       string              `json:"access_token"`
 	AccessTokenSecret string              `json:"access_token_secret"`
 	Hashtags          []map[string]string `json:"hashtags"`
+	EnableReplies     bool                `json:"enable_replies"`
 }
 
 // TelegramConfig to store Telegram API key
@@ -32,6 +33,7 @@ type TelegramConfig struct {
 	Token         string `json:"token"`
 	ChatID        int64  `json:"chat_id"`
 	ChannelName   string `json:"channel_name"`
+	EnableReplies bool   `json:"enable_replies"`
 }
 
 // ApiConfig to store HTTP API configuration

notifier_telegram.go

@@ -23,7 +23,7 @@ type TelegramNotifier struct {
 	bot           *telegram.BotAPI
 	chatID        int64
 	channelName   string
-	timeout     int
+	enableReplies bool
 }
 
 // NewTelegramNotifier to create a Notifier with Telegram capabilities
@@ -46,6 +46,7 @@ func NewTelegramNotifier(config *TelegramConfig, db *gorm.DB) (*TelegramNotifier
 		bot:           bot,
 		chatID:        config.ChatID,
 		channelName:   config.ChannelName,
+		enableReplies: config.EnableReplies,
 	}, nil
 }
 
@@ -92,6 +93,7 @@ func (n *TelegramNotifier) NotifyWhenNotAvailable(productURL string, duration ti
 		return nil
 	}
 
+	if n.enableReplies {
 		// format message
 		text := fmt.Sprintf("And it's gone (%s)", duration)
 
@@ -101,6 +103,7 @@ func (n *TelegramNotifier) NotifyWhenNotAvailable(productURL string, duration ti
 			return fmt.Errorf("failed to reply on telegram: %s", err)
 		}
 		log.Infof("reply to telegram message %d sent", m.MessageID)
+	}
 
 	// remove message from database
 	trx = n.db.Unscoped().Delete(&m)

notifier_twitter.go

@@ -30,6 +30,7 @@ type TwitterNotifier struct {
 	client        *twitter.Client
 	user          *twitter.User
 	hashtagsMap   []map[string]string
+	enableReplies bool
 }
 
 // NewTwitterNotifier creates a TwitterNotifier
@@ -57,7 +58,7 @@ func NewTwitterNotifier(c *TwitterConfig, db *gorm.DB) (*TwitterNotifier, error)
 	}
 	log.Debugf("connected to twitter as @%s", user.ScreenName)
 
-	return &TwitterNotifier{client: client, user: user, hashtagsMap: c.Hashtags, db: db}, nil
+	return &TwitterNotifier{client: client, user: user, hashtagsMap: c.Hashtags, db: db, enableReplies: c.EnableReplies}, nil
 }
 
 // create a brand new tweet
@@ -147,6 +148,7 @@ func (c *TwitterNotifier) NotifyWhenNotAvailable(productURL string, duration tim
 		return nil
 	}
 
+	if c.enableReplies {
 		// format message
 		message := fmt.Sprintf("And it's gone (%s)", duration)
 
@@ -156,6 +158,7 @@ func (c *TwitterNotifier) NotifyWhenNotAvailable(productURL string, duration tim
 			return fmt.Errorf("failed to create reply tweet: %s", err)
 		}
 		log.Infof("reply to tweet %d sent", tweet.TweetID)
+	}
 
 	// remove tweet from database
 	trx = c.db.Unscoped().Delete(&tweet)