📰 Refresh the tiles when a removal is undone

chrome/android/java/src/org/chromium/chrome/browser/suggestions/TileGroupDelegateImpl.java

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.suggestions;
 
 import android.annotation.TargetApi;
 import android.content.Context;
 import android.os.Build;
 
+import org.chromium.base.Callback;
 import org.chromium.base.CommandLine;
 import org.chromium.base.VisibleForTesting;
 import org.chromium.base.metrics.RecordUserAction;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.ChromeActivity;
 import org.chromium.chrome.browser.ChromeSwitches;
 import org.chromium.chrome.browser.ntp.NewTabPageUma;
 import org.chromium.chrome.browser.profiles.Profile;
 import org.chromium.chrome.browser.snackbar.Snackbar;
 import org.chromium.chrome.browser.snackbar.SnackbarManager;
@@ skipping 25 matching lines @@
     public TileGroupDelegateImpl(ChromeActivity activity, Profile profile,
             TabModelSelector tabModelSelector, SuggestionsNavigationDelegate navigationDelegate) {
         mContext = activity;
         mSnackbarManager = activity.getSnackbarManager();
         mTabModelSelector = tabModelSelector;
         mNavigationDelegate = navigationDelegate;
         mMostVisitedSites = buildMostVisitedSites(profile);
     }
 
     @Override
-    public void removeMostVisitedItem(Tile item) {
+    public void removeMostVisitedItem(Tile item, Callback<String> removalUndoneCallback) {
         assert !mIsDestroyed;
 
         mMostVisitedSites.addBlacklistedUrl(item.getUrl());
-        showTileRemovedSnackbar(item.getUrl());
+        showTileRemovedSnackbar(item.getUrl(), removalUndoneCallback);
     }
 
     @Override
     public void openMostVisitedItem(int windowDisposition, Tile item) {
         assert !mIsDestroyed;
 
         String url = item.getUrl();
 
         // TODO(treib): Should we call recordOpenedMostVisitedItem here?
         if (windowDisposition != WindowOpenDisposition.NEW_WINDOW) {
@@ skipping 49 matching lines @@
         } else {
             return new MostVisitedSites(profile);
         }
     }
 
     @VisibleForTesting
     public static void setMostVisitedSitesForTests(MostVisitedSites mostVisitedSitesForTests) {
         sMostVisitedSitesForTests = mostVisitedSitesForTests;
     }
 
-    private void showTileRemovedSnackbar(String url) {
+    private void showTileRemovedSnackbar(String url, final Callback<String> removalUndoneCallback) {
         if (mTileRemovedSnackbarController == null) {
             mTileRemovedSnackbarController = new SnackbarController() {
                 @Override
                 public void onDismissNoAction(Object actionData) {}
 
                 /** Undoes the tile removal. */
                 @Override
                 public void onAction(Object actionData) {
                     if (mIsDestroyed) return;
                     String url = (String) actionData;
+                    removalUndoneCallback.onResult(url);
                     mMostVisitedSites.removeBlacklistedUrl(url);
                 }
             };
         }
         Snackbar snackbar = Snackbar.make(mContext.getString(R.string.most_visited_item_removed),
                                             mTileRemovedSnackbarController, Snackbar.TYPE_ACTION,
                                             Snackbar.UMA_NTP_MOST_VISITED_DELETE_UNDO)
                                     .setAction(mContext.getString(R.string.undo), url);
         mSnackbarManager.showSnackbar(snackbar);
     }
@@ skipping 29 matching lines @@
             }
         }
         return false;
     }
 
     private static boolean matchURLs(String url1, String url2, boolean matchByHost) {
         if (url1 == null || url2 == null) return false;
         return matchByHost ? UrlUtilities.sameHost(url1, url2) : url1.equals(url2);
     }
 }