From 2375fcee52e55cdfef9786b93c2cfe5ba22443f9 Mon Sep 17 00:00:00 2001
From: Alfonso Campodonico <acampodonico@heroku.com>
Date: Fri, 6 Sep 2024 09:48:22 -0300
Subject: [PATCH] Add customDiff for sensitive config vars

---
 heroku/resource_heroku_app.go | 38 +++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/heroku/resource_heroku_app.go b/heroku/resource_heroku_app.go
index 4c63419c..267ba14d 100644
--- a/heroku/resource_heroku_app.go
+++ b/heroku/resource_heroku_app.go
@@ -181,6 +181,7 @@ func resourceHerokuApp() *schema.Resource {
 				Computed: true,
 			},
 		},
+		CustomizeDiff: customDiffSensitiveConfigVars(),
 		SchemaVersion: 1,
 		StateUpgraders: []schema.StateUpgrader{
 			{
@@ -1011,6 +1012,7 @@ func resourceHerokuAppV0() *schema.Resource {
 				Computed: true,
 			},
 		},
+		CustomizeDiff: customDiffSensitiveConfigVars(),
 	}
 }
 
@@ -1032,3 +1034,39 @@ func resourceHerokuAppStateUpgradeV0(ctx context.Context, rawState map[string]in
 
 	return rawState, nil
 }
+
+func customDiffSensitiveConfigVars() schema.CustomizeDiffFunc {
+	return func(ctx context.Context, d *schema.ResourceDiff, m interface{}) error {
+		old, new := d.GetChange("sensitive_config_vars")
+
+		changedKeys := diffMapKeys(old.(map[string]interface{}), new.(map[string]interface{}))
+
+		if len(changedKeys) > 0 {
+			d.SetNewComputed("sensitive_config_vars")
+			d.SetNew("changed_sensitive_config_var_keys", changedKeys)
+		}
+
+		return nil
+	}
+}
+
+// diffMapKeys returns the list of keys that differ between two maps
+func diffMapKeys(oldMap, newMap map[string]interface{}) []string {
+	changedKeys := []string{}
+
+	// Check for added or updated keys
+	for k, newVal := range newMap {
+		if oldVal, ok := oldMap[k]; !ok || oldVal != newVal {
+			changedKeys = append(changedKeys, k)
+		}
+	}
+
+	// Check for removed keys
+	for k := range oldMap {
+		if _, ok := newMap[k]; !ok {
+			changedKeys = append(changedKeys, k)
+		}
+	}
+
+	return changedKeys
+}