Home | History | Annotate | Download | only in search_engines 
      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