[WIP] Add support for Android SuggestionSpans when editing text

content/public/android/java/src/org/chromium/content/browser/input/ImeAdapter.java

 // Copyright 2012 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.
 
 package org.chromium.content.browser.input;
 
 import android.content.res.Configuration;
 import android.os.Build;
 import android.os.ResultReceiver;
 import android.os.SystemClock;
 import android.text.SpannableString;
 import android.text.TextUtils;
 import android.text.style.BackgroundColorSpan;
 import android.text.style.CharacterStyle;
+import android.text.style.SuggestionSpan;
 import android.text.style.UnderlineSpan;
 import android.view.KeyCharacterMap;
 import android.view.KeyEvent;
 import android.view.View;
 import android.view.inputmethod.BaseInputConnection;
 import android.view.inputmethod.EditorInfo;
 import android.view.inputmethod.InputConnection;
 
 import org.chromium.base.Log;
 import org.chromium.base.VisibleForTesting;
@@ skipping 26 matching lines @@
  * Note that the native peer object does not take any strong reference onto the
  * instance of this java object, hence it is up to the client of this class (e.g.
  * the ViewEmbedder implementor) to hold a strong reference to it for the required
  * lifetime of the object.
  */
 @JNINamespace("content")
 public class ImeAdapter {
     private static final String TAG = "cr_Ime";
     private static final boolean DEBUG_LOGS = false;
 
+    private static final int SUGGESTION_SPAN_UNDERLINE_COLOR = 0x88C8C8C8;
+
     public static final int COMPOSITION_KEY_CODE = 229;
 
     /**
      * Interface for the delegate that needs to be notified of IME changes.
      */
     public interface ImeAdapterDelegate {
         /**
          * Called to notify the delegate about synthetic/real key events before sending to renderer.
          */
         void onImeEvent();
@@ skipping 27 matching lines @@
     private ChromiumBaseInputConnection mInputConnection;
     private ChromiumBaseInputConnection.Factory mInputConnectionFactory;
 
     private final ImeAdapterDelegate mViewEmbedder;
     // This holds the information necessary for constructing CursorAnchorInfo, and notifies to
     // InputMethodManager on appropriate timing, depending on how IME requested the information
     // via InputConnection. The update request is per InputConnection, hence for each time it is
     // re-created, the monitoring status will be reset.
     private final CursorAnchorInfoController mCursorAnchorInfoController;
 
+    private SuggestionsPopupWindow mSuggestionsPopupWindow;
+
     private int mTextInputType = TextInputType.NONE;
     private int mTextInputFlags;
     private int mTextInputMode = WebTextInputMode.kDefault;
 
     // Keep the current configuration to detect the change when onConfigurationChanged() is called.
     private Configuration mCurrentConfig;
 
     private int mLastSelectionStart;
     private int mLastSelectionEnd;
     private String mLastText;
@@ skipping 10 matching lines @@
      * @param wrapper InputMethodManagerWrapper that should receive all the call directed to
      *                InputMethodManager.
      * @param embedder The view that is used for callbacks from ImeAdapter.
      */
     public ImeAdapter(Configuration initialConfiguration, WebContents webContents,
             InputMethodManagerWrapper wrapper, ImeAdapterDelegate embedder) {
         mInputMethodManagerWrapper = wrapper;
         mViewEmbedder = embedder;
         // Deep copy newConfig so that we can notice the difference.
         mCurrentConfig = new Configuration(initialConfiguration);
-        // CursorAnchroInfo is supported only after L.
+        // CursorAnchorInfo is supported only after L.
         if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.LOLLIPOP) {
-            mCursorAnchorInfoController = CursorAnchorInfoController.create(wrapper,
-                    new CursorAnchorInfoController.ComposingTextDelegate() {
+            mCursorAnchorInfoController = CursorAnchorInfoController.create(
+                    wrapper, new CursorAnchorInfoController.ComposingTextDelegate() {
                         @Override
                         public CharSequence getText() {
                             return mLastText;
                         }
                         @Override
                         public int getSelectionStart() {
                             return mLastSelectionStart;
                         }
                         @Override
                         public int getSelectionEnd() {
@@ skipping 585 matching lines @@
                 insertionMarkerBottom, mViewEmbedder.getAttachedView());
     }
 
     @CalledByNative
     private void populateUnderlinesFromSpans(CharSequence text, long underlines) {
         if (DEBUG_LOGS) {
             Log.i(TAG, "populateUnderlinesFromSpans: text [%s], underlines [%d]", text, underlines);
         }
         if (!(text instanceof SpannableString)) return;
 
-        SpannableString spannableString = ((SpannableString) text);
+        SpannableString spannableString = (SpannableString) text;
         CharacterStyle spans[] =
                 spannableString.getSpans(0, text.length(), CharacterStyle.class);
         for (CharacterStyle span : spans) {
             if (span instanceof BackgroundColorSpan) {
                 nativeAppendBackgroundColorSpan(underlines, spannableString.getSpanStart(span),
                         spannableString.getSpanEnd(span),
                         ((BackgroundColorSpan) span).getBackgroundColor());
+            } else if (span instanceof SuggestionSpan) {
+                if (Build.VERSION.SDK_INT < Build.VERSION_CODES.LOLLIPOP) {
+                    // The suggestion menu is only supported on Lollipop and newer
+                    continue;
+                }
+
+                SuggestionSpan suggestionSpan = (SuggestionSpan) span;
+                nativeAppendSuggestionSpan(underlines, spannableString.getSpanStart(suggestionSpan),
+                        spannableString.getSpanEnd(suggestionSpan), SUGGESTION_SPAN_UNDERLINE_COLOR,
+                        suggestionSpan.getFlags(), suggestionSpan.getSuggestions());
+
             } else if (span instanceof UnderlineSpan) {
                 nativeAppendUnderlineSpan(underlines, spannableString.getSpanStart(span),
                         spannableString.getSpanEnd(span));
             }
         }
     }
 
     @CalledByNative
     private void cancelComposition() {
         if (DEBUG_LOGS) Log.i(TAG, "cancelComposition");
         if (mInputConnection != null) restartInput();
     }
 
     @CalledByNative
     private void setCharacterBounds(float[] characterBounds) {
         if (mCursorAnchorInfoController == null) return;
         mCursorAnchorInfoController.setCompositionCharacterBounds(characterBounds,
                 mViewEmbedder.getAttachedView());
     }
 
     @CalledByNative
     private void onConnectedToRenderProcess() {
         if (DEBUG_LOGS) Log.i(TAG, "onConnectedToRenderProcess");
         mIsConnected = true;
         createInputConnectionFactory();
         resetAndHideKeyboard();
     }
 
+    @CalledByNative
+    private void showSuggestionMenu(SuggestionInfo[] suggestionInfos) {
+        if (mSuggestionsPopupWindow == null) {
+            mSuggestionsPopupWindow =
+                    new SuggestionsPopupWindow(mInputMethodManagerWrapper.getContext(), this,
+                            mViewEmbedder.getAttachedView(), mCursorAnchorInfoController);
+        }
+
+        mSuggestionsPopupWindow.setSuggestionInfos(suggestionInfos);
+        mSuggestionsPopupWindow.show();
+    }
+
+    public void applySuggestionReplacement(long suggestionMarkerID, int suggestionIndex) {
+        nativeApplySuggestionReplacement(
+                mNativeImeAdapterAndroid, suggestionMarkerID, suggestionIndex);
+    }
+
+    public void deleteSuggestionHighlight() {
+        nativeDeleteSuggestionHighlight(mNativeImeAdapterAndroid);
+    }
+
+    public void suggestionMenuClosed() {
+        nativeSuggestionMenuClosed(mNativeImeAdapterAndroid);
+    }
+
     private native long nativeInit(WebContents webContents);
     private native boolean nativeSendKeyEvent(long nativeImeAdapterAndroid, KeyEvent event,
             int type, int modifiers, long timestampMs, int keyCode, int scanCode,
             boolean isSystemKey, int unicodeChar);
-    private static native void nativeAppendUnderlineSpan(long underlinePtr, int start, int end);
     private static native void nativeAppendBackgroundColorSpan(long underlinePtr, int start,
             int end, int backgroundColor);
+    private static native void nativeAppendSuggestionSpan(long underlinePtr, int start, int end,
+            int underlineColor, int flags, String[] suggestions);
+    private static native void nativeAppendUnderlineSpan(long underlinePtr, int start, int end);
     private native void nativeSetComposingText(long nativeImeAdapterAndroid, CharSequence text,
             String textStr, int newCursorPosition);
     private native void nativeCommitText(
             long nativeImeAdapterAndroid, CharSequence text, String textStr, int newCursorPosition);
     private native void nativeFinishComposingText(long nativeImeAdapterAndroid);
     private native void nativeSetEditableSelectionOffsets(long nativeImeAdapterAndroid,
             int start, int end);
     private native void nativeSetComposingRegion(long nativeImeAdapterAndroid, int start, int end);
     private native void nativeDeleteSurroundingText(long nativeImeAdapterAndroid,
             int before, int after);
     private native void nativeDeleteSurroundingTextInCodePoints(
             long nativeImeAdapterAndroid, int before, int after);
     private native boolean nativeRequestTextInputStateUpdate(long nativeImeAdapterAndroid);
     private native void nativeRequestCursorUpdate(long nativeImeAdapterAndroid,
             boolean immediateRequest, boolean monitorRequest);
+    private native void nativeApplySuggestionReplacement(
+            long nativeImeAdapterAndroid, long suggestionMarkerID, int suggestionIndex);
+    private native void nativeDeleteSuggestionHighlight(long nativeImeAdapterAndroid);
+    private native void nativeSuggestionMenuClosed(long nativeImeAdapterAndroid);
 }