Truncate long URL fragments in Android page info popup

chrome/android/java/src/org/chromium/chrome/browser/WebsiteSettingsPopup.java

 // Copyright 2013 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.chrome.browser;
 
 import android.app.Dialog;
 import android.content.Context;
 import android.content.DialogInterface;
 import android.graphics.Color;
 import android.graphics.drawable.ColorDrawable;
-import android.net.Uri;
 import android.text.Layout;
 import android.text.Spannable;
 import android.text.SpannableStringBuilder;
 import android.text.style.ForegroundColorSpan;
 import android.text.style.StyleSpan;
 import android.util.AttributeSet;
 import android.view.Gravity;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.View.OnClickListener;
@@ skipping 45 matching lines @@
             this.type = type;
             this.value = value;
         }
 
         @Override
         public String toString() {
             return name;
         }
     }
 
+    /**
+     * A TextView which truncates and displays a URL such that the origin is always visible.
+     * The URL can be expanded by clicking on the it.
+     */
     public static class ElidedUrlTextView extends TextView {
         // The number of lines to display when the URL is truncated. This number
         // should still allow the origin to be displayed. NULL before
         // setUrlAfterLayout() is called.
         private Integer mTruncatedUrlLinesToDisplay;
 
+        // The number of lines to display when the URL is expanded. This should be enough to display
+        // at most two lines of the fragment if there is one in the URL.
+        private Integer mFullLinesToDisplay;
+
         // If true, the text view will show the truncated text. If false, it
         // will show the full, expanded text.
         private boolean mIsShowingTruncatedText = true;
 
         // The profile to use when getting the end index for the origin.
         private Profile mProfile = null;
 
         // The maximum number of lines currently shown in the view
         private int mCurrentMaxLines = Integer.MAX_VALUE;
 
         /** Constructor for inflating from XML. */
         public ElidedUrlTextView(Context context, AttributeSet attrs) {
             super(context, attrs);
         }
 
         @Override
         public void setMaxLines(int maxlines) {
             super.setMaxLines(maxlines);
             mCurrentMaxLines = maxlines;
         }
 
+        /**
+         * Find the number of lines of text which must be shown in order to display the character at
+         * a given index.
+         */
+        private int getLineForIndex(int index) {
+            Layout layout = getLayout();
+            int endLine = 0;
+            while (endLine < layout.getLineCount() && layout.getLineEnd(endLine) < index) {
+                endLine++;
+            }
+            // Since endLine is an index, add 1 to get the number of lines.
+            return endLine + 1;
+        }
+
         @Override
         protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {
             setMaxLines(Integer.MAX_VALUE);
             super.onMeasure(widthMeasureSpec, heightMeasureSpec);
             assert mProfile != null : "setProfile() must be called before layout.";
+            String urlText = getText().toString();
 
             // Lay out the URL in a StaticLayout that is the same size as our final
             // container.
-            Layout layout = getLayout();
-            int originEndIndex =
-                    OmniboxUrlEmphasizer.getOriginEndIndex(getText().toString(), mProfile);
+            int originEndIndex = OmniboxUrlEmphasizer.getOriginEndIndex(urlText, mProfile);
 
             // Find the range of lines containing the origin.
-            int originEndLineIndex = 0;
-            while (originEndLineIndex < layout.getLineCount()
-                    && layout.getLineEnd(originEndLineIndex) < originEndIndex) {
-                originEndLineIndex++;
-            }
+            int originEndLine = getLineForIndex(originEndIndex);
 
             // Display an extra line so we don't accidentally hide the origin with
             // ellipses
-            int lastLineIndexToDisplay = originEndLineIndex + 1;
+            mTruncatedUrlLinesToDisplay = originEndLine + 1;
 
-            // Since lastLineToDisplay is an index, add 1 to get the maximum number
-            // of lines. This will always be at least 2 lines (when the origin is
-            // fully contained on line 0).
-            mTruncatedUrlLinesToDisplay = lastLineIndexToDisplay + 1;
+            // Find the line where the fragment starts. Since # is a reserved character, it is safe
+            // to just search for the first # to appear in the url.
+            int fragmentStartIndex = urlText.indexOf('#');
+            if (fragmentStartIndex == -1) fragmentStartIndex = urlText.length();
+
+            int fragmentStartLine = getLineForIndex(fragmentStartIndex);
+            mFullLinesToDisplay = fragmentStartLine + 1;
+
+            // If there is no origin (according to OmniboxUrlEmphasizer), make sure the fragment is
+            // still hidden correctly.
+            if (mFullLinesToDisplay < mTruncatedUrlLinesToDisplay) {
+                mTruncatedUrlLinesToDisplay = mFullLinesToDisplay;
+            }
 
             if (updateMaxLines()) super.onMeasure(widthMeasureSpec, heightMeasureSpec);
         }
 
         /**
          * Sets the profile to use when calculating the end index of the origin.
          * Must be called before layout.
          *
          * @param profile The profile to use when coloring the URL.
          */
         public void setProfile(Profile profile) {
             mProfile = profile;
         }
 
         /**
          * Toggles truncating/expanding the URL text. If the URL text is not
          * truncated, has no effect.
          */
         public void toggleTruncation() {
             mIsShowingTruncatedText = !mIsShowingTruncatedText;
             updateMaxLines();
         }
 
         private boolean updateMaxLines() {
-            int maxLines = Integer.MAX_VALUE;
+            int maxLines = mFullLinesToDisplay;
             if (mIsShowingTruncatedText) maxLines = mTruncatedUrlLinesToDisplay;
             if (maxLines != mCurrentMaxLines) {
                 setMaxLines(maxLines);
                 return true;
             }
             return false;
         }
     }
 
     private static final int MAX_TABLET_DIALOG_WIDTH_DP = 400;
 
