|
@@ -4,6 +4,7 @@ import (
|
|
"embed"
|
|
"embed"
|
|
"fmt"
|
|
"fmt"
|
|
"net"
|
|
"net"
|
|
|
|
+ "net/url"
|
|
"os"
|
|
"os"
|
|
"strconv"
|
|
"strconv"
|
|
"strings"
|
|
"strings"
|
|
@@ -21,6 +22,8 @@ import (
|
|
"github.com/mymmrac/telego"
|
|
"github.com/mymmrac/telego"
|
|
th "github.com/mymmrac/telego/telegohandler"
|
|
th "github.com/mymmrac/telego/telegohandler"
|
|
tu "github.com/mymmrac/telego/telegoutil"
|
|
tu "github.com/mymmrac/telego/telegoutil"
|
|
|
|
+ "github.com/valyala/fasthttp"
|
|
|
|
+ "github.com/valyala/fasthttp/fasthttpproxy"
|
|
)
|
|
)
|
|
|
|
|
|
var bot *telego.Bot
|
|
var bot *telego.Bot
|
|
@@ -90,7 +93,12 @@ func (t *Tgbot) Start(i18nFS embed.FS) error {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
- bot, err = telego.NewBot(tgBottoken)
|
|
|
|
|
|
+ tgBotProxy, err := t.settingService.GetTgBotProxy()
|
|
|
|
+ if err != nil {
|
|
|
|
+ logger.Warning("Failed to get ProxyUrl:", err)
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ bot, err = t.NewBot(tgBottoken, tgBotProxy)
|
|
if err != nil {
|
|
if err != nil {
|
|
fmt.Println("Get tgbot's api error:", err)
|
|
fmt.Println("Get tgbot's api error:", err)
|
|
return err
|
|
return err
|
|
@@ -106,6 +114,23 @@ func (t *Tgbot) Start(i18nFS embed.FS) error {
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+func (t *Tgbot) NewBot(token string, proxyUrl string) (*telego.Bot, error) {
|
|
|
|
+ if proxyUrl == "" || !strings.HasPrefix(proxyUrl, "socks5://") {
|
|
|
|
+ logger.Warning("invalid socks5 url, start with default")
|
|
|
|
+ return telego.NewBot(token)
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ _, err := url.Parse(proxyUrl)
|
|
|
|
+ if err != nil {
|
|
|
|
+ logger.Warning("cant parse proxy url, use default instance for tgbot:", err)
|
|
|
|
+ return telego.NewBot(token)
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ return telego.NewBot(token, telego.WithFastHTTPClient(&fasthttp.Client{
|
|
|
|
+ Dial: fasthttpproxy.FasthttpSocksDialer(proxyUrl),
|
|
|
|
+ }))
|
|
|
|
+}
|
|
|
|
+
|
|
func (t *Tgbot) IsRunning() bool {
|
|
func (t *Tgbot) IsRunning() bool {
|
|
return isRunning
|
|
return isRunning
|
|
}
|
|
}
|