feat: 增强消息发送功能，支持发送照片和编辑消息的标题及媒体

Author: TBXark

telegram/bot.go

@@ -125,13 +125,32 @@ func SendMessage(ctx context.Context, b *bot.Bot, update *Update, m *Message) er
 	}
 	if update.CallbackQuery != nil {
 		origin := update.CallbackQuery.Message.Message
-		param := m.toEditMessageTextParams(origin.Chat.ID, origin.ID)
-		_, err := b.EditMessageText(ctx, param)
-		return err
-	} else if update.Message != nil {
-		param := m.toSendMessageParams(update.Message.Chat.ID)
-		_, err := b.SendMessage(ctx, param)
-		return err
+		if len(origin.Photo) == 0 {
+			param := m.toEditMessageTextParams(origin.Chat.ID, origin.ID)
+			_, err := b.EditMessageText(ctx, param)
+			return err
+		} else {
+			if m.Media == nil {
+				param := m.toEditMessageCaptionParams(origin.Chat.ID, origin.ID)
+				_, err := b.EditMessageCaption(ctx, param)
+				return err
+			} else {
+				param := m.toEditMessageMediaParams(origin.Chat.ID, origin.ID)
+				_, err := b.EditMessageMedia(ctx, param)
+				return err
+			}
+		}
+	}
+	if update.Message != nil {
+		if m.Media == nil {
+			param := m.toSendMessageParams(update.Message.Chat.ID)
+			_, err := b.SendMessage(ctx, param)
+			return err
+		} else {
+			param := m.toSendPhotoParams(update.Message.Chat.ID)
+			_, err := b.SendPhoto(ctx, param)
+			return err
+		}
 	}
 	return nil
 }

telegram/message.go

@@ -21,8 +21,9 @@ func NewButton[T any](text, dType string, data T) Button {
 
 type Message struct {
 	Text      string
-	Button    [][]Button
+	Media     models.InputFile
 	ParseMode models.ParseMode
+	Button    [][]Button
 }
 
 func (m *Message) toInlineKeyboardMarkup() *models.InlineKeyboardMarkup {
@@ -54,6 +55,19 @@ func (m *Message) toSendMessageParams(chatID int64) *bot.SendMessageParams {
 	return params
 }
 
+func (m *Message) toSendPhotoParams(chatID int64) *bot.SendPhotoParams {
+	params := &bot.SendPhotoParams{
+		ChatID:    chatID,
+		Caption:   m.Text,
+		ParseMode: m.ParseMode,
+		Photo:     m.Media,
+	}
+	if len(m.Button) > 0 {
+		params.ReplyMarkup = m.toInlineKeyboardMarkup()
+	}
+	return params
+}
+
 func (m *Message) toEditMessageTextParams(chatID int64, messageID int) *bot.EditMessageTextParams {
 	params := &bot.EditMessageTextParams{
 		ChatID:    chatID,
@@ -66,3 +80,39 @@ func (m *Message) toEditMessageTextParams(chatID int64, messageID int) *bot.Edit
 	}
 	return params
 }
+
+func (m *Message) toEditMessageCaptionParams(chatID int64, messageID int) *bot.EditMessageCaptionParams {
+	params := &bot.EditMessageCaptionParams{
+		ChatID:    chatID,
+		MessageID: messageID,
+		Caption:   m.Text,
+	}
+	if len(m.Button) > 0 {
+		params.ReplyMarkup = m.toInlineKeyboardMarkup()
+	}
+	return params
+}
+
+func (m *Message) toEditMessageMediaParams(chatID int64, messageID int) *bot.EditMessageMediaParams {
+	params := &bot.EditMessageMediaParams{
+		ChatID:    chatID,
+		MessageID: messageID,
+		Media: &models.InputMediaPhoto{
+			Caption:   m.Text,
+			ParseMode: m.ParseMode,
+		},
+	}
+	photo := &models.InputMediaPhoto{}
+	if upload, ok := m.Media.(*models.InputFileUpload); ok {
+		photo.Media = "attach://" + upload.Filename
+		photo.MediaAttachment = upload.Data
+	}
+	if url, ok := m.Media.(*models.InputFileString); ok {
+		photo.Media = url.Data
+	}
+	params.Media = photo
+	if len(m.Button) > 0 {
+		params.ReplyMarkup = m.toInlineKeyboardMarkup()
+	}
+	return params
+}