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 339 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 length1 = str1.length();
+  size_t length2 = str2.length();
+  size_t commonPrefixLength = 0;

[yosin_UTC9, 2016/10/11 03:58:10]

for-loop is better

const size_t maxCommonPrefixLength = std::min(length1, length2);
for (size_t commonPrefixLength = 0; commonPrefixLength < maxCommonPrefixLength;
++commonPrefixLength) {
  if (str1[commonPrefixLength] != str2[commonPrefixLength])
   break;
}

[yabinh, 2016/10/11 05:07:54]

Done.

+
+  while (commonPrefixLength < length1 && commonPrefixLength < length2 &&
+         str1[commonPrefixLength] == str2[commonPrefixLength])
+    ++commonPrefixLength;
+
+  return commonPrefixLength;
+}
+
+static size_t computeCommonSuffixLength(const String& str1,
+                                        const String& str2) {
+  size_t length1 = str1.length();
+  size_t length2 = str2.length();
+  size_t commonSuffixLength = 0;
+
+  while (commonSuffixLength < length1 && commonSuffixLength < length2 &&

[yosin_UTC9, 2016/10/11 03:58:10]

for-loop is better

const size_t maxCommonPrefixLength = std::min(length1, length2);
for (size_t commonPrefixLength = 0; commonPrefixLength < maxCommonPrefixLength;
++commonPrefixLength) {
  if (str1[length1 - commonPrefixLength] != str2[length2 - commonPrefixLength])
   break;
}

[yabinh, 2016/10/11 05:07:54]

Done.

+         str1[length1 - commonSuffixLength - 1] ==
+             str2[length2 - commonSuffixLength - 1])
+    ++commonSuffixLength;
+
+  return commonSuffixLength;
+}
+
+static size_t computeCommonGraphemeClusterPrefixLengthForSetComposition(
+    const String& oldText,
+    const String& newText,
+    const Element* rootEditableElement) {
+  size_t commonPrefixLength = computeCommonPrefixLength(oldText, newText);
+
+  // For multi-code text, we should adjust it for grapheme boundary.

[yosin_UTC9, 2016/10/11 03:58:10]

s/multi-code text/grapheme cluster/

[yabinh, 2016/10/11 05:07:54]

Done.

+  const EphemeralRange& range =
+      PlainTextRange(0, commonPrefixLength).createRange(*rootEditableElement);
+  if (range.isNull())
+    return 0;
+  const Position& position = range.endPosition();
+  const Position& adjustedPosition = previousPositionOf(
+      nextPositionOf(position, PositionMoveType::GraphemeCluster),
+      PositionMoveType::GraphemeCluster);
+
+  DCHECK(position.anchorNode()->isEqualNode(adjustedPosition.anchorNode()));

[yosin_UTC9, 2016/10/11 03:58:10]

Let's use DCHECK_EQ(position.anchroNode(), adjustedPosition.anchorNode()) to
print nodes at assertion failure.

[yabinh, 2016/10/11 05:07:54]

Done.

+
+  int diff = position.computeOffsetInContainerNode() -

[yosin_UTC9, 2016/10/11 03:58:10]

Could you add DCHECK_GE(position.computeOffsetInContanerNode(),
adjustedPositon.computeOffsetInContainerNode())?
to make sure |diff >= 0|

[yabinh, 2016/10/11 05:07:55]

Done.

+             adjustedPosition.computeOffsetInContainerNode();
+  return commonPrefixLength - static_cast<size_t>(diff);
+}
+
+static size_t computeCommonGraphemeClusterSuffixLengthForSetComposition(
+    const String& oldText,
+    const String& newText,
+    const Element* rootEditableElement) {
+  size_t commonSuffixLength = computeCommonSuffixLength(oldText, newText);
+
+  // For multi-code text, we should adjust it for grapheme boundary.
+  const EphemeralRange& range =
+      PlainTextRange(0, oldText.length() - commonSuffixLength)
+          .createRange(*rootEditableElement);
+  if (range.isNull())
+    return 0;
+  const Position& position = range.endPosition();
+  const Position& adjustedPosition = nextPositionOf(
+      previousPositionOf(position, PositionMoveType::GraphemeCluster),
+      PositionMoveType::GraphemeCluster);
+
+  DCHECK(position.anchorNode()->isEqualNode(adjustedPosition.anchorNode()));

[yosin_UTC9, 2016/10/11 03:58:10]

See comment in computeCommonGraphemeClusterPrefixLengthForSetComposition()

[yabinh, 2016/10/11 05:07:54]

Done.

+
+  int diff = adjustedPosition.computeOffsetInContainerNode() -
+             position.computeOffsetInContainerNode();
+  return commonSuffixLength - static_cast<size_t>(diff);
+}
+
+void InputMethodController::setCompositionWithIncrementalText(
+    const String& text,
+    const Vector<CompositionUnderline>& underlines,
+    int selectionStart,
+    int selectionEnd) {
+  Element* rootEditableElement = frame().selection().rootEditableElement();
+  if (!rootEditableElement)
+    return;
+
+  String composing = composingText();
+  size_t commonPrefixLength =
+      computeCommonGraphemeClusterPrefixLengthForSetComposition(
+          composing, text, rootEditableElement);
+
+  // We should ignore common prefix when finding common suffix.
+  size_t commonSuffixLength =

[yosin_UTC9, 2016/10/11 03:58:10]

|const size_t|?

[yabinh, 2016/10/11 05:07:54]

Done.

+      computeCommonGraphemeClusterSuffixLengthForSetComposition(
+          composing.right(composing.length() - commonPrefixLength),
+          text.right(text.length() - commonPrefixLength), rootEditableElement);
+
+  const bool appending =
+      text.length() > commonPrefixLength + commonSuffixLength;
+  const bool subtracting =
+      composing.length() > commonPrefixLength + +commonSuffixLength;
+
+  if (appending || subtracting) {
+    Element* editable = frame().selection().rootEditableElement();
+    if (!editable)

[yosin_UTC9, 2016/10/11 03:58:10]

We've already check |editable| isn't nullptr at L441.

[yabinh, 2016/10/11 05:07:54]

Done.

+      return;
+
+    // Select the text to be subtracted.
+    size_t compositionStart =
+        PlainTextRange::create(*editable, compositionEphemeralRange()).start();
+    size_t subtractionStart = compositionStart + commonPrefixLength;
+    size_t subtractionEnd =
+        compositionStart + composing.length() - commonSuffixLength;
+    const EphemeralRange& subtractionRange =
+        PlainTextRange(subtractionStart, subtractionEnd).createRange(*editable);
+    VisibleSelection selection;
+    selection.setWithoutValidation(subtractionRange.startPosition(),
+                                   subtractionRange.endPosition());
+    frame().selection().setSelection(selection, 0);
+    clear();
+
+    Element* target = frame().document()->focusedElement();

[yosin_UTC9, 2016/10/11 03:58:10]

FrameSeleciton::setSelection() can change document associate to |frame|.
So, we should do:

const Document& currentDocument = frame().document();
...
setSeleciton();
if (currentDocument != frame().document())
  return;
Element* target = frame().document()->focusedElement();
...

[yabinh, 2016/10/11 05:07:54]

Done.

+    if (!target)
+      return;
+
+    // Append the incremental text.
+    size_t appendingLength =
+        text.length() - commonPrefixLength - commonSuffixLength;
+    String appendingText = text.substring(commonPrefixLength, appendingLength);
+    insertTextDuringCompositionWithEvents(frame(), appendingText,
+                                          TypingCommand::PreventSpellChecking,
+                                          TypingCommand::TextCompositionUpdate);
+
+    // Event handlers might destroy document.
+    if (!frame().document())

[yosin_UTC9, 2016/10/11 03:58:10]

Similar to L478 comment.

[yabinh, 2016/10/11 05:07:54]

Done.

+      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, compositionStart,
+                                   compositionStart + text.length());
+  }
+
+  selectComposition();
+  const PlainTextRange& selectedRange = createSelectionRangeForSetComposition(
+      selectionStart, selectionEnd, text.length());
+  // We shouldn't close typing in the middle of setComposition.
+  setEditableSelectionOffsets(selectedRange, NotUserTriggered);
+  m_isDirty = true;
+}
+
 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()) {
+    return setCompositionWithIncrementalText(text, underlines, selectionStart,
+                                             selectionEnd);
+  }
+
   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 96 matching lines @@
     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());
 
@@ skipping 168 matching lines @@
       new RangeVector(1, m_frame->selection().firstRange()));
   TypingCommand::deleteSelection(*frame().document());
 }
 
 DEFINE_TRACE(InputMethodController) {
   visitor->trace(m_frame);
   visitor->trace(m_compositionRange);
 }
 
 }  // namespace blink