Make Service own ServiceContext.

mash/session/session.cc

 // Copyright 2015 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/session/session.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "mash/login/public/interfaces/login.mojom.h"
 #include "services/shell/public/cpp/connection.h"
 #include "services/shell/public/cpp/connector.h"
 
 namespace {
 
 void LogAndCallServiceRestartCallback(const std::string& url,
                                       const base::Closure& callback) {
   LOG(ERROR) << "Restarting service: " << url;
   callback.Run();
 }
 
 }  // namespace
 
 namespace mash {
 namespace session {
 
-Session::Session() : connector_(nullptr), screen_locked_(false) {}
+Session::Session() : screen_locked_(false) {}
 Session::~Session() {}
 
-void Session::OnStart(shell::Connector* connector,
-                      const shell::Identity& identity,
-                      uint32_t id) {
-  connector_ = connector;
+void Session::OnStart(const shell::Identity& identity) {
   StartAppDriver();
   StartWindowManager();
   StartSystemUI();
   StartQuickLaunch();
 }
 
 bool Session::OnConnect(shell::Connection* connection) {
   connection->AddInterface<mojom::Session>(this);
   return true;
 }
 
 void Session::Logout() {
   // TODO(beng): Notify connected listeners that login is happening, potentially
   // give them the option to stop it.
   mash::login::mojom::LoginPtr login;
-  connector_->ConnectToInterface("mojo:login", &login);
+  connector()->ConnectToInterface("mojo:login", &login);
   login->ShowLoginUI();
   // This kills the user environment.
   base::MessageLoop::current()->QuitWhenIdle();
 }
 
 void Session::SwitchUser() {
   mash::login::mojom::LoginPtr login;
-  connector_->ConnectToInterface("mojo:login", &login);
+  connector()->ConnectToInterface("mojo:login", &login);
   login->SwitchUser();
 }
 
 void Session::AddScreenlockStateListener(
     mojom::ScreenlockStateListenerPtr listener) {
   listener->ScreenlockStateChanged(screen_locked_);
   screenlock_listeners_.AddPtr(std::move(listener));
 }
 
 void Session::LockScreen() {
@@ skipping 59 matching lines @@
   auto connection = connections_.find("mojo:screenlock");
   DCHECK(connections_.end() != connection);
   connections_.erase(connection);
 }
 
 void Session::StartRestartableService(
     const std::string& url,
     const base::Closure& restart_callback) {
   // TODO(beng): This would be the place to insert logic that counted restarts
   //             to avoid infinite crash-restart loops.
-  std::unique_ptr<shell::Connection> connection = connector_->Connect(url);
+  std::unique_ptr<shell::Connection> connection =
+      connector()->Connect(url);
   // Note: |connection| may be null if we've lost our connection to the shell.
   if (connection) {
     connection->SetConnectionLostClosure(
         base::Bind(&LogAndCallServiceRestartCallback, url, restart_callback));
     connection->AddInterface<mojom::Session>(this);
     connections_[url] = std::move(connection);
   }
 }
 
 }  // namespace session
 }  // namespace main