feat: enhance artwork caption with username and website link

krau · Jan 27, 2025 · 18256f0 · 18256f0
1 parent 26ff8a2
commit 18256f0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/telegram/utils/post_artwork.go b/telegram/utils/post_artwork.go
@@ -369,10 +369,13 @@ func recaptionArtwork(ctx context.Context, artwork *types.Artwork, bot *telego.B
 		return
 	}
 	newCaption := GetArtworkHTMLCaption(newArtwork)
-	bot.EditMessageCaption(&telego.EditMessageCaptionParams{
+	_, err = bot.EditMessageCaption(&telego.EditMessageCaptionParams{
 		ChatID:    ChannelChatID,
 		MessageID: newArtwork.Pictures[0].TelegramInfo.MessageID,
 		Caption:   newCaption,
 		ParseMode: telego.ModeHTML,
 	})
+	if err != nil {
+		common.Logger.Warnf("error when recaption artwork: %s", err)
+	}
 }
diff --git a/telegram/utils/utils.go b/telegram/utils/utils.go
@@ -99,7 +99,16 @@ func GetArtworkHTMLCaption(artwork *types.Artwork) string {
 		tag = tagCharsReplacer.Replace(tag)
 		tags += "#" + strings.TrimSpace(common.EscapeHTML(tag)) + " "
 	}
-	caption += fmt.Sprintf("\n<blockquote expandable=true>%s</blockquote>", tags)
+	caption += fmt.Sprintf("\n<blockquote expandable=true>%s</blockquote>\n", tags)
+	if ChannelChatID.Username != "" {
+		caption += common.EscapeHTML(ChannelChatID.Username)
+	}
+	if artwork.ID != "" && config.Cfg.API.SiteURL != "" {
+		if ChannelChatID.Username != "" {
+			caption += " | "
+		}
+		caption += fmt.Sprintf("<a href=\"%s/artwork/%s\">在网站查看</a>", config.Cfg.API.SiteURL, artwork.ID)
+	}
 	return caption
 }