diff --git a/drivers/alidrive.go b/drivers/alidrive.go index ecb94549..251cebac 100644 --- a/drivers/alidrive.go +++ b/drivers/alidrive.go @@ -339,6 +339,8 @@ func (a AliDrive) RefreshToken(account *model.Account) error { if e.Code != "" { account.Status = e.Message return fmt.Errorf("failed to refresh token: %s", e.Message) + }else { + account.Status = "work" } account.RefreshToken, account.AccessToken = resp.RefreshToken, resp.AccessToken return nil diff --git a/drivers/native.go b/drivers/native.go index ab59c872..3f5d1d0b 100644 --- a/drivers/native.go +++ b/drivers/native.go @@ -40,6 +40,11 @@ func (n Native) Save(account *model.Account, old *model.Account) error { if !utils.Exists(account.RootFolder) { return fmt.Errorf("[%s] not exist", account.RootFolder) } + account.Status = "work" + err := model.SaveAccount(account) + if err != nil { + return err + } return nil } diff --git a/drivers/onedrive.go b/drivers/onedrive.go index 72630d62..d75fa8e5 100644 --- a/drivers/onedrive.go +++ b/drivers/onedrive.go @@ -153,6 +153,8 @@ func (o Onedrive) RefreshToken(account *model.Account) error { if e.Error != "" { account.Status = e.ErrorDescription return fmt.Errorf("%s", e.ErrorDescription) + }else { + account.Status = "work" } account.RefreshToken, account.AccessToken = resp.RefreshToken, resp.AccessToken return nil