feat(lenovonas_share): add option to not show root directory (OpenListTeam#772)

YangRucheng · web-flow · commit 59ec1dbc9bad · 2025-07-21T14:38:10.000+08:00
diff --git a/drivers/lenovonas_share/driver.go b/drivers/lenovonas_share/driver.go
@@ -3,6 +3,7 @@ package LenovoNasShare
 import (
 	"context"
 	"net/http"
+	"strings"
 	"time"
 
 	"github.com/go-resty/resty/v2"
@@ -32,6 +33,10 @@ func (d *LenovoNasShare) Init(ctx context.Context) error {
 	if err := d.getStoken(); err != nil {
 		return err
 	}
+	if !d.ShowRootFolder && d.RootFolderPath == "" {
+		list, _ := d.List(ctx, File{}, model.ListArgs{})
+		d.RootFolderPath = list[0].GetPath()
+	}
 	return nil
 }
 
@@ -43,19 +48,26 @@ func (d *LenovoNasShare) List(ctx context.Context, dir model.Obj, args model.Lis
 	d.checkStoken() // 检查stoken是否过期
 	files := make([]File, 0)
 
+	path := dir.GetPath()
+	if path == "" && !d.ShowRootFolder && d.RootFolderPath != "" {
+		path = d.RootFolderPath
+	}
+
 	var resp Files
 	query := map[string]string{
 		"code":   d.ShareId,
 		"num":    "5000",
 		"stoken": d.stoken,
-		"path":   dir.GetPath(),
+		"path":   path,
 	}
 	_, err := d.request(d.Host+"/oneproxy/api/share/v1/files", http.MethodGet, func(req *resty.Request) {
 		req.SetQueryParams(query)
 	}, &resp)
+
 	if err != nil {
 		return nil, err
 	}
+
 	files = append(files, resp.Data.List...)
 
 	return utils.SliceConvert(files, func(src File) (model.Obj, error) {
@@ -73,6 +85,10 @@ func (d *LenovoNasShare) getStoken() error { // 获取stoken
 	if d.Host == "" {
 		d.Host = "https://siot-share.lenovo.com.cn"
 	}
+
+	parts := strings.Split(d.ShareId, "/")
+	d.ShareId = parts[len(parts)-1]
+
 	query := map[string]string{
 		"code":     d.ShareId,
 		"password": d.SharePwd,
diff --git a/drivers/lenovonas_share/meta.go b/drivers/lenovonas_share/meta.go
@@ -7,9 +7,10 @@ import (
 
 type Addition struct {
 	driver.RootPath
-	ShareId  string `json:"share_id" required:"true" help:"The part after the last / in the shared link"`
-	SharePwd string `json:"share_pwd" required:"true" help:"The password of the shared link"`
-	Host     string `json:"host" required:"true" default:"https://siot-share.lenovo.com.cn" help:"You can change it to your local area network"`
+	ShareId        string `json:"share_id" required:"true" help:"The part after the last / in the shared link"`
+	SharePwd       string `json:"share_pwd" required:"true" help:"The password of the shared link"`
+	Host           string `json:"host" required:"true" default:"https://siot-share.lenovo.com.cn" help:"You can change it to your local area network"`
+	ShowRootFolder bool   `json:"show_root_folder" default:"true"`
 }
 
 var config = driver.Config{
