mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2025-10-31 19:26:02 +08:00
61 lines
2 KiB
Go
61 lines
2 KiB
Go
package job
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/1Panel-dev/1Panel/agent/app/model"
|
|
"github.com/1Panel-dev/1Panel/agent/constant"
|
|
"github.com/1Panel-dev/1Panel/agent/global"
|
|
"github.com/1Panel-dev/1Panel/agent/utils/cloud_storage/client"
|
|
)
|
|
|
|
type backup struct{}
|
|
|
|
func NewBackupJob() *backup {
|
|
return &backup{}
|
|
}
|
|
|
|
func (b *backup) Run() {
|
|
var backups []model.BackupAccount
|
|
_ = global.DB.Where("`type` in (?) AND is_public = 0", []string{constant.OneDrive, constant.ALIYUN, constant.GoogleDrive}).Find(&backups)
|
|
if len(backups) == 0 {
|
|
return
|
|
}
|
|
for _, backupItem := range backups {
|
|
if backupItem.ID == 0 {
|
|
continue
|
|
}
|
|
global.LOG.Infof("Start to refresh %s-%s access_token ...", backupItem.Type, backupItem.Name)
|
|
varMap := make(map[string]interface{})
|
|
if err := json.Unmarshal([]byte(backupItem.Vars), &varMap); err != nil {
|
|
global.LOG.Errorf("failed to refresh %s - %s token, please retry, err: %v", backupItem.Type, backupItem.Name, err)
|
|
continue
|
|
}
|
|
var (
|
|
refreshToken string
|
|
err error
|
|
)
|
|
switch backupItem.Type {
|
|
case constant.OneDrive:
|
|
refreshToken, err = client.RefreshToken("refresh_token", "refreshToken", varMap)
|
|
case constant.GoogleDrive:
|
|
refreshToken, err = client.RefreshGoogleToken("refresh_token", "refreshToken", varMap)
|
|
case constant.ALIYUN:
|
|
refreshToken, err = client.RefreshALIToken(varMap)
|
|
}
|
|
if err != nil {
|
|
varMap["refresh_status"] = constant.StatusFailed
|
|
varMap["refresh_msg"] = err.Error()
|
|
global.LOG.Errorf("failed to refresh OneDrive token, please retry, err: %v", err)
|
|
continue
|
|
}
|
|
varMap["refresh_status"] = constant.StatusSuccess
|
|
varMap["refresh_time"] = time.Now().Format(constant.DateTimeLayout)
|
|
varMap["refresh_token"] = refreshToken
|
|
|
|
varsItem, _ := json.Marshal(varMap)
|
|
_ = global.DB.Model(&model.BackupAccount{}).Where("id = ?", backupItem.ID).Updates(map[string]interface{}{"vars": string(varsItem)}).Error
|
|
global.LOG.Infof("Refresh %s-%s access_token successful!", backupItem.Type, backupItem.Name)
|
|
}
|
|
}
|