YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: format message from telegram and upload attachments (#1924)

Browse Source 
* feat: format message from telegram and download documents

* fix: remove bool in expression

* refactor: convert to markdown

* refactor: resolve remarks and add support new message types

* refactor: resolve remarks

* feat: add test for mime type

---------

Co-authored-by: Александр Тумайкин <AATumaykin@tsum.ru>
This commit is contained in:
Alexandr Tumaykin
2023-07-13 19:18:44 +03:00
committed by GitHub
parent f074bb1be2
commit c5a1f4c839
15 changed files with 422 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
plugin/telegram/handle.go
View File

@ -2,24 +2,23 @@ package telegram
import (
"context"
"fmt"
)
// handleSingleMessages handle single messages not belongs to group.
func (b *Bot) handleSingleMessages(ctx context.Context, messages []Message) error {
for _, message := range messages {
var blobs map[string][]byte
var attachments []Attachment
// download blob if provided
if len(message.Photo) > 0 {
filepath, blob, err := b.downloadFileID(ctx, message.GetMaxPhotoFileID())
if err != nil {
return err
}
blobs = map[string][]byte{filepath: blob}
for _, message := range messages {
attachment, err := b.downloadAttachment(ctx, &message)
if err != nil {
return err
}
err := b.handler.MessageHandle(ctx, b, message, blobs)
if attachment != nil {
attachments = append(attachments, *attachment)
}
err = b.handler.MessageHandle(ctx, b, message, attachments)
if err != nil {
return err
}
@ -32,7 +31,7 @@ func (b *Bot) handleSingleMessages(ctx context.Context, messages []Message) erro
func (b *Bot) handleGroupMessages(ctx context.Context, groupMessages []Message) error {
captions := make(map[string]string, len(groupMessages))
messages := make(map[string]Message, len(groupMessages))
blobs := make(map[string]map[string][]byte, len(groupMessages))
attachments := make(map[string][]Attachment, len(groupMessages))
// Group all captions, blobs and messages
for _, message := range groupMessages {
@ -44,14 +43,14 @@ func (b *Bot) handleGroupMessages(ctx context.Context, groupMessages []Message)
captions[groupID] += *message.Caption
}
filepath, blob, err := b.downloadFileID(ctx, message.GetMaxPhotoFileID())
attachment, err := b.downloadAttachment(ctx, &message)
if err != nil {
return fmt.Errorf("fail to downloadFileID")
return err
}
if _, found := blobs[groupID]; !found {
blobs[groupID] = make(map[string][]byte)
if attachment != nil {
attachments[groupID] = append(attachments[groupID], *attachment)
}
blobs[groupID][filepath] = blob
}
// Handle each group message
@ -59,7 +58,7 @@ func (b *Bot) handleGroupMessages(ctx context.Context, groupMessages []Message)
// replace Caption with all Caption in the group
caption := captions[groupID]
message.Caption = &caption
err := b.handler.MessageHandle(ctx, b, message, blobs[groupID])
err := b.handler.MessageHandle(ctx, b, message, attachments[groupID])
if err != nil {
return err
}