creds: Support line breaks in values and shorten too long values

Brawl345 · Oct 5, 2024 · 06a36d8 · 06a36d8
1 parent 1cd07c7
commit 06a36d8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/plugin/creds/creds.go b/plugin/creds/creds.go
@@ -43,7 +43,7 @@ func (p *Plugin) Handlers(botInfo *gotgbot.User) []plugin.Handler {
 			AdminOnly:   true,
 		},
 		&plugin.CommandHandler{
-			Trigger:     regexp.MustCompile(fmt.Sprintf(`(?i)^/creds_add(?:@%s)? ([^\s]+) (.+)$`, botInfo.Username)),
+			Trigger:     regexp.MustCompile(fmt.Sprintf(`(?i)^/creds_add(?:@%s)? ([^\s]+) ([\s\S]+)$`, botInfo.Username)),
 			HandlerFunc: p.OnAdd,
 			AdminOnly:   true,
 		},
@@ -82,7 +82,11 @@ func (p *Plugin) OnGet(b *gotgbot.Bot, c plugin.GobotContext) error {
 	var sb strings.Builder
 
 	for _, key := range keys {
-		sb.WriteString(fmt.Sprintf("<b>%s</b>:\n<code>%s</code>\n", key, creds[key]))
+		if len(creds[key]) > 300 {
+			sb.WriteString(fmt.Sprintf("<b>%s</b>:\n<code>%s...</code>\n", key, creds[key][:297]))
+		} else {
+			sb.WriteString(fmt.Sprintf("<b>%s</b>:\n<code>%s</code>\n", key, creds[key]))
+		}
 	}
 
 	_, err := c.EffectiveMessage.Reply(b, sb.String(), &gotgbot.SendMessageOpts{