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 #include "chrome/browser/ui/webui/predictors/predictors_handler.h" 6 7 #include "base/bind.h" 8 #include "base/values.h" 9 #include "chrome/browser/predictors/autocomplete_action_predictor.h" 10 #include "chrome/browser/predictors/autocomplete_action_predictor_factory.h" 11 #include "chrome/browser/profiles/profile.h" 12 #include "content/public/browser/web_ui.h" 13 14 using predictors::AutocompleteActionPredictor; 15 16 PredictorsHandler::PredictorsHandler(Profile* profile) { 17 autocomplete_action_predictor_ = 18 predictors::AutocompleteActionPredictorFactory::GetForProfile(profile); 19 } 20 21 PredictorsHandler::~PredictorsHandler() { } 22 23 void PredictorsHandler::RegisterMessages() { 24 web_ui()->RegisterMessageCallback("requestAutocompleteActionPredictorDb", 25 base::Bind(&PredictorsHandler::RequestAutocompleteActionPredictorDb, 26 base::Unretained(this))); 27 } 28 29 void PredictorsHandler::RequestAutocompleteActionPredictorDb( 30 const base::ListValue* args) { 31 const bool enabled = (autocomplete_action_predictor_ != NULL); 32 base::DictionaryValue dict; 33 dict.SetBoolean("enabled", enabled); 34 if (enabled) { 35 base::ListValue* db = new base::ListValue(); 36 for (AutocompleteActionPredictor::DBCacheMap::const_iterator it = 37 autocomplete_action_predictor_->db_cache_.begin(); 38 it != autocomplete_action_predictor_->db_cache_.end(); 39 ++it) { 40 base::DictionaryValue* entry = new base::DictionaryValue(); 41 entry->SetString("user_text", it->first.user_text); 42 entry->SetString("url", it->first.url.spec()); 43 entry->SetInteger("hit_count", it->second.number_of_hits); 44 entry->SetInteger("miss_count", it->second.number_of_misses); 45 entry->SetDouble("confidence", 46 autocomplete_action_predictor_->CalculateConfidenceForDbEntry(it)); 47 db->Append(entry); 48 } 49 dict.Set("db", db); 50 } 51 52 web_ui()->CallJavascriptFunction("updateAutocompleteActionPredictorDb", dict); 53 } 54