mirror of
https://github.com/offen/docker-volume-backup.git
synced 2024-11-10 00:30:29 +01:00
279844ccfb
* Added abstract helper interface and implemented it for all storage backends * Moved storage client initializations also to helper classes * Fixed ssh init issue * Moved script parameter to helper struct to simplify script init. * Created sub modules. Enhanced abstract implementation. * Fixed config issue * Fixed declaration issues. Added config to interface. * Added StorageProviders to unify all backends. * Cleanup, optimizations, comments. * Applied discussed changes. See description. Moved modules to internal packages. Replaced StoragePool with slice. Moved conditional for init of storage backends back to script. * Fix docker build issue * Fixed accidentally removed local copy condition. * Delete .gitignore * Renaming/changes according to review Renamed Init functions and interface. Replaced config object with specific config values. Init func returns interface instead of struct. Removed custom import names where possible. * Fixed auto-complete error. * Combined copy instructions into one layer. * Added logging func for storages. * Introduced logging func for errors too. * Missed an error message * Moved config back to main. Optimized prune stats handling. * Move stats back to main package * Code doc stuff * Apply changes from #136 * Replace name field with function. * Changed receiver names from stg to b. * Renamed LogFuncDef to Log * Removed redundant package name. * Renamed storagePool to storages. * Simplified creation of new storage backend. * Added initialization for storage stats map. * Invert .dockerignore patterns. * Fix package typo
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
// Copyright 2022 - Offen Authors <hioffen@posteo.de>
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
_ "embed"
|
|
"fmt"
|
|
"os"
|
|
"text/template"
|
|
"time"
|
|
|
|
sTypes "github.com/containrrr/shoutrrr/pkg/types"
|
|
"github.com/offen/docker-volume-backup/internal/utilities"
|
|
)
|
|
|
|
//go:embed notifications.tmpl
|
|
var defaultNotifications string
|
|
|
|
// NotificationData data to be passed to the notification templates
|
|
type NotificationData struct {
|
|
Error error
|
|
Config *Config
|
|
Stats *Stats
|
|
}
|
|
|
|
// notify sends a notification using the given title and body templates.
|
|
// Automatically creates notification data, adding the given error
|
|
func (s *script) notify(titleTemplate string, bodyTemplate string, err error) error {
|
|
params := NotificationData{
|
|
Error: err,
|
|
Stats: s.stats,
|
|
Config: s.c,
|
|
}
|
|
|
|
titleBuf := &bytes.Buffer{}
|
|
if err := s.template.ExecuteTemplate(titleBuf, titleTemplate, params); err != nil {
|
|
return fmt.Errorf("notifyFailure: error executing %s template: %w", titleTemplate, err)
|
|
}
|
|
|
|
bodyBuf := &bytes.Buffer{}
|
|
if err := s.template.ExecuteTemplate(bodyBuf, bodyTemplate, params); err != nil {
|
|
return fmt.Errorf("notifyFailure: error executing %s template: %w", bodyTemplate, err)
|
|
}
|
|
|
|
if err := s.sendNotification(titleBuf.String(), bodyBuf.String()); err != nil {
|
|
return fmt.Errorf("notifyFailure: error notifying: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// notifyFailure sends a notification about a failed backup run
|
|
func (s *script) notifyFailure(err error) error {
|
|
return s.notify("title_failure", "body_failure", err)
|
|
}
|
|
|
|
// notifyFailure sends a notification about a successful backup run
|
|
func (s *script) notifySuccess() error {
|
|
return s.notify("title_success", "body_success", nil)
|
|
}
|
|
|
|
// sendNotification sends a notification to all configured third party services
|
|
func (s *script) sendNotification(title, body string) error {
|
|
var errs []error
|
|
for _, result := range s.sender.Send(body, &sTypes.Params{"title": title}) {
|
|
if result != nil {
|
|
errs = append(errs, result)
|
|
}
|
|
}
|
|
if len(errs) != 0 {
|
|
return fmt.Errorf("sendNotification: error sending message: %w", utilities.Join(errs...))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var templateHelpers = template.FuncMap{
|
|
"formatTime": func(t time.Time) string {
|
|
return t.Format(time.RFC3339)
|
|
},
|
|
"formatBytesDec": func(bytes uint64) string {
|
|
return formatBytes(bytes, true)
|
|
},
|
|
"formatBytesBin": func(bytes uint64) string {
|
|
return formatBytes(bytes, false)
|
|
},
|
|
"env": os.Getenv,
|
|
}
|
|
|
|
// formatBytes converts an amount of bytes in a human-readable representation
|
|
// the decimal parameter specifies if using powers of 1000 (decimal) or powers of 1024 (binary)
|
|
func formatBytes(b uint64, decimal bool) string {
|
|
unit := uint64(1024)
|
|
format := "%.1f %ciB"
|
|
if decimal {
|
|
unit = uint64(1000)
|
|
format = "%.1f %cB"
|
|
}
|
|
if b < unit {
|
|
return fmt.Sprintf("%d B", b)
|
|
}
|
|
div, exp := unit, 0
|
|
for n := b / unit; n >= unit; n /= unit {
|
|
div *= unit
|
|
exp++
|
|
}
|
|
return fmt.Sprintf(format, float64(b)/float64(div), "kMGTPE"[exp])
|
|
}
|