Show scheme in black and content in gray for data: protocol urls

ios/chrome/browser/ui/omnibox/omnibox_view_ios.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 "ios/chrome/browser/ui/omnibox/omnibox_view_ios.h"
 
 #import <CoreText/CoreText.h>
 #import <MobileCoreServices/MobileCoreServices.h>
 
+#include "base/auto_reset.h"
 #include "base/command_line.h"
 #include "base/ios/device_util.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/user_metrics.h"
 #include "base/metrics/user_metrics_action.h"
 #include "base/strings/string16.h"
 #include "base/strings/sys_string_conversions.h"
 #include "components/omnibox/browser/autocomplete_input.h"
 #include "components/omnibox/browser/autocomplete_match.h"
 #include "components/omnibox/browser/omnibox_edit_model.h"
@@ skipping 23 matching lines @@
 // The color of the rest of the URL (i.e. after the TLD) in the omnibox.
 UIColor* BaseTextColor() {
   return [UIColor colorWithWhite:(161 / 255.0) alpha:1.0];
 }
 
 // The color of the https when there is an error.
 UIColor* ErrorTextColor() {
   return skia::UIColorFromSkColor(gfx::kGoogleRed700);
 }
 
-// The color of the https when there is a warning.
-UIColor* WarningTextColor() {
-  return skia::UIColorFromSkColor(gfx::kGoogleYellow700);
-}
-
 // The color of the https when there is not an error.
 UIColor* SecureTextColor() {
   return skia::UIColorFromSkColor(gfx::kGoogleGreen700);
 }
 
 // The color of the https when highlighted in incognito.
 UIColor* IncognitoSecureTextColor() {
   return [UIColor colorWithWhite:(255 / 255.0) alpha:1.0];
 }
 
-// Helper to make converting url_parse ranges to NSRange easier to
-// read.
-NSRange ComponentToNSRange(const url::Component& component) {
-  return NSMakeRange(static_cast<NSInteger>(component.begin),
-                     static_cast<NSInteger>(component.len));
-}
-
 }  // namespace
 
 // Simple Obj-C object to forward UITextFieldDelegate method calls back to the
 // OmniboxViewIOS.
 @interface AutocompleteTextFieldDelegate : NSObject<OmniboxTextFieldDelegate> {
  @private
   OmniboxViewIOS* editView_;  // weak, owns us
 
   // YES if we are already forwarding an OnDidChange() message to the edit view.
   // Needed to prevent infinite recursion.
@@ skipping 93 matching lines @@
                                ios::ChromeBrowserState* browser_state,
                                id<PreloadProvider> preloader,
                                id<OmniboxPopupPositioner> positioner)
     : OmniboxView(
           controller,
           base::MakeUnique<ChromeOmniboxClientIOS>(controller, browser_state)),
       browser_state_(browser_state),
       field_([field retain]),
       controller_(controller),
       preloader_(preloader),
