ash: Add SessionController/Client mojo interfaces

chrome/browser/ui/ash/session_controller_client.cc

Index: chrome/browser/ui/ash/session_controller_client.cc
diff --git a/chrome/browser/ui/ash/session_controller_client.cc b/chrome/browser/ui/ash/session_controller_client.cc
new file mode 100644
index 0000000000000000000000000000000000000000..12ffd00cc3430be532db247e1961a171a553dbf6
--- /dev/null
+++ b/chrome/browser/ui/ash/session_controller_client.cc
@@ -0,0 +1,278 @@
+// 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 "chrome/browser/ui/ash/session_controller_client.h"
+
+#include <algorithm>
+#include <utility>
+
+#include "ash/public/cpp/session_types.h"
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/chromeos/login/users/multi_profile_user_controller.h"
+#include "chrome/browser/chromeos/profiles/profile_helper.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/ash/ash_util.h"
+#include "chrome/browser/ui/ash/multi_user/user_switch_util.h"
+#include "chrome/common/pref_names.h"
+#include "chrome/grit/theme_resources.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
+#include "chromeos/dbus/session_manager_client.h"
+#include "components/prefs/pref_service.h"
+#include "components/session_manager/core/session_manager.h"
+#include "content/public/common/service_manager_connection.h"
+#include "content/public/common/service_names.mojom.h"
+#include "services/service_manager/public/cpp/connector.h"
+#include "ui/base/resource/resource_bundle.h"
+
+using user_manager::UserManager;
+using user_manager::User;
+using user_manager::UserList;
+
+namespace {
+
+// Limits the number of logged in users to 10 due to memory constraints.
+constexpr uint32_t kMaxUsers = 10;
+
+uint32_t GetSessionId(const User* user) {
+  const UserList logged_in_users = UserManager::Get()->GetLoggedInUsers();

[James Cook, 2016/12/06 18:08:25]

Just to double-check: This order in this list doesn't change, so the IDs you
return are stable?

[xiyuan, 2016/12/06 19:24:21]

Correct.

GetLoggedInUsers() reflects the order of user logins. UserManager should only
append to the list. It would be a bug if UserManager change the list order. I
will be moving the user session tracking code from UserManager to
SessionManager. After that, the user session id would be explicitly generated.

+  // TODO(xiyuan): Update with real session id when user session tracking
+  //     code is moved from UserManager to SessionManager.
+  for (size_t i = 0; i < logged_in_users.size(); ++i) {
+    if (logged_in_users[i] == user)
+      return i + 1;
+  }
+
+  NOTREACHED();
+  return 0u;
+}
+
+ash::mojom::UserSessionPtr UserToUserSession(const User& user) {
+  ash::mojom::UserSessionPtr session = ash::mojom::UserSession::New();
+  session->session_id = GetSessionId(&user);
+  session->type = user.GetType();
+  // TODO(xiyuan): Add type map for AccountId.
+  session->account_id = user.GetAccountId().Serialize();
+  session->display_name = base::UTF16ToUTF8(user.display_name());
+  session->display_email = user.display_email();
+
+  // TODO(xiyuan): Observe user image change and update.
+  //     Tracked in http://crbug.com/670422
+  // TODO(xiyuan): Support multiple scale factor.
+  session->avatar = *user.GetImage().bitmap();
+  if (session->avatar.isNull()) {
+    session->avatar = *ResourceBundle::GetSharedInstance()
+                           .GetImageSkiaNamed(IDR_PROFILE_PICTURE_LOADING)
+                           ->bitmap();
+  }
+
+  return session;
+}
+
+void DoSwitchUser(const AccountId& account_id) {
+  UserManager::Get()->SwitchActiveUser(account_id);
+}
+
+}  // namespace
+
+SessionControllerClient::SessionControllerClient() : binding_(this) {
+  session_manager::SessionManager::Get()->AddObserver(this);
+  UserManager::Get()->AddSessionStateObserver(this);
+
+  ConnectToSessionControllerAndSetClient();
+  SendSessionInfoIfChanged();
+  // User sessions and their order will be sent via UserSessionStateObserver
+  // even for crash-n-restart.
+}
+
+SessionControllerClient::~SessionControllerClient() {
+  session_manager::SessionManager::Get()->RemoveObserver(this);
+  UserManager::Get()->RemoveSessionStateObserver(this);
+}
+
+void SessionControllerClient::RequestLockScreen() {
+  DoLockScreen();
+}
+
+void SessionControllerClient::SwitchActiveUser(
+    const std::string& serialized_account_id) {
+  // TODO(xiyuan): Add type map for AccountId.
+  AccountId account_id(EmptyAccountId());
+  if (!AccountId::Deserialize(serialized_account_id, &account_id)) {
+    LOG(ERROR) << "Bad account id for SwitchActiveUser.";
+    return;
+  }
+
+  DoSwitchActiveUser(account_id);
+}
+
+void SessionControllerClient::CycleActiveUser(bool next_user) {
+  DoCycleActiveUser(next_user);
+}
+
+void SessionControllerClient::ActiveUserChanged(const User* active_user) {
+  SendSessionInfoIfChanged();
+
+  // UserAddedToSession is not called for the primary user session so send its
+  // meta data here once.
+  if (!primary_user_session_sent_ &&
+      UserManager::Get()->GetPrimaryUser() == active_user) {
+    primary_user_session_sent_ = true;
+    SendUserSession(*active_user);
+  }
+
+  SendUserSessionOrder();
+}
+
+void SessionControllerClient::UserAddedToSession(const User* added_user) {
+  SendSessionInfoIfChanged();
+  SendUserSession(*added_user);
+}
+
+// static
+bool SessionControllerClient::CanLockScreen() {
+  return !UserManager::Get()->GetUnlockUsers().empty();
+}
+
+// static
+bool SessionControllerClient::ShouldLockScreenAutomatically() {
+  // TODO(xiyuan): Observe prefs::kEnableAutoScreenLock and update ash.
+  // Tracked in http://crbug.com/670423
+  const UserList logged_in_users = UserManager::Get()->GetLoggedInUsers();
+  for (auto* user : logged_in_users) {
+    Profile* profile = chromeos::ProfileHelper::Get()->GetProfileByUser(user);
+    if (profile &&
+        profile->GetPrefs()->GetBoolean(prefs::kEnableAutoScreenLock)) {
+      return true;
+    }
+  }
+  return false;
+}
+
+// static
+ash::AddUserSessionPolicy SessionControllerClient::GetAddUserSessionPolicy() {
+  UserManager* const user_manager = UserManager::Get();
+  if (user_manager->GetUsersAllowedForMultiProfile().empty())
+    return ash::AddUserSessionPolicy::ERROR_NO_ELIGIBLE_USERS;
+
+  if (chromeos::MultiProfileUserController::GetPrimaryUserPolicy() !=
+      chromeos::MultiProfileUserController::ALLOWED) {
+    return ash::AddUserSessionPolicy::ERROR_NOT_ALLOWED_PRIMARY_USER;
+  }
+
+  if (UserManager::Get()->GetLoggedInUsers().size() >= kMaxUsers)
+    return ash::AddUserSessionPolicy::ERROR_MAXIMUM_USERS_REACHED;
+
+  return ash::AddUserSessionPolicy::ALLOWED;
+}
+
+// static
+void SessionControllerClient::DoLockScreen() {
+  if (!CanLockScreen())
+    return;
+
+  VLOG(1) << "Requesting screen lock from SessionControllerClient";
+  chromeos::DBusThreadManager::Get()
+      ->GetSessionManagerClient()
+      ->RequestLockScreen();
+}
+
+// static
+void SessionControllerClient::DoSwitchActiveUser(const AccountId& account_id) {
+  // Disallow switching to an already active user since that might crash.
+  // Also check that we got a user id and not an email address.
+  DCHECK_EQ(
+      account_id.GetUserEmail(),
+      gaia::CanonicalizeEmail(gaia::SanitizeEmail(account_id.GetUserEmail())));
+  if (account_id == UserManager::Get()->GetActiveUser()->GetAccountId())
+    return;
+
+  TrySwitchingActiveUser(base::Bind(&DoSwitchUser, account_id));
+}
+
+// static
+void SessionControllerClient::DoCycleActiveUser(bool next_user) {
+  const UserList& logged_in_users = UserManager::Get()->GetLoggedInUsers();
+  if (logged_in_users.size() <= 1)
+    return;
+
+  AccountId account_id = UserManager::Get()->GetActiveUser()->GetAccountId();
+
+  // Get an iterator positioned at the active user.
+  auto it = std::find_if(logged_in_users.begin(), logged_in_users.end(),
+                         [account_id](const User* user) {
+                           return user->GetAccountId() == account_id;
+                         });
+
+  // Active user not found.
+  if (it == logged_in_users.end())
+    return;
+
+  // Get the user's email to select, wrapping to the start/end of the list if
+  // necessary.
+  if (next_user) {
+    if (++it == logged_in_users.end())
+      account_id = (*logged_in_users.begin())->GetAccountId();
+    else
+      account_id = (*it)->GetAccountId();
+  } else {
+    if (it == logged_in_users.begin())
+      it = logged_in_users.end();
+    account_id = (*(--it))->GetAccountId();
+  }
+
+  DoSwitchActiveUser(account_id);
+}
+
+void SessionControllerClient::OnSessionStateChanged() {
+  SendSessionInfoIfChanged();
+}
+
+void SessionControllerClient::ConnectToSessionControllerAndSetClient() {
+  service_manager::Connector* connector =
+      content::ServiceManagerConnection::GetForProcess()->GetConnector();
+
+  // Under mash the SessionController interface is in the ash process. In
+  // classic ash we provide it to ourself.
+  if (chrome::IsRunningInMash()) {
+    connector->ConnectToInterface("ash", &session_controller_);
+  } else {
+    connector->ConnectToInterface(content::mojom::kBrowserServiceName,
+                                  &session_controller_);
+  }
+
+  // Set as |session_controller_|'s client.
+  session_controller_->SetClient(binding_.CreateInterfacePtrAndBind());
+}
+
+void SessionControllerClient::SendSessionInfoIfChanged() {
+  ash::mojom::SessionInfoPtr info = ash::mojom::SessionInfo::New();
+  info->max_users = kMaxUsers;
+  info->can_lock_screen = CanLockScreen();
+  info->should_lock_screen_automatically = ShouldLockScreenAutomatically();
+  info->add_user_session_policy = GetAddUserSessionPolicy();
+  info->state = session_manager::SessionManager::Get()->session_state();
+
+  if (info != last_sent_session_info_) {
+    last_sent_session_info_ = info->Clone();
+    session_controller_->SetSessionInfo(std::move(info));
+  }
+}
+
+void SessionControllerClient::SendUserSession(const User& user) {
+  session_controller_->UpdateUserSession(UserToUserSession(user));
+}
+
+void SessionControllerClient::SendUserSessionOrder() {
+  UserManager* const user_manager = UserManager::Get();
+
+  const UserList logged_in_users = user_manager->GetLoggedInUsers();
+  std::vector<uint32_t> order;
+  for (auto* user : user_manager->GetLRULoggedInUsers())
+    order.push_back(GetSessionId(user));
+
+  ConnectToSessionControllerAndSetClient();

[James Cook, 2016/12/06 18:08:25]

Do you need this? You're not calling it in the other Send* methods.

[xiyuan, 2016/12/06 19:24:21]

Nope. Left overs from refactoring. Removed now.

+  session_controller_->SetUserSessionOrder(order);
+}