Test distillability without JavaScript

components/dom_distiller/content/browser/distillability_driver.h

+// 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_