Render extension URLs with chips

chrome/browser/ui/location_bar/location_bar.cc

 // Copyright 2014 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/location_bar/location_bar.h"
 
+#include "base/strings/string_util.h"
+#include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/extensions/manifest_handlers/ui_overrides_handler.h"
+#include "content/public/browser/web_contents.h"
 #include "extensions/browser/extension_registry.h"
+#include "extensions/common/constants.h"
 #include "extensions/common/extension_set.h"
 #include "extensions/common/feature_switch.h"
 #include "extensions/common/permissions/permissions_data.h"
 
 const char LocationBar::kSecurityChipFeatureVisibilityParam[] = "visibility";
 const char LocationBar::kSecurityChipFeatureAnimationParam[] = "animation";
 
+namespace {
+const base::char16 kLineBreakChars[] = {'\n',   '\r', '\t',
+                                        0x2028,  // Line separator
+                                        0x2029,  // Paragraph separator
+                                        0};
+}  // namespace
+
 LocationBar::LocationBar(Profile* profile) : profile_(profile) {
 }
 
 LocationBar::~LocationBar() {
 }
 
 bool LocationBar::IsBookmarkStarHiddenByExtension() const {
   const extensions::ExtensionSet& extension_set =
       extensions::ExtensionRegistry::Get(profile_)->enabled_extensions();
   for (extensions::ExtensionSet::const_iterator i = extension_set.begin();
        i != extension_set.end(); ++i) {
     if (extensions::UIOverrides::RemovesBookmarkButton(i->get()) &&
         ((*i)->permissions_data()->HasAPIPermission(
              extensions::APIPermission::kBookmarkManagerPrivate) ||
          extensions::FeatureSwitch::enable_override_bookmarks_ui()
              ->IsEnabled()))
       return true;
   }
 
   return false;
 }
+
+base::string16 LocationBar::GetExtensionName(
+    const GURL& url,
+    content::WebContents* web_contents) const {
+  if (web_contents && url.SchemeIs(extensions::kExtensionScheme)) {

[Peter Kasting, 2016/12/14 22:47:07]

Can there be no web_contents?  Consider adding a comment about when.

[meacer, 2016/12/16 18:46:17]

I don't see an obvious code path that would led to no web_contents, but
LocationBarView code assumes it could be null in several places. Honestly, not
sure what to document here.

[Peter Kasting, 2016/12/20 01:59:31]

IIRC, the places where LocationBarView does this are generally for tests.  If
you are being called from a chain that is doing such a check, you may need one
too.  Otherwise, I would avoid this until you prove you need it.  (You can try
CHECK(web_contents) temporarily to see if the trybots complain?  If you're
really unsure, you can ship that to users too, as long as you change it to a
DCHECK after enough time has past that you're sure you've flushed out crashes.)

[meacer, 2016/12/21 01:38:56]

Done, added a CHECK.

+    content::BrowserContext* browser_context =
+        web_contents->GetBrowserContext();
+    extensions::ExtensionRegistry* extension_registry =
+        extensions::ExtensionRegistry::Get(browser_context);
+    if (extension_registry) {

[Peter Kasting, 2016/12/14 22:47:07]

Can this actually fail?

[meacer, 2016/12/16 18:46:17]

Looks like it can't, removed the check.

+      const extensions::Extension* extension =
+          extension_registry->enabled_extensions().GetByID(url.host());
+      if (extension) {
+        base::string16 extension_name(base::UTF8ToUTF16(extension->name()));
+        base::ReplaceChars(extension_name, kLineBreakChars,
+                           base::ASCIIToUTF16(" "), &extension_name);

[Peter Kasting, 2016/12/14 22:47:07]

Why do you need this?  It seems like CollapseWhitespace() will do this for you.

[meacer, 2016/12/16 18:46:17]

This is the rebase of an old CL, and I seem to remember a case which
CollapseWhitespace didn't handle. I can neither recall what it was nor reproduce
it now, so removed.

+        return base::CollapseWhitespace(extension_name, false);
+      }
+    }
+  }
+  return base::string16();
+}