Refactor DocumentMarkerController

third_party/WebKit/Source/core/editing/markers/SpellCheckMarkerList.cpp

Index: third_party/WebKit/Source/core/editing/markers/SpellCheckMarkerList.cpp
diff --git a/third_party/WebKit/Source/core/editing/markers/SpellCheckMarkerList.cpp b/third_party/WebKit/Source/core/editing/markers/SpellCheckMarkerList.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..58342b4fe8de9e21c45e1ad81b694254e88dfc75
--- /dev/null
+++ b/third_party/WebKit/Source/core/editing/markers/SpellCheckMarkerList.cpp
@@ -0,0 +1,64 @@
+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "core/editing/markers/SpellCheckMarkerList.h"
+
+#include <algorithm>
+#include "core/dom/Text.h"
+
+namespace blink {
+
+SpellCheckMarkerList::SpellCheckMarkerList(
+    DocumentMarkerController* documentMarkerController,
+    DocumentMarker::MarkerType type)
+    : EditingMarkerList(documentMarkerController), m_type(type) {}
+
+static bool doesNotOverlap(const Member<DocumentMarker>& lhv,
+                           const DocumentMarker* rhv) {
+  return lhv->endOffset() < rhv->startOffset();
+}
+
+DocumentMarker::MarkerType SpellCheckMarkerList::allowedMarkerType() const {
+  return m_type;
+}
+
+void SpellCheckMarkerList::insert(DocumentMarker* marker) {
+  if (!m_markersAreSorted)
+    sortMarkerList();
+
+  auto firstOverlappingIt = std::lower_bound(m_markers.begin(), m_markers.end(),
+                                             marker, doesNotOverlap);
+  size_t index = firstOverlappingIt - m_markers.begin();
+  m_markers.insert(index, marker);
+  auto insertedIt = m_markers.begin() + index;
+  for (auto it = insertedIt + 1;
+       it != m_markers.end() &&
+       (*it)->startOffset() <= (*insertedIt)->endOffset();) {
+    (*insertedIt)
+        ->setStartOffset(
+            std::min((*insertedIt)->startOffset(), (*it)->startOffset()));
+    (*insertedIt)
+        ->setEndOffset(
+            std::max((*insertedIt)->endOffset(), (*it)->endOffset()));
+    m_markers.remove(it - m_markers.begin());
+  }
+}
+
+void SpellCheckMarkerList::removeMarkersForWords(const Text& textNode,
+                                                 const Vector<String>& words) {
+  // Build a second vector and swap with m_markers to avoid O(n^2) performance
+  HeapVector<Member<DocumentMarker>> newMarkerList;
+
+  for (Member<DocumentMarker> marker : m_markers) {
+    unsigned start = marker->startOffset();
+    unsigned length = marker->endOffset() - marker->startOffset();
+    String markerText = textNode.data().substring(start, length);
+    if (!words.contains(markerText))
+      newMarkerList.push_back(marker);
+  }
+
+  std::swap(m_markers, newMarkerList);
+}
+
+}  // namespace blink