Refactor SecurityStateModel/Clients for simplicity and reusability.

chrome/browser/ui/browser.cc

Index: chrome/browser/ui/browser.cc
diff --git a/chrome/browser/ui/browser.cc b/chrome/browser/ui/browser.cc
index 4c02948d4ae9543a60c89f9fa0b97db30fb0ee48..89f16cb398cfe801a7dfd928c3de968f466cd83b 100644
--- a/chrome/browser/ui/browser.cc
+++ b/chrome/browser/ui/browser.cc
@@ -79,7 +79,7 @@
 #include "chrome/browser/sessions/session_service_factory.h"
 #include "chrome/browser/sessions/session_tab_helper.h"
 #include "chrome/browser/sessions/tab_restore_service_factory.h"
-#include "chrome/browser/ssl/chrome_security_state_model_client.h"
+#include "chrome/browser/ssl/chrome_security_state_model.h"
 #include "chrome/browser/sync/profile_sync_service_factory.h"
 #include "chrome/browser/sync/sync_ui_util.h"
 #include "chrome/browser/tab_contents/retargeting_details.h"
@@ -167,7 +167,8 @@
 #include "components/history/core/browser/top_sites.h"
 #include "components/prefs/pref_service.h"
 #include "components/search/search.h"
-#include "components/security_state/security_state_model.h"
+#include "components/security_state/content/web_contents_security_state_model.h"
+#include "components/security_state/core/security_state_model.h"
 #include "components/sessions/core/session_types.h"
 #include "components/sessions/core/tab_restore_service.h"
 #include "components/startup_metric_utils/browser/startup_metric_utils.h"
@@ -1279,13 +1280,13 @@ bool Browser::CanDragEnter(content::WebContents* source,
 blink::WebSecurityStyle Browser::GetSecurityStyle(
     WebContents* web_contents,
     content::SecurityStyleExplanations* security_style_explanations) {
-  ChromeSecurityStateModelClient* model_client =
-      ChromeSecurityStateModelClient::FromWebContents(web_contents);
-  DCHECK(model_client);
+  ChromeSecurityStateModel* model =
+      ChromeSecurityStateModel::FromWebContents(web_contents);
+  DCHECK(model);
   security_state::SecurityStateModel::SecurityInfo security_info;
-  model_client->GetSecurityInfo(&security_info);
-  return model_client->GetSecurityStyle(security_info,
-                                        security_style_explanations);
+  model->GetSecurityInfo(&security_info);
+  return security_state::WebContentsSecurityStateModel::GetSecurityStyle(
+      security_info, security_style_explanations);
 }
 
 void Browser::ShowCertificateViewerInDevTools(
@@ -1431,8 +1432,8 @@ void Browser::VisibleSSLStateChanged(WebContents* source) {
   if (tab_strip_model_->GetActiveWebContents() == source)
     UpdateToolbar(false);
 
-  ChromeSecurityStateModelClient* security_model =
-      ChromeSecurityStateModelClient::FromWebContents(source);
+  ChromeSecurityStateModel* security_model =
+      ChromeSecurityStateModel::FromWebContents(source);
   security_model->VisibleSSLStateChanged();
 }