Add BeforeCopy and AfterCopy to Retention

Author: hgraeber

config/profile.go

@@ -243,6 +243,8 @@ type RetentionSection struct {
 	OtherFlagsSection   `mapstructure:",squash"`
 	BeforeBackup        maybe.Bool `mapstructure:"before-backup" description:"Apply retention before starting the backup command"`
 	AfterBackup         maybe.Bool `mapstructure:"after-backup" description:"Apply retention after the backup command succeeded. Defaults to true in configuration format v2 if any \"keep-*\" flag is set and \"before-backup\" is unset"`
+	BeforeCopy          maybe.Bool `mapstructure:"before-copy" description:"Apply retention before starting the copy command"`
+	AfterCopy           maybe.Bool `mapstructure:"after-copy" description:"Apply retention after the backup copy succeeded. Defaults to true in configuration format v2 if any \"keep-*\" flag is set and \"before-copy\" is unset"`
 }
 
 func (r *RetentionSection) IsEmpty() bool { return r == nil }
@@ -262,7 +264,7 @@ func (r *RetentionSection) resolve(profile *Profile) {
 	// Extras, only enabled for Version >= 2 (to remain backward compatible in version 1)
 	if profile.config != nil && profile.config.version >= Version02 {
 		// Auto-enable "after-backup" if nothing was specified explicitly and any "keep-" was configured
-		if r.AfterBackup.IsUndefined() && r.BeforeBackup.IsUndefined() {
+		if r.AfterBackup.IsUndefined() && r.BeforeBackup.IsUndefined() && r.AfterCopy.IsUndefined() && r.BeforeCopy.IsUndefined() {
 			for name := range r.OtherFlags {
 				if strings.HasPrefix(name, "keep-") {
 					r.AfterBackup = maybe.True()

wrapper.go

@@ -141,13 +141,35 @@ func (r *resticWrapper) getCommandAction(command string) func() error {
 func (r *resticWrapper) getCopyAction() func() error {
 	copyAction := r.getCommandAction(constants.CommandCopy)
 
-	return func() error {
+	return func() (err error) {
 		// we might need to initialize the secondary repository (the copy target)
 		if r.global.Initialize || (r.profile.Copy != nil && r.profile.Copy.Initialize) {
 			_ = r.runInitializeCopy() // it's ok if the initialization returned an error
 		}
 
-		return copyAction()
+		// Retention before
+		if r.profile.Retention != nil && r.profile.Retention.BeforeCopy.IsTrue() {
+			err = r.runRetention()
+			if err != nil {
+				return
+			}
+		}
+
+		// Copy command
+		err = copyAction()
+		if err != nil {
+			return
+		}
+
+		// Retention after
+		if r.profile.Retention != nil && r.profile.Retention.AfterCopy.IsTrue() {
+			err = r.runRetention()
+			if err != nil {
+				return
+			}
+		}
+
+		return
 	}
 }