123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214 |
- package server
- import (
- "bytes"
- "context"
- "encoding/json"
- "fmt"
- "strconv"
- "unicode/utf16"
- "github.com/pkg/errors"
- apiv1 "github.com/usememos/memos/api/v1"
- "github.com/usememos/memos/plugin/telegram"
- "github.com/usememos/memos/store"
- )
- type telegramHandler struct {
- store *store.Store
- }
- func newTelegramHandler(store *store.Store) *telegramHandler {
- return &telegramHandler{store: store}
- }
- func (t *telegramHandler) BotToken(ctx context.Context) string {
- return t.store.GetSystemSettingValueWithDefault(&ctx, apiv1.SystemSettingTelegramBotTokenName.String(), "")
- }
- const (
- workingMessage = "Working on send your memo..."
- successMessage = "Success"
- )
- func (t *telegramHandler) MessageHandle(ctx context.Context, bot *telegram.Bot, message telegram.Message, attachments []telegram.Attachment) error {
- reply, err := bot.SendReplyMessage(ctx, message.Chat.ID, message.MessageID, workingMessage)
- if err != nil {
- return fmt.Errorf("fail to SendReplyMessage: %s", err)
- }
- var creatorID int
- userSettingList, err := t.store.ListUserSettings(ctx, &store.FindUserSetting{
- Key: apiv1.UserSettingTelegramUserIDKey.String(),
- })
- if err != nil {
- return errors.Wrap(err, "Failed to find userSettingList")
- }
- for _, userSetting := range userSettingList {
- var value string
- if err := json.Unmarshal([]byte(userSetting.Value), &value); err != nil {
- continue
- }
- if value == strconv.Itoa(message.From.ID) {
- creatorID = userSetting.UserID
- }
- }
- if creatorID == 0 {
- _, err := bot.EditMessage(ctx, message.Chat.ID, reply.MessageID, fmt.Sprintf("Please set your telegram userid %d in UserSetting of Memos", message.From.ID), nil)
- return err
- }
- create := &store.Memo{
- CreatorID: creatorID,
- Visibility: store.Private,
- }
- if message.Text != nil {
- create.Content = convertToMarkdown(*message.Text, message.Entities)
- }
- if message.Caption != nil {
- create.Content = convertToMarkdown(*message.Caption, message.CaptionEntities)
- }
- if message.ForwardFromChat != nil {
- create.Content += fmt.Sprintf("\n\n[Message link](%s)", message.GetMessageLink())
- }
- memoMessage, err := t.store.CreateMemo(ctx, create)
- if err != nil {
- _, err := bot.EditMessage(ctx, message.Chat.ID, reply.MessageID, fmt.Sprintf("failed to CreateMemo: %s", err), nil)
- return err
- }
- // create resources
- for _, attachment := range attachments {
- // Fill the common field of create
- create := store.Resource{
- CreatorID: creatorID,
- Filename: attachment.FileName,
- Type: attachment.GetMimeType(),
- Size: attachment.FileSize,
- }
- err := apiv1.SaveResourceBlob(ctx, t.store, &create, bytes.NewReader(attachment.Data))
- if err != nil {
- _, err := bot.EditMessage(ctx, message.Chat.ID, reply.MessageID, fmt.Sprintf("failed to SaveResourceBlob: %s", err), nil)
- return err
- }
- resource, err := t.store.CreateResource(ctx, &create)
- if err != nil {
- _, err := bot.EditMessage(ctx, message.Chat.ID, reply.MessageID, fmt.Sprintf("failed to CreateResource: %s", err), nil)
- return err
- }
- _, err = t.store.UpsertMemoResource(ctx, &store.UpsertMemoResource{
- MemoID: memoMessage.ID,
- ResourceID: resource.ID,
- })
- if err != nil {
- _, err := bot.EditMessage(ctx, message.Chat.ID, reply.MessageID, fmt.Sprintf("failed to UpsertMemoResource: %s", err), nil)
- return err
- }
- }
- keyboard := generateKeyboardForMemoID(memoMessage.ID)
- _, err = bot.EditMessage(ctx, message.Chat.ID, reply.MessageID, fmt.Sprintf("Saved as %s Memo %d", memoMessage.Visibility, memoMessage.ID), keyboard)
- return err
- }
- func (t *telegramHandler) CallbackQueryHandle(ctx context.Context, bot *telegram.Bot, callbackQuery telegram.CallbackQuery) error {
- var memoID int
- var visibility store.Visibility
- n, err := fmt.Sscanf(callbackQuery.Data, "%s %d", &visibility, &memoID)
- if err != nil || n != 2 {
- return bot.AnswerCallbackQuery(ctx, callbackQuery.ID, fmt.Sprintf("fail to parse callbackQuery.Data %s", callbackQuery.Data))
- }
- update := store.UpdateMemo{
- ID: memoID,
- Visibility: &visibility,
- }
- err = t.store.UpdateMemo(ctx, &update)
- if err != nil {
- return bot.AnswerCallbackQuery(ctx, callbackQuery.ID, fmt.Sprintf("fail to call UpdateMemo %s", err))
- }
- keyboard := generateKeyboardForMemoID(memoID)
- _, err = bot.EditMessage(ctx, callbackQuery.Message.Chat.ID, callbackQuery.Message.MessageID, fmt.Sprintf("Saved as %s Memo %d", visibility, memoID), keyboard)
- if err != nil {
- return bot.AnswerCallbackQuery(ctx, callbackQuery.ID, fmt.Sprintf("fail to EditMessage %s", err))
- }
- return bot.AnswerCallbackQuery(ctx, callbackQuery.ID, fmt.Sprintf("Success change Memo %d to %s", memoID, visibility))
- }
- func generateKeyboardForMemoID(id int) [][]telegram.InlineKeyboardButton {
- allVisibility := []store.Visibility{
- store.Public,
- store.Protected,
- store.Private,
- }
- buttons := make([]telegram.InlineKeyboardButton, 0, len(allVisibility))
- for _, v := range allVisibility {
- button := telegram.InlineKeyboardButton{
- Text: v.String(),
- CallbackData: fmt.Sprintf("%s %d", v, id),
- }
- buttons = append(buttons, button)
- }
- return [][]telegram.InlineKeyboardButton{buttons}
- }
- func convertToMarkdown(text string, messageEntities []telegram.MessageEntity) string {
- insertions := make(map[int]string)
- for _, e := range messageEntities {
- var before, after string
- // this is supported by the current markdown
- switch e.Type {
- case telegram.Bold:
- before = "**"
- after = "**"
- case telegram.Italic:
- before = "*"
- after = "*"
- case telegram.Strikethrough:
- before = "~~"
- after = "~~"
- case telegram.Code:
- before = "`"
- after = "`"
- case telegram.Pre:
- before = "```" + e.Language
- after = "```"
- case telegram.TextLink:
- before = "["
- after = fmt.Sprintf(`](%s)`, e.URL)
- }
- if before != "" {
- insertions[e.Offset] += before
- insertions[e.Offset+e.Length] = after + insertions[e.Offset+e.Length]
- }
- }
- input := []rune(text)
- var output []rune
- utf16pos := 0
- for i := 0; i < len(input); i++ {
- output = append(output, []rune(insertions[utf16pos])...)
- output = append(output, input[i])
- utf16pos += len(utf16.Encode([]rune{input[i]}))
- }
- output = append(output, []rune(insertions[utf16pos])...)
- return string(output)
- }
|