push backup
This commit is contained in:
@@ -35,6 +35,10 @@ type (
|
||||
}
|
||||
)
|
||||
|
||||
var (
|
||||
ErrNotFound error = errors.New("not found")
|
||||
)
|
||||
|
||||
func New(baseURL, username, password string) *Client {
|
||||
return &Client{
|
||||
baseURL: baseURL,
|
||||
@@ -141,59 +145,45 @@ func (c *Client) Metadata(gameID string) (repository.Metadata, error) {
|
||||
return repository.Metadata{}, errors.New("invalid payload sent by the server")
|
||||
}
|
||||
|
||||
func (c *Client) Push(gameID, archivePath string, m repository.Metadata) error {
|
||||
u, err := url.JoinPath(c.baseURL, "api", "v1", "games", gameID, "data")
|
||||
func (c *Client) PushSave(archivePath string, m repository.Metadata) error {
|
||||
u, err := url.JoinPath(c.baseURL, "api", "v1", "games", m.ID, "data")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
f, err := os.OpenFile(archivePath, os.O_RDONLY, 0)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to open file: %w", err)
|
||||
}
|
||||
defer f.Close()
|
||||
return c.push(u, archivePath, m)
|
||||
}
|
||||
|
||||
buf := new(bytes.Buffer)
|
||||
writer := multipart.NewWriter(buf)
|
||||
|
||||
part, err := writer.CreateFormFile("payload", "data.tar.gz")
|
||||
func (c *Client) PushBackup(archiveMetadata repository.Backup, m repository.Metadata) error {
|
||||
u, err := url.JoinPath(c.baseURL, "api", "v1", "games", m.ID, "hist", archiveMetadata.UUID, "data")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if _, err := io.Copy(part, f); err != nil {
|
||||
return fmt.Errorf("failed to copy data: %w", err)
|
||||
}
|
||||
return c.push(u, archiveMetadata.ArchivePath, m)
|
||||
}
|
||||
|
||||
writer.WriteField("name", m.Name)
|
||||
writer.WriteField("version", strconv.Itoa(m.Version))
|
||||
writer.WriteField("date", m.Date.Format(time.RFC3339))
|
||||
|
||||
if err := writer.Close(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cli := http.Client{}
|
||||
|
||||
req, err := http.NewRequest("POST", u, buf)
|
||||
func (c *Client) ArchiveInfo(gameID, uuid string) (repository.Backup, error) {
|
||||
u, err := url.JoinPath(c.baseURL, "api", "v1", "games", gameID, "hist", uuid, "info")
|
||||
if err != nil {
|
||||
return err
|
||||
return repository.Backup{}, err
|
||||
}
|
||||
|
||||
req.SetBasicAuth(c.username, c.password)
|
||||
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||||
|
||||
res, err := cli.Do(req)
|
||||
o, err := c.get(u)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer res.Body.Close()
|
||||
|
||||
if res.StatusCode != 201 {
|
||||
return fmt.Errorf("server returns an unexpected status code: %s (expected 201)", res.Status)
|
||||
return repository.Backup{}, err
|
||||
}
|
||||
|
||||
return nil
|
||||
if m, ok := (o.Data).(map[string]any); ok {
|
||||
b := repository.Backup{
|
||||
UUID: m["uuid"].(string),
|
||||
CreatedAt: customtime.MustParse(time.RFC3339, m["created_at"].(string)),
|
||||
MD5: m["md5"].(string),
|
||||
}
|
||||
return b, nil
|
||||
}
|
||||
|
||||
return repository.Backup{}, errors.New("invalid payload sent by the server")
|
||||
}
|
||||
|
||||
func (c *Client) Pull(gameID, archivePath string) error {
|
||||
@@ -318,6 +308,10 @@ func (c *Client) get(url string) (obj.HTTPObject, error) {
|
||||
}
|
||||
defer res.Body.Close()
|
||||
|
||||
if res.StatusCode == 404 {
|
||||
return obj.HTTPObject{}, ErrNotFound
|
||||
}
|
||||
|
||||
if res.StatusCode != 200 {
|
||||
return obj.HTTPObject{}, fmt.Errorf("server returns an unexpected status code: %d %s (expected 200)", res.StatusCode, res.Status)
|
||||
}
|
||||
@@ -331,3 +325,53 @@ func (c *Client) get(url string) (obj.HTTPObject, error) {
|
||||
|
||||
return httpObject, nil
|
||||
}
|
||||
|
||||
func (c *Client) push(u, archivePath string, m repository.Metadata) error {
|
||||
f, err := os.OpenFile(archivePath, os.O_RDONLY, 0)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to open file: %w", err)
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
buf := new(bytes.Buffer)
|
||||
writer := multipart.NewWriter(buf)
|
||||
|
||||
part, err := writer.CreateFormFile("payload", "data.tar.gz")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if _, err := io.Copy(part, f); err != nil {
|
||||
return fmt.Errorf("failed to copy data: %w", err)
|
||||
}
|
||||
|
||||
writer.WriteField("name", m.Name)
|
||||
writer.WriteField("version", strconv.Itoa(m.Version))
|
||||
writer.WriteField("date", m.Date.Format(time.RFC3339))
|
||||
|
||||
if err := writer.Close(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cli := http.Client{}
|
||||
|
||||
req, err := http.NewRequest("POST", u, buf)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
req.SetBasicAuth(c.username, c.password)
|
||||
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||||
|
||||
res, err := cli.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer res.Body.Close()
|
||||
|
||||
if res.StatusCode != 201 {
|
||||
return fmt.Errorf("server returns an unexpected status code: %s (expected 201)", res.Status)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -1,16 +1,17 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"cloudsave/pkg/tools/hash"
|
||||
"cloudsave/pkg/tools/id"
|
||||
"crypto/md5"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/fs"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type (
|
||||
@@ -21,6 +22,13 @@ type (
|
||||
Version int `json:"version"`
|
||||
Date time.Time `json:"date"`
|
||||
}
|
||||
|
||||
Backup struct {
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
MD5 string `json:"md5"`
|
||||
UUID string `json:"uuid"`
|
||||
ArchivePath string `json:"-"`
|
||||
}
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -142,46 +150,20 @@ func Archive(gameID string) error {
|
||||
// open old
|
||||
f, err := os.OpenFile(path, os.O_RDONLY, 0)
|
||||
if err != nil {
|
||||
if errors.Is(err, os.ErrNotExist) {
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("failed to open old file: %w", err)
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
histDirPath := filepath.Join(datastorepath, gameID, "hist")
|
||||
histDirPath := filepath.Join(datastorepath, gameID, "hist", uuid.NewString())
|
||||
if err := os.MkdirAll(histDirPath, 0740); err != nil {
|
||||
return fmt.Errorf("failed to make 'hist' directory")
|
||||
}
|
||||
|
||||
d, err := os.ReadDir(histDirPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to open 'hist' directory")
|
||||
}
|
||||
|
||||
// keep the dir under 6 files
|
||||
if len(d) > 5 {
|
||||
var oldest *fs.FileInfo
|
||||
for _, hfile := range d {
|
||||
finfo, err := hfile.Info()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to read backup file: %w", err)
|
||||
}
|
||||
|
||||
if oldest == nil {
|
||||
oldest = &finfo
|
||||
continue
|
||||
}
|
||||
|
||||
if finfo.ModTime().Before((*oldest).ModTime()) {
|
||||
oldest = &finfo
|
||||
}
|
||||
}
|
||||
|
||||
if err := os.Remove((*oldest).Name()); err != nil {
|
||||
return fmt.Errorf("failed to remove the oldest backup file: %w", err)
|
||||
}
|
||||
return fmt.Errorf("failed to make directory: %w", err)
|
||||
}
|
||||
|
||||
// open new
|
||||
nf, err := os.OpenFile(filepath.Join(datastorepath, gameID, "hist", time.Now().Format("2006-01-02T15-04-05Z07-00")+".data.tar.gz"), os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0740)
|
||||
nf, err := os.OpenFile(filepath.Join(histDirPath, "data.tar.gz"), os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0740)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to open new file: %w", err)
|
||||
}
|
||||
@@ -195,6 +177,43 @@ func Archive(gameID string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func Archives(gameID string) ([]Backup, error) {
|
||||
histDirPath := filepath.Join(datastorepath, gameID, "hist")
|
||||
if err := os.MkdirAll(histDirPath, 0740); err != nil {
|
||||
return nil, fmt.Errorf("failed to make 'hist' directory")
|
||||
}
|
||||
|
||||
d, err := os.ReadDir(histDirPath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to open 'hist' directory")
|
||||
}
|
||||
|
||||
var res []Backup
|
||||
for _, f := range d {
|
||||
finfo, err := f.Info()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("corrupted datastore: %w", err)
|
||||
}
|
||||
path := filepath.Join(histDirPath, finfo.Name())
|
||||
archivePath := filepath.Join(path, "data.tar.gz")
|
||||
|
||||
h, err := hash.FileMD5(archivePath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to calculate md5 hash: %w", err)
|
||||
}
|
||||
|
||||
b := Backup{
|
||||
CreatedAt: finfo.ModTime(),
|
||||
UUID: filepath.Base(finfo.Name()),
|
||||
MD5: h,
|
||||
ArchivePath: archivePath,
|
||||
}
|
||||
|
||||
res = append(res, b)
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
func DatastorePath() string {
|
||||
return datastorepath
|
||||
}
|
||||
@@ -210,18 +229,7 @@ func Remove(gameID string) error {
|
||||
func Hash(gameID string) (string, error) {
|
||||
path := filepath.Join(datastorepath, gameID, "data.tar.gz")
|
||||
|
||||
f, err := os.OpenFile(path, os.O_RDONLY, 0)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
hasher := md5.New()
|
||||
if _, err := io.Copy(hasher, f); err != nil {
|
||||
return "", err
|
||||
}
|
||||
sum := hasher.Sum(nil)
|
||||
return hex.EncodeToString(sum), nil
|
||||
return hash.FileMD5(path)
|
||||
}
|
||||
|
||||
func Version(gameID string) (int, error) {
|
||||
|
||||
@@ -73,7 +73,7 @@ func Untar(file io.Reader, path string) error {
|
||||
}
|
||||
}
|
||||
|
||||
func Tar(file io.Writer, path string) error {
|
||||
func Tar(file io.Writer, root string) error {
|
||||
gw := gzip.NewWriter(file)
|
||||
defer gw.Close()
|
||||
|
||||
@@ -81,34 +81,38 @@ func Tar(file io.Writer, path string) error {
|
||||
defer tw.Close()
|
||||
|
||||
// Walk again to add files
|
||||
err := filepath.Walk(path, func(path string, info os.FileInfo, walkErr error) error {
|
||||
err := filepath.Walk(root, func(path string, info os.FileInfo, walkErr error) error {
|
||||
if walkErr != nil {
|
||||
return walkErr
|
||||
}
|
||||
|
||||
path, err := filepath.Rel(root, path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Create tar header
|
||||
header, err := tar.FileInfoHeader(info, path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// Preserve directory structure relative to srcDir
|
||||
relPath, err := filepath.Rel(filepath.Dir(path), path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
header.Name = relPath
|
||||
header.Name = path
|
||||
|
||||
if err := tw.WriteHeader(header); err != nil {
|
||||
return err
|
||||
}
|
||||
if info.Mode().IsRegular() {
|
||||
file, err := os.Open(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer file.Close()
|
||||
if _, err := io.Copy(tw, file); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if !info.Mode().IsRegular() {
|
||||
return nil
|
||||
}
|
||||
|
||||
file, err := os.Open(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer file.Close()
|
||||
if _, err := io.Copy(tw, file); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
23
pkg/tools/hash/hash.go
Normal file
23
pkg/tools/hash/hash.go
Normal file
@@ -0,0 +1,23 @@
|
||||
package hash
|
||||
|
||||
import (
|
||||
"crypto/md5"
|
||||
"encoding/hex"
|
||||
"io"
|
||||
"os"
|
||||
)
|
||||
|
||||
func FileMD5(fp string) (string, error) {
|
||||
f, err := os.OpenFile(fp, os.O_RDONLY, 0)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
hasher := md5.New()
|
||||
if _, err := io.Copy(hasher, f); err != nil {
|
||||
return "", err
|
||||
}
|
||||
sum := hasher.Sum(nil)
|
||||
return hex.EncodeToString(sum), nil
|
||||
}
|
||||
Reference in New Issue
Block a user