From 65c5ec0c34d5f027a65933fe89af53791747bdd4 Mon Sep 17 00:00:00 2001
From: itsHenry <2671230065@qq.com>
Date: Sat, 4 Nov 2023 13:35:09 +0800
Subject: [PATCH] feat(cloudreve): folder size count and switch (#5457 close
 #5395)

---
 drivers/cloudreve/driver.go |  8 ++++++++
 drivers/cloudreve/meta.go   | 11 ++++++-----
 drivers/cloudreve/types.go  |  4 ++++
 drivers/cloudreve/util.go   |  3 +++
 4 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/drivers/cloudreve/driver.go b/drivers/cloudreve/driver.go
index 2a22380e..49c2d5f0 100644
--- a/drivers/cloudreve/driver.go
+++ b/drivers/cloudreve/driver.go
@@ -53,6 +53,14 @@ func (d *Cloudreve) List(ctx context.Context, dir model.Obj, args model.ListArgs
 		if err != nil {
 			return nil, err
 		}
+		if src.Type == "dir" && d.EnableThumbAndFolderSize {
+			var dprop DirectoryProp
+			err = d.request(http.MethodGet, "/object/property/"+src.Id+"?is_folder=true", nil, &dprop)
+			if err != nil {
+				return nil, err
+			}
+			src.Size = dprop.Size
+		}
 		return objectToObj(src, thumb), nil
 	})
 }
diff --git a/drivers/cloudreve/meta.go b/drivers/cloudreve/meta.go
index d01d5479..92c0b9fb 100644
--- a/drivers/cloudreve/meta.go
+++ b/drivers/cloudreve/meta.go
@@ -9,11 +9,12 @@ type Addition struct {
 	// Usually one of two
 	driver.RootPath
 	// define other
-	Address  string `json:"address" required:"true"`
-	Username string `json:"username"`
-	Password string `json:"password"`
-	Cookie   string `json:"cookie"`
-	CustomUA string `json:"custom_ua"`
+	Address                  string `json:"address" required:"true"`
+	Username                 string `json:"username"`
+	Password                 string `json:"password"`
+	Cookie                   string `json:"cookie"`
+	CustomUA                 string `json:"custom_ua"`
+	EnableThumbAndFolderSize bool   `json:"enable_thumb_and_folder_size"`
 }
 
 var config = driver.Config{
diff --git a/drivers/cloudreve/types.go b/drivers/cloudreve/types.go
index e2567382..241d993e 100644
--- a/drivers/cloudreve/types.go
+++ b/drivers/cloudreve/types.go
@@ -44,6 +44,10 @@ type Object struct {
 	SourceEnabled bool      `json:"source_enabled"`
 }
 
+type DirectoryProp struct {
+	Size int `json:"size"`
+}
+
 func objectToObj(f Object, t model.Thumbnail) *model.ObjThumb {
 	return &model.ObjThumb{
 		Object: model.Object{
diff --git a/drivers/cloudreve/util.go b/drivers/cloudreve/util.go
index ed879466..284e3289 100644
--- a/drivers/cloudreve/util.go
+++ b/drivers/cloudreve/util.go
@@ -151,6 +151,9 @@ func convertSrc(obj model.Obj) map[string]interface{} {
 }
 
 func (d *Cloudreve) GetThumb(file Object) (model.Thumbnail, error) {
+	if !d.Addition.EnableThumbAndFolderSize {
+		return model.Thumbnail{}, nil
+	}
 	ua := d.CustomUA
 	if ua == "" {
 		ua = base.UserAgent