Implements the dialog view for logout button tray in public sessions.

ash/system/logout_button/logout_button_tray.cc

 // Copyright 2013 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 "ash/system/logout_button/logout_button_tray.h"
 
 #include "ash/shelf/shelf_types.h"
 #include "ash/shell.h"
+#include "ash/system/logout_button/logout_confirmation_dialog_view.h"
 #include "ash/system/status_area_widget.h"
 #include "ash/system/tray/system_tray_delegate.h"
 #include "ash/system/tray/system_tray_notifier.h"
 #include "ash/system/tray/tray_constants.h"
 #include "ash/system/tray/tray_utils.h"
 #include "base/logging.h"
 #include "grit/ash_resources.h"
 #include "third_party/skia/include/core/SkColor.h"
 #include "ui/events/event.h"
 #include "ui/gfx/font.h"
@@ skipping 70 matching lines @@
   set_animate_on_state_change(false);
 
   set_min_size(gfx::Size(0, GetShelfItemHeight()));
 }
 
 LogoutButton::~LogoutButton() {
 }
 
 LogoutButtonTray::LogoutButtonTray(StatusAreaWidget* status_area_widget)
     : TrayBackgroundView(status_area_widget),
-      button_(NULL),
       login_status_(user::LOGGED_IN_NONE),
-      show_logout_button_in_tray_(false) {
+      show_logout_button_in_tray_(false),
+      weak_factory_(this) {
   button_ = new LogoutButton(this);
   tray_container()->AddChildView(button_);
   tray_container()->set_border(NULL);
-  Shell::GetInstance()->system_tray_notifier()->AddLogoutButtonObserver(this);
+  // For testing purpose.
+  if (Shell::HasInstance())
+    Shell::GetInstance()->system_tray_notifier()->AddLogoutButtonObserver(this);
 }
 
 LogoutButtonTray::~LogoutButtonTray() {
-  Shell::GetInstance()->system_tray_notifier()->
-      RemoveLogoutButtonObserver(this);
+  EnsureConfirmationDialogIsClosed();
+  if (Shell::HasInstance())
+    Shell::GetInstance()->system_tray_notifier()->
+        RemoveLogoutButtonObserver(this);
+}
+
+void LogoutButtonTray::EnsureConfirmationDialogIsShowing() {
+  if (!confirmation_dialog_) {
+    confirmation_dialog_.reset(new LogoutConfirmationDialogView(
+          weak_factory_.GetWeakPtr()));
+    confirmation_dialog_->Show(dialog_duration_);
+  }
+}
+
+void LogoutButtonTray::EnsureConfirmationDialogIsClosed() {
+  if (confirmation_dialog_ && Shell::HasInstance())
+    confirmation_dialog_->GetWidget()->Close();
+}
+
+bool LogoutButtonTray::IsConfirmationDialogShowing() {
+  return (bool)confirmation_dialog_;

[bartfab (slow), 2013/12/03 19:46:04]

Nit: We use C++-style casts. In this case though, no explicit casting is needed
at all. You can just do:

return confirmation_dialog_;

[binjin, 2013/12/04 10:47:03]

Done.

 }
 
 void LogoutButtonTray::SetShelfAlignment(ShelfAlignment alignment) {
   TrayBackgroundView::SetShelfAlignment(alignment);
   tray_container()->set_border(NULL);
 }
 
 base::string16 LogoutButtonTray::GetAccessibleNameForTray() {
   return button_->GetText();
 }
 
 void LogoutButtonTray::HideBubbleWithView(
     const views::TrayBubbleView* bubble_view) {
 }
 
 bool LogoutButtonTray::ClickedOutsideBubble() {
   return false;
 }
 
 void LogoutButtonTray::OnShowLogoutButtonInTrayChanged(bool show) {
   show_logout_button_in_tray_ = show;
   UpdateVisibility();
 }
 
+void LogoutButtonTray::OnLogoutDialogDurationChanged(base::TimeDelta duration) {
+  dialog_duration_ = duration;
+  if (confirmation_dialog_)
+    confirmation_dialog_->UpdateDialogDuration(dialog_duration_);
+}
+
 void LogoutButtonTray::ButtonPressed(views::Button* sender,
                                      const ui::Event& event) {
   DCHECK_EQ(sender, button_);
-  Shell::GetInstance()->system_tray_delegate()->SignOut();
+  OnUserLogoutEvent();
 }
 
 void LogoutButtonTray::UpdateAfterLoginStatusChange(
     user::LoginStatus login_status) {
   login_status_ = login_status;
   const base::string16 title =
       GetLocalizedSignOutStringForStatus(login_status, false);
   button_->SetText(title);
   button_->SetAccessibleName(title);
   UpdateVisibility();
 }
 
+void LogoutButtonTray::DeleteConfirmationDialog() {
+  confirmation_dialog_.reset();
+}
+
+void LogoutButtonTray::OnUserLogoutEvent() {
+  // Sign out immediately if kLogoutDialogDurationMs is non-positive.
+  if (dialog_duration_ <= base::TimeDelta::FromSeconds(0))
+    LogoutConfirmationDialogView::GetProvider()->LogoutCurrentUser(NULL);
+  else
+    EnsureConfirmationDialogIsShowing();
+}
+
+LogoutConfirmationDialogView *LogoutButtonTray::GetConfirmationDialog() {
+  return confirmation_dialog_.get();
+}
+
 void LogoutButtonTray::UpdateVisibility() {
   SetVisible(show_logout_button_in_tray_ &&
              login_status_ != user::LOGGED_IN_NONE &&
              login_status_ != user::LOGGED_IN_LOCKED);
+  if (!show_logout_button_in_tray_)
+    EnsureConfirmationDialogIsClosed();
 }
 
 }  // namespace internal
+

[bartfab (slow), 2013/12/03 19:46:04]

Nit: No blank line needed.

[binjin, 2013/12/04 10:47:03]

Done.

 }  // namespace ash