1 // Copyright (c) 2012 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 #ifndef CHROME_BROWSER_SEARCH_ENGINES_UTIL_H_ 6 #define CHROME_BROWSER_SEARCH_ENGINES_UTIL_H_ 7 8 // This file contains utility functions for search engine functionality. 9 #include <set> 10 #include <string> 11 #include <vector> 12 13 #include "base/memory/scoped_ptr.h" 14 #include "base/strings/string16.h" 15 #include "chrome/browser/search_engines/template_url_service.h" 16 17 class Profile; 18 class TemplateURL; 19 class WDTypedResult; 20 class WebDataService; 21 22 // Returns the short name of the default search engine, or the empty string if 23 // none is set. |profile| may be NULL. 24 base::string16 GetDefaultSearchEngineName(Profile* profile); 25 26 // Returns a GURL that searches for |terms| using the default search engine of 27 // |profile|. 28 GURL GetDefaultSearchURLForSearchTerms(Profile* profile, 29 const base::string16& terms); 30 31 // Returns matching URL from |template_urls| or NULL. 32 TemplateURL* FindURLByPrepopulateID( 33 const TemplateURLService::TemplateURLVector& template_urls, 34 int prepopulate_id); 35 36 // Modifies |prepopulated_url| so that it contains user-modified fields from 37 // |original_turl|. Both URLs must have the same prepopulate_id. 38 void MergeIntoPrepopulatedEngineData(const TemplateURL* original_turl, 39 TemplateURLData* prepopulated_url); 40 41 // CreateActionsFromCurrentPrepopulateData() (see below) takes in the current 42 // prepopulated URLs as well as the user's current URLs, and returns an instance 43 // of the following struct representing the changes necessary to bring the 44 // user's URLs in line with the prepopulated URLs. 45 // 46 // There are three types of changes: 47 // (1) Previous prepopulated engines that no longer exist in the current set of 48 // prepopulated engines and thus should be removed from the user's current 49 // URLs. 50 // (2) Previous prepopulated engines whose data has changed. The existing 51 // entries for these engines should be updated to reflect the new data, 52 // except for any user-set names and keywords, which can be preserved. 53 // (3) New prepopulated engines not in the user's engine list, which should be 54 // added. 55 56 // The pair of current search engine and its new value. 57 typedef std::pair<TemplateURL*, TemplateURLData> EditedSearchEngine; 58 typedef std::vector<EditedSearchEngine> EditedEngines; 59 60 struct ActionsFromPrepopulateData { 61 ActionsFromPrepopulateData(); 62 ~ActionsFromPrepopulateData(); 63 64 TemplateURLService::TemplateURLVector removed_engines; 65 EditedEngines edited_engines; 66 std::vector<TemplateURLData> added_engines; 67 }; 68 69 // Given the user's current URLs and the current set of prepopulated URLs, 70 // produces the set of actions (see above) required to make the user's URLs 71 // reflect the prepopulate data. |default_search_provider| is used to avoid 72 // placing the current default provider on the "to be removed" list. 73 // 74 // NOTE: Takes ownership of, and clears, |prepopulated_urls|. 75 ActionsFromPrepopulateData CreateActionsFromCurrentPrepopulateData( 76 ScopedVector<TemplateURLData>* prepopulated_urls, 77 const TemplateURLService::TemplateURLVector& existing_urls, 78 const TemplateURL* default_search_provider); 79 80 // Processes the results of WebDataService::GetKeywords, combining it with 81 // prepopulated search providers to result in: 82 // * a set of template_urls (search providers). The caller owns the 83 // TemplateURL* returned in template_urls. 84 // * whether there is a new resource keyword version (and the value). 85 // |*new_resource_keyword_version| is set to 0 if no new value. Otherwise, 86 // it is the new value. 87 // Only pass in a non-NULL value for service if the WebDataService should be 88 // updated. If |removed_keyword_guids| is not NULL, any TemplateURLs removed 89 // from the keyword table in the WebDataService will have their Sync GUIDs 90 // added to it. |default_search_provider| will be used to prevent removing the 91 // current user-selected DSE, regardless of changes in prepopulate data. 92 void GetSearchProvidersUsingKeywordResult( 93 const WDTypedResult& result, 94 WebDataService* service, 95 Profile* profile, 96 TemplateURLService::TemplateURLVector* template_urls, 97 TemplateURL* default_search_provider, 98 const SearchTermsData& search_terms_data, 99 int* new_resource_keyword_version, 100 std::set<std::string>* removed_keyword_guids); 101 102 // Like GetSearchProvidersUsingKeywordResult(), but allows the caller to pass in 103 // engines in |template_urls| instead of getting them via processing a web data 104 // service request. 105 // |resource_keyword_version| should contain the version number of the current 106 // keyword data, i.e. the version number of the most recent prepopulate data 107 // that has been merged into the current keyword data. On exit, this will be 108 // set as in GetSearchProvidersUsingKeywordResult(). 109 void GetSearchProvidersUsingLoadedEngines( 110 WebDataService* service, 111 Profile* profile, 112 TemplateURLService::TemplateURLVector* template_urls, 113 TemplateURL* default_search_provider, 114 const SearchTermsData& search_terms_data, 115 int* resource_keyword_version, 116 std::set<std::string>* removed_keyword_guids); 117 118 // Due to a bug, the |input_encodings| field of TemplateURLData could have 119 // contained duplicate entries. This removes those entries and returns whether 120 // any were found. 121 bool DeDupeEncodings(std::vector<std::string>* encodings); 122 123 // Removes (and deletes) TemplateURLs from |template_urls| and |service| if they 124 // have duplicate prepopulate ids. If |removed_keyword_guids| is not NULL, the 125 // Sync GUID of each item removed from the DB will be added to it. This is a 126 // helper used by GetSearchProvidersUsingKeywordResult(), but is declared here 127 // so it's accessible by unittests. 128 void RemoveDuplicatePrepopulateIDs( 129 WebDataService* service, 130 const ScopedVector<TemplateURLData>& prepopulated_urls, 131 TemplateURL* default_search_provider, 132 TemplateURLService::TemplateURLVector* template_urls, 133 const SearchTermsData& search_terms_data, 134 std::set<std::string>* removed_keyword_guids); 135 136 #endif // CHROME_BROWSER_SEARCH_ENGINES_UTIL_H_ 137