Index: components/dom_distiller/content/browser/distillability_driver.h |
diff --git a/components/dom_distiller/content/browser/distillability_driver.h b/components/dom_distiller/content/browser/distillability_driver.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..5fe2d59d3e2a58f60efb57e8cdfe0f8da89b0e84 |
--- /dev/null |
+++ b/components/dom_distiller/content/browser/distillability_driver.h |
@@ -0,0 +1,58 @@ |
+// Copyright 2015 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 COMPONENTS_DOM_DISTILLER_CONTENT_BROWSER_DISTILLIBILITY_DRIVER_H_ |
+#define COMPONENTS_DOM_DISTILLER_CONTENT_BROWSER_DISTILLIBILITY_DRIVER_H_ |
+ |
+#include "content/public/browser/web_contents_observer.h" |
+#include "content/public/browser/web_contents_user_data.h" |
+ |
+struct FrameHostMsg_ExtractFeatureResponse_Params; |
+ |
+namespace dom_distiller { |
+ |
+// Extract the features of a page for DOM distiller. |
+typedef base::Callback<void(bool isOGArticle, |
+ const std::string& url, |
+ int numElements, |
+ int numAnchors, |
+ int numForms, |
+ const std::string& innerText, |
+ const std::string& textContent, |
+ const std::string& innerHTML)> |
+ ExtractFeatureCallback; |
+ |
+// This is an IPC helper for determining whether a page should be distilled. |
+class WebContentsDistillabilityDriver |
+ : public content::WebContentsObserver, |
+ public content::WebContentsUserData<WebContentsDistillabilityDriver> { |
+ public: |
+ ~WebContentsDistillabilityDriver() override; |
+ |
+ // content::WebContentsObserver implementation. |
+ bool OnMessageReceived(const IPC::Message& message, |
+ content::RenderFrameHost* rfh) override; |
+ void RenderProcessGone(base::TerminationStatus status) override; |
+ |
+ void OnExtractFeatureResponse(int id, |
+ FrameHostMsg_ExtractFeatureResponse_Params results); |
+ |
+ void ExtractFeatures(const ExtractFeatureCallback& callback); |
+ |
+ private: |
+ explicit WebContentsDistillabilityDriver(content::WebContents* web_contents); |
+ friend class content::WebContentsUserData<WebContentsDistillabilityDriver>; |
+ |
+ // Removes the observer and clears the WebContents member. |
+ void CleanUp(); |
+ |
+ int callback_id_; |
+ std::map<int, ExtractFeatureCallback> callbacks_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(WebContentsDistillabilityDriver); |
+}; |
+ |
+} // namespace dom_distiller |
+ |
+#endif // COMPONENTS_DOM_DISTILLER_CONTENT_BROWSER_DISTILLIBILITY_DRIVER_H_ |