2022-09-16 18:53:45 +08:00
|
|
|
package client
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2024-01-19 16:48:41 +08:00
|
|
|
"io"
|
2022-09-16 18:53:45 +08:00
|
|
|
"net"
|
|
|
|
"os"
|
|
|
|
"path"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/pkg/sftp"
|
|
|
|
"golang.org/x/crypto/ssh"
|
|
|
|
)
|
|
|
|
|
|
|
|
type sftpClient struct {
|
2023-12-01 17:50:09 +08:00
|
|
|
bucket string
|
|
|
|
connInfo string
|
|
|
|
config *ssh.ClientConfig
|
2022-09-16 18:53:45 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func NewSftpClient(vars map[string]interface{}) (*sftpClient, error) {
|
2023-12-01 17:50:09 +08:00
|
|
|
address := loadParamFromVars("address", true, vars)
|
|
|
|
port := loadParamFromVars("port", false, vars)
|
|
|
|
password := loadParamFromVars("password", true, vars)
|
|
|
|
username := loadParamFromVars("username", true, vars)
|
|
|
|
bucket := loadParamFromVars("bucket", true, vars)
|
|
|
|
|
|
|
|
auth := []ssh.AuthMethod{ssh.Password(password)}
|
|
|
|
clientConfig := &ssh.ClientConfig{
|
|
|
|
User: username,
|
|
|
|
Auth: auth,
|
|
|
|
Timeout: 30 * time.Second,
|
|
|
|
HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
|
|
|
|
return nil
|
|
|
|
},
|
2022-09-16 18:53:45 +08:00
|
|
|
}
|
2023-11-30 21:50:08 +08:00
|
|
|
|
2023-12-01 17:50:09 +08:00
|
|
|
return &sftpClient{bucket: bucket, connInfo: fmt.Sprintf("%s:%s", address, port), config: clientConfig}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s sftpClient) Upload(src, target string) (bool, error) {
|
|
|
|
sshClient, err := ssh.Dial("tcp", s.connInfo, s.config)
|
2022-09-16 18:53:45 +08:00
|
|
|
if err != nil {
|
2023-12-01 17:50:09 +08:00
|
|
|
return false, err
|
2022-09-16 18:53:45 +08:00
|
|
|
}
|
2023-12-01 17:50:09 +08:00
|
|
|
client, err := sftp.NewClient(sshClient)
|
2022-09-16 18:53:45 +08:00
|
|
|
if err != nil {
|
2023-12-01 17:50:09 +08:00
|
|
|
return false, err
|
2022-09-16 18:53:45 +08:00
|
|
|
}
|
2023-12-01 17:50:09 +08:00
|
|
|
defer client.Close()
|
|
|
|
defer sshClient.Close()
|
2023-11-30 21:50:08 +08:00
|
|
|
|
2022-09-16 18:53:45 +08:00
|
|
|
srcFile, err := os.Open(src)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer srcFile.Close()
|
|
|
|
|
2024-01-24 17:41:56 +08:00
|
|
|
targetFilePath := path.Join(s.bucket, target)
|
|
|
|
targetDir, _ := path.Split(targetFilePath)
|
|
|
|
if _, err = client.Stat(targetDir); err != nil {
|
|
|
|
if os.IsNotExist(err) {
|
|
|
|
if err = client.MkdirAll(targetDir); err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
}
|
2024-01-19 16:48:41 +08:00
|
|
|
dstFile, err := client.Create(path.Join(s.bucket, target))
|
2022-09-16 18:53:45 +08:00
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer dstFile.Close()
|
2023-06-08 18:58:15 +08:00
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
if _, err := io.Copy(dstFile, srcFile); err != nil {
|
|
|
|
return false, err
|
2022-09-16 18:53:45 +08:00
|
|
|
}
|
|
|
|
return true, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s sftpClient) ListBuckets() ([]interface{}, error) {
|
|
|
|
var result []interface{}
|
|
|
|
return result, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s sftpClient) Download(src, target string) (bool, error) {
|
2023-12-01 17:50:09 +08:00
|
|
|
sshClient, err := ssh.Dial("tcp", s.connInfo, s.config)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
client, err := sftp.NewClient(sshClient)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer client.Close()
|
|
|
|
defer sshClient.Close()
|
2024-01-19 16:48:41 +08:00
|
|
|
|
2023-12-01 17:50:09 +08:00
|
|
|
srcFile, err := client.Open(s.bucket + "/" + src)
|
2022-09-16 18:53:45 +08:00
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer srcFile.Close()
|
|
|
|
|
|
|
|
dstFile, err := os.Create(target)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer dstFile.Close()
|
|
|
|
|
2024-01-25 18:17:13 +08:00
|
|
|
if _, err = io.Copy(srcFile, dstFile); err != nil {
|
2022-09-16 18:53:45 +08:00
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
return true, err
|
|
|
|
}
|
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
func (s sftpClient) Exist(filePath string) (bool, error) {
|
2023-12-01 17:50:09 +08:00
|
|
|
sshClient, err := ssh.Dial("tcp", s.connInfo, s.config)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
client, err := sftp.NewClient(sshClient)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer client.Close()
|
|
|
|
defer sshClient.Close()
|
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
srcFile, err := client.Open(path.Join(s.bucket, filePath))
|
2022-09-16 18:53:45 +08:00
|
|
|
if err != nil {
|
|
|
|
if os.IsNotExist(err) {
|
|
|
|
return false, nil
|
|
|
|
} else {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
defer srcFile.Close()
|
|
|
|
return true, err
|
|
|
|
}
|
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
func (s sftpClient) Size(filePath string) (int64, error) {
|
2023-12-01 17:50:09 +08:00
|
|
|
sshClient, err := ssh.Dial("tcp", s.connInfo, s.config)
|
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
client, err := sftp.NewClient(sshClient)
|
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
defer client.Close()
|
|
|
|
defer sshClient.Close()
|
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
files, err := client.Stat(path.Join(s.bucket, filePath))
|
2023-12-01 17:50:09 +08:00
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
return files.Size(), nil
|
|
|
|
}
|
|
|
|
|
2022-09-16 18:53:45 +08:00
|
|
|
func (s sftpClient) Delete(filePath string) (bool, error) {
|
2023-12-01 17:50:09 +08:00
|
|
|
sshClient, err := ssh.Dial("tcp", s.connInfo, s.config)
|
|
|
|
if err != nil {
|
2022-09-16 18:53:45 +08:00
|
|
|
return false, err
|
|
|
|
}
|
2023-12-01 17:50:09 +08:00
|
|
|
client, err := sftp.NewClient(sshClient)
|
|
|
|
if err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
defer client.Close()
|
|
|
|
defer sshClient.Close()
|
2022-09-16 18:53:45 +08:00
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
if err := client.Remove(path.Join(s.bucket, filePath)); err != nil {
|
2023-12-01 17:50:09 +08:00
|
|
|
return false, err
|
2022-09-16 18:53:45 +08:00
|
|
|
}
|
2023-12-01 17:50:09 +08:00
|
|
|
return true, nil
|
|
|
|
}
|
2022-09-16 18:53:45 +08:00
|
|
|
|
2023-12-01 17:50:09 +08:00
|
|
|
func (s sftpClient) ListObjects(prefix string) ([]string, error) {
|
|
|
|
sshClient, err := ssh.Dial("tcp", s.connInfo, s.config)
|
|
|
|
if err != nil {
|
2022-09-16 18:53:45 +08:00
|
|
|
return nil, err
|
|
|
|
}
|
2023-12-01 17:50:09 +08:00
|
|
|
client, err := sftp.NewClient(sshClient)
|
|
|
|
if err != nil {
|
2022-09-16 18:53:45 +08:00
|
|
|
return nil, err
|
|
|
|
}
|
2023-12-01 17:50:09 +08:00
|
|
|
defer client.Close()
|
|
|
|
defer sshClient.Close()
|
2022-09-16 18:53:45 +08:00
|
|
|
|
2024-01-19 16:48:41 +08:00
|
|
|
files, err := client.ReadDir(path.Join(s.bucket, prefix))
|
2022-09-28 18:11:36 +08:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2023-08-08 16:44:12 +08:00
|
|
|
var result []string
|
2022-09-28 18:11:36 +08:00
|
|
|
for _, file := range files {
|
|
|
|
result = append(result, file.Name())
|
|
|
|
}
|
|
|
|
return result, nil
|
2022-09-28 00:08:21 +08:00
|
|
|
}
|