ai-brain: try to convert markdown to slack markup

kubeshop · Feb 29, 2024 · 68f1496 · 68f1496
1 parent 509c321
commit 68f1496
Showing 1 changed file with 74 additions and 24 deletions.
diff --git a/internal/source/ai-brain/response.go b/internal/source/ai-brain/response.go
@@ -2,34 +2,57 @@ package aibrain
 
 import (
 	"math/rand"
+	"regexp"
+	"strings"
 	"time"
 
 	"github.com/kubeshop/botkube/pkg/api"
 )
 
-var quickResponses = []string{
-	"Just a moment, please...",
-	"Thinking about this one...",
-	"Let me check on that for you.",
-	"Processing your request...",
-	"Working on it!",
-	"This one needs some extra thought.",
-	"I'm carefully considering your request.",
-	"Consulting my super-smart brain...",
-	"Cogs are turning...",
-	"Accessing the knowledge archives...",
-	"Running calculations at lightning speed!",
-	"Hold on tight, I'm diving into the details.",
-	"I'm here to help!",
-	"Happy to look into this for you.",
-	"Always learning to do this better.",
-	"I want to get this right for you.",
-	"Let me see what I can find out.",
-	"My circuits are buzzing!",
-	"Let me consult with my owl advisor...",
-	"Consider it done (or at least, I'll try my best!)",
-	"I'll get back to you with the best possible answer.",
-}
+const (
+	teamsMessageIDSubstr = "thread.tacv2"
+)
+
+var (
+	quickResponses = []string{
+		"Just a moment, please...",
+		"Thinking about this one...",
+		"Let me check on that for you.",
+		"Processing your request...",
+		"Working on it!",
+		"This one needs some extra thought.",
+		"I'm carefully considering your request.",
+		"Consulting my super-smart brain...",
+		"Cogs are turning...",
+		"Accessing the knowledge archives...",
+		"Running calculations at lightning speed!",
+		"Hold on tight, I'm diving into the details.",
+		"I'm here to help!",
+		"Happy to look into this for you.",
+		"Always learning to do this better.",
+		"I want to get this right for you.",
+		"Let me see what I can find out.",
+		"My circuits are buzzing!",
+		"Let me consult with my owl advisor...",
+		"Consider it done (or at least, I'll try my best!)",
+		"I'll get back to you with the best possible answer.",
+	}
+
+	reSlackEscapeCodeBlocks  = regexp.MustCompile("`(.*?)`")
+	reSlackBoldText          = regexp.MustCompile(`\*\*(.*?)\*\*`)
+	reSlackItalicText        = regexp.MustCompile(`_([^_]+)_`)
+	reSlackStrikethroughText = regexp.MustCompile(`~~([^~]+)~~`)
+	reSlackHeadings          = regexp.MustCompile(`^#+ (.*)$`)
+	reSlackLinks             = regexp.MustCompile(`\[([^\]]+)]\(([^)]+)\)`)
+	reSlackImages            = regexp.MustCompile(`!\[([^\]]+)]\(([^)]+)\)`)
+
+	reTeamsBold          = regexp.MustCompile(`\*\*(.*?)\*\*`)
+	reTeamsItalic        = regexp.MustCompile(`_([^_]+)_`)
+	reTeamsStrikethrough = regexp.MustCompile(`~~([^~]+)~~`)
+	reTeamsHeading       = regexp.MustCompile(`^#+ (.*)$`)
+	reTeamsLink          = regexp.MustCompile(`\[([^\]]+)]\(([^)]+)\)`) // Basic link support
+	reTeamsImageLink     = regexp.MustCompile(`!\[([^\]]+)]\(([^)]+)\)`)
+)
 
 func pickQuickResponse(messageID string) api.Message {
 	rand.New(rand.NewSource(time.Now().UnixNano())) // #nosec G404
@@ -74,12 +97,18 @@ func msgNoAIAnswer(messageID string) api.Message {
 }
 
 func msgAIAnswer(messageID, text string) api.Message {
+	convertedText := markdownToSlack(text)
+
+	if strings.Contains(messageID, teamsMessageIDSubstr) {
+		convertedText = markdownToTeams(text)
+	}
+
 	return api.Message{
 		ParentActivityID: messageID,
 		Sections: []api.Section{
 			{
 				Base: api.Base{
-					Body: api.Body{Plaintext: text},
+					Body: api.Body{Plaintext: convertedText},
 				},
 				Context: []api.ContextItem{
 					{Text: "AI-generated content may be incorrect."},
@@ -88,3 +117,24 @@ func msgAIAnswer(messageID, text string) api.Message {
 		},
 	}
 }
+
+func markdownToSlack(markdownText string) string {
+	text := reSlackEscapeCodeBlocks.ReplaceAllString(markdownText, "`$1`")
+	text = reSlackBoldText.ReplaceAllString(text, "*$1*")
+	text = reSlackItalicText.ReplaceAllString(text, "_$1_")
+	text = reSlackStrikethroughText.ReplaceAllString(text, "~$1~")
+	text = reSlackHeadings.ReplaceAllString(text, "*$1*")
+	text = reSlackLinks.ReplaceAllString(text, "<$2|$1>")
+	return reSlackImages.ReplaceAllString(text, "<$2|$1>")
+}
+
+func markdownToTeams(markdownText string) string {
+	text := reTeamsBold.ReplaceAllString(markdownText, "**$1**")
+	text = reTeamsItalic.ReplaceAllString(text, "_$1_")
+	text = reTeamsStrikethrough.ReplaceAllString(text, "<s>$1</s>")
+	text = reTeamsHeading.ReplaceAllString(text, "**$1**")
+
+	// Images in Teams require specific upload mechanisms;
+	//  here, we'll just preserve the image link
+	return reTeamsImageLink.ReplaceAllString(text, "(Image: $2)")
+}