chore: Allow linking condition to states

Author: PythonGermany

config/endpoint/condition.go

@@ -24,7 +24,7 @@ const (
 type Condition string
 
 // Validate checks if the Condition is valid
-func (c Condition) Validate() error {
+func (c Condition) Validate() error { // TODO#227 Validate conditions with linked states have valid states
 	r := &Result{}
 	c.evaluate(r, false, nil)
 	if len(r.Errors) != 0 {
@@ -37,6 +37,18 @@ func (c Condition) Validate() error {
 func (c Condition) evaluate(result *Result, dontResolveFailedConditions bool, context *gontext.Gontext) bool {
 	condition := string(c)
 	success := false
+
+	var linkedState string
+	if strings.Contains(condition, "::") {
+		conditionParts := strings.Split(condition, "::")
+		if len(conditionParts) != 2 { // TODO#227 Not sure if this makes sense. Checking that it is 2 or more should be enough. Then there is no character restriction in the remaining condition.
+			result.AddError(fmt.Sprintf("invalid linked state syntax: %s", condition))
+			return false
+		}
+		linkedState = conditionParts[0]
+		condition = conditionParts[1]
+	}
+
 	conditionToDisplay := condition
 	if strings.Contains(condition, " == ") {
 		parameters, resolvedParameters := sanitizeAndResolveWithContext(strings.Split(condition, " == "), result, context)
@@ -81,7 +93,7 @@ func (c Condition) evaluate(result *Result, dontResolveFailedConditions bool, co
 	if !success {
 		//logr.Debugf("[Condition.evaluate] Condition '%s' did not succeed because '%s' is false", condition, condition)
 	}
-	result.ConditionResults = append(result.ConditionResults, &ConditionResult{Condition: conditionToDisplay, Success: success})
+	result.ConditionResults = append(result.ConditionResults, &ConditionResult{Condition: conditionToDisplay, Success: success, LinkedState: linkedState})
 	return success
 }
 

config/endpoint/condition_result.go

@@ -7,4 +7,7 @@ type ConditionResult struct {
 
 	// Success whether the condition was met (successful) or not (failed)
 	Success bool `json:"success"`
+
+	// The state the condition is linked to
+	LinkedState string `json:"-"`
 }

config/endpoint/endpoint.go

@@ -68,7 +68,7 @@ var (
 	ErrUnknownEndpointType = errors.New("unknown endpoint type")
 
 	// ErrInvalidConditionFormat is the error with which Gatus will panic if a condition has an invalid format
-	ErrInvalidConditionFormat = errors.New("invalid condition format: does not match '<VALUE> <COMPARATOR> <VALUE>'")
+	ErrInvalidConditionFormat = errors.New("invalid condition format: does not match '[STATE::]<VALUE> <COMPARATOR> <VALUE>'")
 
 	// ErrInvalidEndpointIntervalForDomainExpirationPlaceholder is the error with which Gatus will panic if an endpoint
 	// has both an interval smaller than 5 minutes and a condition with DomainExpirationPlaceholder.
@@ -338,10 +338,18 @@ func (e *Endpoint) EvaluateHealthWithContext(context *gontext.Gontext) *Result {
 			result.Success = false
 		}
 	}
-	if result.Success { // TODO#227 Make dynymic and also check conditions e.g. method somewhere at the end that evaluates the state (conditions and condition results + is maintenance needs to be available)
+	if result.Success {
 		result.State = state.DefaultHealthyStateName
-	} else {
-		result.State = state.DefaultUnhealthyStateName
+	} else { // Go over condition results to see if any of them has a specific state to set
+		for _, conditionResult := range result.ConditionResults {
+			if !conditionResult.Success && len(conditionResult.LinkedState) > 0 {
+				result.State = conditionResult.LinkedState // TODO#227 Only set if no other state with h
+				break
+			}
+		}
+		if len(result.State) == 0 {
+			result.State = state.DefaultUnhealthyStateName
+		}
 	}
 	result.Timestamp = time.Now()
 	// Clean up parameters that we don't need to keep in the results