[ios] Switches ToolbarModelDelegateIOS to use WebStateList.

ios/chrome/browser/ui/toolbar/toolbar_model_delegate_ios.mm

 // Copyright 2012 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/toolbar/toolbar_model_delegate_ios.h"
 
 #include "base/logging.h"
 #include "components/omnibox/browser/autocomplete_classifier.h"
 #include "components/omnibox/browser/autocomplete_input.h"
 #include "components/omnibox/browser/autocomplete_match.h"
 #include "components/prefs/pref_service.h"
 #include "ios/chrome/browser/autocomplete/autocomplete_scheme_classifier_impl.h"
 #include "ios/chrome/browser/browser_state/chrome_browser_state.h"
 #include "ios/chrome/browser/chrome_url_constants.h"
 #include "ios/chrome/browser/pref_names.h"
 #include "ios/chrome/browser/ssl/ios_security_state_tab_helper.h"
-#import "ios/chrome/browser/tabs/tab.h"
-#import "ios/chrome/browser/tabs/tab_model.h"
+#include "ios/shared/chrome/browser/tabs/web_state_list.h"
 #import "ios/web/public/navigation_item.h"
+#import "ios/web/public/navigation_manager.h"
+#import "ios/web/public/ssl_status.h"
 #import "ios/web/public/web_state/web_state.h"
 
-ToolbarModelDelegateIOS::ToolbarModelDelegateIOS(TabModel* tab_model)
-    : tab_model_([tab_model retain]) {}
+ToolbarModelDelegateIOS::ToolbarModelDelegateIOS(WebStateList* web_state_list)
+    : web_state_list_(web_state_list) {}

[sdefresne, 2017/03/24 15:09:52]

I think we should DCHECK that web_state_list_ is not null here (or remove the
DCHECK in SetWebStateList). WDYT?

[rohitrao (ping after 24h), 2017/03/24 15:40:21]

I don't think anyone calls SetWebStateList(), so I've deleted it.  I didn't
bother DCHECKing().

 
 ToolbarModelDelegateIOS::~ToolbarModelDelegateIOS() {}
 
-void ToolbarModelDelegateIOS::SetTabModel(TabModel* tab_model) {
-  DCHECK(tab_model);
-  tab_model_.reset([tab_model retain]);
+void ToolbarModelDelegateIOS::SetWebStateList(WebStateList* web_state_list) {
+  DCHECK(web_state_list);
+  web_state_list_ = web_state_list;
 }
 
-Tab* ToolbarModelDelegateIOS::GetCurrentTab() const {
-  return [tab_model_ currentTab];
+web::WebState* ToolbarModelDelegateIOS::GetActiveWebState() const {
+  return web_state_list_->GetActiveWebState();
 }
 
 web::NavigationItem* ToolbarModelDelegateIOS::GetNavigationItem() const {
-  web::WebState* web_state = [GetCurrentTab() webState];
+  web::WebState* web_state = GetActiveWebState();
   web::NavigationManager* navigation_manager =
       web_state ? web_state->GetNavigationManager() : nullptr;
   return navigation_manager ? navigation_manager->GetVisibleItem() : nullptr;
 }
 
 base::string16 ToolbarModelDelegateIOS::FormattedStringWithEquivalentMeaning(
     const GURL& url,
     const base::string16& formatted_url) const {
   return AutocompleteInput::FormattedStringWithEquivalentMeaning(
       url, formatted_url, AutocompleteSchemeClassifierImpl());
@@ skipping 19 matching lines @@
         url = virtual_url;
       const std::string host = url.host();
       return host != kChromeUIBookmarksHost && host != kChromeUINewTabHost;
     }
   }
   return true;
 }
 
 security_state::SecurityLevel ToolbarModelDelegateIOS::GetSecurityLevel()
     const {
-  web::WebState* web_state = [GetCurrentTab() webState];
+  web::WebState* web_state = GetActiveWebState();
   // If there is no active WebState (which can happen during toolbar
   // initialization), assume no security style.
   if (!web_state)
     return security_state::NONE;
   auto* client = IOSSecurityStateTabHelper::FromWebState(web_state);
   security_state::SecurityInfo result;
   client->GetSecurityInfo(&result);
   return result.security_level;
 }
 
 scoped_refptr<net::X509Certificate> ToolbarModelDelegateIOS::GetCertificate()
     const {
   web::NavigationItem* item = GetNavigationItem();
   if (item)
     return item->GetSSL().certificate;
   return scoped_refptr<net::X509Certificate>();
 }
 
 bool ToolbarModelDelegateIOS::FailsMalwareCheck() const {
-  web::WebState* web_state = [GetCurrentTab() webState];
+  web::WebState* web_state = GetActiveWebState();
   // If there is no active WebState (which can happen during toolbar
   // initialization), so nothing can fail.
   if (!web_state)
     return NO;
   auto* client = IOSSecurityStateTabHelper::FromWebState(web_state);
   security_state::SecurityInfo result;
   client->GetSecurityInfo(&result);
   return result.malicious_content_status !=
          security_state::MALICIOUS_CONTENT_STATUS_NONE;
 }
 
 const gfx::VectorIcon* ToolbarModelDelegateIOS::GetVectorIconOverride() const {
   return nullptr;
 }