feat: add telegram proxy support (#1764)

* Add support for reverse proxy of telegram API

* Add Telegram API proxy hint

---------

Co-authored-by: Athurg Feng <athurg@gooth.org>

api/system_setting.go

@@ -182,6 +182,14 @@ func (upsert SystemSettingUpsert) Validate() error {
 		if upsert.Value == "" {
 			return nil
 		}
+		// Robot Token with Reverse Proxy shoule like `http.../bot<token>`
+		if strings.HasPrefix(upsert.Value, "http") {
+			slashIndex := strings.LastIndexAny(upsert.Value, "/")
+			if strings.HasPrefix(upsert.Value[slashIndex:], "/bot") {
+				return nil
+			}
+			return fmt.Errorf("token start with `http` must end with `/bot<token>`")
+		}
 		fragments := strings.Split(upsert.Value, ":")
 		if len(fragments) != 2 {
 			return fmt.Errorf(systemSettingUnmarshalError, settingName)

plugin/telegram/download.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"strings"
 )
 
 // downloadFileId download file with fileID, return the filepath and blob.
@@ -23,13 +24,19 @@ func (r *Robot) downloadFileID(ctx context.Context, fileID string) (string, []by
 
 // downloadFilepath download file with filepath, you can get filepath by calling GetFile.
 func (r *Robot) downloadFilepath(ctx context.Context, filePath string) ([]byte, error) {
-	token := r.handler.RobotToken(ctx)
+	apiURL, err := r.apiURL(ctx)
-	if token == "" {
+	if err != nil {
-		return nil, ErrNoToken
+		return nil, err
 	}
 
-	uri := "https://api.telegram.org/file/bot" + token + "/" + filePath
+	idx := strings.LastIndex(apiURL, "/bot")
-	resp, err := http.Get(uri)
+	if idx < 0 {
+		return nil, ErrInvalidToken
+	}
+
+	fileURL := apiURL[:idx] + "/file" + apiURL[idx:]
+
+	resp, err := http.Get(fileURL + "/" + filePath)
 	if err != nil {
 		return nil, fmt.Errorf("fail to http.Get: %s", err)
 	}

plugin/telegram/request.go

@@ -3,23 +3,19 @@ package telegram
 import (
 	"context"
 	"encoding/json"
-	"errors"
 	"fmt"
 	"io"
 	"net/http"
 	"net/url"
 )
 
-var ErrNoToken = errors.New("token is empty")
-
 func (r *Robot) postForm(ctx context.Context, apiPath string, formData url.Values, result any) error {
-	token := r.handler.RobotToken(ctx)
+	apiURL, err := r.apiURL(ctx)
-	if token == "" {
+	if err != nil {
-		return ErrNoToken
+		return err
 	}
 
-	uri := "https://api.telegram.org/bot" + token + apiPath
+	resp, err := http.PostForm(apiURL+apiPath, formData)
-	resp, err := http.PostForm(uri, formData)
 	if err != nil {
 		return fmt.Errorf("fail to http.PostForm: %s", err)
 	}

plugin/telegram/robot.go

@@ -2,7 +2,9 @@ package telegram
 
 import (
 	"context"
+	"errors"
 	"fmt"
+	"strings"
 	"time"
 
 	"github.com/usememos/memos/common/log"
@@ -32,7 +34,7 @@ func (r *Robot) Start(ctx context.Context) {
 
 	for {
 		updates, err := r.GetUpdates(ctx, offset)
-		if err == ErrNoToken {
+		if err == ErrInvalidToken {
 			time.Sleep(noTokenWait)
 			continue
 		}
@@ -79,3 +81,18 @@ func (r *Robot) Start(ctx context.Context) {
 		}
 	}
 }
+
+var ErrInvalidToken = errors.New("token is invalid")
+
+func (r *Robot) apiURL(ctx context.Context) (string, error) {
+	token := r.handler.RobotToken(ctx)
+	if token == "" {
+		return "", ErrInvalidToken
+	}
+
+	if strings.HasPrefix(token, "http") {
+		return token, nil
+	}
+
+	return "https://api.telegram.org/bot" + token, nil
+}

web/src/components/Settings/SystemSection.tsx

@@ -293,7 +293,10 @@ const SystemSection = () => {
         <div className="flex flex-row items-center">
           <div className="w-auto flex items-center">
             <span className="text-sm mr-1">{t("setting.system-section.telegram-robot-token")}</span>
-            <HelpButton icon="help" url="https://usememos.com/docs/integration/telegram-bot" />
+            <HelpButton
+              hint={t("setting.system-section.telegram-robot-token-description")}
+              url="https://usememos.com/docs/integration/telegram-bot"
+            />
           </div>
         </div>
         <Button onClick={handleSaveTelegramRobotToken}>{t("common.save")}</Button>

web/src/locales/en.json

@@ -259,7 +259,7 @@
       "additional-style-placeholder": "Additional CSS code",
       "additional-script-placeholder": "Additional JavaScript code",
       "telegram-robot-token": "Telegram Robot Token",
-      "telegram-robot-token-description": "Get from @BotFather of Telegram",
+      "telegram-robot-token-description": "Telegram Robot Token or API Proxy like `http.../bot<token>`",
       "telegram-robot-token-placeholder": "Your Telegram Robot token",
       "openai-api-key": "OpenAI: API Key",
       "openai-api-key-description": "Get API key",

web/src/locales/zh-Hans.json

@@ -412,6 +412,7 @@
       "ignore-version-upgrade": "忽略版本升级",
       "telegram-robot-token": "Telegram 机器人 Token",
       "telegram-robot-token-description": "从 Telegram 的 @BotFather 处获取",
+      "telegram-robot-token-description": "Telegram 机器人Token或`http.../bot<token>`格式的代理地址",
       "telegram-robot-token-placeholder": "Telegram 的机器人 Token",
       "openai-api-host": "OpenAI：API Host",
       "openai-api-host-placeholder": "默认：https://api.openai.com/",