cros: Use SessionController for lock starting code

ash/session/session_controller.h

 // 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.
 
 #ifndef ASH_SESSION_SESSION_CONTROLLER_H_
 #define ASH_SESSION_SESSION_CONTROLLER_H_
 
 #include <stdint.h>
 
 #include <vector>
 
 #include "ash/ash_export.h"
 #include "ash/login_status.h"
 #include "ash/public/cpp/session_types.h"
 #include "ash/public/interfaces/session_controller.mojom.h"
+#include "base/callback.h"
 #include "base/macros.h"
+#include "base/memory/weak_ptr.h"
 #include "base/observer_list.h"
 #include "mojo/public/cpp/bindings/binding_set.h"
 
 class AccountId;
 
 namespace ash {
 
 class SessionObserver;
 
 // Implements mojom::SessionController to cache session related info such as
@@ skipping 76 matching lines @@
   // NOTE: Prefer GetSessionState() in new code because the concept of
   // SessionState more closes matches the state in chrome.
   LoginStatus login_status() const { return login_status_; }
 
   // mojom::SessionController
   void SetClient(mojom::SessionControllerClientPtr client) override;
   void SetSessionInfo(mojom::SessionInfoPtr info) override;
   void UpdateUserSession(mojom::UserSessionPtr user_session) override;
   void SetUserSessionOrder(
       const std::vector<uint32_t>& user_session_order) override;
+  void StartLock(const StartLockCallback& callback) override;
   void RunUnlockAnimation(const RunUnlockAnimationCallback& callback) override;
   void NotifyChromeTerminating() override;
 
   // Test helpers.
   void ClearUserSessionsForTest();
   void FlushMojoForTest();
   void LockScreenAndFlushForTest();
 
  private:
   void SetSessionState(session_manager::SessionState state);
   void AddUserSession(mojom::UserSessionPtr user_session);
 
   // Calculate login status based on session state and active user session.
   LoginStatus CalculateLoginStatus() const;
 
   // Helper that returns login status when the session state is ACTIVE.
   LoginStatus CalculateLoginStatusForActiveSession() const;
 
   // Update the |login_status_| and notify observers.
   void UpdateLoginStatus();
 
+  // Used as lock screen displayed callback of LockStateController and invoked
+  // when post lock animation finishes and ash is fully locked. It would then
+  // run |start_lock_callback_| to indicate ash is locked successfully.
+  void OnLockAnimationFinished();
+
   // Bindings for mojom::SessionController interface.
   mojo::BindingSet<mojom::SessionController> bindings_;
 
   // Client interface to session manager code (chrome).
   mojom::SessionControllerClientPtr client_;
 
   // Cached session info.
   bool can_lock_ = false;
   bool should_lock_screen_automatically_ = false;
   AddUserSessionPolicy add_user_session_policy_ = AddUserSessionPolicy::ALLOWED;
   session_manager::SessionState state_;
 
   // Cached user session info sorted by the order from SetUserSessionOrder.
   // Currently the session manager code (chrome) sets a LRU order with the
   // active session being the first.
   std::vector<mojom::UserSessionPtr> user_sessions_;
 
   // The user session id of the current active user session. User session id
   // is managed by session manager code, starting at 1. 0u is an invalid id
   // to detect first active user session.
   uint32_t active_session_id_ = 0u;
 
   // Last known login status. Used to track login status changes.
   LoginStatus login_status_ = LoginStatus::NOT_LOGGED_IN;
 
   // Whether unlocking is in progress. The flag is set when the pre-unlock
   // animation starts and reset when session state is no longer LOCKED.
   bool is_unlocking_ = false;
 
+  // Pending callback for the StartLock request.
+  base::OnceCallback<void(bool)> start_lock_callback_;
+
   base::ObserverList<ash::SessionObserver> observers_;
 
+  base::WeakPtrFactory<SessionController> weak_ptr_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(SessionController);
 };
 
 }  // namespace ash
 
 #endif  // ASH_SESSION_SESSION_CONTROLLER_H_