mash: Remove StatusAreaWidget references from system tray classes

ash/common/system/web_notification/web_notification_tray.cc

 // Copyright (c) 2012 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/web_notification/web_notification_tray.h"
+#include "ash/common/system/web_notification/web_notification_tray.h"
 
 #include "ash/common/ash_switches.h"
 #include "ash/common/material_design/material_design_controller.h"
 #include "ash/common/session/session_state_delegate.h"
 #include "ash/common/shelf/shelf_constants.h"
 #include "ash/common/shelf/wm_shelf.h"
 #include "ash/common/shelf/wm_shelf_util.h"
 #include "ash/common/shell_window_ids.h"
 #include "ash/common/system/tray/system_tray_delegate.h"
 #include "ash/common/system/tray/tray_bubble_wrapper.h"
 #include "ash/common/system/tray/tray_constants.h"
 #include "ash/common/system/tray/tray_utils.h"
+#include "ash/common/system/web_notification/ash_popup_alignment_delegate.h"
 #include "ash/common/wm_lookup.h"
 #include "ash/common/wm_root_window_controller.h"
 #include "ash/common/wm_shell.h"
 #include "ash/common/wm_window.h"
-#include "ash/system/status_area_widget.h"
 #include "ash/system/tray/system_tray.h"
-#include "ash/system/web_notification/ash_popup_alignment_delegate.h"
 #include "base/auto_reset.h"
 #include "base/i18n/number_formatting.h"
 #include "base/i18n/rtl.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "grit/ash_strings.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/display/display.h"
 #include "ui/display/screen.h"
 #include "ui/gfx/paint_vector_icon.h"
@@ skipping 161 matching lines @@
 
   bool is_bubble_visible_;
   int unread_count_;
 
   views::ImageView no_unread_icon_;
   views::Label unread_label_;
 
   DISALLOW_COPY_AND_ASSIGN(WebNotificationButton);
 };
 
-WebNotificationTray::WebNotificationTray(StatusAreaWidget* status_area_widget)
-    : TrayBackgroundView(status_area_widget->wm_shelf()),
-      status_area_widget_(status_area_widget),
-      button_(NULL),
+WebNotificationTray::WebNotificationTray(WmShelf* shelf,
+                                         WmWindow* status_area_window,
+                                         SystemTray* system_tray)
+    : TrayBackgroundView(shelf),
+      status_area_window_(status_area_window),
+      system_tray_(system_tray),
+      button_(nullptr),
       show_message_center_on_unlock_(false),
       should_update_tray_content_(false),
       should_block_shelf_auto_hide_(false) {
-  DCHECK(status_area_widget_);
+  DCHECK(shelf);
+  DCHECK(status_area_window_);
+  DCHECK(system_tray_);
   button_ = new WebNotificationButton(this);
   button_->set_triggerable_event_flags(ui::EF_LEFT_MOUSE_BUTTON |
                                        ui::EF_RIGHT_MOUSE_BUTTON);
   tray_container()->AddChildView(button_);
   button_->SetFocusBehavior(FocusBehavior::NEVER);
   SetContentsBackground();
   tray_container()->SetBorder(views::Border::NullBorder());
   message_center_tray_.reset(new message_center::MessageCenterTray(
       this, message_center::MessageCenter::Get()));
-  WmShelf* shelf = WmLookup::Get()
-                       ->GetWindowForWidget(status_area_widget)
-                       ->GetRootWindowController()
-                       ->GetShelf();
   popup_alignment_delegate_.reset(new AshPopupAlignmentDelegate(shelf));
   popup_collection_.reset(new message_center::MessagePopupCollection(
       message_center(), message_center_tray_.get(),
       popup_alignment_delegate_.get()));
-  const display::Display& display = WmLookup::Get()
-                                        ->GetWindowForWidget(status_area_widget)
-                                        ->GetDisplayNearestWindow();
+  const display::Display& display =
+      status_area_window_->GetDisplayNearestWindow();
   popup_alignment_delegate_->StartObserving(display::Screen::GetScreen(),
                                             display);
   OnMessageCenterTrayChanged();
 }
 
 WebNotificationTray::~WebNotificationTray() {
   // Release any child views that might have back pointers before ~View().
   message_center_bubble_.reset();
   popup_alignment_delegate_.reset();
   popup_collection_.reset();
 }
 
 // Public methods.
 
 bool WebNotificationTray::ShowMessageCenterInternal(bool show_settings) {
   if (!ShouldShowMessageCenter())
     return false;
 
   should_block_shelf_auto_hide_ = true;
   message_center::MessageCenterBubble* message_center_bubble =
       new message_center::MessageCenterBubble(message_center(),
                                               message_center_tray_.get(), true);
 
   int max_height;
   if (IsHorizontalAlignment(shelf()->GetAlignment())) {
     max_height = shelf()->GetIdealBounds().y();
   } else {
     // Assume the status area and bubble bottoms are aligned when vertical.
-    WmWindow* status_area_window =
-        WmLookup::Get()->GetWindowForWidget(status_area_widget_);
     gfx::Rect bounds_in_root =
-        status_area_window->GetRootWindow()->ConvertRectFromScreen(
-            status_area_window->GetBoundsInScreen());
+        status_area_window_->GetRootWindow()->ConvertRectFromScreen(
+            status_area_window_->GetBoundsInScreen());
     max_height = bounds_in_root.bottom();
   }
   message_center_bubble->SetMaxHeight(
       std::max(0, max_height - GetTrayConstant(TRAY_SPACING)));
   if (show_settings)
     message_center_bubble->SetSettingsVisible();
   message_center_bubble_.reset(
       new WebNotificationBubbleWrapper(this, message_center_bubble));
 
-  status_area_widget_->SetHideSystemNotifications(true);
+  system_tray_->SetHideNotifications(true);
   shelf()->UpdateAutoHideState();
   button_->SetBubbleVisible(true);
   SetDrawBackgroundAsActive(true);
   return true;
 }
 
 bool WebNotificationTray::ShowMessageCenter() {
   return ShowMessageCenterInternal(false /* show_settings */);
 }
 
 void WebNotificationTray::HideMessageCenter() {
   if (!message_center_bubble())
     return;
   SetDrawBackgroundAsActive(false);
   message_center_bubble_.reset();
   should_block_shelf_auto_hide_ = false;
   show_message_center_on_unlock_ = false;
-  status_area_widget_->SetHideSystemNotifications(false);
+  system_tray_->SetHideNotifications(false);
   shelf()->UpdateAutoHideState();
   button_->SetBubbleVisible(false);
 }
 
