Show a delete shortcut checkbox when editing a profile that has a shortcut.

chrome/browser/ui/webui/options/manage_profile_handler.cc

 // Copyright (c) 2012 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/manage_profile_handler.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/string_number_conversions.h"
 #include "base/utf_string_conversions.h"
@@ skipping 16 matching lines @@
 #include "content/public/browser/web_ui.h"
 #include "grit/generated_resources.h"
 
 namespace options {
 
 namespace {
 
 const char kCreateProfileIconGridName[] = "create-profile-icon-grid";
 const char kManageProfileIconGridName[] = "manage-profile-icon-grid";
 
+// Given |args| from the WebUI, parses value 0 as a FilePath |profile_file_path|
+// and returns true on success.
+bool GetProfilePathFromArgs(const ListValue* args,
+                            FilePath* profile_file_path) {
+  const Value* file_path_value;
+  if (!args->Get(0, &file_path_value))
+    return false;
+  return base::GetValueAsFilePath(*file_path_value, profile_file_path);
+}
+
 }  // namespace
 
-ManageProfileHandler::ManageProfileHandler() {
+ManageProfileHandler::ManageProfileHandler()
+    : ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) {
 }
 
 ManageProfileHandler::~ManageProfileHandler() {
 }
 
 void ManageProfileHandler::GetLocalizedValues(
     DictionaryValue* localized_strings) {
   DCHECK(localized_strings);
 
   static OptionsStringResource resources[] = {
     { "manageProfilesNameLabel", IDS_PROFILES_MANAGE_NAME_LABEL },
     { "manageProfilesDuplicateNameError",
         IDS_PROFILES_MANAGE_DUPLICATE_NAME_ERROR },
     { "manageProfilesIconLabel", IDS_PROFILES_MANAGE_ICON_LABEL },
     { "manageProfilesManagedUserSettings",
         IDS_PROFILES_MANAGE_MANAGED_USER_SETTINGS_BUTTON },
     { "manageProfilesManagedLabel", IDS_PROFILES_CREATE_MANAGED_CHECKBOX },
     { "deleteProfileTitle", IDS_PROFILES_DELETE_TITLE },
     { "deleteProfileOK", IDS_PROFILES_DELETE_OK_BUTTON_LABEL },
     { "deleteProfileMessage", IDS_PROFILES_DELETE_MESSAGE },
     { "createProfileTitle", IDS_PROFILES_CREATE_TITLE },
     { "createProfileInstructions", IDS_PROFILES_CREATE_INSTRUCTIONS },
     { "createProfileConfirm", IDS_PROFILES_CREATE_CONFIRM },
-    { "createProfileShortcut", IDS_PROFILES_CREATE_SHORTCUT_CHKBOX },
+    { "createProfileShortcut", IDS_PROFILES_CREATE_SHORTCUT_CHECKBOX },
+    { "removeProfileShortcut", IDS_PROFILES_REMOVE_SHORTCUT_CHECKBOX },
   };
 
   RegisterStrings(localized_strings, resources, arraysize(resources));
   RegisterTitle(localized_strings, "manageProfile",
                 IDS_PROFILES_MANAGE_TITLE);
   RegisterTitle(localized_strings, "createProfile",
                 IDS_PROFILES_CREATE_TITLE);
+
+  localized_strings->SetBoolean("profileShortcutsEnabled",
+                                ProfileShortcutManager::IsFeatureEnabled());
 }
 
 void ManageProfileHandler::InitializeHandler() {
   registrar_.Add(this, chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED,
                  content::NotificationService::AllSources());
 }
 
 void ManageProfileHandler::InitializePage() {
   SendProfileNames();
 }
 
 void ManageProfileHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback("setProfileNameAndIcon",
       base::Bind(&ManageProfileHandler::SetProfileNameAndIcon,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("deleteProfile",
       base::Bind(&ManageProfileHandler::DeleteProfile,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("requestDefaultProfileIcons",
       base::Bind(&ManageProfileHandler::RequestDefaultProfileIcons,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("requestNewProfileDefaults",
       base::Bind(&ManageProfileHandler::RequestNewProfileDefaults,
                  base::Unretained(this)));
+  web_ui()->RegisterMessageCallback("requestHasProfileShortcuts",
+      base::Bind(&ManageProfileHandler::RequestHasProfileShortcuts,
+                 base::Unretained(this)));
   web_ui()->RegisterMessageCallback("profileIconSelectionChanged",
       base::Bind(&ManageProfileHandler::ProfileIconSelectionChanged,
                  base::Unretained(this)));
 }
 
 void ManageProfileHandler::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   if (type == chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED) {
@@ skipping 63 matching lines @@
     profile_name_dict.SetBoolean(UTF16ToUTF8(cache.GetNameOfProfileAtIndex(i)),
                                  true);
 
   web_ui()->CallJavascriptFunction("ManageProfileOverlay.receiveProfileNames",
                                    profile_name_dict);
 }
 
 void ManageProfileHandler::SetProfileNameAndIcon(const ListValue* args) {
   DCHECK(args);
 
-  const Value* file_path_value;
   FilePath profile_file_path;
-  if (!args->Get(0, &file_path_value) ||
-      !base::GetValueAsFilePath(*file_path_value, &profile_file_path))
+  if (!GetProfilePathFromArgs(args, &profile_file_path))
     return;
 
   ProfileInfoCache& cache =
       g_browser_process->profile_manager()->GetProfileInfoCache();
   size_t profile_index = cache.GetIndexOfProfileWithPath(profile_file_path);
   if (profile_index == std::string::npos)
     return;
 
   Profile* profile =
       g_browser_process->profile_manager()->GetProfile(profile_file_path);
   if (!profile)
     return;
 
-  bool shortcut_checked;
-  if (!args->GetBoolean(3, &shortcut_checked))
+  std::string shortcut_mode;
+  if (!args->GetString(3, &shortcut_mode))
     return;
-  if (shortcut_checked) {
+  if (!shortcut_mode.empty()) {
+    DCHECK(ProfileShortcutManager::IsFeatureEnabled());
     ProfileShortcutManager* shortcut_manager =
         g_browser_process->profile_manager()->profile_shortcut_manager();
-    if (shortcut_manager) {
-       shortcut_manager->CreateProfileShortcut(
-           cache.GetPathOfProfileAtIndex(profile_index));
+    DCHECK(shortcut_manager);
+
+    const FilePath profile_path = cache.GetPathOfProfileAtIndex(profile_index);
+    if (shortcut_mode == "create") {
+      shortcut_manager->CreateProfileShortcut(profile_path);
+    } else if (shortcut_mode == "remove") {
+      shortcut_manager->RemoveProfileShortcuts(profile_path);
+    } else {
+      NOTREACHED() << shortcut_mode;
     }
   }
 
   string16 new_profile_name;
   if (!args->GetString(1, &new_profile_name))
     return;
   if (new_profile_name == cache.GetGAIANameOfProfileAtIndex(profile_index)) {
     // Set the profile to use the GAIA name as the profile name. Note, this
     // is a little weird if the user typed their GAIA name manually but
     // it's not a big deal.
@@ skipping 51 matching lines @@
 
 void ManageProfileHandler::DeleteProfile(const ListValue* args) {
   DCHECK(args);
   // This handler could have been called in managed mode, for example because
   // the user fiddled with the web inspector. Silently return in this case.
   if (!ProfileManager::IsMultipleProfilesEnabled())
     return;
 
   ProfileMetrics::LogProfileDeleteUser(ProfileMetrics::PROFILE_DELETED);
 
-  const Value* file_path_value;
   FilePath profile_file_path;
-  if (!args->Get(0, &file_path_value) ||
-      !base::GetValueAsFilePath(*file_path_value, &profile_file_path))
+  if (!GetProfilePathFromArgs(args, &profile_file_path))
     return;
 
   Browser* browser =
       chrome::FindBrowserWithWebContents(web_ui()->GetWebContents());
   chrome::HostDesktopType desktop_type = chrome::HOST_DESKTOP_TYPE_NATIVE;
   if (browser)
     desktop_type = browser->host_desktop_type();
 
   g_browser_process->profile_manager()->ScheduleProfileForDeletion(
       profile_file_path, desktop_type);
 }
 
 void ManageProfileHandler::ProfileIconSelectionChanged(
     const base::ListValue* args) {
   DCHECK(args);
 
-  const Value* file_path_value;
-  FilePath file_path;
-  if (!args->Get(0, &file_path_value) ||
-      !base::GetValueAsFilePath(*file_path_value, &file_path)) {
+  FilePath profile_file_path;
+  if (!GetProfilePathFromArgs(args, &profile_file_path))
     return;
-  }
 
   // Currently this only supports editing the current profile's info.
-  if (file_path != Profile::FromWebUI(web_ui())->GetPath())
+  if (profile_file_path != Profile::FromWebUI(web_ui())->GetPath())
     return;
 
   std::string icon_url;
   if (!args->GetString(1, &icon_url))
     return;
 
   if (icon_url != gaia_picture_url_)
     return;
 
   // If the selection is the GAIA picture then also show the GAIA name in the
   // text field.
   ProfileInfoCache& cache =
       g_browser_process->profile_manager()->GetProfileInfoCache();
-  size_t i = cache.GetIndexOfProfileWithPath(file_path);
-  if (i == std::string::npos)
+  size_t profile_index = cache.GetIndexOfProfileWithPath(profile_file_path);
+  if (profile_index == std::string::npos)
     return;
-  string16 gaia_name = cache.GetGAIANameOfProfileAtIndex(i);
+  string16 gaia_name = cache.GetGAIANameOfProfileAtIndex(profile_index);
   if (gaia_name.empty())
     return;
 
   StringValue gaia_name_value(gaia_name);
   web_ui()->CallJavascriptFunction("ManageProfileOverlay.setProfileName",
                                    gaia_name_value);
 }
 
+void ManageProfileHandler::RequestHasProfileShortcuts(const ListValue* args) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  DCHECK(ProfileShortcutManager::IsFeatureEnabled());
+
+  FilePath profile_file_path;
+  if (!GetProfilePathFromArgs(args, &profile_file_path))
+    return;
+
+  const ProfileInfoCache& cache =
+      g_browser_process->profile_manager()->GetProfileInfoCache();
+  size_t profile_index = cache.GetIndexOfProfileWithPath(profile_file_path);
+  if (profile_index == std::string::npos)
+    return;
+
+  const FilePath profile_path = cache.GetPathOfProfileAtIndex(profile_index);
+  ProfileShortcutManager* shortcut_manager =
+      g_browser_process->profile_manager()->profile_shortcut_manager();
+  shortcut_manager->HasProfileShortcuts(
+      profile_path, base::Bind(&ManageProfileHandler::OnHasProfileShortcuts,
+                               weak_factory_.GetWeakPtr()));
+}
+
+void ManageProfileHandler::OnHasProfileShortcuts(bool has_shortcuts) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  const base::FundamentalValue has_shortcuts_value(has_shortcuts);
+  web_ui()->CallJavascriptFunction(
+      "ManageProfileOverlay.receiveHasProfileShortcuts", has_shortcuts_value);
+}
+
 }  // namespace options