feat: calculate picture info

Author: krau

bot/bot.go

@@ -45,6 +45,8 @@ func RunPolling() {
 	baseGroup.HandleMessageCtx(randomPicture, telegohandler.Or(telegohandler.CommandEqual("setu"), telegohandler.CommandEqual("random")), mentionIsBot)
 	baseGroup.HandleMessageCtx(help, telegohandler.CommandEqual("help"), mentionIsBot)
 	baseGroup.HandleMessageCtx(searchPicture, telegohandler.CommandEqual("search"), mentionIsBot)
+	baseGroup.HandleMessageCtx(calculatePicture, telegohandler.CommandEqual("info"), mentionIsBot)
+
 	baseGroup.HandleMessageCtx(setAdmin, telegohandler.CommandEqual("set_admin"))
 	baseGroup.HandleMessageCtx(deletePicture, telegohandler.Or(telegohandler.CommandEqual("del"), telegohandler.CommandEqual("delete")))
 	baseGroup.HandleMessageCtx(fetchArtwork, telegohandler.CommandEqual("fetch"))

bot/handlers_common.go

@@ -426,3 +426,50 @@ func inlineQuery(ctx context.Context, bot *telego.Bot, query telego.InlineQuery)
 		Logger.Errorf("回复查询失败: %s", err)
 	}
 }
+
+func calculatePicture(ctx context.Context, bot *telego.Bot, message telego.Message) {
+	if message.ReplyToMessage == nil {
+		telegram.ReplyMessage(bot, message, "请使用该命令回复一条图片消息")
+		return
+	}
+	var waitMessageID int
+	msg, err := telegram.ReplyMessage(bot, message, "少女做高数中...(((φ(◎ロ◎;)φ)))")
+	if err == nil {
+		waitMessageID = msg.MessageID
+	}
+	fileBytes, err := telegram.GetMessagePhotoFileBytes(bot, message.ReplyToMessage)
+	if err != nil {
+		telegram.ReplyMessage(bot, message, "获取图片文件失败: "+err.Error())
+		return
+	}
+	hash, err := common.GetPhash(fileBytes)
+	if err != nil {
+		telegram.ReplyMessage(bot, message, "计算图片信息失败: "+err.Error())
+		return
+	}
+	blurScore, err := common.GetBlurScore(fileBytes)
+	if err != nil {
+		telegram.ReplyMessage(bot, message, "计算图片信息失败: "+err.Error())
+		return
+	}
+	width, height, err := common.GetImageSize(fileBytes)
+	if err != nil {
+		telegram.ReplyMessage(bot, message, "计算图片信息失败: "+err.Error())
+		return
+	}
+	text := fmt.Sprintf(
+		"<b>Hash</b>: <code>%s</code>\n<b>模糊度</b>: %.2f\n<b>尺寸</b>: %d x %d",
+		hash, blurScore, width, height,
+	)
+	if waitMessageID == 0 {
+		telegram.ReplyMessageWithHTML(bot, message, text)
+		return
+	}
+	bot.EditMessageText(&telego.EditMessageTextParams{
+		ChatID:    message.Chat.ChatID(),
+		MessageID: waitMessageID,
+		Text:      text,
+		ParseMode: telego.ModeHTML,
+	})
+
+}

cmd/version.go

@@ -7,7 +7,7 @@ import (
 )
 
 const (
-	Version string = "0.10.3"
+	Version string = "0.10.4"
 )
 
 var VersionCmd = &cobra.Command{

telegram/telegram.go

@@ -45,6 +45,10 @@ var (
 			Command:     "search",
 			Description: "搜索图片",
 		},
+		{
+			Command:     "info",
+			Description: "计算图片信息",
+		},
 		{
 			Command:     "help",
 			Description: "食用指南",

telegram/utils.go

@@ -156,6 +156,13 @@ func ReplyMessageWithMarkdown(bot *telego.Bot, message telego.Message, text stri
 		}).WithParseMode(telego.ModeMarkdownV2))
 }
 
+func ReplyMessageWithHTML(bot *telego.Bot, message telego.Message, text string) (*telego.Message, error) {
+	return bot.SendMessage(telegoutil.Message(message.Chat.ChatID(), text).
+		WithReplyParameters(&telego.ReplyParameters{
+			MessageID: message.MessageID,
+		}).WithParseMode(telego.ModeHTML))
+}
+
 func GetArtworkPostMessageURL(messageID int) string {
 	return fmt.Sprintf("https://t.me/%s/%d", strings.ReplaceAll(ChannelChatID.String(), "@", ""), messageID)
 }