add failure_or_done notification stragegy

Signed-off-by: Wesley GALIPO <wesley.galipo@ovhcloud.com>

pkg/notify/init/init.go

@@ -148,6 +148,10 @@ func validateStrategyForAction(action, strategy string) error {
 		if action == notify.TaskValidationKey {
 			return errNotAllowed
 		}
+	case utask.NotificationStrategyFailureOrDone:
+		if action == notify.TaskValidationKey {
+			return errNotAllowed
+		}
 	default:
 		return errUnknown
 	}

pkg/notify/messages.go

@@ -236,6 +236,10 @@ func checkIfDeliverMessageFromTaskState(m *Message, strategy string) bool {
 		if v, ok := m.Fields["state"]; ok && v == stateBlocked {
 			send = true
 		}
+	case utask.NotificationStrategyFailureOrDone:
+		if v, ok := m.Fields["state"]; ok && (v == stateBlocked || v == step.StateDone) {
+			send = true
+		}
 	case utask.NotificationStrategySilent:
 	}
 
@@ -254,6 +258,13 @@ func checkIfDeliverMessageFromStepState(m *Message, strategy string) bool {
 				send = true
 			}
 		}
+	case utask.NotificationStrategyFailureOrDone:
+		if v, ok := m.Fields["step_state"]; ok {
+			switch v {
+			case step.StateFatalError, step.StateCrashed, step.StateAfterrunError, step.StateDone:
+				send = true
+			}
+		}
 	case utask.NotificationStrategySilent:
 	}
 

utask.go

@@ -100,6 +100,8 @@ const (
 	NotificationStrategyAlways = "always"
 	// NotificationStrategyFailureOnly corresponds to the mode where notifications will only be sent if the state is BLOCKED
 	NotificationStrategyFailureOnly = "failure_only"
+	// NotificationStrategyFailureOrDone corresponds to the mode where notifications will only be sent if the state is BLOCKED or DONE
+	NotificationStrategyFailureOrDone = "failure_or_done"
 
 	// UsernamesSeparator corresponds to the separator used to break a string into a list of usernames and vice versa.
 	UsernamesSeparator = ","