feat(style): add driver icons and disk usage (#1274)

KirCute · xrgzs · web-flow · commit cc16cb35bf67 · 2025-09-19T11:59:11.000+08:00
* feat(style): add driver icons and disk usage

* feat(driver): add disk usage for 115_open, 123_open, aliyundrive_open and baidu_netdisk

* feat(driver): add disk usage for crypt, sftp and smb

* chore: clean unused variable

* feat(driver): add disk usage for cloudreve_v4

Signed-off-by: MadDogOwner &lt;xiaoran@xrgzs.top&gt;

* fix(local): disk label check when getting disk usage

* feat(style): return details when accessing the manage page

---------

Signed-off-by: MadDogOwner &lt;xiaoran@xrgzs.top&gt;
Co-authored-by: MadDogOwner &lt;xiaoran@xrgzs.top&gt;
diff --git a/drivers/115_open/driver.go b/drivers/115_open/driver.go
@@ -337,6 +337,27 @@ func (d *Open115) OfflineList(ctx context.Context) (*sdk.OfflineTaskListResp, er
 	return resp, nil
 }
 
+func (d *Open115) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	userInfo, err := d.client.UserInfo(ctx)
+	if err != nil {
+		return nil, err
+	}
+	total, err := userInfo.RtSpaceInfo.AllTotal.Size.Int64()
+	if err != nil {
+		return nil, err
+	}
+	free, err := userInfo.RtSpaceInfo.AllRemain.Size.Int64()
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{
+		DiskUsage: model.DiskUsage{
+			TotalSpace: uint64(total),
+			FreeSpace:  uint64(free),
+		},
+	}, nil
+}
+
 // func (d *Open115) GetArchiveMeta(ctx context.Context, obj model.Obj, args model.ArchiveArgs) (model.ArchiveMeta, error) {
 // 	// TODO get archive file meta-info, return errs.NotImplement to use an internal archive tool, optional
 // 	return nil, errs.NotImplement
diff --git a/drivers/123_open/driver.go b/drivers/123_open/driver.go
@@ -214,5 +214,20 @@ func (d *Open123) Put(ctx context.Context, dstDir model.Obj, file model.FileStre
 	return nil, fmt.Errorf("upload complete timeout")
 }
 
+func (d *Open123) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	userInfo, err := d.getUserInfo()
+	if err != nil {
+		return nil, err
+	}
+	total := userInfo.Data.SpacePermanent + userInfo.Data.SpaceTemp
+	free := total - userInfo.Data.SpaceUsed
+	return &model.StorageDetails{
+		DiskUsage: model.DiskUsage{
+			TotalSpace: total,
+			FreeSpace:  free,
+		},
+	}, nil
+}
+
 var _ driver.Driver = (*Open123)(nil)
 var _ driver.PutResult = (*Open123)(nil)
diff --git a/drivers/123_open/types.go b/drivers/123_open/types.go
@@ -133,9 +133,9 @@ type UserInfoResp struct {
 		// HeadImage      string `json:"headImage"`
 		// Passport       string `json:"passport"`
 		// Mail           string `json:"mail"`
-		// SpaceUsed      int64  `json:"spaceUsed"`
-		// SpacePermanent int64  `json:"spacePermanent"`
-		// SpaceTemp      int64  `json:"spaceTemp"`
+		SpaceUsed      uint64 `json:"spaceUsed"`
+		SpacePermanent uint64 `json:"spacePermanent"`
+		SpaceTemp      uint64 `json:"spaceTemp"`
 		// SpaceTempExpr  int64  `json:"spaceTempExpr"`
 		// Vip            bool   `json:"vip"`
 		// DirectTraffic  int64  `json:"directTraffic"`
diff --git a/drivers/aliyundrive_open/driver.go b/drivers/aliyundrive_open/driver.go
@@ -291,6 +291,21 @@ func (d *AliyundriveOpen) Other(ctx context.Context, args model.OtherArgs) (inte
 	return resp, nil
 }
 
+func (d *AliyundriveOpen) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	res, err := d.request(ctx, limiterOther, "/adrive/v1.0/user/getSpaceInfo", http.MethodPost, nil)
+	if err != nil {
+		return nil, err
+	}
+	total := utils.Json.Get(res, "personal_space_info", "total_size").ToUint64()
+	used := utils.Json.Get(res, "personal_space_info", "used_size").ToUint64()
+	return &model.StorageDetails{
+		DiskUsage: model.DiskUsage{
+			TotalSpace: total,
+			FreeSpace:  total - used,
+		},
+	}, nil
+}
+
 var _ driver.Driver = (*AliyundriveOpen)(nil)
 var _ driver.MkdirResult = (*AliyundriveOpen)(nil)
 var _ driver.MoveResult = (*AliyundriveOpen)(nil)
