| OLD | NEW |
| (Empty) |
| 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 #include "content/shell/renderer/test_runner/spell_check_client.h" | |
| 6 | |
| 7 #include "content/shell/renderer/test_runner/mock_grammar_check.h" | |
| 8 #include "content/shell/renderer/test_runner/web_test_delegate.h" | |
| 9 #include "content/shell/renderer/test_runner/web_test_proxy.h" | |
| 10 #include "third_party/WebKit/public/web/WebTextCheckingCompletion.h" | |
| 11 #include "third_party/WebKit/public/web/WebTextCheckingResult.h" | |
| 12 | |
| 13 namespace content { | |
| 14 | |
| 15 namespace { | |
| 16 | |
| 17 class HostMethodTask : public WebMethodTask<SpellCheckClient> { | |
| 18 public: | |
| 19 typedef void (SpellCheckClient::*CallbackMethodType)(); | |
| 20 HostMethodTask(SpellCheckClient* object, CallbackMethodType callback) | |
| 21 : WebMethodTask<SpellCheckClient>(object), callback_(callback) {} | |
| 22 | |
| 23 ~HostMethodTask() override {} | |
| 24 | |
| 25 void RunIfValid() override { (object_->*callback_)(); } | |
| 26 | |
| 27 private: | |
| 28 CallbackMethodType callback_; | |
| 29 | |
| 30 DISALLOW_COPY_AND_ASSIGN(HostMethodTask); | |
| 31 }; | |
| 32 | |
| 33 } // namespace | |
| 34 | |
| 35 SpellCheckClient::SpellCheckClient(WebTestProxyBase* web_test_proxy) | |
| 36 : last_requested_text_checking_completion_(0), | |
| 37 web_test_proxy_(web_test_proxy) { | |
| 38 } | |
| 39 | |
| 40 SpellCheckClient::~SpellCheckClient() { | |
| 41 } | |
| 42 | |
| 43 void SpellCheckClient::SetDelegate(WebTestDelegate* delegate) { | |
| 44 delegate_ = delegate; | |
| 45 } | |
| 46 | |
| 47 // blink::WebSpellCheckClient | |
| 48 void SpellCheckClient::spellCheck( | |
| 49 const blink::WebString& text, | |
| 50 int& misspelled_offset, | |
| 51 int& misspelled_length, | |
| 52 blink::WebVector<blink::WebString>* optional_suggestions) { | |
| 53 // Check the spelling of the given text. | |
| 54 spell_check_.SpellCheckWord(text, &misspelled_offset, &misspelled_length); | |
| 55 } | |
| 56 | |
| 57 void SpellCheckClient::checkTextOfParagraph( | |
| 58 const blink::WebString& text, | |
| 59 blink::WebTextCheckingTypeMask mask, | |
| 60 blink::WebVector<blink::WebTextCheckingResult>* web_results) { | |
| 61 std::vector<blink::WebTextCheckingResult> results; | |
| 62 if (mask & blink::WebTextCheckingTypeSpelling) { | |
| 63 size_t offset = 0; | |
| 64 base::string16 data = text; | |
| 65 while (offset < data.length()) { | |
| 66 int misspelled_position = 0; | |
| 67 int misspelled_length = 0; | |
| 68 spell_check_.SpellCheckWord( | |
| 69 data.substr(offset), &misspelled_position, &misspelled_length); | |
| 70 if (!misspelled_length) | |
| 71 break; | |
| 72 blink::WebTextCheckingResult result; | |
| 73 result.decoration = blink::WebTextDecorationTypeSpelling; | |
| 74 result.location = offset + misspelled_position; | |
| 75 result.length = misspelled_length; | |
| 76 results.push_back(result); | |
| 77 offset += misspelled_position + misspelled_length; | |
| 78 } | |
| 79 } | |
| 80 if (mask & blink::WebTextCheckingTypeGrammar) | |
| 81 MockGrammarCheck::CheckGrammarOfString(text, &results); | |
| 82 web_results->assign(results); | |
| 83 } | |
| 84 | |
| 85 void SpellCheckClient::requestCheckingOfText( | |
| 86 const blink::WebString& text, | |
| 87 const blink::WebVector<uint32_t>& markers, | |
| 88 const blink::WebVector<unsigned>& marker_offsets, | |
| 89 blink::WebTextCheckingCompletion* completion) { | |
| 90 if (text.isEmpty()) { | |
| 91 if (completion) | |
| 92 completion->didCancelCheckingText(); | |
| 93 return; | |
| 94 } | |
| 95 | |
| 96 if (last_requested_text_checking_completion_) | |
| 97 last_requested_text_checking_completion_->didCancelCheckingText(); | |
| 98 | |
| 99 last_requested_text_checking_completion_ = completion; | |
| 100 last_requested_text_check_string_ = text; | |
| 101 if (spell_check_.HasInCache(text)) | |
| 102 FinishLastTextCheck(); | |
| 103 else | |
| 104 delegate_->PostDelayedTask( | |
| 105 new HostMethodTask(this, &SpellCheckClient::FinishLastTextCheck), 0); | |
| 106 } | |
| 107 | |
| 108 void SpellCheckClient::FinishLastTextCheck() { | |
| 109 if (!last_requested_text_checking_completion_) | |
| 110 return; | |
| 111 std::vector<blink::WebTextCheckingResult> results; | |
| 112 int offset = 0; | |
| 113 base::string16 text = last_requested_text_check_string_; | |
| 114 if (!spell_check_.IsMultiWordMisspelling(blink::WebString(text), &results)) { | |
| 115 while (text.length()) { | |
| 116 int misspelled_position = 0; | |
| 117 int misspelled_length = 0; | |
| 118 spell_check_.SpellCheckWord( | |
| 119 blink::WebString(text), &misspelled_position, &misspelled_length); | |
| 120 if (!misspelled_length) | |
| 121 break; | |
| 122 blink::WebVector<blink::WebString> suggestions; | |
| 123 spell_check_.FillSuggestionList( | |
| 124 blink::WebString(text.substr(misspelled_position, misspelled_length)), | |
| 125 &suggestions); | |
| 126 results.push_back(blink::WebTextCheckingResult( | |
| 127 blink::WebTextDecorationTypeSpelling, | |
| 128 offset + misspelled_position, | |
| 129 misspelled_length, | |
| 130 suggestions.isEmpty() ? blink::WebString() : suggestions[0])); | |
| 131 text = text.substr(misspelled_position + misspelled_length); | |
| 132 offset += misspelled_position + misspelled_length; | |
| 133 } | |
| 134 MockGrammarCheck::CheckGrammarOfString(last_requested_text_check_string_, | |
| 135 &results); | |
| 136 } | |
| 137 last_requested_text_checking_completion_->didFinishCheckingText(results); | |
| 138 last_requested_text_checking_completion_ = 0; | |
| 139 | |
| 140 web_test_proxy_->PostSpellCheckEvent(blink::WebString("FinishLastTextCheck")); | |
| 141 } | |
| 142 | |
| 143 blink::WebString SpellCheckClient::autoCorrectWord( | |
| 144 const blink::WebString& word) { | |
| 145 // Returns an empty string as Mac WebKit ('WebKitSupport/WebEditorClient.mm') | |
| 146 // does. (If this function returns a non-empty string, WebKit replaces the | |
| 147 // given misspelled string with the result one. This process executes some | |
| 148 // editor commands and causes layout-test failures.) | |
| 149 return blink::WebString(); | |
| 150 } | |
| 151 | |
| 152 } // namespace content | |
| OLD | NEW |