Settings : Recompute highlight index upon changes

Change-Id: I7d738f5114fc1eea564360ab5e823abc167caee9

1 files changed, 6 insertions, 5 deletions

diff --git a/src/com/android/settings/SettingsPreferenceFragment.java b/src/com/android/settings/SettingsPreferenceFragment.java
index 1711671..02f8a03 100644
--- a/src/com/android/settings/SettingsPreferenceFragment.java
+++ b/src/com/android/settings/SettingsPreferenceFragment.java
@@ -72,12 +72,12 @@ public abstract class SettingsPreferenceFragment extends InstrumentedPreferenceF
     private DataSetObserver mDataSetObserver = new DataSetObserver() {
         @Override
         public void onChanged() {
-            highlightPreferenceIfNeeded();
+            highlightPreferenceIfNeeded(true);
         }
 
         @Override
         public void onInvalidated() {
-            highlightPreferenceIfNeeded();
+            highlightPreferenceIfNeeded(true);
         }
     };
 
@@ -149,7 +149,7 @@ public abstract class SettingsPreferenceFragment extends InstrumentedPreferenceF
         final Bundle args = getArguments();
         if (args != null) {
             mPreferenceKey = args.getString(SettingsActivity.EXTRA_FRAGMENT_ARG_KEY);
-            highlightPreferenceIfNeeded();
+            highlightPreferenceIfNeeded(false);
         }
     }
 
@@ -196,8 +196,9 @@ public abstract class SettingsPreferenceFragment extends InstrumentedPreferenceF
         }
     }
 
-    public void highlightPreferenceIfNeeded() {
-        if (isAdded() && !mPreferenceHighlighted &&!TextUtils.isEmpty(mPreferenceKey)) {
+    public void highlightPreferenceIfNeeded(boolean forceHighlight) {
+        if (isAdded() && (!mPreferenceHighlighted || forceHighlight)
+                && !TextUtils.isEmpty(mPreferenceKey)) {
             highlightPreference(mPreferenceKey);
         }
     }