Android WebView visited link highlighting implementation part 1

android_webview/browser/aw_browser_context.h

 // Copyright (c) 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.
 
 #ifndef ANDROID_WEBVIEW_BROWSER_AW_BROWSER_CONTEXT_H_
 #define ANDROID_WEBVIEW_BROWSER_AW_BROWSER_CONTEXT_H_
 
+#include <vector>
+
 #include "android_webview/browser/aw_download_manager_delegate.h"
 #include "base/file_path.h"
 #include "base/memory/ref_counted.h"
+#include "components/visitedlink/browser/visitedlink_delegate.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/geolocation_permission_context.h"
 
+class GURL;
+
+namespace components {
+class VisitedLinkMaster;
+}  // namespace components
+
+namespace content {
+class WebContents;
+}  // namespace content
+
 namespace android_webview {
 
 class AwURLRequestContextGetter;
 
 typedef content::GeolocationPermissionContext* GeolocationPermissionFactoryFn();
 
-class AwBrowserContext : public content::BrowserContext {
+class AwBrowserContext : public content::BrowserContext,
+                         public components::VisitedLinkDelegate {
  public:
+
   AwBrowserContext(
       const FilePath path,
       GeolocationPermissionFactoryFn* geolocation_permission_factory);
   virtual ~AwBrowserContext();
 
+  // Convenience method to returns the AwBrowserContext corresponding to the
+  // given WebContents.
+  static AwBrowserContext* FromWebContents(
+      content::WebContents* web_contents);
+
   // Called before BrowserThreads are created.
   void InitializeBeforeThreadCreation();
 
+  // Maps to BrowserMainParts::PreMainMessageLoopRun.
+  void PreMainMessageLoopRun();
+
+  // These methods map to Add methods in components::VisitedLinkMaster.
+  void AddVisitedURL(const GURL& url);
+  void AddVisitedURLs(const std::vector<GURL>& urls);
+
   // content::BrowserContext implementation.
   virtual FilePath GetPath() OVERRIDE;
   virtual bool IsOffTheRecord() const OVERRIDE;
   virtual net::URLRequestContextGetter* GetRequestContext() OVERRIDE;
   virtual net::URLRequestContextGetter* GetRequestContextForRenderProcess(
       int renderer_child_id) OVERRIDE;
   virtual net::URLRequestContextGetter* GetRequestContextForStoragePartition(
       const FilePath& partition_path, bool in_memory) OVERRIDE;
   virtual net::URLRequestContextGetter* GetMediaRequestContext() OVERRIDE;
   virtual net::URLRequestContextGetter* GetMediaRequestContextForRenderProcess(
       int renderer_child_id) OVERRIDE;
   virtual net::URLRequestContextGetter*
       GetMediaRequestContextForStoragePartition(
           const FilePath& partition_path, bool in_memory) OVERRIDE;
   virtual content::ResourceContext* GetResourceContext() OVERRIDE;
   virtual content::DownloadManagerDelegate*
       GetDownloadManagerDelegate() OVERRIDE;
   virtual content::GeolocationPermissionContext*
       GetGeolocationPermissionContext() OVERRIDE;
   virtual content::SpeechRecognitionPreferences*
       GetSpeechRecognitionPreferences() OVERRIDE;
   virtual quota::SpecialStoragePolicy* GetSpecialStoragePolicy() OVERRIDE;
 
+  // components::VisitedLinkDelegate implementation.
+  virtual void RebuildTable(
+      const scoped_refptr<URLEnumerator>& enumerator) OVERRIDE;
+
  private:
 
   // The file path where data for this context is persisted.
   FilePath context_storage_path_;
 
   scoped_refptr<AwURLRequestContextGetter> url_request_context_getter_;
   GeolocationPermissionFactoryFn* geolocation_permission_factory_;
   scoped_refptr<content::GeolocationPermissionContext>
       geolocation_permission_context_;
 
   AwDownloadManagerDelegate download_manager_delegate_;
 
+  scoped_ptr<components::VisitedLinkMaster> visitedlink_master_;
+
   DISALLOW_COPY_AND_ASSIGN(AwBrowserContext);
 };
 
 }  // namespace android_webview
 
 #endif  // ANDROID_WEBVIEW_BROWSER_AW_BROWSER_CONTEXT_H_