content/shell/renderer/test_runner/SpellCheckClient.cpp

   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/SpellCheckClient.h"
   6
   7 #include "content/shell/renderer/test_runner/MockGrammarCheck.h"
   8 #include "content/shell/renderer/test_runner/WebTestDelegate.h"
   9 #include "content/shell/renderer/test_runner/WebTestProxy.h"
  10 #include "third_party/WebKit/public/web/WebTextCheckingCompletion.h"
  11 #include "third_party/WebKit/public/web/WebTextCheckingResult.h"
  12
  13 using namespace blink;
  14 using namespace content;
  15 using namespace std;
  16
  17 namespace WebTestRunner {
  18
  19 namespace {
  20
  21 class HostMethodTask : public WebMethodTask<SpellCheckClient> {
  22 public:
  23     typedef void (SpellCheckClient::*CallbackMethodType)();
  24     HostMethodTask(SpellCheckClient* object, CallbackMethodType callback)
  25         : WebMethodTask<SpellCheckClient>(object)
  26         , m_callback(callback)
  27     { }
  28
  29     virtual void runIfValid() OVERRIDE { (m_object->*m_callback)(); }
  30
  31 private:
  32     CallbackMethodType m_callback;
  33 };
  34
  35 }
  36
  37 SpellCheckClient::SpellCheckClient(WebTestProxyBase* webTestProxy)
  38     : m_lastRequestedTextCheckingCompletion(0)
  39     , m_webTestProxy(webTestProxy)
  40 {
  41 }
  42
  43 SpellCheckClient::~SpellCheckClient()
  44 {
  45 }
  46
  47 void SpellCheckClient::setDelegate(WebTestDelegate* delegate)
  48 {
  49     m_delegate = delegate;
  50 }
  51
  52 // blink::WebSpellCheckClient
  53 void SpellCheckClient::spellCheck(const WebString& text, int& misspelledOffset, int& misspelledLength, WebVector<WebString>* optionalSuggestions)
  54 {
  55     // Check the spelling of the given text.
  56     m_spellcheck.spellCheckWord(text, &misspelledOffset, &misspelledLength);
  57 }
  58
  59 void SpellCheckClient::checkTextOfParagraph(const WebString& text, WebTextCheckingTypeMask mask, WebVector<WebTextCheckingResult>* webResults)
  60 {
  61     vector<WebTextCheckingResult> results;
  62     if (mask & WebTextCheckingTypeSpelling) {
  63         size_t offset = 0;
  64         base::string16 data = text;
  65         while (offset < data.length()) {
  66             int misspelledPosition = 0;
  67             int misspelledLength = 0;
  68             m_spellcheck.spellCheckWord(data.substr(offset), &misspelledPosition, &misspelledLength);
  69             if (!misspelledLength)
  70                 break;
  71             WebTextCheckingResult result;
  72             result.decoration = WebTextDecorationTypeSpelling;
  73             result.location = offset + misspelledPosition;
  74             result.length = misspelledLength;
  75             results.push_back(result);
  76             offset += misspelledPosition + misspelledLength;
  77         }
  78     }
  79     if (mask & WebTextCheckingTypeGrammar)
  80         MockGrammarCheck::checkGrammarOfString(text, &results);
  81     webResults->assign(results);
  82 }
  83
  84 void SpellCheckClient::requestCheckingOfText(
  85         const WebString& text,
  86         const WebVector<uint32_t>& markers,
  87         const WebVector<unsigned>& markerOffsets,
  88         WebTextCheckingCompletion* completion)
  89 {
  90     if (text.isEmpty()) {
  91         if (completion)
  92             completion->didCancelCheckingText();
  93         return;
  94     }
  95
  96     if (m_lastRequestedTextCheckingCompletion)
  97         m_lastRequestedTextCheckingCompletion->didCancelCheckingText();
  98
  99     m_lastRequestedTextCheckingCompletion = completion;
 100     m_lastRequestedTextCheckString = text;
 101     if (m_spellcheck.hasInCache(text))
 102         finishLastTextCheck();
 103     else
 104         m_delegate->postDelayedTask(new HostMethodTask(this, &SpellCheckClient::finishLastTextCheck), 0);
 105 }
 106
 107 void SpellCheckClient::finishLastTextCheck()
 108 {
 109     if (!m_lastRequestedTextCheckingCompletion)
 110         return;
 111     vector<WebTextCheckingResult> results;
 112     int offset = 0;
 113     base::string16 text = m_lastRequestedTextCheckString;
 114     if (!m_spellcheck.isMultiWordMisspelling(WebString(text), &results)) {
 115         while (text.length()) {
 116             int misspelledPosition = 0;
 117             int misspelledLength = 0;
 118             m_spellcheck.spellCheckWord(WebString(text), &misspelledPosition, &misspelledLength);
 119             if (!misspelledLength)
 120                 break;
 121             WebVector<WebString> suggestions;
 122             m_spellcheck.fillSuggestionList(WebString(text.substr(misspelledPosition, misspelledLength)), &suggestions);
 123             results.push_back(WebTextCheckingResult(WebTextDecorationTypeSpelling, offset + misspelledPosition, misspelledLength, suggestions.isEmpty() ? WebString() : suggestions[0]));
 124             text = text.substr(misspelledPosition + misspelledLength);
 125             offset += misspelledPosition + misspelledLength;
 126         }
 127         MockGrammarCheck::checkGrammarOfString(m_lastRequestedTextCheckString, &results);
 128     }
 129     m_lastRequestedTextCheckingCompletion->didFinishCheckingText(results);
 130     m_lastRequestedTextCheckingCompletion = 0;
 131
 132     m_webTestProxy->postSpellCheckEvent(WebString("finishLastTextCheck"));
 133 }
 134
 135 WebString SpellCheckClient::autoCorrectWord(const WebString&)
 136 {
 137     // Returns an empty string as Mac WebKit ('WebKitSupport/WebEditorClient.mm')
 138     // does. (If this function returns a non-empty string, WebKit replaces the
 139     // given misspelled string with the result one. This process executes some
 140     // editor commands and causes layout-test failures.)
 141     return WebString();
 142 }
 143
 144 }