[cros] Display emails of users are stored in a separate dictionary in Local State.

chrome/browser/ui/webui/options/chromeos/accounts_options_handler.cc

 // Copyright (c) 2011 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/webui/options/chromeos/accounts_options_handler.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/json/json_reader.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/cros_settings.h"
 #include "chrome/browser/chromeos/cros_settings_names.h"
 #include "chrome/browser/chromeos/login/authenticator.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/user_cros_settings_provider.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/policy/browser_policy_connector.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace chromeos {
 
+namespace {
+
+// Adds specified user to the whitelist. Returns false if that user is already
+// in the whitelist.
+bool WhitelistUser(const std::string& username) {
+  CrosSettings* cros_settings = CrosSettings::Get();
+  if (cros_settings->FindEmailInList(kAccountsPrefUsers, username))
+    return false;
+  base::StringValue username_value(username);
+  cros_settings->AppendToList(kAccountsPrefUsers, &username_value);
+  return true;
+}
+
+}  // namespace
+
 AccountsOptionsHandler::AccountsOptionsHandler() {
 }
 
 AccountsOptionsHandler::~AccountsOptionsHandler() {
 }
 
 void AccountsOptionsHandler::RegisterMessages() {
   DCHECK(web_ui_);
   web_ui_->RegisterMessageCallback("whitelistUser",
-      base::Bind(&AccountsOptionsHandler::WhitelistUser,
+      base::Bind(&AccountsOptionsHandler::HandleWhitelistUser,
                  base::Unretained(this)));
   web_ui_->RegisterMessageCallback("unwhitelistUser",
-      base::Bind(&AccountsOptionsHandler::UnwhitelistUser,
+      base::Bind(&AccountsOptionsHandler::HandleUnwhitelistUser,
                  base::Unretained(this)));
   web_ui_->RegisterMessageCallback("whitelistExistingUsers",
-      base::Bind(&AccountsOptionsHandler::WhitelistExistingUsers,
+      base::Bind(&AccountsOptionsHandler::HandleWhitelistExistingUsers,
                  base::Unretained(this)));
 }
 
 void AccountsOptionsHandler::GetLocalizedValues(
     base::DictionaryValue* localized_strings) {
   DCHECK(localized_strings);
 
   RegisterTitle(localized_strings, "accountsPage",
                 IDS_OPTIONS_ACCOUNTS_TAB_LABEL);
 
   localized_strings->SetString("allow_BWSI", l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_ALLOW_BWSI_DESCRIPTION));
   localized_strings->SetString("use_whitelist",l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_USE_WHITELIST_DESCRIPTION));
   localized_strings->SetString("show_user_on_signin",l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_SHOW_USER_NAMES_ON_SINGIN_DESCRIPTION));
   localized_strings->SetString("username_edit_hint",l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_USERNAME_EDIT_HINT));
   localized_strings->SetString("username_format",l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_USERNAME_FORMAT));
   localized_strings->SetString("add_users",l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_ADD_USERS));
   localized_strings->SetString("owner_only", l10n_util::GetStringUTF16(
       IDS_OPTIONS_ACCOUNTS_OWNER_ONLY));
 
-  std::string owner;
-  CrosSettings::Get()->GetString(kDeviceOwner, &owner);
-  localized_strings->SetString("owner_user_id", UTF8ToUTF16(owner));
+  std::string owner_email;
+  CrosSettings::Get()->GetString(kDeviceOwner, &owner_email);
+  // Translate owner's email to the display email.
+  std::string display_email =
+      UserManager::Get()->GetUserDisplayEmail(owner_email);
+  localized_strings->SetString("owner_user_id", UTF8ToUTF16(display_email));
 
   localized_strings->SetString("current_user_is_owner",
       UserManager::Get()->current_user_is_owner() ?
       ASCIIToUTF16("true") : ASCIIToUTF16("false"));
   localized_strings->SetString("logged_in_as_guest",
       UserManager::Get()->IsLoggedInAsGuest() ?
       ASCIIToUTF16("true") : ASCIIToUTF16("false"));
   localized_strings->SetString("whitelist_is_managed",
       g_browser_process->browser_policy_connector()->IsEnterpriseManaged() ?
           ASCIIToUTF16("true") : ASCIIToUTF16("false"));
 }
 
