wood/Q58Bot
1
0
Fork 0
mirror of https://github.com/woodchen-ink/Q58Bot.git synced 2025-07-18 13:52:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

初始化时获取环境变量

Browse Source
This commit is contained in:
wood chen 2024-09-19 22:07:41 +08:00
parent 20238fab2e
commit 5d05e0a343
2 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
core/init.go
View File

@ -39,7 +39,20 @@ func mustParseInt64(s string) (int64, error) {
return value, nil return value, nil
} }
func Init(botToken string, adminID int64) error { func Init() error {
// 从环境变量获取 BOT_TOKEN
BOT_TOKEN = os.Getenv("BOT_TOKEN")
if BOT_TOKEN == "" {
return fmt.Errorf("BOT_TOKEN 环境变量未设置")
}
// 从环境变量获取 ADMIN_ID
adminIDStr := os.Getenv("ADMIN_ID")
var err error
ADMIN_ID, err = mustParseInt64(adminIDStr)
if err != nil {
return fmt.Errorf("Invalid ADMIN_ID: %v", err)
}
// 设置数据库文件路径 // 设置数据库文件路径
DB_FILE = filepath.Join("/app/data", "q58.db") DB_FILE = filepath.Join("/app/data", "q58.db")
@ -55,7 +68,7 @@ func Init(botToken string, adminID int64) error {
chatIDStr := os.Getenv("CHAT_ID") chatIDStr := os.Getenv("CHAT_ID")
ChatID, err = mustParseInt64(chatIDStr) ChatID, err = mustParseInt64(chatIDStr)
if err != nil { if err != nil {
log.Fatalf("Invalid CHAT_ID: %v", err) return fmt.Errorf("Invalid CHAT_ID: %v", err)
} }
// 初始化 Symbols // 初始化 Symbols
@ -74,9 +87,9 @@ func Init(botToken string, adminID int64) error {
} }
// 初始化 Bot API // 初始化 Bot API
Bot, err = tgbotapi.NewBotAPI(botToken) Bot, err = tgbotapi.NewBotAPI(BOT_TOKEN)
if err != nil { if err != nil {
log.Fatal(err) return fmt.Errorf("创建 Bot API 失败: %v", err)
} }
log.Printf("账户已授权 %s", Bot.Self.UserName) log.Printf("账户已授权 %s", Bot.Self.UserName)

12
main.go
View File

@ -2,8 +2,6 @@ package main
import ( import (
"log" "log"
"os"
"strconv"
"github.com/woodchen-ink/Q58Bot/core" "github.com/woodchen-ink/Q58Bot/core"
"github.com/woodchen-ink/Q58Bot/service" "github.com/woodchen-ink/Q58Bot/service"
@ -13,15 +11,7 @@ import (
func main() { func main() {
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile) log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
botToken := os.Getenv("BOT_TOKEN") err := core.Init()
adminIDStr := os.Getenv("ADMIN_ID")
adminID, err := strconv.ParseInt(adminIDStr, 10, 64)
if err != nil {
log.Fatalf("Failed to get ADMIN_ID: %v", err)
}
err = core.Init(botToken, adminID)
if err != nil { if err != nil {
log.Fatalf("Failed to initialize service: %v", err) log.Fatalf("Failed to initialize service: %v", err)
} }