From 6dbb4b173567cbb60b47092df4aa0181cc053ca5 Mon Sep 17 00:00:00 2001
From: vaaski <admin@vaa.ski>
Date: Sat, 3 Apr 2021 00:38:16 +0200
Subject: [PATCH] add disable_notification parameter (#99)

---
 DOCS.md        | 14 +++++++++++++-
 main.go        |  6 ++++++
 plugin.go      |  2 ++
 plugin_test.go | 21 +++++++++++++++++++++
 4 files changed, 42 insertions(+), 1 deletion(-)

diff --git a/DOCS.md b/DOCS.md
index 67fbdcf..6265de8 100644
--- a/DOCS.md
+++ b/DOCS.md
@@ -233,10 +233,22 @@ Disables link previews for links in this message
     settings:
       token: xxxxxxxxxx
       to: telegram_user_id
-      message: send message using custom socks5 URL
+      message: send message without a link preview
 +     disable_web_page_preview: true
 ```
 
+Disables notifications for this message
+
+```diff
+  - name: send telegram notification
+    image: appleboy/drone-telegram
+    settings:
+      token: xxxxxxxxxx
+      to: telegram_user_id
+      message: send message message silently
++     disable_notification: true
+```
+
 ## Parameter Reference
 
 token
diff --git a/main.go b/main.go
index 6177535..3c5e1d8 100644
--- a/main.go
+++ b/main.go
@@ -114,6 +114,11 @@ func main() {
 			Usage:  "disables link previews for links in this message",
 			EnvVar: "PLUGIN_DISABLE_WEB_PAGE_PREVIEW,INPUT_DISABLE_WEB_PAGE_PREVIEW",
 		},
+		cli.BoolFlag{
+			Name:   "disable.notification",
+			Usage:  "sends the message silently. users will receive a notification with no sound.",
+			EnvVar: "PLUGIN_DISABLE_NOTIFICATION,INPUT_DISABLE_NOTIFICATION",
+		},
 		cli.StringFlag{
 			Name:   "format",
 			Value:  formatMarkdown,
@@ -322,6 +327,7 @@ func run(c *cli.Context) error {
 			Socks5:           c.String("socks5"),
 
 			DisableWebPagePreview: c.Bool("disable.webpage.preview"),
+			DisableNotification:   c.Bool("disable.notification"),
 		},
 	}
 
diff --git a/plugin.go b/plugin.go
index 8b9adc6..0fe85ee 100644
--- a/plugin.go
+++ b/plugin.go
@@ -88,6 +88,7 @@ type (
 		Socks5           string
 
 		DisableWebPagePreview bool
+		DisableNotification   bool
 	}
 
 	// Plugin values.
@@ -365,6 +366,7 @@ func (p Plugin) Exec() (err error) {
 			msg := tgbotapi.NewMessage(user, txt)
 			msg.ParseMode = p.Config.Format
 			msg.DisableWebPagePreview = p.Config.DisableWebPagePreview
+			msg.DisableNotification = p.Config.DisableNotification
 			if err := p.Send(bot, msg); err != nil {
 				return err
 			}
diff --git a/plugin_test.go b/plugin_test.go
index 250a02d..b2fa26c 100644
--- a/plugin_test.go
+++ b/plugin_test.go
@@ -145,6 +145,27 @@ func TestDisableWebPagePreviewMessage(t *testing.T) {
 	assert.Nil(t, err)
 }
 
+func TestDisableNotificationMessage(t *testing.T) {
+	plugin := Plugin{
+		Config: Config{
+			Token:               os.Getenv("TELEGRAM_TOKEN"),
+			To:                  []string{os.Getenv("TELEGRAM_TO")},
+			DisableNotification: true,
+			Debug:               false,
+		},
+	}
+
+	plugin.Config.Message = "DisableNotification https://www.google.com.tw"
+	err := plugin.Exec()
+	assert.Nil(t, err)
+
+	// disable message
+	plugin.Config.DisableNotification = false
+	plugin.Config.Message = "EnableNotification https://www.google.com.tw"
+	err = plugin.Exec()
+	assert.Nil(t, err)
+}
+
 func TestBotError(t *testing.T) {
 	plugin := Plugin{
 		Repo: Repo{