67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/patrickmn/go-cache"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
// backupUploadIDs stores a cache of active S3 backups.
|
|
var backupUploadIDs = cache.New(time.Hour*3, time.Minute*5)
|
|
|
|
type BackupRemoteUploadResponse struct {
|
|
UploadID string `json:"upload_id"`
|
|
Parts []string `json:"parts"`
|
|
PartSize int64 `json:"part_size"`
|
|
}
|
|
|
|
func (r *Request) GetBackupRemoteUploadURLs(backup string, size int64) (*BackupRemoteUploadResponse, error) {
|
|
resp, err := r.Get(fmt.Sprintf("/backups/%s", backup), Q{"size": strconv.FormatInt(size, 10)})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.HasError() {
|
|
return nil, resp.Error()
|
|
}
|
|
|
|
var res BackupRemoteUploadResponse
|
|
if err := resp.Bind(&res); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Store the backup upload id for later use, this is a janky way to be able to use it later with SendBackupStatus.
|
|
// Yes, the timeout of 3 hours is intentional, if this value is removed before the backup completes,
|
|
// the backup will fail even if it uploaded properly.
|
|
backupUploadIDs.Set(backup, res.UploadID, 0)
|
|
|
|
return &res, nil
|
|
}
|
|
|
|
type BackupRequest struct {
|
|
UploadID string `json:"upload_id"`
|
|
Checksum string `json:"checksum"`
|
|
ChecksumType string `json:"checksum_type"`
|
|
Size int64 `json:"size"`
|
|
Successful bool `json:"successful"`
|
|
}
|
|
|
|
// Notifies the panel that a specific backup has been completed and is now
|
|
// available for a user to view and download.
|
|
func (r *Request) SendBackupStatus(backup string, data BackupRequest) error {
|
|
// Set the UploadID on the data.
|
|
if v, ok := backupUploadIDs.Get(backup); ok {
|
|
data.UploadID = v.(string)
|
|
}
|
|
|
|
resp, err := r.Post(fmt.Sprintf("/backups/%s", backup), data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
return resp.Error()
|
|
}
|