mirror of
https://github.com/offen/docker-volume-backup.git
synced 2024-11-25 14:40:28 +01:00
121 lines
3.4 KiB
Go
121 lines
3.4 KiB
Go
// Copyright 2022 - Offen Authors <hioffen@posteo.de>
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package webdav
|
|
|
|
import (
|
|
"fmt"
|
|
"io/fs"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/offen/docker-volume-backup/internal/errwrap"
|
|
"github.com/offen/docker-volume-backup/internal/storage"
|
|
"github.com/studio-b12/gowebdav"
|
|
)
|
|
|
|
type webDavStorage struct {
|
|
*storage.StorageBackend
|
|
client *gowebdav.Client
|
|
url string
|
|
}
|
|
|
|
// Config allows to configure a WebDAV storage backend.
|
|
type Config struct {
|
|
URL string
|
|
RemotePath string
|
|
Username string
|
|
Password string
|
|
URLInsecure bool
|
|
}
|
|
|
|
// NewStorageBackend creates and initializes a new WebDav storage backend.
|
|
func NewStorageBackend(opts Config, logFunc storage.Log) (storage.Backend, error) {
|
|
if opts.Username == "" || opts.Password == "" {
|
|
return nil, errwrap.Wrap(nil, "WEBDAV_URL is defined, but no credentials were provided")
|
|
} else {
|
|
webdavClient := gowebdav.NewClient(opts.URL, opts.Username, opts.Password)
|
|
|
|
if opts.URLInsecure {
|
|
defaultTransport, ok := http.DefaultTransport.(*http.Transport)
|
|
if !ok {
|
|
return nil, errwrap.Wrap(nil, "unexpected error when asserting type for http.DefaultTransport")
|
|
}
|
|
webdavTransport := defaultTransport.Clone()
|
|
webdavTransport.TLSClientConfig.InsecureSkipVerify = opts.URLInsecure
|
|
webdavClient.SetTransport(webdavTransport)
|
|
}
|
|
|
|
return &webDavStorage{
|
|
StorageBackend: &storage.StorageBackend{
|
|
DestinationPath: opts.RemotePath,
|
|
Log: logFunc,
|
|
},
|
|
client: webdavClient,
|
|
}, nil
|
|
}
|
|
}
|
|
|
|
// Name returns the name of the storage backend
|
|
func (b *webDavStorage) Name() string {
|
|
return "WebDAV"
|
|
}
|
|
|
|
// Copy copies the given file to the WebDav storage backend.
|
|
func (b *webDavStorage) Copy(file string) error {
|
|
_, name := path.Split(file)
|
|
if err := b.client.MkdirAll(b.DestinationPath, 0644); err != nil {
|
|
return errwrap.Wrap(err, fmt.Sprintf("error creating directory '%s' on server", b.DestinationPath))
|
|
}
|
|
|
|
r, err := os.Open(file)
|
|
if err != nil {
|
|
return errwrap.Wrap(err, "error opening the file to be uploaded")
|
|
}
|
|
|
|
if err := b.client.WriteStream(filepath.Join(b.DestinationPath, name), r, 0644); err != nil {
|
|
return errwrap.Wrap(err, "error uploading the file")
|
|
}
|
|
b.Log(storage.LogLevelInfo, b.Name(), "Uploaded a copy of backup '%s' to '%s' at path '%s'.", file, b.url, b.DestinationPath)
|
|
|
|
return nil
|
|
}
|
|
|
|
// Prune rotates away backups according to the configuration and provided deadline for the WebDav storage backend.
|
|
func (b *webDavStorage) Prune(deadline time.Time, pruningPrefix string) (*storage.PruneStats, error) {
|
|
candidates, err := b.client.ReadDir(b.DestinationPath)
|
|
if err != nil {
|
|
return nil, errwrap.Wrap(err, "error looking up candidates from remote storage")
|
|
}
|
|
var matches []fs.FileInfo
|
|
var lenCandidates int
|
|
for _, candidate := range candidates {
|
|
if !strings.HasPrefix(candidate.Name(), pruningPrefix) {
|
|
continue
|
|
}
|
|
lenCandidates++
|
|
if candidate.ModTime().Before(deadline) {
|
|
matches = append(matches, candidate)
|
|
}
|
|
}
|
|
|
|
stats := &storage.PruneStats{
|
|
Total: uint(lenCandidates),
|
|
Pruned: uint(len(matches)),
|
|
}
|
|
|
|
pruneErr := b.DoPrune(b.Name(), len(matches), lenCandidates, deadline, func() error {
|
|
for _, match := range matches {
|
|
if err := b.client.Remove(filepath.Join(b.DestinationPath, match.Name())); err != nil {
|
|
return errwrap.Wrap(err, "error removing file")
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
return stats, pruneErr
|
|
}
|