Re-factor Ash Message Center code part 4/4

chrome/browser/ui/views/ash/balloon_collection_impl_ash.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 "chrome/browser/ui/views/ash/balloon_collection_impl_ash.h"
 
 #include "ash/shell.h"
+#include "ash/system/web_notification/web_notification_tray.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/notifications/balloon.h"
 #include "chrome/browser/notifications/desktop_notification_service.h"
 #include "chrome/browser/notifications/desktop_notification_service_factory.h"
 #include "chrome/browser/notifications/notification.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/chrome_pages.h"
 #include "chrome/browser/ui/views/ash/balloon_view_ash.h"
 #include "chrome/browser/ui/views/notifications/balloon_view_host.h"
 #include "chrome/browser/ui/views/notifications/balloon_view_views.h"
 
 BalloonCollectionImplAsh::BalloonCollectionImplAsh() {
-  ash::Shell::GetInstance()->GetWebNotificationTray()->SetDelegate(this);
+  ash::Shell::GetInstance()->GetWebNotificationTray()->message_center()->
+      SetDelegate(this);
 }
 
 BalloonCollectionImplAsh::~BalloonCollectionImplAsh() {
 }
 
 bool BalloonCollectionImplAsh::HasSpace() const {
   return true;  // Overflow is handled by ash::WebNotificationTray.
 }
 
 void BalloonCollectionImplAsh::Add(const Notification& notification,
@@ skipping 103 matching lines @@
   return extension_service->extensions()->GetExtensionOrAppByURL(
       ExtensionURLInfo(origin));
 }
 
 #if defined(OS_CHROMEOS)
 // static
 BalloonCollection* BalloonCollection::Create() {
   return new BalloonCollectionImplAsh();
 }
 #endif