OLD | NEW |
(Empty) | |
| 1 // Copyright 2013 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/WebTestDelegate.h" |
| 8 #include "content/shell/renderer/test_runner/mock_grammar_check.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 virtual ~HostMethodTask() {} |
| 24 |
| 25 virtual void runIfValid() OVERRIDE { (m_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 |