mirror of https://github.com/Xhofe/alist
119 lines
3.0 KiB
Go
119 lines
3.0 KiB
Go
package sftp
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/alist-org/alist/v3/internal/driver"
|
|
"github.com/alist-org/alist/v3/internal/errs"
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/pkg/utils"
|
|
"github.com/pkg/sftp"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type SFTP struct {
|
|
model.Storage
|
|
Addition
|
|
client *sftp.Client
|
|
clientConnectionError error
|
|
}
|
|
|
|
func (d *SFTP) Config() driver.Config {
|
|
return config
|
|
}
|
|
|
|
func (d *SFTP) GetAddition() driver.Additional {
|
|
return &d.Addition
|
|
}
|
|
|
|
func (d *SFTP) Init(ctx context.Context) error {
|
|
return d.initClient()
|
|
}
|
|
|
|
func (d *SFTP) Drop(ctx context.Context) error {
|
|
if d.client != nil {
|
|
_ = d.client.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *SFTP) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return nil, err
|
|
}
|
|
log.Debugf("[sftp] list dir: %s", dir.GetPath())
|
|
files, err := d.client.ReadDir(dir.GetPath())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
objs, err := utils.SliceConvert(files, func(src os.FileInfo) (model.Obj, error) {
|
|
return d.fileToObj(src, dir.GetPath())
|
|
})
|
|
return objs, err
|
|
}
|
|
|
|
func (d *SFTP) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return nil, err
|
|
}
|
|
remoteFile, err := d.client.Open(file.GetPath())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
link := &model.Link{
|
|
MFile: remoteFile,
|
|
}
|
|
return link, nil
|
|
}
|
|
|
|
func (d *SFTP) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return err
|
|
}
|
|
return d.client.MkdirAll(path.Join(parentDir.GetPath(), dirName))
|
|
}
|
|
|
|
func (d *SFTP) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return err
|
|
}
|
|
return d.client.Rename(srcObj.GetPath(), path.Join(dstDir.GetPath(), srcObj.GetName()))
|
|
}
|
|
|
|
func (d *SFTP) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return err
|
|
}
|
|
return d.client.Rename(srcObj.GetPath(), path.Join(path.Dir(srcObj.GetPath()), newName))
|
|
}
|
|
|
|
func (d *SFTP) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
|
|
return errs.NotSupport
|
|
}
|
|
|
|
func (d *SFTP) Remove(ctx context.Context, obj model.Obj) error {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return err
|
|
}
|
|
return d.remove(obj.GetPath())
|
|
}
|
|
|
|
func (d *SFTP) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
|
|
if err := d.clientReconnectOnConnectionError(); err != nil {
|
|
return err
|
|
}
|
|
dstFile, err := d.client.Create(path.Join(dstDir.GetPath(), stream.GetName()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
_ = dstFile.Close()
|
|
}()
|
|
err = utils.CopyWithCtx(ctx, dstFile, stream, stream.GetSize(), up)
|
|
return err
|
|
}
|
|
|
|
var _ driver.Driver = (*SFTP)(nil)
|