chrome/browser/ui/blocked_content/app_modal_dialog_helper.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/ui/blocked_content/app_modal_dialog_helper.h"
   6
   7 #include "chrome/browser/ui/browser.h"
   8 #include "chrome/browser/ui/browser_finder.h"
   9 #include "chrome/browser/ui/browser_list.h"
  10 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  11 #include "content/public/browser/web_contents.h"
  12 #include "content/public/browser/web_contents_delegate.h"
  13
  14 AppModalDialogHelper::AppModalDialogHelper(content::WebContents* dialog_host)
  15     : popup_(nullptr) {
  16   // If the WebContents that triggered this dialog is not currently focused, we
  17   // want to store a potential popup here to restore it after the dialog was
  18   // closed.
  19   chrome::HostDesktopType desktop_type =
  20       chrome::GetHostDesktopTypeForNativeView(dialog_host->GetNativeView());
  21   Browser* active_browser =
  22       BrowserList::GetInstance(desktop_type)->GetLastActive();
  23   if (active_browser) {
  24     content::WebContents* active_web_contents =
  25         active_browser->tab_strip_model()->GetActiveWebContents();
  26     if (active_browser->is_type_popup() && active_web_contents &&
  27         active_web_contents->GetOpener() == dialog_host) {
  28       // It's indeed a popup from the dialog opening WebContents. Store it, so
  29       // we can focus it later.
  30       popup_ = active_web_contents;
  31       Observe(popup_);
  32     }
  33   }
  34 }
  35
  36 AppModalDialogHelper::~AppModalDialogHelper() {
  37   if (popup_)
  38     popup_->GetDelegate()->ActivateContents(popup_);
  39 }
  40
  41 void AppModalDialogHelper::WebContentsDestroyed() {
  42   popup_ = nullptr;
  43 }