Add condition for indicating restore completion

apis/constants.go

@@ -204,6 +204,10 @@ const (
 	StashInitContainerInjected = "StashInitContainerInjected"
 	// RestoreJobCreated indicates whether the restore job was created
 	RestoreJobCreated = "RestoreJobCreated"
+	// RestoreCompleted condition indicates whether the restore process has been completed or not.
+	// This condition is particularly helpful when the restore addon require some additional operations to perform
+	// before marking the RestoreSession Succeeded/Failed.
+	RestoreCompleted = "RestoreCompleted"
 
 	// GlobalPreBackupHookSucceeded indicates whether the global PreBackupHook was executed successfully or not
 	GlobalPreBackupHookSucceeded = "GlobalPreBackupHookSucceeded"

pkg/conditions/restore.go

@@ -106,3 +106,21 @@ func SetInitContainerInjectedConditionToFalse(inv invoker.RestoreInvoker, tref a
 		Message: fmt.Sprintf("Failed to inject Stash init-container. Reason: %v", err.Error()),
 	})
 }
+
+func SetRestoreCompletedConditionToTrue(inv invoker.RestoreInvoker, tref api_v1beta1.TargetRef, msg string) error {
+	return inv.SetCondition(&tref, kmapi.Condition{
+		Type:    apis.RestoreCompleted,
+		Status:  core.ConditionTrue,
+		Reason:  "PostRestoreTasksExecuted",
+		Message: msg,
+	})
+}
+
+func SetRestoreCompletedConditionToFalse(inv invoker.RestoreInvoker, tref api_v1beta1.TargetRef, msg string) error {
+	return inv.SetCondition(&tref, kmapi.Condition{
+		Type:    apis.RestoreCompleted,
+		Status:  core.ConditionFalse,
+		Reason:  "PostRestoreTasksNotExecuted",
+		Message: msg,
+	})
+}