Remove WebToolbarDelegate from GoogleLandingDataSource.

ios/chrome/browser/ui/ntp/google_landing_mediator.mm

 // 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.
 
 #import "ios/chrome/browser/ui/ntp/google_landing_mediator.h"
 
 #import "base/ios/weak_nsobject.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/metrics/user_metrics.h"
 #include "base/metrics/user_metrics_action.h"
@@ skipping 17 matching lines @@
 #import "ios/chrome/browser/ui/commands/UIKit+ChromeExecuteCommand.h"
 #import "ios/chrome/browser/ui/commands/generic_chrome_command.h"
 #import "ios/chrome/browser/ui/ntp/google_landing_consumer.h"
 #import "ios/chrome/browser/ui/ntp/notification_promo_whats_new.h"
 #import "ios/chrome/browser/ui/toolbar/web_toolbar_controller.h"
 #import "ios/chrome/browser/ui/url_loader.h"
 #import "ios/chrome/browser/web_state_list/web_state_list.h"
 #import "ios/chrome/browser/web_state_list/web_state_list_observer_bridge.h"
 #include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
 #include "ios/public/provider/chrome/browser/voice/voice_search_provider.h"
+#include "ios/web/public/web_state/web_state.h"
 
 using base::UserMetricsAction;
 
 namespace {
 
 const NSInteger kMaxNumMostVisitedFavicons = 8;
 
 }  // namespace
 
 @interface GoogleLandingMediator (UsedBySearchEngineObserver)
@@ skipping 120 matching lines @@
 }
 
 - (void)setUp {
   [_consumer setIsOffTheRecord:_browserState->IsOffTheRecord()];
   [_consumer setVoiceSearchIsEnabled:ios::GetChromeBrowserProvider()
                                          ->GetVoiceSearchProvider()
                                          ->IsVoiceSearchEnabled()];
   [_consumer
       setMaximumMostVisitedSitesShown:[GoogleLandingMediator maxSitesShown]];
   [_consumer setTabCount:self.webStateList->count()];
+  web::WebState* webState = _webStateList->GetActiveWebState();
+  if (webState) {
+    web::NavigationManager* nav = webState->GetNavigationManager();
+    [_consumer setCanGoForward:nav->CanGoForward()];
+    [_consumer setCanGoBack:nav->CanGoBack()];
+  }
 
   // Set up template URL service to listen for default search engine changes.
   _templateURLService =
       ios::TemplateURLServiceFactory::GetForBrowserState(_browserState);
   _observer.reset(
       new google_landing::SearchEngineObserver(self, _templateURLService));
   _templateURLService->Load();
   _doodleController.reset(ios::GetChromeBrowserProvider()->CreateLogoVendor(
       _browserState, _loader));
   [_consumer setLogoVendor:_doodleController];
@@ skipping 83 matching lines @@
               atIndex:(int)index {
   [self.consumer setTabCount:self.webStateList->count()];
 }
 
 - (void)webStateList:(WebStateList*)webStateList
     didDetachWebState:(web::WebState*)webState
               atIndex:(int)atIndex {
   [self.consumer setTabCount:self.webStateList->count()];
 }
 
+// If the actual webState associated with this mediator were passed in, this
+// would not be necessary.  However, since the active webstate can change when
+// the new tab page is created (and animated in), listen for changes here and
+// always display what's active.
+- (void)webStateList:(WebStateList*)webStateList
+    didChangeActiveWebState:(web::WebState*)newWebState
+                oldWebState:(web::WebState*)oldWebState
+                    atIndex:(int)atIndex
+                 userAction:(BOOL)userAction {
+  if (newWebState) {
+    web::NavigationManager* nav = newWebState->GetNavigationManager();
+    [self.consumer setCanGoForward:nav->CanGoForward()];
+    [self.consumer setCanGoBack:nav->CanGoBack()];
+  }
+}
+
 #pragma mark - GoogleLandingDataSource
 
 - (void)addBlacklistedURL:(const GURL&)url {
   _mostVisitedSites->AddOrRemoveBlacklistedUrl(url, true);
 }
 
 - (void)removeBlacklistedURL:(const GURL&)url {
   _mostVisitedSites->AddOrRemoveBlacklistedUrl(url, false);
 }
 
@@ skipping 19 matching lines @@
 }
 
 - (LargeIconCache*)largeIconCache {
   return IOSChromeLargeIconCacheFactory::GetForBrowserState(_browserState);
 }
 
 - (favicon::LargeIconService*)largeIconService {
   return IOSChromeLargeIconServiceFactory::GetForBrowserState(_browserState);
 }
 
-- (id<WebToolbarDelegate>)toolbarDelegate {
-  return _webToolbarDelegate;
-}
-
 - (void)promoViewed {
   DCHECK(_notification_promo);
   _notification_promo->HandleViewed();
   [self.consumer setPromoCanShow:_notification_promo->CanShow()];
 }
 
 - (void)promoTapped {
   DCHECK(_notification_promo);
   _notification_promo->HandleClosed();
   [self.consumer setPromoCanShow:_notification_promo->CanShow()];
 
   if (_notification_promo->IsURLPromo()) {
     [_loader webPageOrderedOpen:_notification_promo->url()
                        referrer:web::Referrer()
                    inBackground:NO
                        appendTo:kCurrentTab];
     return;
   }
 
   if (_notification_promo->IsChromeCommand()) {
     base::scoped_nsobject<GenericChromeCommand> command(
         [[GenericChromeCommand alloc]
             initWithTag:_notification_promo->command_id()]);
     [self.consumer chromeExecuteCommand:command];
     return;
   }
   NOTREACHED();
 }
 
+- (void)prepareToEnterTabSwitcher:(id)sender {
+  [_webToolbarDelegate prepareToEnterTabSwitcher:sender];
+}
+
 #pragma mark - UrlLoader
 
 - (void)loadURL:(const GURL&)url
              referrer:(const web::Referrer&)referrer
            transition:(ui::PageTransition)transition
     rendererInitiated:(BOOL)rendererInitiated {
   [_loader loadURL:url
                referrer:referrer
              transition:transition
       rendererInitiated:rendererInitiated];
@@ skipping 45 matching lines @@
 
 - (void)onFakeboxBlur {
   [_focuser onFakeboxBlur];
 }
 
 - (void)onFakeboxAnimationComplete {
   [_focuser onFakeboxAnimationComplete];
 }
 
 @end