[WIP] Add support for Android SuggestionSpans when editing text

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

 /*
  * Copyright (C) 2011 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 72 matching lines @@
   return match ? &trueInstance : &falseInstance;
 }
 
 inline DocumentMarkerTextMatch* toDocumentMarkerTextMatch(
     DocumentMarkerDetails* details) {
   if (details && details->isTextMatch())
     return static_cast<DocumentMarkerTextMatch*>(details);
   return 0;
 }
 
-class TextCompositionMarkerDetails final : public DocumentMarkerDetails {
+class TextCompositionMarkerDetails : public DocumentMarkerDetails {
  public:
   static TextCompositionMarkerDetails* create(Color underlineColor,
                                               bool thick,
                                               Color backgroundColor);
-
-  bool isComposition() const override { return true; }
-  Color underlineColor() const { return m_underlineColor; }
-  bool thick() const { return m_thick; }
-  Color backgroundColor() const { return m_backgroundColor; }
-
- private:
   TextCompositionMarkerDetails(Color underlineColor,
                                bool thick,
                                Color backgroundColor)
       : m_underlineColor(underlineColor),
         m_backgroundColor(backgroundColor),
         m_thick(thick) {}
 
+  bool isComposition() const override { return true; }
+  Color underlineColor() const { return m_underlineColor; }
+  bool thick() const { return m_thick; }
+  Color backgroundColor() const { return m_backgroundColor; }
+
+ private:
   Color m_underlineColor;
   Color m_backgroundColor;
   bool m_thick;
 };
 
 TextCompositionMarkerDetails* TextCompositionMarkerDetails::create(
     Color underlineColor,
     bool thick,
     Color backgroundColor) {
   return new TextCompositionMarkerDetails(underlineColor, thick,
                                           backgroundColor);
 }
 
 inline TextCompositionMarkerDetails* toTextCompositionMarkerDetails(
     DocumentMarkerDetails* details) {
   if (details && details->isComposition())
     return static_cast<TextCompositionMarkerDetails*>(details);
   return nullptr;
 }
 
+class TextSuggestionMarkerDetails final : public TextCompositionMarkerDetails {
+ public:
+  static TextSuggestionMarkerDetails* create(
+      Color underlineColor,
+      bool thick,
+      Color backgroundColor,
+      const std::vector<std::string>& suggestions,
+      int id);
+
+  bool isSuggestion() const override { return true; }
+  std::vector<std::string>& suggestions() { return m_suggestions; }
+  int id() { return m_id; }
+
+ private:
+  TextSuggestionMarkerDetails(Color underlineColor,
+                              bool thick,
+                              Color backgroundColor,
+                              const std::vector<std::string>& suggestions,
+                              int id)
+      : TextCompositionMarkerDetails(underlineColor, thick, backgroundColor),
+        m_suggestions(suggestions),
+        m_id(id) {}
+
+  std::vector<std::string> m_suggestions;
+  int m_id;
+};
+
+TextSuggestionMarkerDetails* TextSuggestionMarkerDetails::create(
+    Color underlineColor,
+    bool thick,
+    Color backgroundColor,
+    const std::vector<std::string>& suggestions,
+    int id) {
+  return new TextSuggestionMarkerDetails(underlineColor, thick, backgroundColor,
+                                         suggestions, id);
+}
+
+inline TextSuggestionMarkerDetails* toTextSuggestionMarkerDetails(
+    DocumentMarkerDetails* details) {
+  if (details && details->isSuggestion())
+    return static_cast<TextSuggestionMarkerDetails*>(details);
+  return nullptr;
+}
+
 DocumentMarker::DocumentMarker(MarkerType type,
                                unsigned startOffset,
                                unsigned endOffset,
                                const String& description,
                                uint32_t hash)
     : m_type(type),
       m_startOffset(startOffset),
       m_endOffset(endOffset),
       m_details(description.isEmpty()
                     ? nullptr
                     : DocumentMarkerDescription::create(description)),
       m_hash(hash) {}
 
 DocumentMarker::DocumentMarker(unsigned startOffset,
                                unsigned endOffset,
                                bool activeMatch)
     : m_type(DocumentMarker::TextMatch),
       m_startOffset(startOffset),
       m_endOffset(endOffset),
       m_details(DocumentMarkerTextMatch::create(activeMatch)),
       m_hash(0) {}
 
-DocumentMarker::DocumentMarker(unsigned startOffset,
+DocumentMarker::DocumentMarker(MarkerType type,
+                               unsigned startOffset,
                                unsigned endOffset,
                                Color underlineColor,
                                bool thick,
                                Color backgroundColor)
-    : m_type(DocumentMarker::Composition),
+    : m_type(type),
       m_startOffset(startOffset),
       m_endOffset(endOffset),
       m_details(TextCompositionMarkerDetails::create(underlineColor,
                                                      thick,
                                                      backgroundColor)),
       m_hash(0) {}
 
+DocumentMarker::DocumentMarker(unsigned startOffset,
+                               unsigned endOffset,
+                               Color underlineColor,
+                               bool thick,
+                               Color backgroundColor,
+                               const std::vector<std::string>& suggestions,
+                               int suggestionMarkerID)
+    : m_startOffset(startOffset), m_endOffset(endOffset), m_hash(0) {
+  if (suggestions.empty()) {
+    m_type = DocumentMarker::Composition;
+    m_details = TextCompositionMarkerDetails::create(underlineColor, thick,
+                                                     backgroundColor);
+  } else {
+    m_type = DocumentMarker::Suggestion;
+    m_details = TextSuggestionMarkerDetails::create(
+        underlineColor, thick, backgroundColor, suggestions,
+        suggestionMarkerID);
+  }
+}
+
 DocumentMarker::DocumentMarker(const DocumentMarker& marker)
     : m_type(marker.type()),
       m_startOffset(marker.startOffset()),
       m_endOffset(marker.endOffset()),
       m_details(marker.details()),
       m_hash(marker.hash()) {}
 
 void DocumentMarker::shiftOffsets(int delta) {
   m_startOffset += delta;
   m_endOffset += delta;
@@ skipping 31 matching lines @@
   return false;
 }
 
 Color DocumentMarker::backgroundColor() const {
   if (TextCompositionMarkerDetails* details =
           toTextCompositionMarkerDetails(m_details.get()))
     return details->backgroundColor();
   return Color::transparent;
 }
 
+const std::vector<std::string> DocumentMarker::suggestions() const {
+  if (TextSuggestionMarkerDetails* details =
+          toTextSuggestionMarkerDetails(m_details.get()))
+    return details->suggestions();
+  return std::vector<std::string>();
+}
+
+int DocumentMarker::suggestionMarkerID() const {
+  if (TextSuggestionMarkerDetails* details =
+          toTextSuggestionMarkerDetails(m_details.get()))
+    return details->id();
+  return -1;
+}
+
+void DocumentMarker::replaceSuggestion(int index,
+                                       const std::string& newSuggestion) {
+  if (TextSuggestionMarkerDetails* details =
+          toTextSuggestionMarkerDetails(m_details.get()))
+    details->suggestions()[index] = newSuggestion;
+}
+
 DEFINE_TRACE(DocumentMarker) {
   visitor->trace(m_details);
 }
 
 }  // namespace blink