feat: ssh生成本地RSA,支持本地直连授权配置

src/modules/network_element/controller/ne_host.go

@@ -354,3 +354,53 @@ func (s *NeHostController) CheckBySSH(c *gin.Context) {
 
 	c.JSON(200, result.OkData(data))
 }
+
+// 网元主机SSH方式授权免密发送
+//
+// POST /authorizedBySSH
+func (s *NeHostController) AuthorizedBySSH(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body model.NeHost
+	err := c.ShouldBindBodyWith(&body, binding.JSON)
+	if err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 本地免密创建链接直连
+	sshLink := false
+	lcoalConnSSH := ssh.ConnSSH{
+		User: body.User,
+		Addr: body.Addr,
+		Port: body.Port,
+	}
+	lcoalClient, err := lcoalConnSSH.NewClientByLocalPrivate()
+	if err == nil {
+		sshLink = true
+	}
+	defer lcoalClient.Close()
+	if sshLink {
+		// 连接主机成功，无需重复免密授权认证
+		c.JSON(200, result.OkMsg(i18n.TKey(language, "neHost.okBySSHLink")))
+		return
+	}
+
+	// 创建链接SSH客户端
+	var connSSH ssh.ConnSSH
+	body.CopyTo(&connSSH)
+	client, err := connSSH.NewClient()
+	if err != nil {
+		// 连接主机失败，请检查连接参数后重试
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "neHost.errByHostInfo")))
+		return
+	}
+	defer client.Close()
+
+	// 发送密钥
+	err = client.SendToAuthorizedKeys()
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	c.JSON(200, result.Ok(nil))
+}