diff --git a/src/com/android/settings/location/RecentLocationAccessPreferenceController.java b/src/com/android/settings/location/RecentLocationAccessPreferenceController.java index 18221d0f409..f4044ed03d1 100644 --- a/src/com/android/settings/location/RecentLocationAccessPreferenceController.java +++ b/src/com/android/settings/location/RecentLocationAccessPreferenceController.java @@ -131,12 +131,16 @@ public class RecentLocationAccessPreferenceController extends LocationBasePrefer banner.setSelectable(false); mCategoryRecentLocationRequests.addPreference(banner); } + + if (mUiBlockListener != null) { + mUiBlockListener.onBlockerWorkFinished(this); + } } @Override public void onLocationModeChanged(int mode, boolean restricted) { boolean enabled = mLocationEnabler.isEnabled(mode); - mCategoryRecentLocationRequests.setVisible(enabled); + updatePreferenceVisibilityDelegate(mCategoryRecentLocationRequests, enabled); } /** diff --git a/src/com/android/settings/location/RecentLocationAccessSeeAllButtonPreferenceController.java b/src/com/android/settings/location/RecentLocationAccessSeeAllButtonPreferenceController.java index 68cde637eeb..b0dfae40c31 100644 --- a/src/com/android/settings/location/RecentLocationAccessSeeAllButtonPreferenceController.java +++ b/src/com/android/settings/location/RecentLocationAccessSeeAllButtonPreferenceController.java @@ -20,11 +20,13 @@ import android.content.Context; import androidx.preference.Preference; import androidx.preference.PreferenceScreen; +import com.android.settings.core.BasePreferenceController; + /** * Preference controller that handles the "See All" button for recent location access. */ public class RecentLocationAccessSeeAllButtonPreferenceController extends - LocationBasePreferenceController { + LocationBasePreferenceController implements BasePreferenceController.UiBlocker { private Preference mPreference; @@ -44,6 +46,6 @@ public class RecentLocationAccessSeeAllButtonPreferenceController extends @Override public void onLocationModeChanged(int mode, boolean restricted) { boolean enabled = mLocationEnabler.isEnabled(mode); - mPreference.setVisible(enabled); + updatePreferenceVisibilityDelegate(mPreference, enabled); } }