-void AccountsOptionsHandler::WhitelistUser(const base::ListValue* args) {
+base::DictionaryValue* AccountsOptionsHandler::CreateUserInfo(
+    const std::string& username,
+    const std::string& display_email,
+    const std::string& display_name) {
+  base::DictionaryValue* user_dict = new DictionaryValue;
+  user_dict->SetString("username", username);
+  user_dict->SetString("name", display_email);
+  user_dict->SetString("email", display_name);
+  user_dict->SetBoolean("owner", IsLoggedInOwner(username));
+  return user_dict;
+}
+
+void AccountsOptionsHandler::HandleWhitelistUser(const base::ListValue* args) {
+  std::string email;
+  std::string name;
+  if (!args->GetString(0, &email) ||
+      !args->GetString(1, &name)) {
+    return;
+  }
+
+  std::string canonical_email = Authenticator::Canonicalize(email);
+  // Record a display email for the whitelisted user if it doesn't exist yet.
+  UserManager::Get()->SaveUserDisplayEmail(canonical_email, email, false);
+
+  if (WhitelistUser(canonical_email)) {
+    base::ListValue single_user;
+    single_user.Append(CreateUserInfo(canonical_email, email, name));
+    web_ui_->CallJavascriptFunction("AccountsOptions.addUsers", single_user);
+  }
+}
+
+void AccountsOptionsHandler::HandleUnwhitelistUser(
+    const base::ListValue* args) {
   std::string email;
   if (!args->GetString(0, &email)) {
     return;
   }
 
-  scoped_ptr<base::StringValue> canonical_email(
-      base::Value::CreateStringValue(Authenticator::Canonicalize(email)));
-  CrosSettings::Get()->AppendToList(kAccountsPrefUsers, canonical_email.get());
-}
-
-void AccountsOptionsHandler::UnwhitelistUser(const base::ListValue* args) {
-  std::string email;
-  if (!args->GetString(0, &email)) {
-    return;
-  }
-
-  scoped_ptr<base::StringValue> canonical_email(
-      base::Value::CreateStringValue(Authenticator::Canonicalize(email)));
-  CrosSettings::Get()->RemoveFromList(kAccountsPrefUsers,
-                                      canonical_email.get());
+  base::StringValue canonical_email(Authenticator::Canonicalize(email));
+  CrosSettings::Get()->RemoveFromList(kAccountsPrefUsers, &canonical_email);
+  // TODO(ivankr): this will not remove user's diplayed email if there is
+  // no such user.
   UserManager::Get()->RemoveUser(email, NULL);
 }
 
-void AccountsOptionsHandler::WhitelistExistingUsers(
+void AccountsOptionsHandler::HandleWhitelistExistingUsers(
     const base::ListValue* args) {
   base::ListValue whitelist_users;
   const UserList& users = UserManager::Get()->GetUsers();
   for (UserList::const_iterator it = users.begin(); it < users.end(); ++it) {
     const std::string& email = (*it)->email();
-    if (!CrosSettings::Get()->FindEmailInList(kAccountsPrefUsers, email)) {
-      base::DictionaryValue* user_dict = new DictionaryValue;
-      user_dict->SetString("name", (*it)->GetDisplayName());
-      user_dict->SetString("email", email);
-      user_dict->SetBoolean("owner", false);
-
-      whitelist_users.Append(user_dict);
+    if (WhitelistUser(email)) {
+      whitelist_users.Append(CreateUserInfo(
+          email, (*it)->display_email(), (*it)->GetDisplayName()));
     }
   }
 
   web_ui_->CallJavascriptFunction("AccountsOptions.addUsers", whitelist_users);
 }
 
+bool AccountsOptionsHandler::IsLoggedInOwner(const std::string& username) {

[James Hawkins, 2011/12/02 19:52:30]

This should be able to be moved to unnamed namespace.

[Ivan Korotkov, 2011/12/05 14:12:48]

Done.

+  UserManager* user_manager = UserManager::Get();
+  return user_manager->current_user_is_owner() &&
+      user_manager->logged_in_user().email() == username;
+}
+
 }  // namespace chromeos