Infinite Suggest for mac

chrome/browser/ui/cocoa/omnibox/omnibox_popup_view_mac.mm

 // Copyright (c) 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.
 
 #include "chrome/browser/ui/cocoa/omnibox/omnibox_popup_view_mac.h"
 
 #include <cmath>
 
 #include "base/stl_util.h"
 #include "base/strings/sys_string_conversions.h"
@@ skipping 105 matching lines @@
   gfx::Font result_font(OmniboxViewMac::GetFieldFont());
 
   // Calculate the width of the matrix based on backing out the popup's border
   // from the width of the field.
   const CGFloat matrix_width = NSWidth([field_ bounds]);
   DCHECK_GT(matrix_width, 0.0);
 
   // Load the results into the popup's matrix.
   DCHECK_GT(rows, 0U);
   [matrix_ renewRows:rows columns:1];
+
+  CGFloat max_match_width = 0;
+  CGFloat max_required_width = 0;
+  CGFloat ignored_prefix_width = 0;
+  CGFloat ellipsis_prefix_width = 0;
   for (size_t ii = 0; ii < rows; ++ii) {
     OmniboxPopupCell* cell = [matrix_ cellAtRow:ii column:0];
     const AutocompleteMatch& match = GetResult().match_at(ii + start_match);
     [cell setImage:ImageForMatch(match)];
-    [cell setAttributedTitle:MatchText(match, result_font, matrix_width)];
+    NSAttributedString* match_text =
+        MatchText(match, result_font, matrix_width);
+
+    if (match.type == AutocompleteMatchType::SEARCH_SUGGEST_INFINITE) {
+      // Use ignored_prefix from the fill_into_edit to compute the
+      // required_width and max_required_width.
+      size_t input_start =
+          (match.transition == content::PAGE_TRANSITION_KEYWORD) ?
+              match.keyword.length() + 1 : 0;
+      string16 ignored_prefix = match.fill_into_edit.substr(input_start,
+          match.fill_into_edit.rfind(match.contents) - input_start);
+      NSMutableAttributedString* ignored_prefix_as = DecorateMatchedString(
+          ignored_prefix, ACMatchClassifications(), ContentTextColor(),
+          DimContentTextColor(), result_font);
+      if (ignored_prefix_width == 0)
+        ignored_prefix_width = [ignored_prefix_as size].width;
+      [ignored_prefix_as appendAttributedString:match_text];
+      CGFloat required_width = [ignored_prefix_as size].width;
+      if (required_width > max_required_width)
+        max_required_width = required_width;
+
+      // Use ellipsis prefix to compute the match_width and max_match_width.
+      NSMutableAttributedString* ellipsis_prefix = DecorateMatchedString(
+          base::string16(gfx::kEllipsisUTF16).append(UTF8ToUTF16(" ")),
+          ACMatchClassifications(), ContentTextColor(), DimContentTextColor(),
+          result_font);
+      if (ellipsis_prefix_width == 0)
+        ellipsis_prefix_width = [ellipsis_prefix size].width;
+      [ellipsis_prefix appendAttributedString:match_text];
+      CGFloat match_width = [ellipsis_prefix size].width;
+      if (match_width > max_match_width)
+        max_match_width = match_width;
+      match_text = ellipsis_prefix;
+    }
+    [cell setAttributedTitle:match_text];
+  }
+
+  const float text_width = matrix_width - kTextXOffset;
+  CGFloat infinite_suggest_offset = (max_required_width < text_width) ?
+    std::max(ignored_prefix_width - ellipsis_prefix_width, 0.0f) :
+    std::max(text_width - max_match_width, 0.0f);
+
+  for (size_t ii = 0; ii < rows; ++ii) {
+    const AutocompleteMatch& match = GetResult().match_at(ii + start_match);
+    OmniboxPopupCell* cell = [matrix_ cellAtRow:ii column:0];
+    CGFloat additional_offset =
+        (match.type != AutocompleteMatchType::SEARCH_SUGGEST_INFINITE) ?
+            0 : infinite_suggest_offset;
+    [cell setAdditionalOffset:additional_offset];
   }
 
   // Set the cell size to fit a line of text in the cell's font.  All
   // cells should use the same font and each should layout in one
   // line, so they should all be about the same height.
   const NSSize cell_size = [[matrix_ cellAtRow:0 column:0] cellSize];
   DCHECK_GT(cell_size.height, 0.0);
   const CGFloat cell_height = cell_size.height + kCellHeightAdjust;
   [matrix_ setCellSize:NSMakeSize(matrix_width, cell_height)];
 
@@ skipping 351 matching lines @@
   return OmniboxViewMac::ImageForResource(resource_id);
 }
 
 void OmniboxPopupViewMac::OpenURLForRow(size_t row,
                                         WindowOpenDisposition disposition) {
   size_t start_match = model_->result().ShouldHideTopMatch() ? 1 : 0;
   row += start_match;
   DCHECK_LT(row, GetResult().size());
   omnibox_view_->OpenMatch(GetResult().match_at(row), disposition, GURL(), row);
 }