feat(pikpak): support disk usage (#1426)

KirCute · web-flow · commit 4ba7696032d3 · 2025-10-06T16:37:00.000+08:00
* feat(pikpak): support disk usage

* fix(alias): cannot list with details

* refactor: rename `NewDiskUsageFromUsedAndTotal`

* fix(disk-usage): get details of storages that is not initialized
diff --git a/drivers/123/driver.go b/drivers/123/driver.go
@@ -260,7 +260,7 @@ func (d *Pan123) GetDetails(ctx context.Context) (*model.StorageDetails, error)
 	}
 	total := userInfo.Data.SpacePermanent + userInfo.Data.SpaceTemp
 	return &model.StorageDetails{
-		DiskUsage: *model.NewDiskUsageFromUsedAndTotal(userInfo.Data.SpaceUsed, total),
+		DiskUsage: driver.DiskUsageFromUsedAndTotal(userInfo.Data.SpaceUsed, total),
 	}, nil
 }
 
diff --git a/drivers/alias/driver.go b/drivers/alias/driver.go
@@ -146,22 +146,27 @@ func (d *Alias) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([
 		})
 		if err == nil {
 			tmp, err = utils.SliceConvert(tmp, func(obj model.Obj) (model.Obj, error) {
-				thumb, ok := model.GetThumb(obj)
 				objRes := model.Object{
 					Name:     obj.GetName(),
 					Size:     obj.GetSize(),
 					Modified: obj.ModTime(),
 					IsFolder: obj.IsDir(),
 				}
-				if !ok {
-					return &objRes, nil
+				if thumb, ok := model.GetThumb(obj); ok {
+					return &model.ObjThumb{
+						Object: objRes,
+						Thumbnail: model.Thumbnail{
+							Thumbnail: thumb,
+						},
+					}, nil
 				}
-				return &model.ObjThumb{
-					Object: objRes,
-					Thumbnail: model.Thumbnail{
-						Thumbnail: thumb,
-					},
-				}, nil
+				if details, ok := model.GetStorageDetails(obj); ok {
+					return &model.ObjStorageDetails{
+						Obj:                    &objRes,
+						StorageDetailsWithName: *details,
+					}, nil
+				}
+				return &objRes, nil
 			})
 		}
 		if err == nil {
diff --git a/drivers/alias/util.go b/drivers/alias/util.go
@@ -6,6 +6,7 @@ import (
 	stdpath "path"
 	"strings"
 	"sync"
+	"time"
 
 	"github.com/OpenListTeam/OpenList/v4/internal/driver"
 	"github.com/OpenListTeam/OpenList/v4/internal/errs"
@@ -49,9 +50,11 @@ func (d *Alias) listRoot(ctx context.Context, withDetails bool) []model.Obj {
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
-			details, e := op.GetStorageDetails(ctx, remoteDriver)
+			c, cancel := context.WithTimeout(ctx, time.Second)
+			defer cancel()
+			details, e := op.GetStorageDetails(c, remoteDriver)
 			if e != nil {
-				if !errors.Is(e, errs.NotImplement) {
+				if !errors.Is(e, errs.NotImplement) && !errors.Is(e, errs.StorageNotInit) {
 					log.Errorf("failed get %s storage details: %+v", remoteDriver.GetStorage().MountPath, e)
 				}
 				return
diff --git a/drivers/aliyundrive/driver.go b/drivers/aliyundrive/driver.go
@@ -337,7 +337,7 @@ func (d *AliDrive) GetDetails(ctx context.Context) (*model.StorageDetails, error
 	used := utils.Json.Get(res, "drive_used_size").ToUint64()
 	total := utils.Json.Get(res, "drive_total_size").ToUint64()
 	return &model.StorageDetails{
-		DiskUsage: *model.NewDiskUsageFromUsedAndTotal(used, total),
+		DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),
 	}, nil
 }
 
diff --git a/drivers/baidu_netdisk/driver.go b/drivers/baidu_netdisk/driver.go
@@ -369,7 +369,7 @@ func (d *BaiduNetdisk) GetDetails(ctx context.Context) (*model.StorageDetails, e
 	if err != nil {
 		return nil, err
 	}
-	return &model.StorageDetails{DiskUsage: *du}, nil
+	return &model.StorageDetails{DiskUsage: du}, nil
 }
 
 var _ driver.Driver = (*BaiduNetdisk)(nil)
diff --git a/drivers/baidu_netdisk/util.go b/drivers/baidu_netdisk/util.go
@@ -12,6 +12,7 @@ import (
 	"unicode"
 
 	"github.com/OpenListTeam/OpenList/v4/drivers/base"
+	"github.com/OpenListTeam/OpenList/v4/internal/driver"
 	"github.com/OpenListTeam/OpenList/v4/internal/errs"
 	"github.com/OpenListTeam/OpenList/v4/internal/model"
 	"github.com/OpenListTeam/OpenList/v4/internal/op"
@@ -382,15 +383,15 @@ func (d *BaiduNetdisk) getSliceSize(filesize int64) int64 {
 	return maxSliceSize
 }
 
-func (d *BaiduNetdisk) quota(ctx context.Context) (*model.DiskUsage, error) {
+func (d *BaiduNetdisk) quota(ctx context.Context) (model.DiskUsage, error) {
 	var resp QuotaResp
 	_, err := d.request("https://pan.baidu.com/api/quota", http.MethodGet, func(req *resty.Request) {
 		req.SetContext(ctx)
 	}, &resp)
 	if err != nil {
-		return nil, err
+		return model.DiskUsage{}, err
 	}
-	return model.NewDiskUsageFromUsedAndTotal(resp.Used, resp.Total), nil
+	return driver.DiskUsageFromUsedAndTotal(resp.Used, resp.Total), nil
 }
 
 // func encodeURIComponent(str string) string {
diff --git a/drivers/cloudreve_v4/driver.go b/drivers/cloudreve_v4/driver.go
@@ -349,7 +349,7 @@ func (d *CloudreveV4) GetDetails(ctx context.Context) (*model.StorageDetails, er
 		return nil, err
 	}
 	return &model.StorageDetails{
-		DiskUsage: *model.NewDiskUsageFromUsedAndTotal(r.Used, r.Total),
+		DiskUsage: driver.DiskUsageFromUsedAndTotal(r.Used, r.Total),
 	}, nil
 }
 
diff --git a/drivers/google_drive/driver.go b/drivers/google_drive/driver.go
@@ -189,7 +189,7 @@ func (d *GoogleDrive) GetDetails(ctx context.Context) (*model.StorageDetails, er
 		return nil, err
 	}
 	return &model.StorageDetails{
-		DiskUsage: *model.NewDiskUsageFromUsedAndTotal(used, total),
+		DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),
 	}, nil
 }
 
diff --git a/drivers/ilanzou/driver.go b/drivers/ilanzou/driver.go
@@ -414,7 +414,7 @@ func (d *ILanZou) GetDetails(ctx context.Context) (*model.StorageDetails, error)
 	total := totalSize + rewardSize
 	used := utils.Json.Get(res, "map", "usedSize").ToUint64() * 1024
 	return &model.StorageDetails{
-		DiskUsage: *model.NewDiskUsageFromUsedAndTotal(used, total),
+		DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),
 	}, nil
 }
 
diff --git a/drivers/pikpak/driver.go b/drivers/pikpak/driver.go
@@ -36,7 +36,6 @@ func (d *PikPak) GetAddition() driver.Additional {
 }
 
 func (d *PikPak) Init(ctx context.Context) (err error) {
-
 	if d.Common == nil {
 		d.Common = &Common{
 			client:       base.NewRestyClient(),
@@ -247,7 +246,7 @@ func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
 	}
 
 	params := resp.Resumable.Params
-	//endpoint := strings.Join(strings.Split(params.Endpoint, ".")[1:], ".")
+	// endpoint := strings.Join(strings.Split(params.Endpoint, ".")[1:], ".")
 	// web 端上传 返回的endpoint 为 `mypikpak.net` | android 端上传 返回的endpoint 为 `vip-lixian-07.mypikpak.net`·
 	if d.Addition.Platform == "android" {
 		params.Endpoint = "mypikpak.net"
@@ -260,6 +259,27 @@ func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStr
 	return d.UploadByMultipart(ctx, &params, stream.GetSize(), stream, up)
 }
 
+func (d *PikPak) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	var about AboutResponse
+	_, err := d.request("https://api-drive.mypikpak.com/drive/v1/about", http.MethodGet, func(req *resty.Request) {
+		req.SetContext(ctx)
+	}, &about)
+	if err != nil {
+		return nil, err
+	}
+	total, err := strconv.ParseUint(about.Quota.Limit, 10, 64)
+	if err != nil {
+		return nil, err
+	}
+	used, err := strconv.ParseUint(about.Quota.Usage, 10, 64)
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{
+		DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),
+	}, nil
+}
+
 // 离线下载文件
 func (d *PikPak) OfflineDownload(ctx context.Context, fileUrl string, parentDir model.Obj, fileName string) (*OfflineTask, error) {
 	requestBody := base.Json{
@@ -278,7 +298,6 @@ func (d *PikPak) OfflineDownload(ctx context.Context, fileUrl string, parentDir
 		req.SetContext(ctx).
 			SetBody(requestBody)
 	}, &resp)
-
 	if err != nil {
 		return nil, err
 	}
@@ -325,7 +344,6 @@ func (d *PikPak) OfflineList(ctx context.Context, nextPageToken string, phase []
 		req.SetContext(ctx).
 			SetQueryParams(params)
 	}, &resp)
-
 	if err != nil {
 		return nil, fmt.Errorf("failed to get offline list: %w", err)
 	}
diff --git a/drivers/pikpak/types.go b/drivers/pikpak/types.go
@@ -78,7 +78,7 @@ type Media struct {
 
 type UploadTaskData struct {
 	UploadType string `json:"upload_type"`
-	//UPLOAD_TYPE_RESUMABLE
+	// UPLOAD_TYPE_RESUMABLE
 	Resumable *struct {
 		Kind     string   `json:"kind"`
 		Params   S3Params `json:"params"`
@@ -195,3 +195,15 @@ type CaptchaTokenResponse struct {
 	ExpiresIn    int64  `json:"expires_in"`
 	Url          string `json:"url"`
 }
+
+type AboutResponse struct {
+	Quota struct {
+		Limit         string `json:"limit"`
+		Usage         string `json:"usage"`
+		UsageInTrash  string `json:"usage_in_trash"`
+		IsUnlimited   bool   `json:"is_unlimited"`
+		Complimentary string `json:"complimentary"`
+	} `json:"quota"`
+	ExpiresAt string `json:"expires_at"`
+	UserType  int    `json:"user_type"`
+}
diff --git a/internal/driver/utils.go b/internal/driver/utils.go
@@ -20,7 +20,7 @@ func (p *Progress) Write(b []byte) (n int, err error) {
 	n = len(b)
 	p.Done += int64(n)
 	p.up(float64(p.Done) / float64(p.Total) * 100)
-	return
+	return n, err
 }
 
 func NewProgress(total int64, up UpdateProgress) *Progress {
@@ -61,3 +61,10 @@ type ReaderWithCtx = stream.ReaderWithCtx
 type ReaderUpdatingProgress = stream.ReaderUpdatingProgress
 
 type SimpleReaderWithSize = stream.SimpleReaderWithSize
+
+func DiskUsageFromUsedAndTotal(used, total uint64) model.DiskUsage {
+	return model.DiskUsage{
+		TotalSpace: max(used, total),
+		FreeSpace:  total - min(used, total),
+	}
+}
diff --git a/internal/errs/errors.go b/internal/errs/errors.go
@@ -12,13 +12,12 @@ var (
 	NotSupport   = errors.New("not support")
 	RelativePath = errors.New("using relative path is not allowed")
 
-	MoveBetweenTwoStorages = errors.New("can't move files between two storages, try to copy")
-	UploadNotSupported     = errors.New("upload not supported")
-
-	MetaNotFound     = errors.New("meta not found")
-	StorageNotFound  = errors.New("storage not found")
-	StreamIncomplete = errors.New("upload/download stream incomplete, possible network issue")
-	StreamPeekFail   = errors.New("StreamPeekFail")
+	UploadNotSupported = errors.New("upload not supported")
+	MetaNotFound       = errors.New("meta not found")
+	StorageNotFound    = errors.New("storage not found")
+	StorageNotInit     = errors.New("storage not init")
+	StreamIncomplete   = errors.New("upload/download stream incomplete, possible network issue")
+	StreamPeekFail     = errors.New("StreamPeekFail")
 
 	UnknownArchiveFormat      = errors.New("unknown archive format")
 	WrongArchivePassword      = errors.New("wrong archive password")
diff --git a/internal/model/storage.go b/internal/model/storage.go
@@ -32,7 +32,7 @@ type Proxy struct {
 	WebdavPolicy string `json:"webdav_policy"`
 	ProxyRange   bool   `json:"proxy_range"`
 	DownProxyURL string `json:"down_proxy_url"`
-	//Disable sign for DownProxyURL
+	// Disable sign for DownProxyURL
 	DisableProxySign bool `json:"disable_proxy_sign"`
 }
 
@@ -61,13 +61,6 @@ type DiskUsage struct {
 	FreeSpace  uint64 `json:"free_space"`
 }
 
-func NewDiskUsageFromUsedAndTotal(used, total uint64) *DiskUsage {
-	return &DiskUsage{
-		TotalSpace: max(used, total),
-		FreeSpace: total - min(used, total),
-	}
-}
-
 type StorageDetails struct {
 	DiskUsage
 }
diff --git a/internal/op/archive.go b/internal/op/archive.go
@@ -27,7 +27,7 @@ var archiveMetaG singleflight.Group[*model.ArchiveMetaProvider]
 
 func GetArchiveMeta(ctx context.Context, storage driver.Driver, path string, args model.ArchiveMetaArgs) (*model.ArchiveMetaProvider, error) {
 	if storage.Config().CheckStatus && storage.GetStorage().Status != WORK {
-		return nil, errors.Errorf("storage not init: %s", storage.GetStorage().Status)
+		return nil, errors.WithMessagef(errs.StorageNotInit, "storage status: %s", storage.GetStorage().Status)
 	}
 	path = utils.FixAndCleanPath(path)
 	key := Key(storage, path)
@@ -163,7 +163,7 @@ var archiveListG singleflight.Group[[]model.Obj]
 
 func ListArchive(ctx context.Context, storage driver.Driver, path string, args model.ArchiveListArgs) ([]model.Obj, error) {
 	if storage.Config().CheckStatus && storage.GetStorage().Status != WORK {
-		return nil, errors.Errorf("storage not init: %s", storage.GetStorage().Status)
+		return nil, errors.WithMessagef(errs.StorageNotInit, "storage status: %s", storage.GetStorage().Status)
 	}
 	path = utils.FixAndCleanPath(path)
 	metaKey := Key(storage, path)
@@ -309,7 +309,7 @@ func splitPath(path string) []string {
 
 func ArchiveGet(ctx context.Context, storage driver.Driver, path string, args model.ArchiveListArgs) (model.Obj, model.Obj, error) {
 	if storage.Config().CheckStatus && storage.GetStorage().Status != WORK {
-		return nil, nil, errors.Errorf("storage not init: %s", storage.GetStorage().Status)
+		return nil, nil, errors.WithMessagef(errs.StorageNotInit, "storage status: %s", storage.GetStorage().Status)
 	}
 	path = utils.FixAndCleanPath(path)
 	af, err := GetUnwrap(ctx, storage, path)
@@ -364,7 +364,7 @@ var extractG = singleflight.Group[*extractLink]{Remember: true}
 
 func DriverExtract(ctx context.Context, storage driver.Driver, path string, args model.ArchiveInnerArgs) (*model.Link, model.Obj, error) {
 	if storage.Config().CheckStatus && storage.GetStorage().Status != WORK {
-		return nil, nil, errors.Errorf("storage not init: %s", storage.GetStorage().Status)
+		return nil, nil, errors.WithMessagef(errs.StorageNotInit, "storage status: %s", storage.GetStorage().Status)
 	}
 	key := stdpath.Join(Key(storage, path), args.InnerPath)
 	if link, ok := extractCache.Get(key); ok {
@@ -480,7 +480,7 @@ func InternalExtract(ctx context.Context, storage driver.Driver, path string, ar
 
 func ArchiveDecompress(ctx context.Context, storage driver.Driver, srcPath, dstDirPath string, args model.ArchiveDecompressArgs, lazyCache ...bool) error {
 	if storage.Config().CheckStatus && storage.GetStorage().Status != WORK {
-		return errors.Errorf("storage not init: %s", storage.GetStorage().Status)
+		return errors.WithMessagef(errs.StorageNotInit, "storage status: %s", storage.GetStorage().Status)
 	}
 	srcPath = utils.FixAndCleanPath(srcPath)
 	dstDirPath = utils.FixAndCleanPath(dstDirPath)
diff --git a/internal/op/fs.go b/internal/op/fs.go
diff --git a/internal/op/storage.go b/internal/op/storage.go
diff --git a/server/handles/storage.go b/server/handles/storage.go

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,7 @@ func (d Pan123) GetDetails(ctx context.Context) (model.StorageDetails, error)`
`260`	`260`	`}`
`261`	`261`	`total := userInfo.Data.SpacePermanent + userInfo.Data.SpaceTemp`
`262`	`262`	`return &model.StorageDetails{`
`263`		`- DiskUsage: *model.NewDiskUsageFromUsedAndTotal(userInfo.Data.SpaceUsed, total),`
	`263`	`+ DiskUsage: driver.DiskUsageFromUsedAndTotal(userInfo.Data.SpaceUsed, total),`
`264`	`264`	`}, nil`
`265`	`265`	`}`
`266`	`266`
Original file line number	Diff line number	Diff line change
`@@ -337,7 +337,7 @@ func (d AliDrive) GetDetails(ctx context.Context) (model.StorageDetails, error`
`337`	`337`	`used := utils.Json.Get(res, "drive_used_size").ToUint64()`
`338`	`338`	`total := utils.Json.Get(res, "drive_total_size").ToUint64()`
`339`	`339`	`return &model.StorageDetails{`
`340`		`- DiskUsage: *model.NewDiskUsageFromUsedAndTotal(used, total),`
	`340`	`+ DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),`
`341`	`341`	`}, nil`
`342`	`342`	`}`
`343`	`343`
Original file line number	Diff line number	Diff line change
`@@ -369,7 +369,7 @@ func (d BaiduNetdisk) GetDetails(ctx context.Context) (model.StorageDetails, e`
`369`	`369`	`if err != nil {`
`370`	`370`	`return nil, err`
`371`	`371`	`}`
`372`		`- return &model.StorageDetails{DiskUsage: *du}, nil`
	`372`	`+ return &model.StorageDetails{DiskUsage: du}, nil`
`373`	`373`	`}`
`374`	`374`
`375`	`375`	`var _ driver.Driver = (*BaiduNetdisk)(nil)`
Original file line number	Diff line number	Diff line change
`@@ -349,7 +349,7 @@ func (d CloudreveV4) GetDetails(ctx context.Context) (model.StorageDetails, er`
`349`	`349`	`return nil, err`
`350`	`350`	`}`
`351`	`351`	`return &model.StorageDetails{`
`352`		`- DiskUsage: *model.NewDiskUsageFromUsedAndTotal(r.Used, r.Total),`
	`352`	`+ DiskUsage: driver.DiskUsageFromUsedAndTotal(r.Used, r.Total),`
`353`	`353`	`}, nil`
`354`	`354`	`}`
`355`	`355`
Original file line number	Diff line number	Diff line change
`@@ -189,7 +189,7 @@ func (d GoogleDrive) GetDetails(ctx context.Context) (model.StorageDetails, er`
`189`	`189`	`return nil, err`
`190`	`190`	`}`
`191`	`191`	`return &model.StorageDetails{`
`192`		`- DiskUsage: *model.NewDiskUsageFromUsedAndTotal(used, total),`
	`192`	`+ DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),`
`193`	`193`	`}, nil`
`194`	`194`	`}`
`195`	`195`
Original file line number	Diff line number	Diff line change
`@@ -414,7 +414,7 @@ func (d ILanZou) GetDetails(ctx context.Context) (model.StorageDetails, error)`
`414`	`414`	`total := totalSize + rewardSize`
`415`	`415`	`used := utils.Json.Get(res, "map", "usedSize").ToUint64() * 1024`
`416`	`416`	`return &model.StorageDetails{`
`417`		`- DiskUsage: *model.NewDiskUsageFromUsedAndTotal(used, total),`
	`417`	`+ DiskUsage: driver.DiskUsageFromUsedAndTotal(used, total),`
`418`	`418`	`}, nil`
`419`	`419`	`}`
`420`	`420`