OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "ios/chrome/browser/search/search_util.h" | |
6 | |
7 #include "components/search/search.h" | |
8 #include "components/search_engines/template_url_service.h" | |
9 #include "ios/chrome/browser/browser_state/chrome_browser_state.h" | |
10 #include "ios/chrome/browser/search_engines/template_url_service_factory.h" | |
11 #include "ios/chrome/browser/search_engines/ui_thread_search_terms_data.h" | |
12 #include "ios/web/public/navigation_item.h" | |
13 #import "ios/web/public/navigation_manager.h" | |
14 #include "ios/web/public/web_state/web_state.h" | |
15 #include "url/gurl.h" | |
16 | |
17 namespace { | |
18 | |
19 TemplateURL* GetDefaultSearchProviderTemplateURL( | |
20 ios::ChromeBrowserState* browser_state) { | |
21 if (!browser_state) | |
22 return nullptr; | |
23 | |
24 TemplateURLService* template_url_service = | |
25 ios::TemplateURLServiceFactory::GetForBrowserState(browser_state); | |
26 return template_url_service ? template_url_service->GetDefaultSearchProvider() | |
27 : nullptr; | |
28 } | |
29 | |
30 } // namespace | |
31 | |
32 base::string16 ExtractSearchTermsFromURL(ios::ChromeBrowserState* browser_state, | |
33 const GURL& url) { | |
34 base::string16 search_terms; | |
35 TemplateURL* template_url = | |
36 GetDefaultSearchProviderTemplateURL(browser_state); | |
37 if (template_url) { | |
38 template_url->ExtractSearchTermsFromURL( | |
39 url, ios::UIThreadSearchTermsData(browser_state), &search_terms); | |
40 } | |
41 return search_terms; | |
42 } | |
43 | |
44 bool IsQueryExtractionAllowedForURL(ios::ChromeBrowserState* browser_state, | |
45 const GURL& url) { | |
46 TemplateURL* template_url = | |
47 GetDefaultSearchProviderTemplateURL(browser_state); | |
48 return template_url && search::IsSuitableURLForInstant(url, template_url); | |
49 } | |
50 | |
51 base::string16 GetSearchTerms(web::WebState* web_state) { | |
52 if (!web_state) | |
53 return base::string16(); | |
54 | |
55 web::NavigationItem* item = | |
56 web_state->GetNavigationManager()->GetVisibleItem(); | |
57 if (!item) | |
58 return base::string16(); | |
59 | |
60 if (!search::IsQueryExtractionEnabled()) | |
61 return base::string16(); | |
62 | |
63 ios::ChromeBrowserState* browser_state = | |
64 ios::ChromeBrowserState::FromBrowserState(web_state->GetBrowserState()); | |
65 if (!IsQueryExtractionAllowedForURL(browser_state, item->GetVirtualURL())) | |
66 return base::string16(); | |
67 | |
68 return ExtractSearchTermsFromURL(browser_state, item->GetVirtualURL()); | |
69 } | |
OLD | NEW |