From 4300502182c646927e77639d534472a0b3f68b6c Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Wed, 13 Nov 2024 10:32:47 +0800
Subject: [PATCH] add update ssh comfig

---
 .../server/controller/configinstance.go       | 45 +++++++++++++++++++
 configmanage/server/service/ssh.go            |  4 ++
 2 files changed, 49 insertions(+)

diff --git a/configmanage/server/controller/configinstance.go b/configmanage/server/controller/configinstance.go
index 1c89baf8..0998059b 100644
--- a/configmanage/server/controller/configinstance.go
+++ b/configmanage/server/controller/configinstance.go
@@ -590,6 +590,51 @@ func UpdateConfigHandler(c *gin.Context) {
 		response.Success(c, nil, "update host config success")
 
 	case global.SSH:
+		// 解析参数
+		sshconfig := &service.SSHConfig{}
+		if err := json.Unmarshal(query.Data, sshconfig); err != nil {
+			logger.Error("failed to parse sshconfig parameter: %s", err.Error())
+			response.Fail(c, "failed to parse sshconfig parameter:", err.Error())
+			return
+		}
+
+		file := common.File{}
+		if err := json.Unmarshal([]byte(sshconfig.Content), &file); err != nil {
+			logger.Error("failed to parse file parameter: %s", err.Error())
+			response.Fail(c, "failed to parse file parameter:", err.Error())
+			return
+		}
+		file.Content = base64.StdEncoding.EncodeToString([]byte(file.Content))
+
+		// 查询此配置的内容
+		sf, err := service.GetSSHFileByInfoUUID(ci.UUID, nil)
+		if err != nil {
+			logger.Error("failed to get sshconfig parameter: %s", err.Error())
+			response.Fail(c, "failed to get sshconfig parameter:", err.Error())
+			return
+		}
+
+		// 更新参数
+		sshconfig.UUID = sf.UUID
+		sshconfig.ConfigInfoUUID = ci.UUID
+		sshconfig.IsActive = false
+		sshconfig.Content, err = json.Marshal(file)
+		if err != nil {
+			logger.Error("Error encoding JSON:: %s", err.Error())
+			response.Fail(c, "Error encoding JSON:", err.Error())
+			return
+		}
+
+		// 将参数添加到数据库
+		err = sshconfig.Record()
+		if err != nil {
+			logger.Error("failed to update sshconfig: %s", err.Error())
+			response.Fail(c, "failed to update sshconfig:", err.Error())
+			return
+		}
+
+		logger.Debug("update sshconfig success")
+		response.Success(c, nil, "update sshconfig success")
 
 	case global.SSHD:
 
diff --git a/configmanage/server/service/ssh.go b/configmanage/server/service/ssh.go
index 7fc7133d..105194b1 100644
--- a/configmanage/server/service/ssh.go
+++ b/configmanage/server/service/ssh.go
@@ -229,6 +229,10 @@ func (sc *SSHConfig) Collect() error {
 	return nil
 }
 
+func GetSSHFileByInfoUUID(uuid string, isindex interface{}) (SSHFile, error) {
+	return internal.GetSSHFileByInfoUUID(uuid, isindex)
+}
+
 // 根据配置uuid获取所有配置文件
 func GetSSHFilesByCinfigUUID(uuid string) ([]SSHFile, error) {
 	return internal.GetSSHFilesByCinfigUUID(uuid)
-- 
Gitee