Add LoginState class to src/chromeos/login

chrome/browser/chromeos/login/user_manager_impl.cc

 // Copyright (c) 2013 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/chromeos/login/user_manager_impl.h"
 
 #include <cstddef>
 #include <set>
 #include <vector>
 
@@ skipping 25 matching lines @@
 #include "chrome/browser/policy/browser_policy_connector.h"
 #include "chrome/browser/prefs/scoped_user_pref_update.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/sync/profile_sync_service_factory.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chromeos/chromeos_switches.h"
 #include "chromeos/cryptohome/async_method_caller.h"
+#include "chromeos/login/login_state.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
 #include "google_apis/gaia/gaia_auth_util.h"
 #include "google_apis/gaia/google_service_auth_error.h"
 #include "policy/policy_constants.h"
 
 using content::BrowserThread;
 
 namespace chromeos {
 
@@ skipping 143 matching lines @@
       merge_session_state_(MERGE_STATUS_NOT_STARTED),
       observed_sync_service_(NULL),
       user_image_manager_(new UserImageManagerImpl) {
   // UserManager instance should be used only on UI thread.
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   registrar_.Add(this, chrome::NOTIFICATION_OWNERSHIP_STATUS_CHANGED,
       content::NotificationService::AllSources());
   registrar_.Add(this, chrome::NOTIFICATION_PROFILE_ADDED,
       content::NotificationService::AllSources());
   RetrieveTrustedDevicePolicies();
+  UpdateLoginState();
 }
 
 UserManagerImpl::~UserManagerImpl() {
   // Can't use STLDeleteElements because of the private destructor of User.
   for (UserList::iterator it = users_.begin(); it != users_.end();
        it = users_.erase(it)) {
     if (logged_in_user_ == *it)
       logged_in_user_ = NULL;
     delete *it;
   }
@@ skipping 187 matching lines @@
   is_current_user_new_ = true;
   is_current_user_ephemeral_regular_user_ = true;
   logged_in_user_ = User::CreateRegularUser(email);
   user_image_manager_->UserLoggedIn(email, is_current_user_new_, false);
   WallpaperManager::Get()->SetInitialUserWallpaper(email, false);
 }
 
 void UserManagerImpl::SessionStarted() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   session_started_ = true;
+  UpdateLoginState();
   content::NotificationService::current()->Notify(
       chrome::NOTIFICATION_SESSION_STARTED,
       content::NotificationService::AllSources(),
       content::NotificationService::NoDetails());
   if (is_current_user_new_) {
     // Make sure that the new user's data is persisted to Local State.
     g_browser_process->local_state()->CommitPendingWrite();
   }
 }
 
@@ skipping 281 matching lines @@
 bool UserManagerImpl::IsCurrentUserOwner() const {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   base::AutoLock lk(is_current_user_owner_lock_);
   return is_current_user_owner_;
 }
 
 void UserManagerImpl::SetCurrentUserIsOwner(bool is_current_user_owner) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   base::AutoLock lk(is_current_user_owner_lock_);
   is_current_user_owner_ = is_current_user_owner;
+  UpdateLoginState();
 }
 
 bool UserManagerImpl::IsCurrentUserNew() const {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   return is_current_user_new_;
 }
 
 bool UserManagerImpl::IsCurrentUserNonCryptohomeDataEphemeral() const {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   return IsUserLoggedIn() &&
@@ skipping 253 matching lines @@
   const UserList& users = GetUsers();
   for (UserList::const_iterator it = users.begin(); it != users.end(); ++it) {
     if ((*it)->email() == email)
       return *it;
   }
   return NULL;
 }
 
 void UserManagerImpl::NotifyOnLogin() {
   CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  UpdateLoginState();
   content::NotificationService::current()->Notify(
       chrome::NOTIFICATION_LOGIN_USER_CHANGED,
       content::Source<UserManager>(this),
       content::Details<const User>(logged_in_user_));
 
   CrosLibrary::Get()->GetCertLibrary()->LoadKeyStore();
 
   // Indicate to DeviceSettingsService that the owner key may have become
   // available.
   DeviceSettingsService::Get()->SetUsername(logged_in_user_->email());
@@ skipping 293 matching lines @@
       content::Source<UserManager>(this),
       content::NotificationService::NoDetails());
 }
 
 void UserManagerImpl::NotifyMergeSessionStateChanged() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   FOR_EACH_OBSERVER(UserManager::Observer, observer_list_,
                     MergeSessionStateChanged(merge_session_state_));
 }
 
+void UserManagerImpl::UpdateLoginState() {
+  if (!LoginState::IsInitialized())
+    return;  // LoginState may not be intialized in tests.
+  LoginState::State login_state;
+  if (!logged_in_user_ || !session_started_)
+    login_state = LoginState::LOGGED_IN_NONE;
+  else if (is_current_user_owner_)
+    login_state = LoginState::LOGGED_IN_OWNER;
+  else if (logged_in_user_->GetType() == User::USER_TYPE_GUEST)
+    login_state = LoginState::LOGGED_IN_GUEST;
+  else if (logged_in_user_->GetType() == User::USER_TYPE_RETAIL_MODE)
+    login_state = LoginState::LOGGED_IN_RETAIL_MODE;
+  else if (logged_in_user_->GetType() == User::USER_TYPE_PUBLIC_ACCOUNT)
+    login_state = LoginState::LOGGED_IN_PUBLIC;
+  else if (logged_in_user_->GetType() == User::USER_TYPE_LOCALLY_MANAGED)
+    login_state = LoginState::LOGGED_IN_LOCALLY_MANAGED;
+  else if (logged_in_user_->GetType() == User::USER_TYPE_KIOSK_APP)
+    login_state = LoginState::LOGGED_IN_KIOSK_APP;
+  else
+    login_state = LoginState::LOGGED_IN_USER;
+  LoginState::Get()->SetLoginState(login_state);
+  LoginState::Get()->SetSessionStarted(session_started_);
+}
+
 }  // namespace chromeos