diff --git a/main.go b/main.go index 98790bb..cc69652 100644 --- a/main.go +++ b/main.go @@ -18,7 +18,10 @@ func main() { log.Fatalf("Invalid ADMIN_ID: %v", err) } - service.Init(botToken, adminID) + err = service.Init(botToken, adminID) + if err != nil { + log.Fatalf("Failed to initialize service: %v", err) + } go service.RunGuard() go service.RunBinance() diff --git a/service/guard.go b/service/guard.go index af50a68..223cc8f 100644 --- a/service/guard.go +++ b/service/guard.go @@ -91,7 +91,7 @@ func startBot() error { return fmt.Errorf("error registering commands: %w", err) } - linkFilter, err := NewLinkFilter(core.DB_FILE) + linkFilter, err := NewLinkFilter() if err != nil { return fmt.Errorf("failed to create LinkFilter: %v", err) } diff --git a/service/init.go b/service/init.go index b6eee12..4b646c4 100644 --- a/service/init.go +++ b/service/init.go @@ -6,10 +6,18 @@ import ( "github.com/woodchen-ink/Q58Bot/core" ) -func Init(botToken string, adminID int64) { +func Init(botToken string, adminID int64) error { core.InitGlobalVariables(botToken, adminID) + // 初始化提示词服务 + err := InitPromptService() + if err != nil { + return err + } + // 设置时区 loc := time.FixedZone("Asia/Singapore", 8*60*60) time.Local = loc + + return nil }