diff --git a/drivers/baidu_netdisk/driver.go b/drivers/baidu_netdisk/driver.go
@@ -364,4 +364,12 @@ func (d *BaiduNetdisk) uploadSlice(ctx context.Context, params map[string]string
 	return nil
 }
 
+func (d *BaiduNetdisk) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	du, err := d.quota()
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{DiskUsage: *du}, nil
+}
+
 var _ driver.Driver = (*BaiduNetdisk)(nil)
diff --git a/drivers/baidu_netdisk/types.go b/drivers/baidu_netdisk/types.go
@@ -189,3 +189,12 @@ type PrecreateResp struct {
 	// return_type=2
 	File File `json:"info"`
 }
+
+type QuotaResp struct {
+	Errno     int    `json:"errno"`
+	RequestId int64  `json:"request_id"`
+	Total     uint64 `json:"total"`
+	Used      uint64 `json:"used"`
+	//Free      uint64 `json:"free"`
+	//Expire    bool   `json:"expire"`
+}
diff --git a/drivers/baidu_netdisk/util.go b/drivers/baidu_netdisk/util.go
@@ -381,6 +381,18 @@ func (d *BaiduNetdisk) getSliceSize(filesize int64) int64 {
 	return maxSliceSize
 }
 
+func (d *BaiduNetdisk) quota() (*model.DiskUsage, error) {
+	var resp QuotaResp
+	_, err := d.request("https://pan.baidu.com/api/quota", http.MethodGet, nil, &resp)
+	if err != nil {
+		return nil, err
+	}
+	return &model.DiskUsage{
+		TotalSpace: resp.Total,
+		FreeSpace:  resp.Total - resp.Used,
+	}, nil
+}
+
 // func encodeURIComponent(str string) string {
 // 	r := url.QueryEscape(str)
 // 	r = strings.ReplaceAll(r, "+", "%20")
diff --git a/drivers/cloudreve_v4/driver.go b/drivers/cloudreve_v4/driver.go
@@ -339,6 +339,21 @@ func (d *CloudreveV4) ArchiveDecompress(ctx context.Context, srcObj, dstDir mode
 	return nil, errs.NotImplement
 }
 
+func (d *CloudreveV4) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	// TODO return storage details (total space, free space, etc.)
+	var r CapacityResp
+	err := d.request(http.MethodGet, "/user/capacity", nil, &r)
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{
+		DiskUsage: model.DiskUsage{
+			TotalSpace: r.Total,
+			FreeSpace:  r.Total - r.Used,
+		},
+	}, nil
+}
+
 //func (d *CloudreveV4) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
 //	return nil, errs.NotSupport
 //}
diff --git a/drivers/cloudreve_v4/types.go b/drivers/cloudreve_v4/types.go
@@ -204,3 +204,9 @@ type FolderSummaryResp struct {
 		CalculatedAt time.Time `json:"calculated_at"`
 	} `json:"folder_summary"`
 }
+
+type CapacityResp struct {
+	Total uint64 `json:"total"`
+	Used  uint64 `json:"used"`
+	// StoragePackTotal uint64 `json:"storage_pack_total"`
+}
diff --git a/drivers/crypt/driver.go b/drivers/crypt/driver.go
@@ -411,6 +411,20 @@ func (d *Crypt) Put(ctx context.Context, dstDir model.Obj, streamer model.FileSt
 	return nil
 }
 
+func (d *Crypt) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	wd, ok := d.remoteStorage.(driver.WithDetails)
+	if !ok {
+		return nil, errs.NotImplement
+	}
+	remoteDetails, err := wd.GetDetails(ctx)
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{
+		DiskUsage: remoteDetails.DiskUsage,
+	}, nil
+}
+
 //func (d *Safe) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
 //	return nil, errs.NotSupport
 //}
diff --git a/drivers/local/driver.go b/drivers/local/driver.go
@@ -434,4 +434,14 @@ func (d *Local) Put(ctx context.Context, dstDir model.Obj, stream model.FileStre
 	return nil
 }
 
