xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-18 06:03:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ feat: support configuration file (#117)

Browse Source 
* ♻️ refactor: move file directory

* ♻️ refactor: move file directory

* ♻️ refactor: support multiple config methods

* 🔥 del: remove unused code

* 💩 refactor: Refactor channel management and synchronization

* 💄 improve: add channel website

* ✨ feat: allow recording 0 consumption
This commit is contained in:
Buer
2024-03-20 14:12:47 +08:00
committed by GitHub
parent 0409de0ea9
commit 71171c63f5
50 changed files with 581 additions and 481 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
common/telegram/common.go
View File

@@ -5,7 +5,6 @@ import (
"fmt"
"one-api/common"
"one-api/model"
"os"
"strings"
"time"
@@ -14,6 +13,7 @@ import (
"github.com/PaulSonOfLars/gotgbot/v2/ext/handlers"
"github.com/PaulSonOfLars/gotgbot/v2/ext/handlers/filters/callbackquery"
"github.com/PaulSonOfLars/gotgbot/v2/ext/handlers/filters/message"
"github.com/spf13/viper"
)
var TGupdater *ext.Updater
@@ -28,13 +28,14 @@ func InitTelegramBot() {
return
}
if os.Getenv("TG_BOT_API_KEY") == "" {
botKey := viper.GetString("TG_BOT_API_KEY")
if botKey == "" {
common.SysLog("Telegram bot is not enabled")
return
}
var err error
TGBot, err = gotgbot.NewBot(os.Getenv("TG_BOT_API_KEY"), nil)
TGBot, err = gotgbot.NewBot(botKey, nil)
if err != nil {
common.SysLog("failed to create new telegram bot: " + err.Error())
return
@@ -47,15 +48,16 @@ func InitTelegramBot() {
}
func StartTelegramBot() {
if os.Getenv("TG_WEBHOOK_SECRET") != "" {
botWebhook := viper.GetString("TG_WEBHOOK_SECRET")
if botWebhook != "" {
if common.ServerAddress == "" {
common.SysLog("Telegram bot is not enabled: Server address is not set")
StopTelegramBot()
return
}
TGWebHookSecret = os.Getenv("TG_WEBHOOK_SECRET")
TGWebHookSecret = botWebhook
serverAddress := strings.TrimSuffix(common.ServerAddress, "/")
urlPath := fmt.Sprintf("/api/telegram/%s", os.Getenv("TG_BOT_API_KEY"))
urlPath := fmt.Sprintf("/api/telegram/%s", viper.GetString("TG_BOT_API_KEY"))
webHookOpts := &ext.AddWebhookOpts{
SecretToken: TGWebHookSecret,