Attempted sanitization cleanup to only run once

Author: frikky

shared.go

@@ -16380,12 +16380,14 @@ func ParsedExecutionResult(ctx context.Context, workflowExecution WorkflowExecut
 		return &workflowExecution, true, nil
 	}
 
+
 	// 1. CHECK cache if it happened in another?
 	// 2. Set cache
 	// 3. Find executed without a result
 	// 4. Ensure the result is NOT set when running an action)
 
 	actionResult = FixActionResultOutput(actionResult)
+	actionResult.Sanitized = false
 	actionCacheId := fmt.Sprintf("%s_%s_result", actionResult.ExecutionId, actionResult.Action.ID)
 
 	// Done elsewhere
@@ -32164,16 +32166,27 @@ func cleanupProtectedKeys(exec WorkflowExecution) WorkflowExecution {
 		return exec
 	}
 
-	for _, protectedKey := range protectedKeys {
 
-		for resultKey, _ := range exec.Results {
+	for resultKey, _ := range exec.Results {
+		if exec.Results[resultKey].Status != "FINISHED" && exec.Results[resultKey].Status != "SUCCESS" {
+			continue
+		}
+
+		if exec.Results[resultKey].Sanitized {
+			continue
+		}
+
+		for _, protectedKey := range protectedKeys {
+
 			if len(protectedKey.Value) <= 8 {
 				exec.Results[resultKey].Result = strings.ReplaceAll(exec.Results[resultKey].Result, protectedKey.Value, "***")
 			} else {
 				exec.Results[resultKey].Result = SanitizeFuzzySubstring(exec.Results[resultKey].Result, protectedKey.Value, 2)
 			}
 		}
 
+		exec.Results[resultKey].Sanitized = true
+
 	}
 
 	return exec

structs.go

@@ -1604,16 +1604,18 @@ type SimilarAction struct {
 }
 
 type ActionResult struct {
-	Action           Action          `json:"action" datastore:"action"`
-	ExecutionId      string          `json:"execution_id" datastore:"execution_id"`
-	Authorization    string          `json:"authorization" datastore:"authorization"`
-	Result           string          `json:"result" datastore:"result,noindex"`
-	StartedAt        int64           `json:"started_at" datastore:"started_at"`
-	CompletedAt      int64           `json:"completed_at" datastore:"completed_at"`
-	Status           string          `json:"status" datastore:"status"`
+	Action        Action `json:"action" datastore:"action"`
+	ExecutionId   string `json:"execution_id" datastore:"execution_id"`
+	Authorization string `json:"authorization" datastore:"authorization"`
+	Result        string `json:"result" datastore:"result,noindex"`
+	StartedAt     int64  `json:"started_at" datastore:"started_at"`
+	CompletedAt   int64  `json:"completed_at" datastore:"completed_at"`
+	Status        string `json:"status" datastore:"status"`
+
 	AttackTechniques []string        `json:"attack_techniques" datastore:"attack_techniques"`
 	AttackTactics    []string        `json:"attack_tactics" datastore:"attack_tactics"`
 	SimilarActions   []SimilarAction `json:"similar_actions" datastore:"similar_actions"`
+	Sanitized        bool            `json:"sanitized" datastore:"sanitized"`
 }
 
 type AuthenticationUsage struct {
@@ -1640,9 +1642,9 @@ type Notification struct {
 	Personal          bool     `json:"personal" datastore:"personal"`
 	Read              bool     `json:"read" datastore:"read"`
 
-	ModifiedBy			string `json:"modified_by" datastore:"modified_by"`
-	Ignored				bool   `json:"ignored" datastore:"ignored"`
-	ExecutionId			string `json:"execution_id" datastore:"execution_id"`
+	ModifiedBy  string `json:"modified_by" datastore:"modified_by"`
+	Ignored     bool   `json:"ignored" datastore:"ignored"`
+	ExecutionId string `json:"execution_id" datastore:"execution_id"`
 }
 
 type NotificationCached struct {
@@ -4292,17 +4294,17 @@ type AppHealth struct {
 }
 
 type DatastoreHealth struct {
-	Create	bool	`json:"create"`
-	Read	bool	`json:"read"`
-	Result	string	`json:"result"`
-	Delete	bool	`json:"delete"`
+	Create bool   `json:"create"`
+	Read   bool   `json:"read"`
+	Result string `json:"result"`
+	Delete bool   `json:"delete"`
 }
 
 type FileHealth struct {
-	Create	bool	`json:"create"`
-	FileId	string	`json:"fileId"`
-	Upload	bool	`json:"get_file"`
-	Delete	bool	`json:"delete"`
+	Create bool   `json:"create"`
+	FileId string `json:"fileId"`
+	Upload bool   `json:"get_file"`
+	Delete bool   `json:"delete"`
 }
 
 type WorkflowHealth struct {
@@ -4336,13 +4338,13 @@ type LiveExecutionStatus struct {
 }
 
 type HealthCheck struct {
-	Success    bool           `json:"success"`
-	Updated    int64          `json:"updated"`
-	Apps       AppHealth      `json:"apps"`
-	Workflows  WorkflowHealth `json:"workflows"`
+	Success    bool            `json:"success"`
+	Updated    int64           `json:"updated"`
+	Apps       AppHealth       `json:"apps"`
+	Workflows  WorkflowHealth  `json:"workflows"`
 	PythonApps AppHealth       `json:"python_apps"`
-	Datastore  DatastoreHealth	`json:"datastore"`
-	FileOps	   FileHealth		`json:"fileops"`
+	Datastore  DatastoreHealth `json:"datastore"`
+	FileOps    FileHealth      `json:"fileops"`
 }
 
 type HealthCheckDB struct {