Display extension name on device chooser title

chrome/browser/chooser_controller/chooser_controller.cc

 // Copyright 2016 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/chooser_controller/chooser_controller.h"
 
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/grit/generated_resources.h"
+#include "components/url_formatter/elide_url.h"
 #include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents.h"
+#include "extensions/browser/extension_registry.h"
+#include "extensions/common/constants.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "url/gurl.h"
 #include "url/origin.h"
 
 ChooserController::ChooserController(content::RenderFrameHost* owner)
     : owning_frame_(owner) {}
 
 ChooserController::~ChooserController() {}
 
-url::Origin ChooserController::GetOrigin() const {
-  return owning_frame_ ? owning_frame_->GetLastCommittedOrigin()
-                       : url::Origin();
+base::string16 ChooserController::GetTitle() const {
+  if (!owning_frame_)
+    return base::string16();
+
+  base::string16 title_prefix;
+  url::Origin origin = owning_frame_->GetLastCommittedOrigin();
+  if (origin.scheme() == extensions::kExtensionScheme) {
+    content::WebContents* web_contents =
+        content::WebContents::FromRenderFrameHost(owning_frame_);
+    content::BrowserContext* browser_context =
+        web_contents->GetBrowserContext();
+    extensions::ExtensionRegistry* extension_registry =
+        extensions::ExtensionRegistry::Get(browser_context);
+    if (extension_registry) {
+      const extensions::Extension* extension =
+          extension_registry->enabled_extensions().GetExtensionOrAppByURL(
+              GURL(origin.Serialize()));

[Reilly Grant (use Gerrit), 2016/07/12 00:12:05]

You can use extension_registry->enabled_extensions().GetByID(origin.host()).

[juncai, 2016/07/12 01:14:43]

Done.

+      if (extension)
+        title_prefix = base::UTF8ToUTF16(extension->name());

[Reilly Grant (use Gerrit), 2016/07/12 00:12:05]

No need to set title_prefix here, just return
l10n_util::GetStringFUTF16(IDS_DEVICE_CHOOSER_PROMPT_EXTENSION_NAME,
base::UTF8ToUTF16(extension->name()).

[juncai, 2016/07/12 01:14:43]

Done.

+    }
+  }
+
+  if (title_prefix.empty()) {
+    title_prefix = url_formatter::FormatOriginForSecurityDisplay(
+        origin, url_formatter::SchemeDisplay::OMIT_CRYPTOGRAPHIC);
+    return l10n_util::GetStringFUTF16(IDS_DEVICE_CHOOSER_PROMPT_ORIGIN,
+                                      title_prefix);
+  } else {
+    return l10n_util::GetStringFUTF16(IDS_DEVICE_CHOOSER_PROMPT_EXTENSION_NAME,
+                                      title_prefix);
+  }
 }