Implemented new channel switcher UI.

chrome/browser/ui/webui/help/version_updater_chromeos.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/help/version_updater_chromeos.h"
 
 #include <cmath>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/login/wizard_controller.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/settings/cros_settings_names.h"
+#include "chrome/browser/policy/browser_policy_connector.h"
 #include "chrome/browser/ui/webui/help/help_utils_chromeos.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/power_manager_client.h"
 #include "chromeos/network/network_handler.h"
 #include "chromeos/network/network_state.h"
 #include "chromeos/network/network_state_handler.h"
 #include "grit/chromium_strings.h"
 #include "grit/generated_resources.h"
 #include "third_party/cros_system_api/dbus/service_constants.h"
 #include "ui/base/l10n/l10n_util.h"
@@ skipping 92 matching lines @@
     update_engine_client->RequestUpdateCheck(
         base::Bind(&VersionUpdaterCros::OnUpdateCheck,
                    weak_ptr_factory_.GetWeakPtr()));
   }
 }
 
 void VersionUpdaterCros::RelaunchBrowser() const {
   DBusThreadManager::Get()->GetPowerManagerClient()->RequestRestart();
 }
 
-void VersionUpdaterCros::SetReleaseChannel(const std::string& channel) {
-  DBusThreadManager::Get()->GetUpdateEngineClient()->SetReleaseTrack(channel);
-  // For local owner set the field in the policy blob too.
-  if (UserManager::Get()->IsCurrentUserOwner())
+void VersionUpdaterCros::SetChannel(const std::string& channel,
+                                    bool is_powerwash_allowed) {
+  // On enterprise machines we can only use SetChannel to store the
+  // user choice in the lsb-release file but we can not modify the
+  // policy blob.  Therefore we only call SetString if the device is
+  // locally owned and the currently logged in user is the owner.
+  if (g_browser_process->browser_policy_connector()->IsEnterpriseManaged()) {
+    DBusThreadManager::Get()->GetUpdateEngineClient()->
+        SetChannel(channel, is_powerwash_allowed);
+  } else if (UserManager::Get()->IsCurrentUserOwner()) {
+    // For local owner set the field in the policy blob.
     CrosSettings::Get()->SetString(chromeos::kReleaseChannel, channel);
+  }
 }
 
-void VersionUpdaterCros::GetReleaseChannel(const ChannelCallback& cb) {
-  channel_callback_ = cb;
-
-  // TODO(jhawkins): Store on this object.
+void VersionUpdaterCros::GetChannel(bool get_current_channel,
+                                    const ChannelCallback& cb) {
   UpdateEngineClient* update_engine_client =
       DBusThreadManager::Get()->GetUpdateEngineClient();
 
-  // Request the channel information. Use the observer to track the help page
-  // handler and ensure it does not get deleted before the callback.
-  update_engine_client->GetReleaseTrack(
-      base::Bind(&VersionUpdaterCros::UpdateSelectedChannel,
-                 weak_ptr_factory_.GetWeakPtr()));
+  // Request the channel information.
+  update_engine_client->GetChannel(get_current_channel, cb);
 }
 
 VersionUpdaterCros::VersionUpdaterCros()
     : last_operation_(UpdateEngineClient::UPDATE_STATUS_IDLE),
       weak_ptr_factory_(this) {
 }
 
 VersionUpdaterCros::~VersionUpdaterCros() {
   UpdateEngineClient* update_engine_client =
       DBusThreadManager::Get()->GetUpdateEngineClient();
@@ skipping 51 matching lines @@
   last_operation_ = status.status;
 }
 
 void VersionUpdaterCros::OnUpdateCheck(
     UpdateEngineClient::UpdateCheckResult result) {
   // If version updating is not implemented, this binary is the most up-to-date
   // possible with respect to automatic updating.
   if (result == UpdateEngineClient::UPDATE_RESULT_NOTIMPLEMENTED)
     callback_.Run(UPDATED, 0, string16());
 }
-
-void VersionUpdaterCros::UpdateSelectedChannel(const std::string& channel) {
-  channel_callback_.Run(channel);
-}