feat: implement telegram bot plugin (#1740)

server/telegram.go

@@ -0,0 +1,102 @@
+package server
+
+import (
+	"context"
+	"fmt"
+	"path"
+	"strconv"
+
+	"github.com/usememos/memos/api"
+	"github.com/usememos/memos/common"
+	"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) RobotToken(ctx context.Context) string {
+	return t.store.GetSystemSettingValueOrDefault(&ctx, api.SystemSettingTelegramRobotTokenName, "")
+}
+
+func (t *telegramHandler) MessageHandle(ctx context.Context, message telegram.Message, blobs map[string][]byte) error {
+	var creatorID int
+	userSettingList, err := t.store.FindUserSettingList(ctx, &api.UserSettingFind{
+		Key: api.UserSettingTelegramUserIDKey,
+	})
+	if err != nil {
+		return fmt.Errorf("Fail to find memo user: %s", err)
+	}
+	for _, userSetting := range userSettingList {
+		if userSetting.Value == strconv.Itoa(message.From.ID) {
+			creatorID = userSetting.UserID
+		}
+	}
+
+	if creatorID == 0 {
+		return fmt.Errorf("Please set your telegram userid %d in UserSetting of Memos", message.From.ID)
+	}
+
+	// create memo
+	memoCreate := api.CreateMemoRequest{
+		CreatorID:  creatorID,
+		Visibility: api.Private,
+	}
+
+	if message.Text != nil {
+		memoCreate.Content = *message.Text
+	}
+	if blobs != nil && message.Caption != nil {
+		memoCreate.Content = *message.Caption
+	}
+
+	memoMessage, err := t.store.CreateMemo(ctx, convertCreateMemoRequestToMemoMessage(&memoCreate))
+	if err != nil {
+		return fmt.Errorf("failed to CreateMemo: %s", err)
+	}
+
+	if err := createMemoCreateActivity(ctx, t.store, memoMessage); err != nil {
+		return fmt.Errorf("failed to createMemoCreateActivity: %s", err)
+	}
+
+	// create resources
+	for filename, blob := range blobs {
+		// TODO support more
+		mime := "application/octet-stream"
+		switch path.Ext(filename) {
+		case ".jpg":
+			mime = "image/jpeg"
+		case ".png":
+			mime = "image/png"
+		}
+		resourceCreate := api.ResourceCreate{
+			CreatorID: creatorID,
+			Filename:  filename,
+			Type:      mime,
+			Size:      int64(len(blob)),
+			Blob:      blob,
+			PublicID:  common.GenUUID(),
+		}
+		resource, err := t.store.CreateResource(ctx, &resourceCreate)
+		if err != nil {
+			return fmt.Errorf("failed to CreateResource: %s", err)
+		}
+		if err := createResourceCreateActivity(ctx, t.store, resource); err != nil {
+			return fmt.Errorf("failed to createResourceCreateActivity: %s", err)
+		}
+
+		_, err = t.store.UpsertMemoResource(ctx, &api.MemoResourceUpsert{
+			MemoID:     memoMessage.ID,
+			ResourceID: resource.ID,
+		})
+		if err != nil {
+			return fmt.Errorf("failed to UpsertMemoResource: %s", err)
+		}
+	}
+	return nil
+}