From: qydysky <qydysky@foxmail.com>
Date: Sun, 20 Oct 2024 01:38:47 +0000 (+0800)
Subject: Improve 命令行参数支持从环境变量获取
X-Git-Tag: v0.14.22~31
X-Git-Url: http://127.0.0.1:8081/?a=commitdiff_plain;h=db1e0c8ece3371a003876aa9fc85f8886886a7e1;p=bili_danmu%2F.git

Improve 命令行参数支持从环境变量获取
---

diff --git a/CV/Var.go b/CV/Var.go
index d72d35b..2528d3d 100644
--- a/CV/Var.go
+++ b/CV/Var.go
@@ -13,6 +13,7 @@ import (
 	"net/url"
 	"os"
 	"runtime"
+	"strconv"
 	"strings"
 	"sync"
 	"testing"
@@ -393,6 +394,18 @@ func (t *Common) Init() *Common {
 		}
 	}
 
+	// load from env
+	if tmp := os.Getenv("ckv"); *ckv == "" && tmp != "" {
+		fmt.Println("加载环境变量ckv:", tmp)
+		*ckv = tmp
+	}
+	if tmp := os.Getenv("r"); *roomIdP == 0 && tmp != "" {
+		fmt.Println("加载环境变量r:", tmp)
+		if r, e := strconv.Atoi(tmp); e == nil {
+			*roomIdP = r
+		}
+	}
+
 	t.Roomid = *roomIdP
 
 	if e := t.loadConf(*ckv); e != nil {
diff --git a/README.md b/README.md
index 038b595..2b329be 100644
--- a/README.md
+++ b/README.md
@@ -956,6 +956,7 @@ go build .
 
 #### 注意事项
 * 其中[]内的内容为可选项
+* 当`-r`,`-ckv`为空时，将尝试从环境变量中获取，分别对应键值`r`,`ckv`(>v0.14.21)
 * 法1，2，3建议使用最新提交
 * 程序受主机时区配置影响，注意正确配置主机时区
 * 弹幕及礼物会记录于danmu.log中