Pass user session type to extension feature checks

chrome/browser/chromeos/login/users/chrome_user_manager_impl.cc

 // Copyright 2014 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/users/chrome_user_manager_impl.h"
 
 #include <stddef.h>
 
 #include <cstddef>
 #include <set>
@@ skipping 52 matching lines @@
 #include "chromeos/login/user_names.h"
 #include "chromeos/settings/cros_settings_names.h"
 #include "chromeos/timezone/timezone_resolver.h"
 #include "components/prefs/pref_registry_simple.h"
 #include "components/prefs/pref_service.h"
 #include "components/prefs/scoped_user_pref_update.h"
 #include "components/session_manager/core/session_manager.h"
 #include "components/signin/core/account_id/account_id.h"
 #include "components/user_manager/known_user.h"
 #include "components/user_manager/remove_user_delegate.h"
+#include "components/user_manager/user.h"
 #include "components/user_manager/user_image/user_image.h"
 #include "components/user_manager/user_type.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
+#include "extensions/common/features/feature_session_type.h"
 #include "policy/policy_constants.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/chromeos/resources/grit/ui_chromeos_resources.h"
 #include "ui/chromeos/strings/grit/ui_chromeos_strings.h"
 #include "ui/gfx/image/image_skia.h"
 #include "ui/wm/core/wm_core_switches.h"
 
 using content::BrowserThread;
 
@@ skipping 51 matching lines @@
   PrefService* const prefs = profile->GetPrefs();
   if (can_lock)
     *can_lock = user->can_lock() && prefs->GetBoolean(prefs::kAllowScreenLock);
   if (multi_profile_behavior) {
     *multi_profile_behavior =
         prefs->GetString(prefs::kMultiProfileUserBehavior);
   }
   return true;
 }
 
+extensions::FeatureSessionType GetFeatureSessionTypeForUser(
+    const user_manager::User* user) {
+  switch (user->GetType()) {
+    case user_manager::USER_TYPE_REGULAR:
+      return extensions::FeatureSessionType::REGULAR;
+    case user_manager::USER_TYPE_KIOSK_APP:
+      return extensions::FeatureSessionType::KIOSK;
+    default:
+      // The rest of user types is not used by extensions features.
+      return extensions::FeatureSessionType::UNKNOWN;
+  }
+}
+
 }  // namespace
 
 // static
 void ChromeUserManagerImpl::RegisterPrefs(PrefRegistrySimple* registry) {
   ChromeUserManager::RegisterPrefs(registry);
 
   registry->RegisterListPref(kDeviceLocalAccounts);
   registry->RegisterStringPref(kDeviceLocalAccountPendingDataRemoval,
                                std::string());
   registry->RegisterListPref(kReportingUsers);
@@ skipping 1072 matching lines @@
   users_update->Remove(
       base::StringValue(FullyCanonicalize(account_id.GetUserEmail())), NULL);
 }
 
 void ChromeUserManagerImpl::UpdateLoginState(
     const user_manager::User* active_user,
     const user_manager::User* primary_user,
     bool is_current_user_owner) const {
   chrome_user_manager_util::UpdateLoginState(active_user, primary_user,
                                              is_current_user_owner);
+  // If a user is logged in, update session type as seen by extensions system.
+  if (active_user) {
+    extensions::SetCurrentFeatureSessionType(
+        GetFeatureSessionTypeForUser(active_user));
+  }
 }
 
 bool ChromeUserManagerImpl::GetPlatformKnownUserId(
     const std::string& user_email,
     const std::string& gaia_id,
     AccountId* out_account_id) const {
   return chrome_user_manager_util::GetPlatformKnownUserId(user_email, gaia_id,
                                                           out_account_id);
 }
 
@@ skipping 73 matching lines @@
       break;
     default:
       NOTREACHED();
       break;
   }
 
   return user;
 }
 
 }  // namespace chromeos