Make extension callbacks with auto-dismissing dialogs

chrome/browser/ui/javascript_dialogs/javascript_dialog_tab_helper.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/javascript_dialogs/javascript_dialog_tab_helper.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/feature_list.h"
 #include "base/metrics/histogram_macros.h"
 #include "chrome/browser/engagement/site_engagement_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/tab_modal_confirm_dialog.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_features.h"
+#include "components/app_modal/javascript_dialog_extensions_client.h"
 #include "components/app_modal/javascript_dialog_manager.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/render_frame_host.h"
 #include "ui/gfx/text_elider.h"
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(JavaScriptDialogTabHelper);
 
 namespace {
 
 bool IsEnabled() {
@@ skipping 136 matching lines @@
                               kMessageTextMaxCols, false,
                               &truncated_message_text);
     base::string16 truncated_default_prompt_text;
     gfx::ElideString(default_prompt_text, kDefaultPromptMaxSize,
                      &truncated_default_prompt_text);
 
     base::string16 title =
         AppModalDialogManager()->GetTitle(alerting_web_contents, origin_url);
     dialog_callback_ = callback;
     dialog_type_ = dialog_type;
+    alerting_web_contents_ = alerting_web_contents;
+    AppModalDialogManager()->GetExtensionsClient()->OnDialogOpened(
+        alerting_web_contents);
     dialog_ = JavaScriptDialog::Create(
         parent_web_contents, alerting_web_contents, title, dialog_type,
         truncated_message_text, truncated_default_prompt_text,
         base::Bind(&JavaScriptDialogTabHelper::OnDialogClosed,
                    base::Unretained(this), callback));
 
     BrowserList::AddObserver(this);
 
     // Message suppression is something that we don't give the user a checkbox
     // for any more. It was useful back in the day when dialogs were app-modal
@@ skipping 139 matching lines @@
       break;
   }
 }
 
 void JavaScriptDialogTabHelper::OnDialogClosed(
     DialogClosedCallback callback,
     bool success,
     const base::string16& user_input) {
   LogDialogDismissalCause(DismissalCause::DIALOG_BUTTON_CLICKED);
   callback.Run(success, user_input);
+  AppModalDialogManager()->GetExtensionsClient()->OnDialogClosed(
+      alerting_web_contents_);
 
   ClearDialogInfo();
 }
 
 void JavaScriptDialogTabHelper::CloseDialog(bool success,
                                             const base::string16& user_input,
                                             DismissalCause cause) {
   DCHECK(dialog_);
   LogDialogDismissalCause(cause);
 
   dialog_->CloseDialogWithoutCallback();
   dialog_callback_.Run(success, user_input);
+  AppModalDialogManager()->GetExtensionsClient()->OnDialogClosed(
+      alerting_web_contents_);
 
   ClearDialogInfo();
 }
 
 void JavaScriptDialogTabHelper::ClearDialogInfo() {
   dialog_.reset();
   dialog_callback_.Reset();
+  alerting_web_contents_ = nullptr;
   BrowserList::RemoveObserver(this);
 }