From db1e0c8ece3371a003876aa9fc85f8886886a7e1 Mon Sep 17 00:00:00 2001
From: qydysky <qydysky@foxmail.com>
Date: Sun, 20 Oct 2024 09:38:47 +0800
Subject: [PATCH] =?utf8?q?Improve=20=E5=91=BD=E4=BB=A4=E8=A1=8C=E5=8F=82?=
 =?utf8?q?=E6=95=B0=E6=94=AF=E6=8C=81=E4=BB=8E=E7=8E=AF=E5=A2=83=E5=8F=98?=
 =?utf8?q?=E9=87=8F=E8=8E=B7=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 CV/Var.go | 13 +++++++++++++
 README.md |  1 +
 2 files changed, 14 insertions(+)

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中
-- 
2.39.2