Inject CanAddURLToHistory into TopSitesImpl

components/history/core/browser/top_sites_impl.h

 // Copyright (c) 2013 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.
 
-#ifndef CHROME_BROWSER_HISTORY_TOP_SITES_IMPL_H_
-#define CHROME_BROWSER_HISTORY_TOP_SITES_IMPL_H_
+#ifndef COMPONENTS_HISTORY_CORE_BROWSER_TOP_SITES_IMPL_H_
+#define COMPONENTS_HISTORY_CORE_BROWSER_TOP_SITES_IMPL_H_
 
 #include <list>
 #include <set>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/gtest_prod_util.h"
@@ skipping 29 matching lines @@
 class HistoryService;
 class TopSitesCache;
 class TopSitesImplTest;
 
 // This class allows requests for most visited urls and thumbnails on any
 // thread. All other methods must be invoked on the UI thread. All mutations
 // to internal state happen on the UI thread and are scheduled to update the
 // db using TopSitesBackend.
 class TopSitesImpl : public TopSites, public HistoryServiceObserver {
  public:
+  // Called to check whether an URL can be added to the history. Must be
+  // callable multiple time and during the whole lifetime of TopSitesImpl.
+  using CanAddURLToHistoryFn = base::Callback<bool(const GURL&)>;
+
   TopSitesImpl(PrefService* pref_service,
                HistoryService* history_service,
                const char* blacklist_pref_name,
-               const PrepopulatedPageList& prepopulated_pages);
+               const PrepopulatedPageList& prepopulated_pages,
+               const CanAddURLToHistoryFn& can_add_url_to_history);
 
   // Initializes TopSitesImpl.
   void Init(const base::FilePath& db_name,
             const scoped_refptr<base::SingleThreadTaskRunner>& db_task_runner);
 
   // TopSites implementation.
   bool SetPageThumbnail(const GURL& url,
                         const gfx::Image& thumbnail,
                         const ThumbnailScore& score) override;
   bool SetPageThumbnailToJPEGBytes(const GURL& url,
@@ skipping 217 matching lines @@
   // TopSitesImpl.
   PrefService* pref_service_;
 
   // Key for the NTP URL blacklist dictionary in PrefService.
   const char* blacklist_pref_name_;
 
   // HistoryService that TopSitesImpl can query. May be null, but if defined it
   // must outlive TopSitesImpl.
   HistoryService* history_service_;
 
+  // Can URL be added to the history?
+  CanAddURLToHistoryFn can_add_url_to_history_;
+
   // Are we loaded?
   bool loaded_;
 
   // Have the SetTopSites execution time related histograms been recorded?
   // The histogram should only be recorded once for each Chrome execution.
   static bool histogram_recorded_;
 
   ScopedObserver<HistoryService, HistoryServiceObserver>
       history_service_observer_;
 
   DISALLOW_COPY_AND_ASSIGN(TopSitesImpl);
 };
 
 }  // namespace history
 
-#endif  // CHROME_BROWSER_HISTORY_TOP_SITES_IMPL_H_
+#endif  // COMPONENTS_HISTORY_CORE_BROWSER_TOP_SITES_IMPL_H_