wings/api/backup_endpoints.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()
}