From 4677d789f2e25015c8df4e24af9cd3c84d3abb95 Mon Sep 17 00:00:00 2001
From: fumiama <s862105088@gmail.com>
Date: Mon, 11 Oct 2021 13:32:23 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=E5=A2=9E=E5=8A=A0-u=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md |  6 ++++--
 main.go   | 12 +++++++++---
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 8e481001..62d277d6 100644
--- a/README.md
+++ b/README.md
@@ -20,9 +20,11 @@
 
 ## 命令行参数
 ```bash
-zerobot -h [-d|w] [-g] qq1 qq2 qq3 ...
+zerobot -h -t token -u url [-d|w] [-g] qq1 qq2 qq3 ...
 ```
-- **-h**: 显示帮助 
+- **-h**: 显示帮助
+- **-t token**: 设置`AccessToken`
+- **-u url**: 设置`Url`
 - **-d|w**: 开启 debug | warning 级别及以上日志输出
 - **-g**: 开启 [webgui](https://github.com/FloatTech/bot-manager)
 - **qqs**: superusers 的 qq 号
diff --git a/main.go b/main.go
index 6d77207e..0ad4876b 100644
--- a/main.go
+++ b/main.go
@@ -61,13 +61,19 @@ var (
 		"* Project: https://github.com/FloatTech/ZeroBot-Plugin",
 	}
 	banner = strings.Join(contents, "\n")
+	token  *string
+	url    *string
 )
 
 func init() {
-	// 解析命令行参数，输入 `-d` 即可开启 debug log
+	// 解析命令行参数
 	d := flag.Bool("d", false, "Enable debug level log and higher.")
 	w := flag.Bool("w", false, "Enable warning level log and higher.")
 	h := flag.Bool("h", false, "Display this help.")
+	// 直接写死 AccessToken 时，请更改下面第二个参数
+	token = flag.String("t", "", "Set AccessToken of WSClient.")
+	// 直接写死 URL 时，请更改下面第二个参数
+	url = flag.String("u", "ws://127.0.0.1:6700", "Set Url of WSClient.")
 	flag.Parse()
 	if *h {
 		printBanner()
@@ -106,8 +112,8 @@ func main() {
 		Driver: []zero.Driver{
 			&driver.WSClient{
 				// OneBot 正向WS 默认使用 6700 端口
-				Url:         "ws://127.0.0.1:6700",
-				AccessToken: "",
+				Url:         *url,
+				AccessToken: *token,
 			},
 		},
 	})