Q58Bot/core/init.go

package core

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"
	"time"

	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)

var (
	Bot         *tgbotapi.BotAPI
	BOT_TOKEN   string
	ChatID      int64
	ADMIN_ID    int64
	Symbols     []string
	SingaporeTZ *time.Location
	DB_FILE     string
	DEBUG_MODE  bool
	err         error
	DB          *Database
)

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")
	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, err = NewDatabase()
	if err != nil {
		return fmt.Errorf("初始化数据库失败: %v", err)
	}

	// 确保所有必要的表都存在
	if err := DB.EnsureTablesExist(); err != nil {
		return fmt.Errorf("确保数据库表存在失败: %v", err)
	}

	// 检查并报告表中的记录数
	keywordCount, err := DB.CountRecords("keywords")
	if err != nil {
		log.Printf("检查 keywords 表记录数时出错: %v", err)
	} else {
		log.Printf("keywords 表中有 %d 条记录", keywordCount)
	}

	whitelistCount, err := DB.CountRecords("whitelist")
	if err != nil {
		log.Printf("检查 whitelist 表记录数时出错: %v", err)
	} else {
		log.Printf("whitelist 表中有 %d 条记录", whitelistCount)
	}

	// 从环境变量中读取调试模式设置
	DEBUG_MODE = os.Getenv("DEBUG_MODE") == "true"

	// 设置时区
	loc := time.FixedZone("Asia/Singapore", 8*60*60)
	time.Local = loc

	// 初始化 Chat ID
	chatIDStr := os.Getenv("CHAT_ID")
	ChatID, err = mustParseInt64(chatIDStr)
	if err != nil {
		return fmt.Errorf("invalid CHAT_ID: %v", err)
	}

	// 初始化 Symbols
	symbolsRaw := strings.Split(os.Getenv("SYMBOLS"), ",")
	Symbols = make([]string, len(symbolsRaw))
	for i, s := range symbolsRaw {
		Symbols[i] = strings.ReplaceAll(s, "/", "")
	}

	// 初始化新加坡时区
	SingaporeTZ, err = time.LoadLocation("Asia/Singapore")
	if err != nil {
		log.Printf("加载新加坡时区时出错: %v", err)
		log.Println("回落至 UTC+8")
		SingaporeTZ = time.FixedZone("Asia/Singapore", 8*60*60)
	}

	// 初始化 Bot API
	Bot, err = tgbotapi.NewBotAPI(BOT_TOKEN)
	if err != nil {
		return fmt.Errorf("创建 Bot API 失败: %v", err)
	}

	log.Printf("账户已授权 %s", Bot.Self.UserName)
	return nil
}