feat(degoo): token improvement (#1149)

* Update driver.go

Signed-off-by: Caspian <[email protected]>

* Update meta.go

Signed-off-by: Caspian <[email protected]>

* Update util.go

Signed-off-by: Caspian <[email protected]>

* Update util.go

Signed-off-by: Caspian <[email protected]>

* Update util.go

Signed-off-by: Caspian <[email protected]>

* Update util.go

Signed-off-by: MadDogOwner <[email protected]>

* make account optional

* ensure username and password

Signed-off-by: MadDogOwner <[email protected]>

---------

Signed-off-by: Caspian <[email protected]>
Signed-off-by: MadDogOwner <[email protected]>
Co-authored-by: MadDogOwner <[email protected]>

Author: CaspianGUAN

drivers/degoo/driver.go

@@ -32,11 +32,9 @@ func (d *Degoo) Init(ctx context.Context) error {
 
 	d.client = base.HttpClient
 
-	if d.Token == "" {
-		err := d.login(ctx)
-		if err != nil {
-			return err
-		}
+	// Ensure we have a valid token (will login if needed or refresh if expired)
+	if err := d.ensureValidToken(ctx); err != nil {
+		return fmt.Errorf("failed to initialize token: %w", err)
 	}
 
 	return d.getDevices(ctx)
@@ -87,7 +85,7 @@ func (d *Degoo) MakeDir(ctx context.Context, parentDir model.Obj, dirName string
 	const query = `mutation SetUploadFile3($Token: String!, $FileInfos: [FileInfoUpload3]!) { setUploadFile3(Token: $Token, FileInfos: $FileInfos) }`
 
 	variables := map[string]interface{}{
-		"Token": d.Token,
+		"Token": d.AccessToken,
 		"FileInfos": []map[string]interface{}{
 			{
 				"Checksum":     folderChecksum,
@@ -111,7 +109,7 @@ func (d *Degoo) Move(ctx context.Context, srcObj, dstDir model.Obj) (model.Obj,
 	const query = `mutation SetMoveFile($Token: String!, $Copy: Boolean, $NewParentID: String!, $FileIDs: [String]!) { setMoveFile(Token: $Token, Copy: $Copy, NewParentID: $NewParentID, FileIDs: $FileIDs) }`
 
 	variables := map[string]interface{}{
-		"Token":       d.Token,
+		"Token":       d.AccessToken,
 		"Copy":        false,
 		"NewParentID": dstDir.GetID(),
 		"FileIDs":     []string{srcObj.GetID()},
@@ -129,7 +127,7 @@ func (d *Degoo) Rename(ctx context.Context, srcObj model.Obj, newName string) er
 	const query = `mutation SetRenameFile($Token: String!, $FileRenames: [FileRenameInfo]!) { setRenameFile(Token: $Token, FileRenames: $FileRenames) }`
 
 	variables := map[string]interface{}{
-		"Token": d.Token,
+		"Token": d.AccessToken,
 		"FileRenames": []DegooFileRenameInfo{
 			{
 				ID:      srcObj.GetID(),
@@ -155,7 +153,7 @@ func (d *Degoo) Remove(ctx context.Context, obj model.Obj) error {
 	const query = `mutation SetDeleteFile5($Token: String!, $IsInRecycleBin: Boolean!, $IDs: [IDType]!) { setDeleteFile5(Token: $Token, IsInRecycleBin: $IsInRecycleBin, IDs: $IDs) }`
 
 	variables := map[string]interface{}{
-		"Token":          d.Token,
+		"Token":          d.AccessToken,
 		"IsInRecycleBin": false,
 		"IDs":            []map[string]string{{"FileID": obj.GetID()}},
 	}

drivers/degoo/meta.go

@@ -7,9 +7,10 @@ import (
 
 type Addition struct {
 	driver.RootID
-	Username string `json:"username" required:"true" help:"Your Degoo account email"`
-	Password string `json:"password" required:"true" help:"Your Degoo account password"`
-	Token    string `json:"token" help:"Access token for Degoo API, will be obtained automatically if not provided"`
+	Username     string `json:"username" help:"Your Degoo account email"`
+	Password     string `json:"password" help:"Your Degoo account password"`
+	RefreshToken string `json:"refresh_token" help:"Refresh token for automatic token renewal, obtained automatically"`
+	AccessToken  string `json:"access_token" help:"Access token for Degoo API, obtained automatically"`
 }
 
 var config = driver.Config{

drivers/degoo/upload.go

@@ -49,7 +49,7 @@ func (d *Degoo) getBucketWriteAuth4(ctx context.Context, file model.FileStreamer
   }`
 
 	variables := map[string]interface{}{
-		"Token":    d.Token,
+		"Token":    d.AccessToken,
 		"ParentID": parentID,
 		"StorageUploadInfos": []map[string]string{{
 			"FileName": file.GetName(),
@@ -174,7 +174,7 @@ func (d *Degoo) SetUploadFile3(ctx context.Context, file model.FileStreamer, par
   }`
 
 	variables := map[string]interface{}{
-		"Token": d.Token,
+		"Token": d.AccessToken,
 		"FileInfos": []map[string]string{{
 			"Checksum":     checksum,
 			"CreationTime": strconv.FormatInt(file.CreateTime().UnixMilli(), 10),