Workaround for setComposition styling clobber

third_party/WebKit/Source/core/editing/InputMethodController.cpp

 /*
  * Copyright (C) 2006, 2007, 2008, 2011 Apple Inc. All rights reserved.
  * Copyright (C) 2008 Nokia Corporation and/or its subsidiary(-ies)
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 320 matching lines @@
     if (!selection.isNone() && !m_compositionRange->collapsed()) {
         if (selection.start().compareTo(m_compositionRange->startPosition()) >= 0
             && selection.end().compareTo(m_compositionRange->endPosition()) <= 0)
             return;
     }
 
     cancelComposition();
     frame().chromeClient().didCancelCompositionOnSelectionChange();
 }
 
+static size_t computeCommonPrefixLength(const String& str1, const String& str2)
+{
+    size_t commonPrefixLength = 0;
+    size_t length1 = str1.length();
+    size_t length2 = str2.length();
+    while (commonPrefixLength < length1 && commonPrefixLength < length2 && str1[commonPrefixLength] == str2[commonPrefixLength])

[Changwan Ryu, 2016/09/27 05:23:03]

Can you skip a loop for surrogate pairs? Things can get weird if we apply
different formats to surrogate head and tail.

[yabinh, 2016/09/27 08:47:20]

Done.

+        commonPrefixLength++;
+    return commonPrefixLength;
+}
+
 void InputMethodController::setComposition(const String& text, const Vector<CompositionUnderline>& underlines, int selectionStart, int selectionEnd)
 {
     Editor::RevealSelectionScope revealSelectionScope(&editor());
 
     // Updates styles before setting selection for composition to prevent
     // inserting the previous composition text into text nodes oddly.
     // See https://bugs.webkit.org/show_bug.cgi?id=46868
     frame().document()->updateStyleAndLayoutTree();
 
+    // When the IME only wants to change a few characters at the end of the
+    // composition, only touch those characters in order to preserve rich text
+    // substructure.
+    if (hasComposition() && text.length()) {
+        String composing = composingText();
+        size_t commonPrefixLength = computeCommonPrefixLength(composing, text);
+
+        bool appending = text.length() > commonPrefixLength;
+        bool backspacing = composing.length() > commonPrefixLength;
+        if (appending || backspacing) {
+            const EphemeralRange range = compositionEphemeralRange();
+            Element* editable = frame().selection().rootEditableElement();
+            if (!editable)
+                return;
+
+            // Move selection to the end of the composition.
+            PlainTextRange compositionPlainOffsets = PlainTextRange::create(*editable, range);
+            VisibleSelection selection;
+            selection.setWithoutValidation(range.endPosition(), range.endPosition());
+            frame().selection().setSelection(selection, 0);
+            clear();
+
+            Element* target = frame().document()->focusedElement();
+            if (!target)
+                return;
+
+            // Apply the incremental change.
+            if (backspacing)
+                extendSelectionAndDelete(composing.length() - commonPrefixLength, 0);

[aelias_OOO_until_Jul13, 2016/09/27 04:55:58]

Hmm, this has the same JS event problem as the other case you just fixed.  How
about selecting all the non-shared suffix characters in the selection logic
above, and then just always calling insertTextDuringCompositionWithEvents()
(possibly with "" for the backspacing case) and relying on the replacement of
selection?

[yabinh, 2016/09/27 08:47:20]

Done.

+            if (appending)
+                insertTextDuringCompositionWithEvents(frame(), text.substring(commonPrefixLength), TypingCommand::PreventSpellChecking, TypingCommand::TextCompositionUpdate);
+
+            // Event handlers might destroy document.
+            if (!frame().document())
+                return;
+
+            // TODO(yosin): The use of updateStyleAndLayoutIgnorePendingStylesheets
+            // needs to be audited. see http://crbug.com/590369 for more details.
+            frame().document()->updateStyleAndLayoutIgnorePendingStylesheets();
+
+            // Now recreate the composition starting at its original start, and
+            // apply the specified final selection offsets.
+            setCompositionFromExistingText(underlines, compositionPlainOffsets.start(), compositionPlainOffsets.start() + text.length());
+            selectComposition();
+            PlainTextRange selectedRange = createSelectionRangeForSetComposition(selectionStart, selectionEnd, text.length());
+            // We shouldn't close typing in the middle of setComposition.
+            setEditableSelectionOffsets(selectedRange, NotUserTriggered);
+            m_isDirty = true;
+            return;
+        }
+    }
+
     selectComposition();
 
     if (frame().selection().isNone())
         return;
 
     Element* target = frame().document()->focusedElement();
     if (!target)
         return;
 
-    // TODO(xiaochengh): The use of updateStyleAndLayoutIgnorePendingStylesheets
-    // needs to be audited. see http://crbug.com/590369 for more details.
-    frame().document()->updateStyleAndLayoutIgnorePendingStylesheets();
-
-    int selectionOffsetsStart = static_cast<int>(getSelectionOffsets().start());
-    int start = selectionOffsetsStart + selectionStart;
-    int end = selectionOffsetsStart + selectionEnd;
-    PlainTextRange selectedRange = createRangeForSelection(start, end, text.length());
+    PlainTextRange selectedRange = createSelectionRangeForSetComposition(selectionStart, selectionEnd, text.length());
 
     // Dispatch an appropriate composition event to the focused node.
     // We check the composition status and choose an appropriate composition event since this
     // function is used for three purposes:
     // 1. Starting a new composition.
     //    Send a compositionstart and a compositionupdate event when this function creates
     //    a new composition node, i.e.
     //    !hasComposition() && !text.isEmpty().
     //    Sending a compositionupdate event at this time ensures that at least one
     //    compositionupdate event is dispatched.
@@ skipping 82 matching lines @@
     for (const auto& underline : underlines) {
         unsigned underlineStart = baseOffset + underline.startOffset();
         unsigned underlineEnd = baseOffset + underline.endOffset();
         EphemeralRange ephemeralLineRange = EphemeralRange(Position(baseNode, underlineStart), Position(baseNode, underlineEnd));
         if (ephemeralLineRange.isNull())
             continue;
         frame().document()->markers().addCompositionMarker(ephemeralLineRange.startPosition(), ephemeralLineRange.endPosition(), underline.color(), underline.thick(), underline.backgroundColor());
     }
 }
 
+PlainTextRange InputMethodController::createSelectionRangeForSetComposition(int selectionStart, int selectionEnd, size_t textLength) const
+{
+    // TODO(xiaochengh): The use of updateStyleAndLayoutIgnorePendingStylesheets
+    // needs to be audited. see http://crbug.com/590369 for more details.
+    frame().document()->updateStyleAndLayoutIgnorePendingStylesheets();
+
+    int selectionOffsetsStart = static_cast<int>(getSelectionOffsets().start());
+    int start = selectionOffsetsStart + selectionStart;
+    int end = selectionOffsetsStart + selectionEnd;
+    return createRangeForSelection(start, end, textLength);
+}
+
 void InputMethodController::setCompositionFromExistingText(const Vector<CompositionUnderline>& underlines, unsigned compositionStart, unsigned compositionEnd)
 {
     Element* editable = frame().selection().rootEditableElement();
     if (!editable)
         return;
 
     DCHECK(!editable->document().needsLayoutTreeUpdate());
 
     const EphemeralRange range = PlainTextRange(compositionStart, compositionEnd).createRange(*editable);
     if (range.isNull())
@@ skipping 150 matching lines @@
     TypingCommand::deleteSelection(*frame().document());
 }
 
 DEFINE_TRACE(InputMethodController)
 {
     visitor->trace(m_frame);
     visitor->trace(m_compositionRange);
 }
 
 } // namespace blink