diff options
Diffstat (limited to 'chrome/browser/spellchecker.cc')
| -rw-r--r-- | chrome/browser/spellchecker.cc | 814 |
1 files changed, 0 insertions, 814 deletions
diff --git a/chrome/browser/spellchecker.cc b/chrome/browser/spellchecker.cc deleted file mode 100644 index 903cdb9..0000000 --- a/chrome/browser/spellchecker.cc +++ /dev/null @@ -1,814 +0,0 @@ -// Copyright (c) 2006-2009 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 "chrome/browser/spellchecker.h" - -#include "app/l10n_util.h" -#include "base/basictypes.h" -#include "base/compiler_specific.h" -#include "base/file_util.h" -#include "base/histogram.h" -#include "base/logging.h" -#include "base/path_service.h" -#include "base/stats_counters.h" -#include "base/string_util.h" -#include "chrome/browser/net/url_fetcher.h" -#include "chrome/browser/profile.h" -#include "chrome/browser/spellchecker_common.h" -#include "chrome/browser/spellchecker_platform_engine.h" -#include "chrome/common/chrome_constants.h" -#include "chrome/common/chrome_counters.h" -#include "chrome/common/chrome_paths.h" -#include "chrome/common/pref_names.h" -#include "chrome/common/pref_service.h" -#include "grit/generated_resources.h" -#include "grit/locale_settings.h" -#include "net/url_request/url_request.h" -#include "third_party/hunspell/src/hunspell/hunspell.hxx" - -using base::TimeTicks; - -namespace { - -static const struct { - // The language. - const char* language; - - // The corresponding language and region, used by the dictionaries. - const char* language_region; -} g_supported_spellchecker_languages[] = { - {"en-US", "en-US"}, - {"en-GB", "en-GB"}, - {"en-AU", "en-AU"}, - {"fr", "fr-FR"}, - {"it", "it-IT"}, - {"de", "de-DE"}, - {"es", "es-ES"}, - {"nl", "nl-NL"}, - {"pt-BR", "pt-BR"}, - {"ru", "ru-RU"}, - {"pl", "pl-PL"}, - // {"th", "th-TH"}, // Not to be included in Spellchecker as per B=1277824 - {"sv", "sv-SE"}, - {"da", "da-DK"}, - {"pt-PT", "pt-PT"}, - {"ro", "ro-RO"}, - // {"hu", "hu-HU"}, // Not to be included in Spellchecker as per B=1277824 - {"he", "he-IL"}, - {"id", "id-ID"}, - {"cs", "cs-CZ"}, - {"el", "el-GR"}, - {"nb", "nb-NO"}, - {"vi", "vi-VN"}, - // {"bg", "bg-BG"}, // Not to be included in Spellchecker as per B=1277824 - {"hr", "hr-HR"}, - {"lt", "lt-LT"}, - {"sk", "sk-SK"}, - {"sl", "sl-SI"}, - {"ca", "ca-ES"}, - {"lv", "lv-LV"}, - // {"uk", "uk-UA"}, // Not to be included in Spellchecker as per B=1277824 - {"hi", "hi-IN"}, - {"et", "et-EE"}, - {"tr", "tr-TR"}, -}; - -// Get the fallback folder (currently chrome::DIR_USER_DATA) where the -// dictionary is downloaded in case of system-wide installations. -FilePath GetFallbackDictionaryDownloadDirectory() { - FilePath dict_dir_userdata; - PathService::Get(chrome::DIR_USER_DATA, &dict_dir_userdata); - dict_dir_userdata = dict_dir_userdata.AppendASCII("Dictionaries"); - return dict_dir_userdata; -} - -bool SaveBufferToFile(const std::string& data, - FilePath file_to_write) { - int num_bytes = data.length(); - return file_util::WriteFile(file_to_write, data.data(), num_bytes) == - num_bytes; -} - -} // namespace - -// Design: The spellchecker initializes hunspell_ in the Initialize() method. -// This is done using the dictionary file on disk, e.g. "en-US_1_1.bdic". -// Initialization of hunspell_ is held off during this process. If the -// dictionary is not available, we first attempt to download and save it. After -// the dictionary is downloaded and saved to disk (or the attempt to do so -// fails)), corresponding flags are set in spellchecker - in the IO thread. -// After the flags are cleared, a (final) attempt is made to initialize -// hunspell_. If it fails even then (dictionary could not download), no more -// attempts are made to initialize it. -class SaveDictionaryTask : public Task { - public: - SaveDictionaryTask(Task* on_dictionary_save_complete_callback_task, - const FilePath& first_attempt_file_name, - const FilePath& fallback_file_name, - const std::string& data) - : on_dictionary_save_complete_callback_task_( - on_dictionary_save_complete_callback_task), - first_attempt_file_name_(first_attempt_file_name), - fallback_file_name_(fallback_file_name), - data_(data) { - } - - private: - void Run(); - - bool SaveBufferToFile(const std::string& data, - FilePath file_to_write) { - int num_bytes = data.length(); - return file_util::WriteFile(file_to_write, data.data(), num_bytes) == - num_bytes; - } - - // factory object to invokelater back to spellchecker in io thread on - // download completion to change appropriate flags. - Task* on_dictionary_save_complete_callback_task_; - - // The file which will be stored in the first attempt. - FilePath first_attempt_file_name_; - - // The file which will be stored as a fallback. - FilePath fallback_file_name_; - - // The buffer which has to be stored to disk. - std::string data_; - - // This invokes back to io loop when downloading is over. - DISALLOW_COPY_AND_ASSIGN(SaveDictionaryTask); -}; - -void SaveDictionaryTask::Run() { - if (!SaveBufferToFile(data_, first_attempt_file_name_)) { - // Try saving it to |fallback_file_name_|, which almost surely has - // write permission. If even this fails, there is nothing to be done. - FilePath fallback_dir = fallback_file_name_.DirName(); - // Create the directory if it does not exist. - if (!file_util::PathExists(fallback_dir)) - file_util::CreateDirectory(fallback_dir); - SaveBufferToFile(data_, fallback_file_name_); - } // Unsuccessful save is taken care of in SpellChecker::Initialize(). - - // Set Flag that dictionary is not downloading anymore. - ChromeThread::PostTask( - ChromeThread::IO, FROM_HERE, on_dictionary_save_complete_callback_task_); -} - -// Design: this task tries to read the dictionary from disk and load it into -// memory. It is executed on the file thread, and posts the results back to -// the IO thread. -// The task first checks for the existence of the dictionary in one of the two -// given locations. If it does not exist, the task informs the SpellChecker, -// which will try to download the directory and run a new ReadDictionaryTask. -class ReadDictionaryTask : public Task { - public: - ReadDictionaryTask(SpellChecker* spellchecker, - const FilePath& dict_file_name_app, - const FilePath& dict_file_name_usr) - : spellchecker_(spellchecker), - hunspell_(NULL), - bdict_file_(NULL), - custom_dictionary_file_name_( - spellchecker->custom_dictionary_file_name_), - dict_file_name_app_(dict_file_name_app), - dict_file_name_usr_(dict_file_name_usr) { - } - - virtual void Run() { - FilePath bdict_file_path; - if (file_util::PathExists(dict_file_name_app_)) { - bdict_file_path = dict_file_name_app_; - } else if (file_util::PathExists(dict_file_name_usr_)) { - bdict_file_path = dict_file_name_usr_; - } else { - Finish(false); - return; - } - - bdict_file_ = new file_util::MemoryMappedFile; - if (bdict_file_->Initialize(bdict_file_path)) { - TimeTicks start_time = TimeTicks::Now(); - - hunspell_ = - new Hunspell(bdict_file_->data(), bdict_file_->length()); - - // Add custom words to Hunspell. - std::string contents; - file_util::ReadFileToString(custom_dictionary_file_name_, &contents); - std::vector<std::string> list_of_words; - SplitString(contents, '\n', &list_of_words); - for (std::vector<std::string>::iterator it = list_of_words.begin(); - it != list_of_words.end(); ++it) { - hunspell_->add(it->c_str()); - } - - DHISTOGRAM_TIMES("Spellcheck.InitTime", - TimeTicks::Now() - start_time); - } else { - delete bdict_file_; - bdict_file_ = NULL; - } - - Finish(true); - } - - private: - void Finish(bool file_existed) { - ChromeThread::PostTask( - ChromeThread::IO, FROM_HERE, - NewRunnableMethod( - spellchecker_.get(), &SpellChecker::HunspellInited, hunspell_, - bdict_file_, file_existed)); - } - - // The SpellChecker we are working for. - scoped_refptr<SpellChecker> spellchecker_; - Hunspell* hunspell_; - file_util::MemoryMappedFile* bdict_file_; - - FilePath custom_dictionary_file_name_; - FilePath dict_file_name_app_; - FilePath dict_file_name_usr_; - - DISALLOW_COPY_AND_ASSIGN(ReadDictionaryTask); -}; - -void SpellChecker::SpellCheckLanguages(std::vector<std::string>* languages) { - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(g_supported_spellchecker_languages); - ++i) { - languages->push_back(g_supported_spellchecker_languages[i].language); - } -} - -// This function returns the language-region version of language name. -// e.g. returns hi-IN for hi. -std::string SpellChecker::GetSpellCheckLanguageRegion( - std::string input_language) { - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(g_supported_spellchecker_languages); - ++i) { - std::string language( - g_supported_spellchecker_languages[i].language); - if (language == input_language) - return std::string( - g_supported_spellchecker_languages[i].language_region); - } - - return input_language; -} - - -std::string SpellChecker::GetLanguageFromLanguageRegion( - std::string input_language) { - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(g_supported_spellchecker_languages); - ++i) { - std::string language( - g_supported_spellchecker_languages[i].language_region); - if (language == input_language) - return std::string(g_supported_spellchecker_languages[i].language); - } - - return input_language; -} - -std::string SpellChecker::GetCorrespondingSpellCheckLanguage( - const std::string& language) { - // Look for exact match in the Spell Check language list. - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(g_supported_spellchecker_languages); - ++i) { - // First look for exact match in the language region of the list. - std::string spellcheck_language( - g_supported_spellchecker_languages[i].language); - if (spellcheck_language == language) - return language; - - // Next, look for exact match in the language_region part of the list. - std::string spellcheck_language_region( - g_supported_spellchecker_languages[i].language_region); - if (spellcheck_language_region == language) - return g_supported_spellchecker_languages[i].language; - } - - // Look for a match by comparing only language parts. All the 'en-RR' - // except for 'en-GB' exactly matched in the above loop, will match - // 'en-US'. This is not ideal because 'en-ZA', 'en-NZ' had - // better be matched with 'en-GB'. This does not handle cases like - // 'az-Latn-AZ' vs 'az-Arab-AZ', either, but we don't use 3-part - // locale ids with a script code in the middle, yet. - // TODO(jungshik): Add a better fallback. - std::string language_part(language, 0, language.find('-')); - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(g_supported_spellchecker_languages); - ++i) { - std::string spellcheck_language( - g_supported_spellchecker_languages[i].language_region); - if (spellcheck_language.substr(0, spellcheck_language.find('-')) == - language_part) - return spellcheck_language; - } - - // No match found - return blank. - return std::string(); -} - -// static -int SpellChecker::GetSpellCheckLanguages( - Profile* profile, - std::vector<std::string>* languages) { - StringPrefMember accept_languages_pref; - StringPrefMember dictionary_language_pref; - accept_languages_pref.Init(prefs::kAcceptLanguages, profile->GetPrefs(), - NULL); - dictionary_language_pref.Init(prefs::kSpellCheckDictionary, - profile->GetPrefs(), NULL); - std::string dictionary_language = - WideToASCII(dictionary_language_pref.GetValue()); - - // The current dictionary language should be there. - languages->push_back(dictionary_language); - - // Now scan through the list of accept languages, and find possible mappings - // from this list to the existing list of spell check languages. - std::vector<std::string> accept_languages; - - if (SpellCheckerPlatform::SpellCheckerAvailable()) { - SpellCheckerPlatform::GetAvailableLanguages(&accept_languages); - } else { - SplitString(WideToASCII(accept_languages_pref.GetValue()), ',', - &accept_languages); - } - for (std::vector<std::string>::const_iterator i = accept_languages.begin(); - i != accept_languages.end(); ++i) { - std::string language = GetCorrespondingSpellCheckLanguage(*i); - if (!language.empty() && - std::find(languages->begin(), languages->end(), language) == - languages->end()) - languages->push_back(language); - } - - for (size_t i = 0; i < languages->size(); ++i) { - if ((*languages)[i] == dictionary_language) - return i; - } - return -1; -} - -FilePath SpellChecker::GetVersionedFileName(const std::string& input_language, - const FilePath& dict_dir) { - // The default dictionary version is 1-2. These versions have been augmented - // with additional words found by the translation team. - static const char kDefaultVersionString[] = "-1-2"; - - // The following dictionaries have either not been augmented with additional - // words (version 1-1) or have new words, as well as an upgraded dictionary - // as of Feb 2009 (version 1-3). - static const struct { - // The language input. - const char* language; - - // The corresponding version. - const char* version; - } special_version_string[] = { - {"en-AU", "-1-1"}, - {"en-GB", "-1-1"}, - {"es-ES", "-1-1"}, - {"nl-NL", "-1-1"}, - {"ru-RU", "-1-1"}, - {"sv-SE", "-1-1"}, - {"he-IL", "-1-1"}, - {"el-GR", "-1-1"}, - {"hi-IN", "-1-1"}, - {"tr-TR", "-1-1"}, - {"et-EE", "-1-1"}, - {"fr-FR", "-1-4"}, // to fix crash, fr dictionary was updated to 1.4 - {"lt-LT", "-1-3"}, - {"pl-PL", "-1-3"} - }; - - // Generate the bdict file name using default version string or special - // version string, depending on the language. - std::string language = GetSpellCheckLanguageRegion(input_language); - std::string versioned_bdict_file_name(language + kDefaultVersionString + - ".bdic"); - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(special_version_string); ++i) { - if (language == special_version_string[i].language) { - versioned_bdict_file_name = - language + special_version_string[i].version + ".bdic"; - break; - } - } - - return dict_dir.AppendASCII(versioned_bdict_file_name); -} - -SpellChecker::SpellChecker(const FilePath& dict_dir, - const std::string& language, - URLRequestContextGetter* request_context_getter, - const FilePath& custom_dictionary_file_name) - : given_dictionary_directory_(dict_dir), - custom_dictionary_file_name_(custom_dictionary_file_name), - tried_to_init_(false), - language_(language), - tried_to_download_dictionary_file_(false), - request_context_getter_(request_context_getter), - obtaining_dictionary_(false), - auto_spell_correct_turned_on_(false), - is_using_platform_spelling_engine_(false), - fetcher_(NULL), - ALLOW_THIS_IN_INITIALIZER_LIST(method_factory_(this)) { - if (SpellCheckerPlatform::SpellCheckerAvailable()) { - SpellCheckerPlatform::Init(); - if (SpellCheckerPlatform::PlatformSupportsLanguage(language)) { - // If we have reached here, then we know that the current platform - // supports the given language and we will use it instead of hunspell. - SpellCheckerPlatform::SetLanguage(language); - is_using_platform_spelling_engine_ = true; - } - } - - // Get the corresponding BDIC file name. - bdic_file_name_ = GetVersionedFileName(language, dict_dir).BaseName(); - - // Get the path to the custom dictionary file. - if (custom_dictionary_file_name_.empty()) { - FilePath personal_file_directory; - PathService::Get(chrome::DIR_USER_DATA, &personal_file_directory); - custom_dictionary_file_name_ = - personal_file_directory.Append(chrome::kCustomDictionaryFileName); - } - - // Use this dictionary language as the default one of the - // SpellcheckCharAttribute object. - character_attributes_.SetDefaultLanguage(language); -} - -SpellChecker::~SpellChecker() { - DCHECK(ChromeThread::CurrentlyOn(ChromeThread::IO)); -} - -void SpellChecker::StartDictionaryDownload(const FilePath& file_name) { - // Determine URL of file to download. - static const char kDownloadServerUrl[] = - "http://cache.pack.google.com/edgedl/chrome/dict/"; - GURL url = GURL(std::string(kDownloadServerUrl) + WideToUTF8( - l10n_util::ToLower(bdic_file_name_.ToWStringHack()))); - fetcher_.reset(new URLFetcher(url, URLFetcher::GET, this)); - fetcher_->set_request_context(request_context_getter_); - obtaining_dictionary_ = true; - fetcher_->Start(); -} - -void SpellChecker::OnURLFetchComplete(const URLFetcher* source, - const GURL& url, - const URLRequestStatus& status, - int response_code, - const ResponseCookies& cookies, - const std::string& data) { - DCHECK(source); - if ((response_code / 100) != 2) { - obtaining_dictionary_ = false; - return; - } - - // Basic sanity check on the dictionary. - // There's the small chance that we might see a 200 status code for a body - // that represents some form of failure. - if (data.size() < 4 || data[0] != 'B' || data[1] != 'D' || data[2] != 'i' || - data[3] != 'c') { - obtaining_dictionary_ = false; - return; - } - - // Save the file in the file thread, and not here, the IO thread. - FilePath first_attempt_file_name = given_dictionary_directory_.Append( - bdic_file_name_); - FilePath user_data_dir = GetFallbackDictionaryDownloadDirectory(); - FilePath fallback_file_name = user_data_dir.Append(bdic_file_name_); - Task* dic_task = method_factory_. - NewRunnableMethod(&SpellChecker::OnDictionarySaveComplete); - ChromeThread::PostTask( - ChromeThread::FILE, FROM_HERE, - new SaveDictionaryTask( - dic_task, first_attempt_file_name, fallback_file_name, data)); -} - -void SpellChecker::OnDictionarySaveComplete() { - obtaining_dictionary_ = false; - // Now that the dictionary is downloaded, continue trying to download. - Initialize(); -} - -// Initialize SpellChecker. In this method, if the dictionary is not present -// in the local disk, it is fetched asynchronously. -bool SpellChecker::Initialize() { - DCHECK(ChromeThread::CurrentlyOn(ChromeThread::IO)); - - // Return false if the dictionary files are downloading. - if (obtaining_dictionary_) - return false; - - // Return false if tried to init and failed - don't try multiple times in - // this session. - if (tried_to_init_) - return hunspell_.get() != NULL; - - StatsScope<StatsCounterTimer> timer(chrome::Counters::spellcheck_init()); - - // The default place whether the spellcheck dictionary can reside is - // chrome::DIR_APP_DICTIONARIES. However, for systemwide installations, - // this directory may not have permissions for download. In that case, the - // alternate directory for download is chrome::DIR_USER_DATA. We have to check - // for the spellcheck dictionaries in both the directories. If not found in - // either one, it has to be downloaded in either of the two. - // TODO(sidchat): Some sort of UI to warn users that spellchecker is not - // working at all (due to failed dictionary download)? - - // File name for downloading in DIR_APP_DICTIONARIES. - FilePath dictionary_file_name_app = GetVersionedFileName(language_, - given_dictionary_directory_); - - // Filename for downloading in the fallback dictionary download directory, - // DIR_USER_DATA. - FilePath dict_dir_userdata = GetFallbackDictionaryDownloadDirectory(); - FilePath dictionary_file_name_usr = GetVersionedFileName(language_, - dict_dir_userdata); - - ChromeThread::PostTask( - ChromeThread::FILE, FROM_HERE, - new ReadDictionaryTask( - this, dictionary_file_name_app, dictionary_file_name_usr)); - - return hunspell_.get() != NULL; -} - -void SpellChecker::HunspellInited(Hunspell* hunspell, - file_util::MemoryMappedFile* bdict_file, - bool file_existed) { - DCHECK(ChromeThread::CurrentlyOn(ChromeThread::IO)); - - if (file_existed) - tried_to_init_ = true; - - if (!hunspell) { - if (!file_existed) { - // File didn't exist. We need to download a dictionary. - DoDictionaryDownload(); - } - return; - } - - - bdict_file_.reset(bdict_file); - hunspell_.reset(hunspell); - // Add all the custom words we've gotten while Hunspell was loading. - while (!custom_words_.empty()) { - hunspell_->add(custom_words_.front().c_str()); - custom_words_.pop(); - } -} - -void SpellChecker::DoDictionaryDownload() { - // Download the dictionary file. - if (request_context_getter_) { - if (!tried_to_download_dictionary_file_) { - FilePath dictionary_file_name_app = GetVersionedFileName(language_, - given_dictionary_directory_); - StartDictionaryDownload(dictionary_file_name_app); - tried_to_download_dictionary_file_ = true; - } else { - // Don't try to download a dictionary more than once. - tried_to_init_ = true; - } - } else { - NOTREACHED(); - } -} - -string16 SpellChecker::GetAutoCorrectionWord(const string16& word, int tag) { - string16 autocorrect_word; - if (!auto_spell_correct_turned_on_) - return autocorrect_word; // Return the empty string. - - int word_length = static_cast<int>(word.size()); - if (word_length < 2 || word_length > kMaxAutoCorrectWordSize) - return autocorrect_word; - - char16 misspelled_word[kMaxAutoCorrectWordSize + 1]; - const char16* word_char = word.c_str(); - for (int i = 0; i <= kMaxAutoCorrectWordSize; i++) { - if (i >= word_length) - misspelled_word[i] = NULL; - else - misspelled_word[i] = word_char[i]; - } - - // Swap adjacent characters and spellcheck. - int misspelling_start, misspelling_len; - for (int i = 0; i < word_length - 1; i++) { - // Swap. - std::swap(misspelled_word[i], misspelled_word[i + 1]); - - // Check spelling. - misspelling_start = misspelling_len = 0; - SpellCheckWord(misspelled_word, word_length, tag, &misspelling_start, - &misspelling_len, NULL); - - // Make decision: if only one swap produced a valid word, then we want to - // return it. If we found two or more, we don't do autocorrection. - if (misspelling_len == 0) { - if (autocorrect_word.empty()) { - autocorrect_word.assign(misspelled_word); - } else { - autocorrect_word.clear(); - break; - } - } - - // Restore the swapped characters. - std::swap(misspelled_word[i], misspelled_word[i + 1]); - } - return autocorrect_word; -} - -void SpellChecker::EnableAutoSpellCorrect(bool turn_on) { - auto_spell_correct_turned_on_ = turn_on; -} - -// Returns whether or not the given string is a valid contraction. -// This function is a fall-back when the SpellcheckWordIterator class -// returns a concatenated word which is not in the selected dictionary -// (e.g. "in'n'out") but each word is valid. -bool SpellChecker::IsValidContraction(const string16& contraction, int tag) { - SpellcheckWordIterator word_iterator; - word_iterator.Initialize(&character_attributes_, contraction.c_str(), - contraction.length(), false); - - string16 word; - int word_start; - int word_length; - while (word_iterator.GetNextWord(&word, &word_start, &word_length)) { - if (!CheckSpelling(word, tag)) - return false; - } - return true; -} - -bool SpellChecker::SpellCheckWord( - const char16* in_word, - int in_word_len, - int tag, - int* misspelling_start, - int* misspelling_len, - std::vector<string16>* optional_suggestions) { - DCHECK(in_word_len >= 0); - DCHECK(misspelling_start && misspelling_len) << "Out vars must be given."; - - // This must always be called on the same thread (normally the I/O thread). - DCHECK(ChromeThread::CurrentlyOn(ChromeThread::IO)); - - // Check if the platform spellchecker is being used. - if (!is_using_platform_spelling_engine_) { - // If it isn't, try and init hunspell. - Initialize(); - - // Check to see if hunspell was successfuly initialized. - if (!hunspell_.get()) - return true; // Unable to spellcheck, return word is OK. - } - - StatsScope<StatsRate> timer(chrome::Counters::spellcheck_lookup()); - - *misspelling_start = 0; - *misspelling_len = 0; - if (in_word_len == 0) - return true; // No input means always spelled correctly. - - SpellcheckWordIterator word_iterator; - string16 word; - int word_start; - int word_length; - word_iterator.Initialize(&character_attributes_, in_word, in_word_len, true); - while (word_iterator.GetNextWord(&word, &word_start, &word_length)) { - // Found a word (or a contraction) that the spellchecker can check the - // spelling of. - bool word_ok = CheckSpelling(word, tag); - if (word_ok) - continue; - - // If the given word is a concatenated word of two or more valid words - // (e.g. "hello:hello"), we should treat it as a valid word. - if (IsValidContraction(word, tag)) - continue; - - *misspelling_start = word_start; - *misspelling_len = word_length; - - // Get the list of suggested words. - if (optional_suggestions) - FillSuggestionList(word, optional_suggestions); - return false; - } - - return true; -} - -// This task is called in the file loop to write the new word to the custom -// dictionary in disc. -class AddWordToCustomDictionaryTask : public Task { - public: - AddWordToCustomDictionaryTask(const FilePath& file_name, - const string16& word) - : file_name_(file_name), - word_(UTF16ToUTF8(word)) { - } - - private: - void Run(); - - FilePath file_name_; - std::string word_; -}; - -void AddWordToCustomDictionaryTask::Run() { - // Add the word with a new line. Note that, although this would mean an - // extra line after the list of words, this is potentially harmless and - // faster, compared to verifying everytime whether to append a new line - // or not. - word_ += "\n"; - FILE* f = file_util::OpenFile(file_name_, "a+"); - if (f != NULL) - fputs(word_.c_str(), f); - file_util::CloseFile(f); -} - -void SpellChecker::AddWord(const string16& word) { - if (is_using_platform_spelling_engine_) { - SpellCheckerPlatform::AddWord(word); - return; - } - - // Check if the |hunspell_| has been initialized at all. - Initialize(); - - // Add the word to hunspell. - std::string word_to_add = UTF16ToUTF8(word); - // Don't attempt to add an empty word, or one larger than Hunspell can handle - if (!word_to_add.empty() && word_to_add.length() < MAXWORDLEN) { - // Either add the word to |hunspell_|, or, if |hunspell_| is still loading, - // defer it till after the load completes. - if (hunspell_.get()) - hunspell_->add(word_to_add.c_str()); - else - custom_words_.push(word_to_add); - } - - // Now add the word to the custom dictionary file. - ChromeThread::PostTask( - ChromeThread::FILE, FROM_HERE, - new AddWordToCustomDictionaryTask(custom_dictionary_file_name_, word)); -} - -bool SpellChecker::CheckSpelling(const string16& word_to_check, int tag) { - bool word_correct = false; - - TimeTicks begin_time = TimeTicks::Now(); - if (is_using_platform_spelling_engine_) { - word_correct = SpellCheckerPlatform::CheckSpelling(word_to_check, tag); - } else { - std::string word_to_check_utf8(UTF16ToUTF8(word_to_check)); - // Hunspell shouldn't let us exceed its max, but check just in case - if (word_to_check_utf8.length() < MAXWORDLEN) { - // |hunspell_->spell| returns 0 if the word is spelled correctly and - // non-zero otherwsie. - word_correct = (hunspell_->spell(word_to_check_utf8.c_str()) != 0); - } - } - DHISTOGRAM_TIMES("Spellcheck.CheckTime", TimeTicks::Now() - begin_time); - - return word_correct; -} - -void SpellChecker::FillSuggestionList( - const string16& wrong_word, - std::vector<string16>* optional_suggestions) { - if (is_using_platform_spelling_engine_) { - SpellCheckerPlatform::FillSuggestionList(wrong_word, optional_suggestions); - return; - } - char** suggestions; - TimeTicks begin_time = TimeTicks::Now(); - int number_of_suggestions = hunspell_->suggest(&suggestions, - UTF16ToUTF8(wrong_word).c_str()); - DHISTOGRAM_TIMES("Spellcheck.SuggestTime", - TimeTicks::Now() - begin_time); - - // Populate the vector of WideStrings. - for (int i = 0; i < number_of_suggestions; i++) { - if (i < kMaxSuggestions) - optional_suggestions->push_back(UTF8ToUTF16(suggestions[i])); - free(suggestions[i]); - } - if (suggestions != NULL) - free(suggestions); -} |