feat(115_share): support 115_share

Author: varg1714

drivers/115_share/driver.go

@@ -3,6 +3,7 @@ package _115_share
 import (
 	"context"
 
+	"github.com/OpenListTeam/OpenList/v4/drivers/virtual_file"
 	"github.com/OpenListTeam/OpenList/v4/internal/driver"
 	"github.com/OpenListTeam/OpenList/v4/internal/errs"
 	"github.com/OpenListTeam/OpenList/v4/internal/model"
@@ -46,38 +47,54 @@ func (d *Pan115Share) Drop(ctx context.Context) error {
 }
 
 func (d *Pan115Share) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
-	if err := d.WaitLimit(ctx); err != nil {
-		return nil, err
-	}
 
-	files := make([]driver115.ShareFile, 0)
-	fileResp, err := d.client.GetShareSnap(d.ShareCode, d.ReceiveCode, dir.GetID(), driver115.QueryLimit(int(d.PageSize)))
-	if err != nil {
-		return nil, err
-	}
-	files = append(files, fileResp.Data.List...)
-	total := fileResp.Data.Count
-	count := len(fileResp.Data.List)
-	for total > count {
-		fileResp, err := d.client.GetShareSnap(
-			d.ShareCode, d.ReceiveCode, dir.GetID(),
-			driver115.QueryLimit(int(d.PageSize)), driver115.QueryOffset(count),
-		)
+	return virtual_file.List(d.ID, dir, func(virtualFile model.VirtualFile, dir model.Obj) ([]model.Obj, error) {
+
+		if err := d.WaitLimit(ctx); err != nil {
+			return nil, err
+		}
+
+		parentId := ""
+		if virDir, ok := dir.(*model.ObjVirtualDir); ok {
+			parentId = virDir.VirtualFile.ParentDir
+		} else {
+			parentId = dir.GetID()
+		}
+
+		files := make([]driver115.ShareFile, 0)
+		fileResp, err := d.client.GetShareSnap(virtualFile.ShareID, virtualFile.SharePwd, parentId, driver115.QueryLimit(int(d.PageSize)))
 		if err != nil {
 			return nil, err
 		}
 		files = append(files, fileResp.Data.List...)
-		count += len(fileResp.Data.List)
-	}
+		total := fileResp.Data.Count
+		count := len(fileResp.Data.List)
+		for total > count {
+			fileResp, err := d.client.GetShareSnap(
+				virtualFile.ShareID, virtualFile.SharePwd, dir.GetID(),
+				driver115.QueryLimit(int(d.PageSize)), driver115.QueryOffset(count),
+			)
+			if err != nil {
+				return nil, err
+			}
+			files = append(files, fileResp.Data.List...)
+			count += len(fileResp.Data.List)
+		}
+
+		return utils.SliceConvert(files, func(src driver115.ShareFile) (model.Obj, error) {
+			return transFunc(dir, src)
+		})
+	})
 
-	return utils.SliceConvert(files, transFunc)
 }
 
 func (d *Pan115Share) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
 	if err := d.WaitLimit(ctx); err != nil {
 		return nil, err
 	}
-	downloadInfo, err := d.client.DownloadByShareCode(d.ShareCode, d.ReceiveCode, file.GetID())
+
+	virtualFile := virtual_file.GetSubscription(d.ID, file.GetPath())
+	downloadInfo, err := d.client.DownloadByShareCode(virtualFile.ShareID, virtualFile.SharePwd, file.GetID())
 	if err != nil {
 		return nil, err
 	}
@@ -86,27 +103,31 @@ func (d *Pan115Share) Link(ctx context.Context, file model.Obj, args model.LinkA
 }
 
 func (d *Pan115Share) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
-	return errs.NotSupport
+	return virtual_file.MakeDir(d.ID, parentDir, dirName)
 }
 
 func (d *Pan115Share) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
-	return errs.NotSupport
+	return virtual_file.Move(d.ID, srcObj, dstDir)
 }
 
 func (d *Pan115Share) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
-	return errs.NotSupport
+	return virtual_file.Rename(d.ID, srcObj, newName)
 }
 
 func (d *Pan115Share) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
 	return errs.NotSupport
 }
 
 func (d *Pan115Share) Remove(ctx context.Context, obj model.Obj) error {
-	return errs.NotSupport
+	return virtual_file.DeleteVirtualFile(d.ID, obj)
 }
 
 func (d *Pan115Share) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
 	return errs.NotSupport
 }
 
+func (d *Pan115Share) MkdirConfig() []driver.Item {
+	return virtual_file.GetMkdirConfig()
+}
+
 var _ driver.Driver = (*Pan115Share)(nil)

drivers/115_share/utils.go

@@ -2,6 +2,7 @@ package _115_share
 
 import (
 	"fmt"
+	"path/filepath"
 	"strconv"
 	"time"
 
@@ -20,6 +21,7 @@ type FileObj struct {
 	FileName string
 	isDir    bool
 	FileID   string
+	Path     string
 }
 
 func (f *FileObj) CreateTime() time.Time {
@@ -51,10 +53,10 @@ func (f *FileObj) GetID() string {
 }
 
 func (f *FileObj) GetPath() string {
-	return ""
+	return f.Path
 }
 
-func transFunc(sf driver115.ShareFile) (model.Obj, error) {
+func transFunc(parent model.Obj, sf driver115.ShareFile) (model.Obj, error) {
 	timeInt, err := strconv.ParseInt(sf.UpdateTime, 10, 64)
 	if err != nil {
 		return nil, err
@@ -74,6 +76,7 @@ func transFunc(sf driver115.ShareFile) (model.Obj, error) {
 		FileName: string(sf.FileName),
 		isDir:    isDir,
 		FileID:   fileID,
+		Path:     filepath.Join(parent.GetPath(), sf.FileID),
 	}, nil
 }
 
@@ -85,9 +88,6 @@ func (d *Pan115Share) login() error {
 		driver115.UA(UserAgent),
 	}
 	d.client = driver115.New(opts...)
-	if _, err := d.client.GetShareSnap(d.ShareCode, d.ReceiveCode, ""); err != nil {
-		return errors.Wrap(err, "failed to get share snap")
-	}
 	cr := &driver115.Credential{}
 	if d.QRCodeToken != "" {
 		s := &driver115.QRCodeSession{