1// Copyright 2014 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 COMPONENTS_SEARCH_ENGINES_SEARCH_HOST_TO_URLS_MAP_H_ 6#define COMPONENTS_SEARCH_ENGINES_SEARCH_HOST_TO_URLS_MAP_H_ 7 8#include <map> 9#include <set> 10#include <string> 11 12#include "base/basictypes.h" 13#include "components/search_engines/template_url_service.h" 14 15// Holds the host to template url mappings for the search providers. WARNING: 16// This class does not own any TemplateURLs passed to it and it is up to the 17// caller to ensure the right lifetime of them. 18class SearchHostToURLsMap { 19 public: 20 typedef std::set<TemplateURL*> TemplateURLSet; 21 22 SearchHostToURLsMap(); 23 ~SearchHostToURLsMap(); 24 25 // Initializes the map. 26 void Init(const TemplateURLService::TemplateURLVector& template_urls, 27 const SearchTermsData& search_terms_data); 28 29 // Adds a new TemplateURL to the map. Since |template_url| is owned 30 // externally, Remove or RemoveAll should be called if it becomes invalid. 31 void Add(TemplateURL* template_url, 32 const SearchTermsData& search_terms_data); 33 34 // Removes the TemplateURL from the lookup. 35 void Remove(TemplateURL* template_url); 36 37 // Returns the first TemplateURL found with a URL using the specified |host|, 38 // or NULL if there are no such TemplateURLs 39 TemplateURL* GetTemplateURLForHost(const std::string& host); 40 41 // Return the TemplateURLSet for the given the |host| or NULL if there are 42 // none. 43 TemplateURLSet* GetURLsForHost(const std::string& host); 44 45 private: 46 friend class SearchHostToURLsMapTest; 47 48 typedef std::map<std::string, TemplateURLSet> HostToURLsMap; 49 50 // Adds many URLs to the map. 51 void Add(const TemplateURLService::TemplateURLVector& template_urls, 52 const SearchTermsData& search_terms_data); 53 54 // Maps from host to set of TemplateURLs whose search url host is host. 55 HostToURLsMap host_to_urls_map_; 56 57 // The security origin for the default search provider. 58 std::string default_search_origin_; 59 60 // Has Init been called? 61 bool initialized_; 62 63 DISALLOW_COPY_AND_ASSIGN(SearchHostToURLsMap); 64}; 65 66#endif // COMPONENTS_SEARCH_ENGINES_SEARCH_HOST_TO_URLS_MAP_H_ 67