Show "Update Chrome OS" in the wrench menu, when the update image is ready.

chrome/browser/dom_ui/options/about_page_handler.cc

 // Copyright (c) 2010 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/dom_ui/options/about_page_handler.h"
 
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #include "base/basictypes.h"
 #include "base/callback.h"
@@ skipping 15 matching lines @@
 #include "grit/locale_settings.h"
 #include "grit/theme_resources.h"
 #include "webkit/glue/webkit_glue.h"
 
 #if defined(CHROME_V8)
 #include "v8/include/v8.h"
 #endif
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/cros/cros_library.h"
+#include "chrome/browser/chromeos/cros/power_library.h"
 #include "chrome/browser/chromeos/cros/update_library.h"
 #endif
 
 namespace {
 
 // These are used as placeholder text around the links in the text in the
 // license.
 const wchar_t kBeginLink[] = L"BEGIN_LINK";
 const wchar_t kEndLink[] = L"END_LINK";
 const wchar_t kBeginLinkChr[] = L"BEGIN_LINK_CHR";
@@ skipping 13 matching lines @@
   int resource;
 };
 
 const LocalizeEntry localize_table[] = {
 #if defined (OS_CHROMEOS)
     { "product", IDS_PRODUCT_OS_NAME },
     { "os", IDS_PRODUCT_OS_NAME },
     { "loading", IDS_ABOUT_PAGE_LOADING },
     { "check_now", IDS_ABOUT_PAGE_CHECK_NOW },
     { "update_status", IDS_UPGRADE_CHECK_STARTED },
+    { "restart_now", IDS_RESTART_AND_UPDATE },
 #else
     { "product", IDS_PRODUCT_NAME },
     { "check_now", IDS_ABOUT_CHROME_UPDATE_CHECK },
 #endif
     { "browser", IDS_PRODUCT_NAME },
     { "more_info", IDS_ABOUT_PAGE_MORE_INFO },
     { "copyright", IDS_ABOUT_VERSION_COPYRIGHT },
     { "channel", IDS_ABOUT_PAGE_CHANNEL },
     { "release", IDS_ABOUT_PAGE_CHANNEL_RELEASE },
     { "beta", IDS_ABOUT_PAGE_CHANNEL_BETA },
@@ skipping 153 matching lines @@
 #endif
 }
 
 void AboutPageHandler::RegisterMessages() {
   dom_ui_->RegisterMessageCallback("PageReady",
       NewCallback(this, &AboutPageHandler::PageReady));
 
 #if defined(OS_CHROMEOS)
   dom_ui_->RegisterMessageCallback("CheckNow",
       NewCallback(this, &AboutPageHandler::CheckNow));
+  dom_ui_->RegisterMessageCallback("RestartNow",
+      NewCallback(this, &AboutPageHandler::RestartNow));
 #endif
 }
 
 void AboutPageHandler::PageReady(const ListValue* args) {
 #if defined(OS_CHROMEOS)
   // Version information is loaded from a callback
   loader_.GetVersion(&consumer_,
                      NewCallback(this, &AboutPageHandler::OnOSVersion),
                      true);
 
@@ skipping 13 matching lines @@
 #endif
 }
 
 #if defined(OS_CHROMEOS)
 
 void AboutPageHandler::CheckNow(const ListValue* args) {
   if (chromeos::InitiateUpdateCheck)
     chromeos::InitiateUpdateCheck();
 }
 
+void AboutPageHandler::RestartNow(const ListValue* args) {
+  chromeos::CrosLibrary::Get()->GetPowerLibrary()->RequestRestart();
+}
+
 void AboutPageHandler::UpdateStatus(
     const chromeos::UpdateLibrary::Status& status) {
   string16 message;
   std::string image = "up-to-date";
   bool enabled = false;
 
   switch (status.status) {
     case chromeos::UPDATE_STATUS_IDLE:
       if (!sticky_) {
         message = l10n_util::GetStringFUTF16(IDS_UPGRADE_ALREADY_UP_TO_DATE,
@@ skipping 55 matching lines @@
                                     *version_string);
 
     scoped_ptr<Value> enabled_value(Value::CreateBooleanValue(enabled));
     dom_ui_->CallJavascriptFunction(L"AboutPage.updateEnableCallback",
                                     *enabled_value);
 
     scoped_ptr<Value> image_string(Value::CreateStringValue(image));
     dom_ui_->CallJavascriptFunction(L"AboutPage.setUpdateImage",
                                     *image_string);
   }
+  // We'll change the "Check For Update" button to "Restart" button.
+  if (status.status == chromeos::UPDATE_STATUS_UPDATED_NEED_REBOOT) {
+    dom_ui_->CallJavascriptFunction(L"AboutPage.changeToRestartButton");
+  }
 }
 
 void AboutPageHandler::OnOSVersion(chromeos::VersionLoader::Handle handle,
                                    std::string version) {
   if (version.size()) {
     scoped_ptr<Value> version_string(Value::CreateStringValue(version));
     dom_ui_->CallJavascriptFunction(L"AboutPage.updateOSVersionCallback",
                                     *version_string);
   }
 }
 #endif