From 8ba44602ae6de64f561edec8b43e8165142da46b Mon Sep 17 00:00:00 2001
From: eatmoreapple <eatmoreorange@gmail.com>
Date: Fri, 9 Dec 2022 13:26:14 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=9C=A8=E9=9D=9E=E7=83=AD?=
 =?UTF-8?q?=E7=99=BB=E5=BD=95=E6=A8=A1=E5=BC=8F=E4=B8=8B=E4=B8=BB=E5=8A=A8?=
 =?UTF-8?q?=E5=AF=BC=E5=87=BA=E7=83=AD=E7=99=BB=E5=BD=95=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/bot.go b/bot.go
index 264d351..b8093c9 100644
--- a/bot.go
+++ b/bot.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
+	"io"
 	"log"
 	"net/url"
 	"os/exec"
@@ -352,6 +353,12 @@ func (b *Bot) DumpHotReloadStorage() error {
 	if b.hotReloadStorage == nil {
 		return errors.New("HotReloadStorage can not be nil")
 	}
+	return b.DumpTo(b.hotReloadStorage)
+}
+
+// DumpTo 将热登录需要的数据写入到指定的 io.Writer 中
+// 注: 写之前最好先清空之前的数据
+func (b *Bot) DumpTo(writer io.Writer) error {
 	cookies := b.Caller.Client.GetCookieMap()
 	item := HotReloadStorageItem{
 		BaseRequest:  b.Storage.Request,
@@ -360,8 +367,7 @@ func (b *Bot) DumpHotReloadStorage() error {
 		WechatDomain: b.Caller.Client.Domain,
 		UUID:         b.uuid,
 	}
-
-	return json.NewEncoder(b.hotReloadStorage).Encode(item)
+	return json.NewEncoder(writer).Encode(item)
 }
 
 // OnLogin is a setter for LoginCallBack