Extract shell methods from ApplicationImpl into a base class, and pass this to Initialize() instead.

mash/screenlock/screenlock.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 "mash/screenlock/screenlock.h"
 
 #include "base/macros.h"
 #include "base/strings/utf_string_conversions.h"
 #include "components/mus/public/cpp/property_type_converters.h"
 #include "mash/shell/public/interfaces/shell.mojom.h"
 #include "mash/wm/public/interfaces/container.mojom.h"
 #include "mojo/public/cpp/bindings/binding.h"
-#include "mojo/shell/public/cpp/application_impl.h"
+#include "mojo/shell/public/cpp/shell.h"
 #include "ui/views/background.h"
 #include "ui/views/controls/button/label_button.h"
 #include "ui/views/mus/aura_init.h"
 #include "ui/views/mus/native_widget_mus.h"
 #include "ui/views/mus/window_manager_connection.h"
 #include "ui/views/widget/widget_delegate.h"
 
 namespace mash {
 namespace screenlock {
 namespace {
 
 class ScreenlockView : public views::WidgetDelegateView,
                        public views::ButtonListener {
  public:
-  explicit ScreenlockView(mojo::ApplicationImpl* app)
-      : app_(app),
+  explicit ScreenlockView(mojo::Shell* shell)
+      : shell_(shell),
         unlock_button_(
             new views::LabelButton(this, base::ASCIIToUTF16("Unlock"))) {
     set_background(views::Background::CreateSolidBackground(SK_ColorYELLOW));
     unlock_button_->SetStyle(views::Button::STYLE_BUTTON);
     AddChildView(unlock_button_);
   }
   ~ScreenlockView() override {}
 
  private:
   // Overridden from views::WidgetDelegate:
@@ skipping 13 matching lines @@
 
     unlock_button_->SetBounds(bounds.width() - ps.width(),
                               bounds.bottom() + 10,
                               ps.width(), ps.height());
   }
 
   // Overridden from views::ButtonListener:
   void ButtonPressed(views::Button* sender, const ui::Event& event) override {
     DCHECK_EQ(sender, unlock_button_);
     mash::shell::mojom::ShellPtr shell;
-    app_->ConnectToService("mojo:mash_shell", &shell);
+    shell_->ConnectToService("mojo:mash_shell", &shell);
     shell->UnlockScreen();
   }
 
-  mojo::ApplicationImpl* app_;
+  mojo::Shell* shell_;
   views::LabelButton* unlock_button_;
 
   DISALLOW_COPY_AND_ASSIGN(ScreenlockView);
 };
 
 }  // namespace
 
-Screenlock::Screenlock() : app_(nullptr) {}
+Screenlock::Screenlock() : shell_(nullptr) {}
 Screenlock::~Screenlock() {}
 
-void Screenlock::Initialize(mojo::ApplicationImpl* app) {
-  app_ = app;
-  tracing_.Initialize(app);
+void Screenlock::Initialize(mojo::Shell* shell, const std::string& url,
+                            uint32_t id) {
+  shell_ = shell;
+  tracing_.Initialize(shell, url);
 
-  mash::shell::mojom::ShellPtr shell;
-  app_->ConnectToService("mojo:mash_shell", &shell);
-  shell->AddScreenlockStateListener(bindings_.CreateInterfacePtrAndBind(this));
+  mash::shell::mojom::ShellPtr mash_shell;
+  shell_->ConnectToService("mojo:mash_shell", &mash_shell);
+  mash_shell->AddScreenlockStateListener(
+      bindings_.CreateInterfacePtrAndBind(this));
 
-  aura_init_.reset(new views::AuraInit(app, "views_mus_resources.pak"));
-  views::WindowManagerConnection::Create(app);
+  aura_init_.reset(new views::AuraInit(shell, "views_mus_resources.pak"));
+  views::WindowManagerConnection::Create(shell);
 
   views::Widget* widget = new views::Widget;
   views::Widget::InitParams params(
       views::Widget::InitParams::TYPE_WINDOW_FRAMELESS);
-  params.delegate = new ScreenlockView(app);
+  params.delegate = new ScreenlockView(shell);
 
   std::map<std::string, std::vector<uint8_t>> properties;
   properties[mash::wm::mojom::kWindowContainer_Property] =
       mojo::TypeConverter<const std::vector<uint8_t>, int32_t>::Convert(
           static_cast<int32_t>(mash::wm::mojom::Container::LOGIN_WINDOWS));
   mus::Window* window =
       views::WindowManagerConnection::Get()->NewWindow(properties);
   params.native_widget = new views::NativeWidgetMus(
-      widget, app->shell(), window, mus::mojom::SurfaceType::DEFAULT);
+      widget, shell, window, mus::mojom::SurfaceType::DEFAULT);
   widget->Init(params);
   widget->Show();
 }
 
 void Screenlock::ScreenlockStateChanged(bool screen_locked) {
   if (!screen_locked)
-    app_->Quit();
+    shell_->Quit();
 }
 
 }  // namespace screenlock
 }  // namespace main