| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/dom_distiller/content/browser/distillable_page_utils.h" | 5 #include "components/dom_distiller/content/browser/distillable_page_utils.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
| 10 #include "base/strings/utf_string_conversions.h" | 10 #include "base/strings/utf_string_conversions.h" |
| 11 #include "base/thread_task_runner_handle.h" | 11 #include "base/thread_task_runner_handle.h" |
| 12 #include "base/values.h" | 12 #include "base/values.h" |
| 13 #include "components/dom_distiller/content/browser/distillability_driver.h" |
| 13 #include "components/dom_distiller/content/browser/distiller_javascript_utils.h" | 14 #include "components/dom_distiller/content/browser/distiller_javascript_utils.h" |
| 15 #include "components/dom_distiller/content/common/distiller_messages.h" |
| 14 #include "components/dom_distiller/core/distillable_page_detector.h" | 16 #include "components/dom_distiller/core/distillable_page_detector.h" |
| 15 #include "components/dom_distiller/core/experiments.h" | 17 #include "components/dom_distiller/core/experiments.h" |
| 16 #include "components/dom_distiller/core/page_features.h" | 18 #include "components/dom_distiller/core/page_features.h" |
| 17 #include "content/public/browser/render_frame_host.h" | 19 #include "content/public/browser/render_frame_host.h" |
| 18 #include "grit/components_resources.h" | 20 #include "grit/components_resources.h" |
| 19 #include "ui/base/resource/resource_bundle.h" | 21 #include "ui/base/resource/resource_bundle.h" |
| 20 | 22 |
| 21 namespace dom_distiller { | 23 namespace dom_distiller { |
| 22 namespace { | 24 namespace { |
| 25 |
| 23 void OnOGArticleJsResult(base::Callback<void(bool)> callback, | 26 void OnOGArticleJsResult(base::Callback<void(bool)> callback, |
| 24 const base::Value* result) { | 27 const base::Value* result) { |
| 25 bool success = false; | 28 bool success = false; |
| 26 if (result) { | 29 if (result) { |
| 27 result->GetAsBoolean(&success); | 30 result->GetAsBoolean(&success); |
| 28 } | 31 } |
| 29 callback.Run(success); | 32 callback.Run(success); |
| 30 } | 33 } |
| 34 /* |
| 35 void OnExtractFeaturesNativeResult( |
| 36 const DistillablePageDetector* detector, |
| 37 base::Callback<void(bool)> callback, |
| 38 bool isOGArticle, |
| 39 const std::string& url, |
| 40 int numElements, |
| 41 int numAnchors, |
| 42 int numForms, |
| 43 const std::string& innerText, |
| 44 const std::string& textContent, |
| 45 const std::string& innerHTML) { |
| 31 | 46 |
| 32 void OnExtractFeaturesJsResult(const DistillablePageDetector* detector, | 47 GURL parsed_url(url); |
| 33 base::Callback<void(bool)> callback, | 48 if (!parsed_url.is_valid()) { |
| 34 const base::Value* result) { | 49 callback.Run(false); |
| 35 callback.Run(detector->Classify(CalculateDerivedFeaturesFromJSON(result))); | 50 return; |
| 51 } |
| 52 |
| 53 auto res = CalculateDerivedFeatures( |
| 54 isOGArticle, |
| 55 parsed_url, |
| 56 numElements, |
| 57 numAnchors, |
| 58 numForms, |
| 59 innerText, |
| 60 textContent, |
| 61 innerHTML |
| 62 ); |
| 63 callback.Run(detector->Classify(res)); |
| 36 } | 64 } |
| 65 */ |
| 37 } // namespace | 66 } // namespace |
| 38 | 67 |
| 39 void IsOpenGraphArticle(content::WebContents* web_contents, | 68 void IsOpenGraphArticle(content::WebContents* web_contents, |
| 40 base::Callback<void(bool)> callback) { | 69 base::Callback<void(bool)> callback) { |
| 41 content::RenderFrameHost* main_frame = web_contents->GetMainFrame(); | 70 content::RenderFrameHost* main_frame = web_contents->GetMainFrame(); |
| 42 if (!main_frame) { | 71 if (!main_frame) { |
| 43 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, | 72 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 44 base::Bind(callback, false)); | 73 base::Bind(callback, false)); |
| 45 return; | 74 return; |
| 46 } | 75 } |
| (...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 82 | 111 |
| 83 void IsDistillablePageForDetector(content::WebContents* web_contents, | 112 void IsDistillablePageForDetector(content::WebContents* web_contents, |
| 84 const DistillablePageDetector* detector, | 113 const DistillablePageDetector* detector, |
| 85 base::Callback<void(bool)> callback) { | 114 base::Callback<void(bool)> callback) { |
| 86 content::RenderFrameHost* main_frame = web_contents->GetMainFrame(); | 115 content::RenderFrameHost* main_frame = web_contents->GetMainFrame(); |
| 87 if (!main_frame) { | 116 if (!main_frame) { |
| 88 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, | 117 base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| 89 base::Bind(callback, false)); | 118 base::Bind(callback, false)); |
| 90 return; | 119 return; |
| 91 } | 120 } |
| 92 std::string extract_features_js = | 121 |
| 93 ResourceBundle::GetSharedInstance() | 122 WebContentsDistillabilityDriver::CreateForWebContents(web_contents); |
| 94 .GetRawDataResource(IDR_EXTRACT_PAGE_FEATURES_JS) | 123 WebContentsDistillabilityDriver *driver = |
| 95 .as_string(); | 124 WebContentsDistillabilityDriver::FromWebContents(web_contents); |
| 96 RunIsolatedJavaScript( | 125 CHECK(driver); |
| 97 main_frame, extract_features_js, | 126 /* |
| 98 base::Bind(OnExtractFeaturesJsResult, detector, callback)); | 127 driver->ExtractFeatures( |
| 128 base::Bind(OnExtractFeaturesNativeResult, detector, callback)); |
| 129 */ |
| 99 } | 130 } |
| 100 | 131 |
| 101 } // namespace dom_distiller | 132 } // namespace dom_distiller |
| OLD | NEW |