Support parse_mode for telegram

muesli · Oct 2, 2020 · 0c9f153 · 0c9f153
1 parent d22455f
commit 0c9f153
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/bees/telegrambee/telegrambee.go b/bees/telegrambee/telegrambee.go
@@ -35,6 +35,12 @@ import (
 	"github.com/muesli/beehive/bees"
 )
 
+const (
+	ModeHTML     = telegram.ModeHTML
+	ModeMarkdown = "MarkdownV2"
+	ModePlain    = ""
+)
+
 // TelegramBee is a Bee that can connect to Telegram.
 type TelegramBee struct {
 	bees.Bee
@@ -53,15 +59,26 @@ func (mod *TelegramBee) Action(action bees.Action) []bees.Placeholder {
 	case "send":
 		chatID := ""
 		text := ""
+		parseMode := ""
 		action.Options.Bind("chat_id", &chatID)
 		action.Options.Bind("text", &text)
+		action.Options.Bind("parse_mode", &parseMode)
 
 		cid, err := strconv.ParseInt(chatID, 10, 64)
 		if err != nil {
 			panic("Invalid telegram chat ID")
 		}
 
+		// use newer version of markdown parser
+		if parseMode == telegram.ModeMarkdown {
+			parseMode = ModeMarkdown
+		}
+		if parseMode != ModePlain && parseMode != ModeHTML && parseMode != ModeMarkdown {
+			panic("Invalid parse mode")
+		}
+
 		msg := telegram.NewMessage(cid, text)
+		msg.ParseMode = parseMode
 		_, err = mod.bot.Send(msg)
 		if err != nil {
 			mod.Logf("Error sending message %v", err)

diff --git a/bees/telegrambee/telegrambeefactory.go b/bees/telegrambee/telegrambeefactory.go
@@ -22,7 +22,11 @@
 
 package telegrambee
 
-import "github.com/muesli/beehive/bees"
+import (
+	"fmt"
+
+	"github.com/muesli/beehive/bees"
+)
 
 // TelegramBeeFactory is a factory for TelegramBees.
 type TelegramBeeFactory struct {
@@ -130,6 +134,12 @@ func (factory *TelegramBeeFactory) Actions() []bees.ActionDescriptor {
 				Type:        "string",
 				Mandatory:   true,
 			},
+			{
+				Name:        "parse_mode",
+				Description: fmt.Sprintf("Parse mode (%s, %s, or leave empty)", ModeHTML, ModeMarkdown),
+				Type:        "string",
+				Mandatory:   false,
+			},
 		},
 	}}
 	return actions