diff --git a/main.go b/main.go index 146b352..44b02c7 100644 --- a/main.go +++ b/main.go @@ -2,6 +2,7 @@ package main import ( "log" + "os" "time" "github.com/CuteReimu/bilibili/v2" @@ -27,6 +28,11 @@ func init() { duration = time.Duration(config.DisconnectDuration) * time.Hour } + // 检查并创建tmp目录 + if err := ensureTmpDirectory(); err != nil { + log.Printf("创建tmp目录失败: %v", err) + } + // 初始化bilibili客户端 biliClient = bilibili.New() initBilibili() @@ -39,6 +45,23 @@ func init() { }() } +// ensureTmpDirectory 检查tmp目录是否存在,如果不存在则创建 +func ensureTmpDirectory() error { + _, err := os.Stat("tmp") + if os.IsNotExist(err) { + err = os.Mkdir("tmp", 0755) + if err != nil { + return err + } + log.Println("已创建tmp目录") + } else if err != nil { + return err + } else { + log.Println("tmp目录已存在") + } + return nil +} + func trigger(config configs.Config) { if config.EnableQQ { go internal.SendQQ()