-    private static final char FIRST_UNICODE_WHITESPACE = '\u2000';
-    private static final char FINAL_UNICODE_WHITESPACE = '\u200F';
-    private static final char UNICODE_NBSP = '\u00A0';
-
     private final Context mContext;
     private final Profile mProfile;
     private final WebContents mWebContents;
 
     // A pointer to the C++ object for this UI.
     private final long mNativeWebsiteSettingsPopup;
 
     // The outer container, filled with the layout from website_settings.xml.
     private final LinearLayout mContainer;
 
@@ skipping 109 matching lines @@
         try {
             mParsedUrl = new URI(mFullUrl);
             mIsInternalPage = UrlUtilities.isInternalScheme(mParsedUrl);
         } catch (URISyntaxException e) {
             mParsedUrl = null;
             mIsInternalPage = false;
         }
         mSecurityLevel = ToolbarModel.getSecurityLevelForWebContents(mWebContents);
         mDeprecatedSHA1Present = ToolbarModel.isDeprecatedSHA1Present(mWebContents);
 
-        String displayUrl = prepareUrlForDisplay(mFullUrl);
-        SpannableStringBuilder urlBuilder = new SpannableStringBuilder(displayUrl);
+        SpannableStringBuilder urlBuilder = new SpannableStringBuilder(mFullUrl);
         OmniboxUrlEmphasizer.emphasizeUrl(urlBuilder, mContext.getResources(), mProfile,
                 mSecurityLevel, mIsInternalPage, true);
         mUrlTitle.setText(urlBuilder);
 
         // Set the URL connection message now, and the URL after layout (so it
         // can calculate its ideal height).
         mUrlConnectionMessage.setText(getUrlConnectionMessage());
     }
 
     /**
-     * Percent-encodes suspicious Unicode whitespace characters in a URL so that it can be safely
-     * displayed.
-     */
-    public static String prepareUrlForDisplay(String urlStr) {
-        StringBuilder urlBuilder = new StringBuilder();
-        for (int i = 0; i < urlStr.length(); i++) {
-            char c = urlStr.charAt(i);
-            if ((c >= FIRST_UNICODE_WHITESPACE
-                        && c <= FINAL_UNICODE_WHITESPACE)
-                    || c == ' '
-                    || c == UNICODE_NBSP) {
-                urlBuilder.append(Uri.encode(Character.toString(c)));
-            } else {
-                urlBuilder.append(c);
-            }
-        }
-        return urlBuilder.toString();
-    }
-
-    /**
      * Sets the visibility of the lower area of the dialog (containing the permissions and 'Site
      * Settings' button).
      *
      * @param isVisible Whether to show or hide the dialog area.
      */
     private void setVisibilityOfLowerDialogArea(boolean isVisible) {
         mHorizontalSeparator.setVisibility(isVisible ? View.VISIBLE : View.GONE);
         mLowerDialogArea.setVisibility(isVisible ? View.VISIBLE : View.GONE);
     }
 
@@ skipping 226 matching lines @@
         new WebsiteSettingsPopup(context, profile, webContents);
     }
 
     private static native long nativeInit(WebsiteSettingsPopup popup, WebContents webContents);
 
     private native void nativeDestroy(long nativeWebsiteSettingsPopupAndroid);
 
     private native void nativeOnPermissionSettingChanged(long nativeWebsiteSettingsPopupAndroid,
             int type, int setting);
 }