Make ChromeSubresourceFilterClient a WebContentsUserData

chrome/browser/subresource_filter/chrome_subresource_filter_client.h

 // Copyright 2016 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_SUBRESOURCE_FILTER_CHROME_SUBRESOURCE_FILTER_CLIENT_H_
 #define CHROME_BROWSER_SUBRESOURCE_FILTER_CHROME_SUBRESOURCE_FILTER_CLIENT_H_
 
 #include <set>
 #include <string>
 
 #include "base/macros.h"
 #include "components/content_settings/core/common/content_settings.h"
 #include "components/subresource_filter/content/browser/subresource_filter_client.h"
+#include "content/public/browser/web_contents_user_data.h"
 
 class GURL;
 
 namespace content {
 class NavigationHandle;
 class WebContents;
 }  // namespace content
 
 // This enum backs a histogram. Make sure new elements are only added to the
 // end. Keep histograms.xml up to date with any changes.
@@ skipping 30 matching lines @@
   // problem because this can only be changed from the settings UI, which is
   // relatively rare.
   // TODO(crbug.com/706061): Fix this once content settings API becomes more
   // flexible.
   kActionContentSettingsWildcardUpdate,
 
   kActionLastEntry
 };
 
 // Chrome implementation of SubresourceFilterClient.
+// TODO(csharrison): Make this a WebContentsObserver and own the throttle
+// manager directly.
 class ChromeSubresourceFilterClient
-    : public subresource_filter::SubresourceFilterClient {
+    : public content::WebContentsUserData<ChromeSubresourceFilterClient>,
+      public subresource_filter::SubresourceFilterClient {
  public:
   explicit ChromeSubresourceFilterClient(content::WebContents* web_contents);
   ~ChromeSubresourceFilterClient() override;
 
   // SubresourceFilterClient:
   void ToggleNotificationVisibility(bool visibility) override;
   bool ShouldSuppressActivation(
       content::NavigationHandle* navigation_handle) override;
   void WhitelistByContentSettings(const GURL& url) override;
   void WhitelistInCurrentWebContents(const GURL& url) override;
   subresource_filter::VerifiedRulesetDealer::Handle* GetRulesetDealer()
       override;
 
   bool did_show_ui_for_navigation() const {
     return did_show_ui_for_navigation_;
   }
 
   static void LogAction(SubresourceFilterAction action);
 
  private:
   ContentSetting GetContentSettingForUrl(const GURL& url);
   std::set<std::string> whitelisted_hosts_;
   content::WebContents* web_contents_;
   bool did_show_ui_for_navigation_;
 
   DISALLOW_COPY_AND_ASSIGN(ChromeSubresourceFilterClient);
 };
 
 #endif  // CHROME_BROWSER_SUBRESOURCE_FILTER_CHROME_SUBRESOURCE_FILTER_CLIENT_H_