Change ChooserController ownership model

chrome/browser/ui/views/extensions/chooser_dialog_view.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/ui/views/extensions/chooser_dialog_view.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/extensions/chrome_extension_chooser_dialog.h"
 #include "chrome/browser/ui/views/chooser_content_view.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/chooser_controller/chooser_controller.h"
 #include "components/constrained_window/constrained_window_views.h"
 #include "components/url_formatter/elide_url.h"
 #include "components/web_modal/web_contents_modal_dialog_manager.h"
 #include "content/public/browser/browser_thread.h"
 #include "extensions/browser/extension_registry.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/views/controls/styled_label.h"
 #include "ui/views/window/dialog_client_view.h"
-#include "url/origin.h"
 
-ChooserDialogView::ChooserDialogView(content::WebContents* web_contents,
-                                     ChooserController* chooser_controller)
-    : chooser_content_view_(nullptr),
-      web_contents_(web_contents),
-      chooser_controller_(chooser_controller) {
+ChooserDialogView::ChooserDialogView(
+    content::WebContents* web_contents,
+    std::unique_ptr<ChooserController> chooser_controller)
+    : web_contents_(web_contents), chooser_content_view_(nullptr) {
   // ------------------------------------
   // | Chooser dialog title             |
   // | -------------------------------- |
   // | | option 0                     | |
   // | | option 1                     | |
   // | | option 2                     | |
   // | |                              | |
   // | |                              | |
   // | |                              | |
   // | -------------------------------- |
   // |           [ Connect ] [ Cancel ] |
   // |----------------------------------|
   // | Not seeing your device? Get help |
   // ------------------------------------
 
   DCHECK(web_contents_);
-
-  chooser_content_view_ = new ChooserContentView(this, chooser_controller_);
+  DCHECK(chooser_controller);
+  origin_ = chooser_controller->GetOrigin();
+  chooser_content_view_ =
+      new ChooserContentView(this, std::move(chooser_controller));
 }
 
 ChooserDialogView::~ChooserDialogView() {}
 
 base::string16 ChooserDialogView::GetWindowTitle() const {
   base::string16 chooser_title;
-  url::Origin origin = chooser_controller_->GetOrigin();
   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()));
+            GURL(origin_.Serialize()));
     if (extension)
       chooser_title = base::UTF8ToUTF16(extension->name());
   }
 
   if (chooser_title.empty()) {
     chooser_title = url_formatter::FormatOriginForSecurityDisplay(
-        origin, url_formatter::SchemeDisplay::OMIT_CRYPTOGRAPHIC);
+        origin_, url_formatter::SchemeDisplay::OMIT_CRYPTOGRAPHIC);
   }
 
   return l10n_util::GetStringFUTF16(IDS_DEVICE_CHOOSER_PROMPT, chooser_title);
 }
 
 bool ChooserDialogView::ShouldShowCloseButton() const {
   return false;
 }
 
 ui::ModalType ChooserDialogView::GetModalType() const {
@@ skipping 48 matching lines @@
 
 void ChooserDialogView::OnSelectionChanged() {
   GetDialogClientView()->UpdateDialogButtons();
 }
 
 views::TableView* ChooserDialogView::table_view_for_test() const {
   return chooser_content_view_->table_view_for_test();
 }
 
 void ChromeExtensionChooserDialog::ShowDialogImpl(
-    ChooserController* chooser_controller) const {
+    std::unique_ptr<ChooserController> chooser_controller) const {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   DCHECK(chooser_controller);
 
   web_modal::WebContentsModalDialogManager* manager =
       web_modal::WebContentsModalDialogManager::FromWebContents(web_contents_);
   if (manager) {
     constrained_window::ShowWebModalDialogViews(
-        new ChooserDialogView(web_contents_, chooser_controller),
+        new ChooserDialogView(web_contents_, std::move(chooser_controller)),
         web_contents_);
   }
 }