SpellingCorrectionCommand.cpp [plain text]
#include "config.h"
#include "SpellingCorrectionCommand.h"
#include "AlternativeTextController.h"
#include "Document.h"
#include "DocumentFragment.h"
#include "Frame.h"
#include "ReplaceSelectionCommand.h"
#include "SetSelectionCommand.h"
#include "TextIterator.h"
#include "markup.h"
namespace WebCore {
#if USE(AUTOCORRECTION_PANEL)
class SpellingCorrectionRecordUndoCommand : public SimpleEditCommand {
public:
static PassRefPtr<SpellingCorrectionRecordUndoCommand> create(Document* document, const String& corrected, const String& correction)
{
return adoptRef(new SpellingCorrectionRecordUndoCommand(document, corrected, correction));
}
private:
SpellingCorrectionRecordUndoCommand(Document* document, const String& corrected, const String& correction)
: SimpleEditCommand(document)
, m_corrected(corrected)
, m_correction(correction)
, m_hasBeenUndone(false)
{
}
virtual void doApply() OVERRIDE
{
}
virtual void doUnapply() OVERRIDE
{
if (!m_hasBeenUndone) {
document()->frame()->editor()->unappliedSpellCorrection(startingSelection(), m_corrected, m_correction);
m_hasBeenUndone = true;
}
}
#ifndef NDEBUG
virtual void getNodesInCommand(HashSet<Node*>&) OVERRIDE
{
}
#endif
String m_corrected;
String m_correction;
bool m_hasBeenUndone;
};
#endif
SpellingCorrectionCommand::SpellingCorrectionCommand(PassRefPtr<Range> rangeToBeCorrected, const String& correction)
: CompositeEditCommand(rangeToBeCorrected->startContainer()->document())
, m_rangeToBeCorrected(rangeToBeCorrected)
, m_selectionToBeCorrected(m_rangeToBeCorrected.get())
, m_correction(correction)
{
}
void SpellingCorrectionCommand::doApply()
{
m_corrected = plainText(m_rangeToBeCorrected.get());
if (!m_corrected.length())
return;
if (!document()->frame()->selection()->shouldChangeSelection(m_selectionToBeCorrected))
return;
RefPtr<DocumentFragment> fragment = createFragmentFromText(m_rangeToBeCorrected.get(), m_correction);
if (!fragment)
return;
applyCommandToComposite(SetSelectionCommand::create(m_selectionToBeCorrected, FrameSelection::SpellCorrectionTriggered | FrameSelection::CloseTyping | FrameSelection::ClearTypingStyle));
#if USE(AUTOCORRECTION_PANEL)
applyCommandToComposite(SpellingCorrectionRecordUndoCommand::create(document(), m_corrected, m_correction));
#endif
applyCommandToComposite(ReplaceSelectionCommand::create(document(), fragment, ReplaceSelectionCommand::MatchStyle | ReplaceSelectionCommand::PreventNesting, EditActionPaste));
}
bool SpellingCorrectionCommand::shouldRetainAutocorrectionIndicator() const
{
return true;
}
}