From 8eccb967f521aee22d44028886503bac247c2b24 Mon Sep 17 00:00:00 2001
From: cdle <798731886@qq.com>
Date: Sat, 18 Sep 2021 16:59:26 +0800
Subject: [PATCH] update

---
 core/im.go   |  4 +++-
 core/test.go | 20 +++++++++++---------
 2 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/core/im.go b/core/im.go
index 6d0e324..22ac915 100644
--- a/core/im.go
+++ b/core/im.go
@@ -26,7 +26,9 @@ type Sender interface {
 	Disappear(lifetime ...time.Duration)
 }
 
-type Edit string
+type Edit int
+
+var E Edit
 
 type Faker struct {
 	Message interface{}
diff --git a/core/test.go b/core/test.go
index e841fee..8c4deaa 100644
--- a/core/test.go
+++ b/core/test.go
@@ -44,7 +44,7 @@ func initSys() {
 			Rules: []string{"raw ^升级$"},
 			Admin: true,
 			Handle: func(s Sender) interface{} {
-				s.Reply(name() + "开始检查核心功能。")
+				s.Reply(name()+"开始检查核心功能。", E)
 				update := false
 				record := func(b bool) {
 					if !update && b {
@@ -56,35 +56,37 @@ func initSys() {
 					return err
 				}
 				if !need {
-					s.Reply(name() + "核心功能已是最新。")
+					s.Reply(name()+"核心功能已是最新。", E)
 				} else {
 					record(need)
-					s.Reply(name() + "核心功能发现更新。")
+					s.Reply(name()+"核心功能发现更新。", E)
 				}
 				files, _ := ioutil.ReadDir(ExecPath + "/develop")
 				for _, f := range files {
 					if f.IsDir() {
 						need, err := GitPull("/develop/" + f.Name())
 						if err != nil {
-							s.Reply(name() + "扩展" + f.Name() + "更新错误" + err.Error() + "。")
+							s.Reply(name()+"扩展"+f.Name()+"更新错误"+err.Error()+"。", E)
 						}
 						if !need {
-							s.Reply(name() + "扩展" + f.Name() + "已是最新。")
+							s.Reply(name()+"扩展"+f.Name()+"已是最新。", E)
 						} else {
 							record(need)
-							s.Reply(name() + "扩展" + f.Name() + "发现更新。")
+							s.Reply(name()+"扩展"+f.Name()+"发现更新。", E)
 						}
 					}
 				}
 				if !update {
-					return name() + "没有更新。"
+					s.Reply(name()+"没有更新。", E)
+					return nil
 				}
-				s.Reply(name() + "正在编译程序。")
+				s.Reply(name()+"正在编译程序。", E)
 				if err := CompileCode(); err != nil {
 					return err
 				}
-				s.Reply(name()+"编译程序完毕。", time.Duration(0))
+				s.Reply(name()+"编译程序完毕。", E)
 				sillyGirl.Set("rebootInfo", fmt.Sprintf("%v %v %v", s.GetImType(), s.GetChatID(), s.GetUserID()))
+				s.Reply(name()+"更新完成，即将重启！", E)
 				Daemon()
 				return nil
 			},