-      ignore_popup_updates_(false) {
+      ignore_popup_updates_(false),
+      attributing_display_string_(nil) {
   popup_view_.reset(new OmniboxPopupViewIOS(this, model(), positioner));
   field_delegate_.reset(
       [[AutocompleteTextFieldDelegate alloc] initWithEditView:this]);
   [field_ setDelegate:field_delegate_];
   [field_ addTarget:field_delegate_
                 action:@selector(textFieldDidChange:)
       forControlEvents:UIControlEventEditingChanged];
 }
 
 OmniboxViewIOS::~OmniboxViewIOS() {
@@ skipping 428 matching lines @@
 }
 
 bool OmniboxViewIOS::CanCopyURL() {
   return false;
 }
 
 void OmniboxViewIOS::WillPaste() {
   model()->OnPaste();
 }
 
+// static
+UIColor* OmniboxViewIOS::GetSecureTextColor(
+    security_state::SecurityLevel security_level,
+    bool in_dark_mode) {
+  if (security_level == security_state::EV_SECURE ||
+      security_level == security_state::SECURE) {
+    return in_dark_mode ? IncognitoSecureTextColor() : SecureTextColor();
+  }
+
+  // Don't color strikethrough in dark mode. See https://crbug.com/635004#c6
+  if (security_level == security_state::DANGEROUS && !in_dark_mode)
+    return ErrorTextColor();
+
+  return nil;
+}
+
+void OmniboxViewIOS::SetEmphasis(bool emphasize, const gfx::Range& range) {
+  NSRange ns_range = range.IsValid()
+                         ? range.ToNSRange()
+                         : NSMakeRange(0, [attributing_display_string_ length]);
+
+  [attributing_display_string_
+      addAttribute:NSForegroundColorAttributeName
+             value:(emphasize) ? [field_ displayedTextColor] : BaseTextColor()
+             range:ns_range];
+}
+
+void OmniboxViewIOS::UpdateSchemeStyle(const gfx::Range& range) {
+  if (!range.IsValid())
+    return;
+
+  const security_state::SecurityLevel security_level =
+      controller()->GetToolbarModel()->GetSecurityLevel(false);
+
+  if ((security_level == security_state::NONE) ||
+      (security_level == security_state::HTTP_SHOW_WARNING)) {
+    return;
+  }
+
+  DCHECK_NE(security_state::SECURITY_WARNING, security_level);
+  DCHECK_NE(security_state::SECURE_WITH_POLICY_INSTALLED_CERT, security_level);
+
+  if (security_level == security_state::DANGEROUS) {
+    // Add a strikethrough through the scheme.
+    [attributing_display_string_
+        addAttribute:NSStrikethroughStyleAttributeName
+               value:[NSNumber numberWithInteger:NSUnderlineStyleSingle]
+               range:range.ToNSRange()];
+  }
+
+  UIColor* color = GetSecureTextColor(security_level, [field_ incognito]);
+  if (color) {
+    [attributing_display_string_ addAttribute:NSForegroundColorAttributeName
+                                        value:color
+                                        range:range.ToNSRange()];
+  }
+}
+
 NSAttributedString* OmniboxViewIOS::ApplyTextAttributes(
     const base::string16& text) {
   NSMutableAttributedString* as = [[[NSMutableAttributedString alloc]
       initWithString:base::SysUTF16ToNSString(text)] autorelease];
-  url::Component scheme, host;
-  AutocompleteInput::ParseForEmphasizeComponents(
-      text, AutocompleteSchemeClassifierImpl(), &scheme, &host);
-
-  const bool emphasize = model()->CurrentTextIsURL() && (host.len > 0);
-  if (emphasize) {
-    [as addAttribute:NSForegroundColorAttributeName
-               value:BaseTextColor()
-               range:NSMakeRange(0, [as length])];
-
-    [as addAttribute:NSForegroundColorAttributeName
-               value:[field_ displayedTextColor]
-               range:ComponentToNSRange(host)];
-
-    if (scheme.len > 0) {
-      UIColor* color = nil;
-      switch (controller_->GetToolbarModel()->GetSecurityLevel(false)) {
-        case security_state::NONE:
-        case security_state::HTTP_SHOW_WARNING:
-          break;
-        case security_state::SECURITY_WARNING:
-          // Don't color strikethrough schemes. See https://crbug.com/635004#c6
-          if (![field_ incognito])
-            color = WarningTextColor();
-          [as addAttribute:NSStrikethroughStyleAttributeName
-                     value:[NSNumber numberWithInteger:NSUnderlineStyleSingle]
-                     range:ComponentToNSRange(scheme)];
-          break;
-        case security_state::SECURE:
-        case security_state::EV_SECURE:
-          color = [field_ incognito] ? IncognitoSecureTextColor()
-                                     : SecureTextColor();
-          break;
-        case security_state::DANGEROUS:
-          // Don't color strikethrough schemes. See https://crbug.com/635004#c6
-          if (![field_ incognito])
-            color = ErrorTextColor();
-          [as addAttribute:NSStrikethroughStyleAttributeName
-                     value:[NSNumber numberWithInteger:NSUnderlineStyleSingle]
-                     range:ComponentToNSRange(scheme)];
-          break;
-        case security_state::SECURE_WITH_POLICY_INSTALLED_CERT:
-          NOTREACHED();
-      }
-      if (color) {
-        [as addAttribute:NSForegroundColorAttributeName
-                   value:color
-                   range:ComponentToNSRange(scheme)];
-      }
-    }
-  }
+  // Cache a pointer to the attributed string to allow the superclass'
+  // virtual method invocations to add attributes.
+  DCHECK(attributing_display_string_ == nil);
+  base::AutoReset<NSMutableAttributedString*> resetter(
+      &attributing_display_string_, as);
+  UpdateTextStyle(text, AutocompleteSchemeClassifierImpl());
   return as;
 }
 
 void OmniboxViewIOS::UpdateAppearance() {
   // If Siri is thinking, treat that as user input being in progress.  It is
   // unsafe to modify the text field while voice entry is pending.
   if (model()->UpdatePermanentText()) {
     // Revert everything to the baseline look.
     RevertAll();
   } else if (!model()->has_focus() &&
@@ skipping 164 matching lines @@
 // is being left out for now because it was not present before the OmniboxView
 // rewrite.
 #if 0
   // When editing is in progress, the url text is not colored, so there is
   // nothing to emphasize.  (Calling SetText() in that situation would also be
   // harmful, as it would reset the carat position to the end of the text.)
   if (!IsEditingOrEmpty())
     SetText(GetText());
 #endif
 }