fix: 网元操作文件列表/上传/下载改为ssh连接

src/modules/network_element/controller/ne_action.go

@@ -3,6 +3,7 @@ package controller
 import (
 	"fmt"
 	"path/filepath"
+	"runtime"
 	"strings"
 
 	"be.ems/src/framework/i18n"
@@ -50,19 +51,30 @@ func (s *NeActionController) PushFile(c *gin.Context) {
 		return
 	}
 
-	// 本地文件
-	localPath := file.ParseUploadFilePath(body.UploadPath)
-	nePath := "/tmp" //config.Get("mml.upload").(string)
-	// 复制到远程
-	err := ssh.FileSCPLocalToNe(neInfo.IP, localPath, nePath)
+	// 网元主机的SSH客户端
+	sshClient, err := s.neInfoService.NeRunSSHclient(neInfo.NeType, neInfo.NeId)
 	if err != nil {
 		c.JSON(200, result.ErrMsg(err.Error()))
 		return
 	}
+	defer sshClient.Close()
+	// 网元主机的SSH客户端进行文件传输
+	sftpClient, err := sshClient.NewClientSFTP()
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer sftpClient.Close()
+
+	// 本地文件
+	localFilePath := file.ParseUploadFilePath(body.UploadPath)
+	neFilePath := fmt.Sprintf("/tmp/%s", filepath.Base(localFilePath))
+	// 复制到远程
+	if err = sftpClient.CopyFileLocalToRemote(localFilePath, neFilePath); err != nil {
+		c.JSON(200, result.ErrMsg(fmt.Sprintf("%s : please check if scp remote copy is allowed", neInfo.NeType)))
+		return
+	}
 
-	// 网元端文件路径
-	fileName := localPath[strings.LastIndex(localPath, "/")+1:]
-	neFilePath := fmt.Sprintf("%s/%s", nePath, fileName)
 	c.JSON(200, result.OkData(filepath.ToSlash(neFilePath)))
 }
 
@@ -89,14 +101,32 @@ func (s *NeActionController) PullFile(c *gin.Context) {
 		return
 	}
 
-	nePath := fmt.Sprintf("%s/%s", querys.Path, querys.FileName)
-	localPath := fmt.Sprintf("/tmp/omc/pullFile/%s", querys.FileName)
-	err := ssh.FileSCPNeToLocal(neInfo.IP, nePath, localPath)
+	// 网元主机的SSH客户端
+	sshClient, err := s.neInfoService.NeRunSSHclient(neInfo.NeType, neInfo.NeId)
 	if err != nil {
 		c.JSON(200, result.ErrMsg(err.Error()))
 		return
 	}
-	c.FileAttachment(localPath, querys.FileName)
+	defer sshClient.Close()
+	// 网元主机的SSH客户端进行文件传输
+	sftpClient, err := sshClient.NewClientSFTP()
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer sftpClient.Close()
+
+	nePath := fmt.Sprintf("%s/%s", querys.Path, querys.FileName)
+	localFilePath := fmt.Sprintf("/tmp/omc/pullFile%s", nePath)
+	if runtime.GOOS == "windows" {
+		localFilePath = fmt.Sprintf("C:%s", localFilePath)
+	}
+	// 复制到本地
+	if err = sftpClient.CopyFileRemoteToLocal(nePath, localFilePath); err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	c.FileAttachment(localFilePath, querys.FileName)
 }
 
 // 网元端文件列表
@@ -124,7 +154,16 @@ func (s *NeActionController) Files(c *gin.Context) {
 		return
 	}
 
-	totalSize, rows, err := ssh.FileList(querys.Path, neInfo.IP, querys.Search)
+	// 网元主机的SSH客户端
+	sshClient, err := s.neInfoService.NeRunSSHclient(neInfo.NeType, neInfo.NeId)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer sshClient.Close()
+
+	// 获取文件列表
+	totalSize, rows, err := ssh.FileList(sshClient, querys.Path, querys.Search)
 	if err != nil {
 		c.JSON(200, result.Ok(map[string]any{
 			"path":      querys.Path,