Cache SearchEngineType of TemplateURL

components/search_engines/template_url_data.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/search_engines/template_url_data.h"
 
 #include "base/guid.h"
 #include "base/i18n/case_conversion.h"
 #include "base/logging.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
+#include "components/search_engines/template_url.h"
+#include "components/search_engines/template_url_prepopulate_data.h"
 
 TemplateURLData::TemplateURLData()
     : show_in_default_list(false),
       safe_for_autoreplace(false),
       id(0),
       date_created(base::Time::Now()),
       last_modified(base::Time::Now()),
       created_by_policy(false),
       usage_count(0),
+      engine_type_(SEARCH_ENGINE_UNKNOWN),
       prepopulate_id(0),
       sync_guid(base::GenerateGUID()),
       keyword_(base::ASCIIToUTF16("dummy")),
       url_("x") {
 }
 
 TemplateURLData::TemplateURLData(const TemplateURLData& other) = default;
 
 TemplateURLData::~TemplateURLData() {
 }
@@ skipping 10 matching lines @@
   DCHECK(!keyword.empty());
 
   // Case sensitive keyword matching is confusing. As such, we force all
   // keywords to be lower case.
   keyword_ = base::i18n::ToLower(keyword);
 }
 
 void TemplateURLData::SetURL(const std::string& url) {
   DCHECK(!url.empty());
   url_ = url;
+  engine_type_ = SEARCH_ENGINE_UNKNOWN;
 }
+
+SearchEngineType TemplateURLData::GetEngineType(
+    const SearchTermsData& search_terms_data,
+    const TemplateURL* template_url_hint) const {
+  if (engine_type_ == SEARCH_ENGINE_UNKNOWN) {
+    std::unique_ptr<TemplateURL> temp_template_url;
+    if (!template_url_hint || (&template_url_hint->data() != this)) {
+      temp_template_url.reset(new TemplateURL(*this));
+      template_url_hint = temp_template_url.get();
+    }
+    const GURL url = template_url_hint->GenerateSearchURL(search_terms_data);
+    if (url.is_valid()) {
+      engine_type_ = TemplateURLPrepopulateData::GetEngineType(url);
+      DCHECK_NE(SEARCH_ENGINE_UNKNOWN, engine_type_);
+    } else {
+      engine_type_ = SEARCH_ENGINE_OTHER;
+    }
+  }
+  return engine_type_;
+}