From 397a233074a62f2676fc85fddd4ddea03e7a918c Mon Sep 17 00:00:00 2001
From: naiba <hi@nai.ba>
Date: Fri, 8 Jan 2021 21:21:06 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=B8=20improve:=20=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=9C=8D=E5=8A=A1=E5=99=A8=E5=AF=BC=E8=87=B4?=
 =?UTF-8?q?=E7=9A=84=E4=B8=BB=E6=9C=BA=E4=BF=A1=E6=81=AF=E4=B8=A2=E5=A4=B1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/dashboard/controller/member_api.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/cmd/dashboard/controller/member_api.go b/cmd/dashboard/controller/member_api.go
index eb3f666..438a104 100644
--- a/cmd/dashboard/controller/member_api.go
+++ b/cmd/dashboard/controller/member_api.go
@@ -93,6 +93,7 @@ func (ma *memberAPI) addOrEditServer(c *gin.Context) {
 	admin := c.MustGet(model.CtxKeyAuthorizedUser).(*model.User)
 	var sf serverForm
 	var s model.Server
+	var isEdit bool
 	err := c.ShouldBindJSON(&sf)
 	if err == nil {
 		dao.ServerLock.Lock()
@@ -106,6 +107,7 @@ func (ma *memberAPI) addOrEditServer(c *gin.Context) {
 			s.Secret = s.Secret[:10]
 			err = dao.DB.Create(&s).Error
 		} else {
+			isEdit = true
 			err = dao.DB.Save(&s).Error
 		}
 	}
@@ -116,8 +118,13 @@ func (ma *memberAPI) addOrEditServer(c *gin.Context) {
 		})
 		return
 	}
-	s.Host = &model.Host{}
-	s.State = &model.State{}
+	if isEdit {
+		s.Host = dao.ServerList[s.ID].Host
+		s.State = dao.ServerList[s.ID].State
+	} else {
+		s.Host = &model.Host{}
+		s.State = &model.State{}
+	}
 	dao.ServerList[s.ID] = &s
 	dao.ReSortServer()
 	c.JSON(http.StatusOK, model.Response{