feat: .env持久化

Author: timetetng

.env

@@ -0,0 +1,2 @@
+BILI_ROOM=7394063
+BILI_COOKIE="buvid3=49FFE070-F309-D1B1-B8FA-00D463C9D3D203375infoc; SESSDATA=5997b3dc%2C1788271713%2C04fd6%2A32CjBs3VQzSERGzBZF9dQe8QuArTf55vMgIAO69yAEHFsdTJ9lz7Wex0e8bTaM5vaBd4cSVk5JUUgxRUE5RHJsLVBXaGZDU2xwWTBkNWRMdEotOVVrS2lBS1FPRWcwVmo1Z3lXZW01MGZJYlZxWFotUHdNQ0V1UE1uUkdHdDhRWTdIX2V5YjJ4dFFRIIEC; bili_jct=791fe73eb5f1f0249f0ce1d1860d5afd"

.env_example

@@ -0,0 +1,6 @@
+# Bilibili 自动监听配置
+# BILI_ROOM: 你的B站直播间房间号
+BILI_ROOM=
+
+# BILI_COOKIE: 你的B站Cookie (可选，如果不需要登录身份则留空)
+BILI_COOKIE=

docker-compose.yaml

@@ -8,6 +8,8 @@ services:
     ports:
       - "8080:8080"
       - "7777:7777"
+    env_file:
+      - .env
     volumes:
       - ./data:/app/data
     restart: always

main.go

@@ -86,6 +86,17 @@ func main() {
 				Value:   0,
 				Usage:   "日志等级 (0: 默认, 1: 简洁, 2: 静默)",
 			},
+			// 新增：B站房间号和Cookie环境变量注入
+			&cli.IntFlag{
+				Name:    "biliRoom",
+				EnvVars: []string{"BILI_ROOM"},
+				Usage:   "Bilibili 房间号 (自动启动监听)",
+			},
+			&cli.StringFlag{
+				Name:    "biliCookie",
+				EnvVars: []string{"BILI_COOKIE"},
+				Usage:   "Bilibili Cookie",
+			},
 		},
 		Action: func(c *cli.Context) error {
 			// 初始化 Trace
@@ -128,6 +139,13 @@ func main() {
 			// 恢复之前保存的服务状态
 			api.RecoverService()
 
+			// 新增：通过环境变量自动启动 Bilibili 服务
+			biliRoom := c.Int("biliRoom")
+			if biliRoom > 0 {
+				biliCookie := c.String("biliCookie")
+				api.AutoStartBilibili(biliRoom, biliCookie)
+			}
+
 			// 处理程序信号以进行优雅退出
 			trace.HandleSignal()
 

services/api/restful.go

@@ -582,3 +582,22 @@ func ServeDashboard(w http.ResponseWriter, r *http.Request) {
 	w.WriteHeader(http.StatusOK)
 	_, _ = w.Write(web.DashboardHTML)
 }
+
+// AutoStartBilibili 提供给 main 包调用的自动启动 Bilibili 服务的方法
+func AutoStartBilibili(roomID int, cookie string) {
+	serviceKey := generateServiceKey("bilibili", strconv.Itoa(roomID))
+	
+	// 如果由于 RecoverService 已经从文件恢复了这个房间，则跳过
+	if _, exists := serviceMap.GetService(serviceKey); exists {
+		log.Printf("INFO", "Bilibili 房间 %d 已在监听(已从持久化配置恢复)，跳过自动启动", roomID)
+		return
+	}
+	
+	stopChan := make(chan struct{})
+	err := startBilibiliService(roomID, cookie, stopChan)
+	if err != nil {
+		log.Printf("ERROR", "通过环境变量自动启动 Bilibili 监听失败: %v", err)
+	} else {
+		log.Printf("INFO", "通过环境变量自动启动 Bilibili 监听成功 (房间: %d)", roomID)
+	}
+}