+func (d *Local) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	du, err := getDiskUsage(d.RootFolderPath)
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{
+		DiskUsage: du,
+	}, nil
+}
+
 var _ driver.Driver = (*Local)(nil)
diff --git a/drivers/local/util_unix.go b/drivers/local/util_unix.go
@@ -5,8 +5,25 @@ package local
 import (
 	"io/fs"
 	"strings"
+	"syscall"
+
+	"github.com/OpenListTeam/OpenList/v4/internal/model"
 )
 
 func isHidden(f fs.FileInfo, _ string) bool {
 	return strings.HasPrefix(f.Name(), ".")
 }
+
+func getDiskUsage(path string) (model.DiskUsage, error) {
+	var stat syscall.Statfs_t
+	err := syscall.Statfs(path, &stat)
+	if err != nil {
+		return model.DiskUsage{}, err
+	}
+	total := stat.Blocks * uint64(stat.Bsize)
+	free := stat.Bfree * uint64(stat.Bsize)
+	return model.DiskUsage{
+		TotalSpace: total,
+		FreeSpace:  free,
+	}, nil
+}
diff --git a/drivers/local/util_windows.go b/drivers/local/util_windows.go
@@ -1,22 +1,51 @@
-//go:build windows
-
-package local
-
-import (
-	"io/fs"
-	"path/filepath"
-	"syscall"
-)
-
-func isHidden(f fs.FileInfo, fullPath string) bool {
-	filePath := filepath.Join(fullPath, f.Name())
-	namePtr, err := syscall.UTF16PtrFromString(filePath)
-	if err != nil {
-		return false
-	}
-	attrs, err := syscall.GetFileAttributes(namePtr)
-	if err != nil {
-		return false
-	}
-	return attrs&syscall.FILE_ATTRIBUTE_HIDDEN != 0
-}
+//go:build windows
+
+package local
+
+import (
+	"errors"
+	"io/fs"
+	"path/filepath"
+	"syscall"
+
+	"github.com/OpenListTeam/OpenList/v4/internal/model"
+	"golang.org/x/sys/windows"
+)
+
+func isHidden(f fs.FileInfo, fullPath string) bool {
+	filePath := filepath.Join(fullPath, f.Name())
+	namePtr, err := syscall.UTF16PtrFromString(filePath)
+	if err != nil {
+		return false
+	}
+	attrs, err := syscall.GetFileAttributes(namePtr)
+	if err != nil {
+		return false
+	}
+	return attrs&syscall.FILE_ATTRIBUTE_HIDDEN != 0
+}
+
+func getDiskUsage(path string) (model.DiskUsage, error) {
+	abs, err := filepath.Abs(path)
+	if err != nil {
+		return model.DiskUsage{}, err
+	}
+	root := filepath.VolumeName(abs)
+	if len(root) != 2 || root[1] != ':' {
+		return model.DiskUsage{}, errors.New("cannot get disk label")
+	}
+	var freeBytes, totalBytes, totalFreeBytes uint64
+	err = windows.GetDiskFreeSpaceEx(
+		windows.StringToUTF16Ptr(root),
+		&freeBytes,
+		&totalBytes,
+		&totalFreeBytes,
+	)
+	if err != nil {
+		return model.DiskUsage{}, err
+	}
+	return model.DiskUsage{
+		TotalSpace: totalBytes,
+		FreeSpace:  freeBytes,
+	}, nil
+}
diff --git a/drivers/sftp/driver.go b/drivers/sftp/driver.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"os"
 	"path"
+	"strings"
 
 	"github.com/OpenListTeam/OpenList/v4/internal/driver"
 	"github.com/OpenListTeam/OpenList/v4/internal/errs"
@@ -127,4 +128,22 @@ func (d *SFTP) Put(ctx context.Context, dstDir model.Obj, stream model.FileStrea
 	return err
 }
 
+func (d *SFTP) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	stat, err := d.client.StatVFS(d.RootFolderPath)
+	if err != nil {
+		if strings.Contains(err.Error(), "unimplemented") {
+			return nil, errs.NotImplement
+		}
+		return nil, err
+	}
+	total := stat.Blocks * stat.Bsize
+	free := stat.Bfree * stat.Bsize
+	return &model.StorageDetails{
+		DiskUsage: model.DiskUsage{
+			TotalSpace: total,
+			FreeSpace:  free,
+		},
+	}, nil
+}
+
 var _ driver.Driver = (*SFTP)(nil)
