feat(cloudreve): support reference

drivers/cloudreve/driver.go

@@ -18,6 +18,7 @@ import (
 type Cloudreve struct {
 	model.Storage
 	Addition
+	ref *Cloudreve
 }
 
 func (d *Cloudreve) Config() driver.Config {
@@ -37,8 +38,18 @@ func (d *Cloudreve) Init(ctx context.Context) error {
 	return d.login()
 }
 
+func (d *Cloudreve) InitReference(storage driver.Driver) error {
+	refStorage, ok := storage.(*Cloudreve)
+	if ok {
+		d.ref = refStorage
+		return nil
+	}
+	return errs.NotSupport
+}
+
 func (d *Cloudreve) Drop(ctx context.Context) error {
 	d.Cookie = ""
+	d.ref = nil
 	return nil
 }
 

drivers/cloudreve/util.go

@@ -36,6 +36,9 @@ func (d *Cloudreve) getUA() string {
 }
 
 func (d *Cloudreve) request(method string, path string, callback base.ReqCallback, out interface{}) error {
+	if d.ref != nil {
+		return d.ref.request(method, path, callback, out)
+	}
 	u := d.Address + "/api/v3" + path
 	req := base.RestyClient.R()
 	req.SetHeaders(map[string]string{

drivers/cloudreve_v4/driver.go

@@ -20,6 +20,7 @@ import (
 type CloudreveV4 struct {
 	model.Storage
 	Addition
+	ref *CloudreveV4
 }
 
 func (d *CloudreveV4) Config() driver.Config {
@@ -34,6 +35,9 @@ func (d *CloudreveV4) Init(ctx context.Context) error {
 	// removing trailing slash
 	d.Address = strings.TrimSuffix(d.Address, "/")
 	op.MustSaveDriverStorage(d)
+	if d.ref != nil {
+		return nil
+	}
 	if d.AccessToken == "" && d.RefreshToken != "" {
 		return d.refreshToken()
 	}
@@ -43,7 +47,17 @@ func (d *CloudreveV4) Init(ctx context.Context) error {
 	return nil
 }
 
+func (d *CloudreveV4) InitReference(storage driver.Driver) error {
+	refStorage, ok := storage.(*CloudreveV4)
+	if ok {
+		d.ref = refStorage
+		return nil
+	}
+	return errs.NotSupport
+}
+
 func (d *CloudreveV4) Drop(ctx context.Context) error {
+	d.ref = nil
 	return nil
 }
 

drivers/cloudreve_v4/util.go

@@ -34,6 +34,9 @@ func (d *CloudreveV4) getUA() string {
 }
 
 func (d *CloudreveV4) request(method string, path string, callback base.ReqCallback, out any) error {
+	if d.ref != nil {
+		return d.ref.request(method, path, callback, out)
+	}
 	u := d.Address + "/api/v4" + path
 	req := base.RestyClient.R()
 	req.SetHeaders(map[string]string{