use markdown for a subset of responses

moolite · Jan 11, 2024 · 855bb41 · 855bb41
1 parent 242697b
commit 855bb41
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/internal/telegram/escape_text.go b/internal/telegram/escape_text.go
@@ -0,0 +1,12 @@
+package telegram
+
+import (
+	"strings"
+)
+
+func escapeText(t string) string {
+	t = strings.ReplaceAll(t, "*", `\*`)
+	t = strings.ReplaceAll(t, "!", `\!`)
+	t = strings.ReplaceAll(t, ".", `\.`)
+	return t
+}
diff --git a/internal/telegram/types.go b/internal/telegram/types.go
@@ -43,7 +43,8 @@ func (w *WebhookResponse) setMethod(method string, isMedia bool) *WebhookRespons
 }
 
 func (w *WebhookResponse) SendDice() *WebhookResponse {
-	return w.setMethod("sendMessage", false)
+	return w.setMethod("sendMessage", false).
+		UseMarkdown()
 }
 
 func (w *WebhookResponse) SendMessage() *WebhookResponse {
@@ -75,6 +76,8 @@ func (w *WebhookResponse) SetChatID(chatID string) *WebhookResponse {
 }
 
 func (w *WebhookResponse) SetText(text string) *WebhookResponse {
+	text = escapeText(text)
+
 	if w.isMedia {
 		w.Caption = &text
 	} else {
@@ -94,6 +97,11 @@ func (w *WebhookResponse) SetLinks(data []URLButton) *WebhookResponse {
 	return w
 }
 
+func (w *WebhookResponse) UseMarkdown() *WebhookResponse {
+	w.ParseMode = "MarkdownV2"
+	return w
+}
+
 func (w *WebhookResponse) SetKeyboard(kb string) *WebhookResponse {
 	w.ReplyMarkup = &kb
 	return w
@@ -123,11 +131,10 @@ func (w *WebhookResponse) Empty() *WebhookResponse {
 
 func (w *WebhookResponse) Marshal() ([]byte, error) {
 	// empty response
-	// spare a 46 bytes
+	// spare 46 bytes
 	if len(w.Method) == 0 && !w.isMedia {
 		return []byte{}, nil
 	}
 
-	w.ParseMode = "MarkdownV2"
 	return json.Marshal(w)
 }