diff --git a/drivers/smb/driver.go b/drivers/smb/driver.go
@@ -205,6 +205,22 @@ func (d *SMB) Put(ctx context.Context, dstDir model.Obj, stream model.FileStream
 	return nil
 }
 
+func (d *SMB) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	if err := d.checkConn(); err != nil {
+		return nil, err
+	}
+	stat, err := d.fs.Statfs(d.RootFolderPath)
+	if err != nil {
+		return nil, err
+	}
+	return &model.StorageDetails{
+		DiskUsage: model.DiskUsage{
+			TotalSpace: stat.BlockSize() * stat.TotalBlockCount(),
+			FreeSpace:  stat.BlockSize() * stat.AvailableBlockCount(),
+		},
+	}, nil
+}
+
 //func (d *SMB) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
 //	return nil, errs.NotSupport
 //}
diff --git a/drivers/template/driver.go b/drivers/template/driver.go
@@ -93,6 +93,11 @@ func (d *Template) ArchiveDecompress(ctx context.Context, srcObj, dstDir model.O
 	return nil, errs.NotImplement
 }
 
+func (d *Template) GetDetails(ctx context.Context) (*model.StorageDetails, error) {
+	// TODO return storage details (total space, free space, etc.)
+	return nil, errs.NotImplement
+}
+
 //func (d *Template) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
 //	return nil, errs.NotSupport
 //}
diff --git a/internal/bootstrap/data/setting.go b/internal/bootstrap/data/setting.go
@@ -114,6 +114,7 @@ func InitialSettings() []model.SettingItem {
 		{Key: "share_icon", Value: "🎁", Type: conf.TypeString, Group: model.STYLE},
 		{Key: "home_container", Value: "max_980px", Type: conf.TypeSelect, Options: "max_980px,hope_container", Group: model.STYLE},
 		{Key: "settings_layout", Value: "list", Type: conf.TypeSelect, Options: "list,responsive", Group: model.STYLE},
+		{Key: conf.HideStorageDetails, Value: "false", Type: conf.TypeBool, Group: model.STYLE, Flag: model.PRIVATE},
 		// preview settings
 		{Key: conf.TextTypes, Value: "txt,htm,html,xml,java,properties,sql,js,md,json,conf,ini,vue,php,py,bat,gitignore,yml,go,sh,c,cpp,h,hpp,tsx,vtt,srt,ass,rs,lrc,strm", Type: conf.TypeText, Group: model.PREVIEW, Flag: model.PRIVATE},
 		{Key: conf.AudioTypes, Value: "mp3,flac,ogg,m4a,wav,opus,wma", Type: conf.TypeText, Group: model.PREVIEW, Flag: model.PRIVATE},
diff --git a/internal/conf/const.go b/internal/conf/const.go
@@ -17,9 +17,10 @@ const (
 	AllowMounted = "allow_mounted"
 	RobotsTxt    = "robots_txt"
 
-	Logo      = "logo" // multi-lines text, L1: light, EOL: dark
-	Favicon   = "favicon"
-	MainColor = "main_color"
+	Logo               = "logo" // multi-lines text, L1: light, EOL: dark
+	Favicon            = "favicon"
+	MainColor          = "main_color"
+	HideStorageDetails = "hide_storage_details"
 
 	// preview
 	TextTypes                = "text_types"
diff --git a/internal/driver/driver.go b/internal/driver/driver.go
@@ -210,6 +210,11 @@ type ArchiveDecompressResult interface {
 	ArchiveDecompress(ctx context.Context, srcObj, dstDir model.Obj, args model.ArchiveDecompressArgs) ([]model.Obj, error)
 }
 
+type WithDetails interface {
+	// GetDetails get storage details (total space, free space, etc.)
+	GetDetails(ctx context.Context) (*model.StorageDetails, error)
+}
+
 type Reference interface {
 	InitReference(storage Driver) error
 }
diff --git a/internal/fs/fs.go b/internal/fs/fs.go
diff --git a/internal/fs/get.go b/internal/fs/get.go
diff --git a/internal/fs/list.go b/internal/fs/list.go
diff --git a/internal/model/storage.go b/internal/model/storage.go
diff --git a/internal/op/storage.go b/internal/op/storage.go
diff --git a/server/handles/fsread.go b/server/handles/fsread.go
diff --git a/server/handles/storage.go b/server/handles/storage.go