-void WebNotificationTray::SetSystemTrayHeight(int height) {
-  popup_alignment_delegate_->SetSystemTrayHeight(height);
+void WebNotificationTray::SetTrayBubbleHeight(int height) {
+  popup_alignment_delegate_->SetTrayBubbleHeight(height);
 }
 
-int WebNotificationTray::system_tray_height_for_test() const {
-  return popup_alignment_delegate_->system_tray_height_for_test();
+int WebNotificationTray::tray_bubble_height_for_test() const {
+  return popup_alignment_delegate_->tray_bubble_height_for_test();
 }
 
 bool WebNotificationTray::ShowPopups() {
   if (message_center_bubble())
     return false;
 
   popup_collection_->DoUpdateIfPossible();
   return true;
 }
 
 void WebNotificationTray::HidePopups() {
   DCHECK(popup_collection_.get());
   popup_collection_->MarkAllPopupsShown();
 }
 
 // Private methods.
 
 bool WebNotificationTray::ShouldShowMessageCenter() {
-  return !(status_area_widget_->system_tray() &&
-           status_area_widget_->system_tray()->HasNotificationBubble());
+  return !system_tray_->HasNotificationBubble();
 }
 
 bool WebNotificationTray::ShouldBlockShelfAutoHide() const {
   return should_block_shelf_auto_hide_;
 }
 
 bool WebNotificationTray::IsMessageCenterBubbleVisible() const {
   return (message_center_bubble() &&
           message_center_bubble()->bubble()->IsVisible());
 }
@@ skipping 160 matching lines @@
   button_->SetUnreadCount(message_center->UnreadNotificationCount());
   if (IsMessageCenterBubbleVisible())
     button_->SetState(views::CustomButton::STATE_PRESSED);
   else
     button_->SetState(views::CustomButton::STATE_NORMAL);
 
   SetVisible(IsLoggedIn());
   Layout();
   SchedulePaint();
   if (IsLoggedIn())
-    status_area_widget_->system_tray()->SetNextFocusableView(this);
+    system_tray_->SetNextFocusableView(this);
 }
 
 void WebNotificationTray::ClickedOutsideBubble() {
   // Only hide the message center
   if (!message_center_bubble())
     return;
 
   message_center_tray_->HideMessageCenterBubble();
 }
 
@@ skipping 17 matching lines @@
 
 message_center::MessageCenterBubble*
 WebNotificationTray::GetMessageCenterBubbleForTest() {
   if (!message_center_bubble())
     return NULL;
   return static_cast<message_center::MessageCenterBubble*>(
       message_center_bubble()->bubble());
 }
 
 }  // namespace ash