Switch about box to web ui on Windows.

chrome/browser/ui/webui/help/version_updater_win.cc

+// Use of this source code is governed by a BSD-style license that can be
+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// found in the LICENSE file.
+
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "base/string16.h"
+#include "base/version.h"
+#include "base/win/windows_version.h"
+#include "base/win/win_util.h"
+#include "chrome/browser/google/google_update_win.h"
+#include "chrome/browser/lifetime/application_lifetime.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/webui/help/version_updater.h"
+#include "chrome/common/chrome_version_info.h"
+#include "chrome/installer/util/browser_distribution.h"
+#include "chrome/installer/util/install_util.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/user_metrics.h"
+#include "grit/chromium_strings.h"
+#include "grit/generated_resources.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/views/widget/widget.h"
+
+using content::BrowserThread;
+using content::UserMetricsAction;
+
+namespace {
+
+// Windows implementation of version update functionality, used by the WebUI
+// About/Help page.
+class VersionUpdaterWin : public VersionUpdater,
+                          public GoogleUpdateStatusListener {
+ private:
+  friend class VersionReader;
+  friend class VersionUpdater;
+
+  // Clients must use VersionUpdater::Create().
+  VersionUpdaterWin();
+  virtual ~VersionUpdaterWin();
+
+  // VersionUpdater implementation.
+  virtual void CheckForUpdate(const StatusCallback& callback) OVERRIDE;
+  virtual void RelaunchBrowser() const OVERRIDE;
+
+  // GoogleUpdateStatusListener implementation.
+  virtual void OnReportResults(GoogleUpdateUpgradeResult result,
+                               GoogleUpdateErrorCode error_code,
+                               const string16& error_message,
+                               const string16& version) OVERRIDE;
+
+  // Update the UI to show the status of the upgrade.
+  void UpdateStatus(GoogleUpdateUpgradeResult result,
+                    GoogleUpdateErrorCode error_code,
+                    const string16& error_message);
+
+  // Got the intalled version so the handling of the UPGRADE_ALREADY_UP_TO_DATE
+  // result case can now be completeb on the UI thread.
+  void GotInstalledVersion(const Version* version);
+
+  // Little helper function to reset google_updater_.
+  void SetGoogleUpdater();
+
+  // Returns a window that can be used for elevation.
+  HWND GetElevationParent();
+
+  // The class that communicates with Google Update to find out if an update is
+  // available and asks it to start an upgrade.
+  scoped_refptr<GoogleUpdate> google_updater_;
+
+  // Used for callbacks.
+  base::WeakPtrFactory<VersionUpdaterWin> weak_factory_;
+
+  // Callback used to communicate update status to the client.
+  StatusCallback callback_;
+
+  // A handle to a window from the UI thread which is used to pass to
+  // GoogleUpdate to potentially be used for the elevation UI.
+  HWND window_;
+
+  DISALLOW_COPY_AND_ASSIGN(VersionUpdaterWin);
+};
+
+// This class is used to read the version on the FILE thread and then call back
+// the version updater in the UI thread. Using a class helps better control
+// the lifespan of the Version independently of the lifespan of the version
+// updater, which may die while asynchonicity is happening, thus the usage of
+// the WeakPtr, which can only be used from the thread that created it.
+class VersionReader
+    : public base::RefCountedThreadSafe<VersionReader> {
+ public:
+  explicit VersionReader(
+      const base::WeakPtr<VersionUpdaterWin>& version_updater)
+      : version_updater_(version_updater) {
+  }
+
+  void GetVersionFromFileThread() {
+    BrowserDistribution* dist = BrowserDistribution::GetDistribution();
+    installed_version_.reset(InstallUtil::GetChromeVersion(dist, false));
+    if (!installed_version_.get()) {
+      // User-level Chrome is not installed, check system-level.
+      installed_version_.reset(InstallUtil::GetChromeVersion(dist, true));
+    }
+    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, base::Bind(
+          &VersionReader::SetVersionInUIThread, this));
+  }
+
+  void SetVersionInUIThread() {
+    if (version_updater_.get() != NULL)
+      version_updater_->GotInstalledVersion(installed_version_.get());
+  }
+
+ private:
+  friend class base::RefCountedThreadSafe<VersionReader>;
+
+  // The version updater that must be called back when we are done.
+  // We use a weak pointer in case the updater gets destroyed while waiting.
+  base::WeakPtr<VersionUpdaterWin> version_updater_;
+
+  // This is the version that gets read in the FILE thread and set on the
+  // the updater in the UI thread.
+  scoped_ptr<Version> installed_version_;
+};
+
+VersionUpdaterWin::VersionUpdaterWin()
+    : ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)),
+      window_(NULL) {
+  SetGoogleUpdater();
+}
+
+VersionUpdaterWin::~VersionUpdaterWin() {
+  // The Google Updater will hold a pointer to the listener until it reports
+  // status, so that pointer must be cleared when the listener is destoyed.
+  if (google_updater_)
+    google_updater_->set_status_listener(NULL);
+}
+
+void VersionUpdaterWin::CheckForUpdate(const StatusCallback& callback) {
+  callback_ = callback;
+
+  // On-demand updates for Chrome don't work in Vista RTM when UAC is turned
+  // off. So, in this case, the version updater must not mention
+  // on-demand updates. Silent updates (in the background) should still
+  // work as before - enabling UAC or installing the latest service pack
+  // for Vista is another option.
+  if (!(base::win::GetVersion() == base::win::VERSION_VISTA &&
+        (base::win::OSInfo::GetInstance()->service_pack().major == 0) &&
+        !base::win::UserAccountControlIsEnabled())) {
+    // This could happen if the page got refreshed after results were returned.
+    if (!google_updater_)
+      SetGoogleUpdater();
+    UpdateStatus(UPGRADE_CHECK_STARTED, GOOGLE_UPDATE_NO_ERROR, string16());
+    // Specify false to not upgrade yet.
+    google_updater_->CheckForUpdate(false, GetElevationParent());
+  }
+}
+
+void VersionUpdaterWin::RelaunchBrowser() const {
+  browser::AttemptRestart();
+}
+
+void VersionUpdaterWin::OnReportResults(
+    GoogleUpdateUpgradeResult result, GoogleUpdateErrorCode error_code,
+    const string16& error_message, const string16& version) {
+  // Drop the last reference to the object so that it gets cleaned up here.
+  google_updater_ = NULL;
+  UpdateStatus(result, error_code, error_message);
+}
+
+void VersionUpdaterWin::UpdateStatus(GoogleUpdateUpgradeResult result,
+                                     GoogleUpdateErrorCode error_code,
+                                     const string16& error_message) {
+  // For Chromium builds it would show an error message.
+  // But it looks weird because in fact there is no error,
+  // just the update server is not available for non-official builds.
+#if defined(GOOGLE_CHROME_BUILD)
+  Status status = UPDATED;
+  string16 message;
+
+  switch (result) {
+    case UPGRADE_CHECK_STARTED: {
+      content::RecordAction(UserMetricsAction("UpgradeCheck_Started"));
+      status = CHECKING;
+      break;
+    }
+    case UPGRADE_STARTED: {
+      content::RecordAction(UserMetricsAction("Upgrade_Started"));
+      status = UPDATING;
+      break;
+    }
+    case UPGRADE_IS_AVAILABLE: {
+      content::RecordAction(
+          UserMetricsAction("UpgradeCheck_UpgradeIsAvailable"));
+      DCHECK(!google_updater_);  // Should have been nulled out already.
+      SetGoogleUpdater();
+      UpdateStatus(UPGRADE_STARTED, GOOGLE_UPDATE_NO_ERROR, string16());
+      // Specify true to upgrade now.
+      google_updater_->CheckForUpdate(true, GetElevationParent());
+      return;
+    }
+    case UPGRADE_ALREADY_UP_TO_DATE: {
+      // Google Update reported that Chrome is up-to-date.
+      // To confirm the updated version is running, the reading
+      // must be done on the file thread. The rest of this case
+      // will be handled within GotInstalledVersion.
+      BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, base::Bind(
+            &VersionReader::GetVersionFromFileThread,
+            new VersionReader(weak_factory_.GetWeakPtr())));
+      return;
+    }
+    case UPGRADE_SUCCESSFUL: {
+      content::RecordAction(UserMetricsAction("UpgradeCheck_Upgraded"));
+      status = NEARLY_UPDATED;
+      break;
+    }
+    case UPGRADE_ERROR: {
+      content::RecordAction(UserMetricsAction("UpgradeCheck_Error"));
+      status = FAILED;
+      if (error_code != GOOGLE_UPDATE_DISABLED_BY_POLICY) {
+        message =
+            l10n_util::GetStringFUTF16Int(IDS_UPGRADE_ERROR, error_code);
+      } else {
+        message =
+            l10n_util::GetStringUTF16(IDS_UPGRADE_DISABLED_BY_POLICY);
+      }
+      if (!error_message.empty()) {
+        message +=
+            l10n_util::GetStringFUTF16(IDS_ABOUT_BOX_ERROR_DURING_UPDATE_CHECK,
+                                       error_message);
+      }
+      break;
+    }
+  }
+
+  // TODO(mad): Get proper progress value instead of passing 0.
+  // http://crbug.com/136117
+  callback_.Run(status, 0, message);
+#endif  //  defined(GOOGLE_CHROME_BUILD)
+}
+
+void VersionUpdaterWin::GotInstalledVersion(const Version* version) {
+  // This must be called on the UI thread so that callback_ can be called.
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+  // Make sure that the latest version is running and if not,
+  // notify the user by setting the status to NEARLY_UPDATED.
+  //
+  // The extra version check is necessary on Windows because the application
+  // may be already up to date on disk though the running app is still
+  // out of date.
+  chrome::VersionInfo version_info;
+  scoped_ptr<Version> running_version(
+      Version::GetVersionFromString(version_info.Version()));
+  if (!version || (version->CompareTo(*running_version) <= 0)) {
+    content::RecordAction(
+        UserMetricsAction("UpgradeCheck_AlreadyUpToDate"));
+    callback_.Run(UPDATED, 0, string16());
+  } else {
+    content::RecordAction(UserMetricsAction("UpgradeCheck_AlreadyUpgraded"));
+    callback_.Run(NEARLY_UPDATED, 0, string16());
+  }
+}
+
+void VersionUpdaterWin::SetGoogleUpdater() {
+  google_updater_ = new GoogleUpdate();
+  google_updater_->set_status_listener(this);
+}
+
+BOOL CALLBACK WindowEnumeration(HWND window, LPARAM param) {
+  if (IsWindowVisible(window)) {
+    HWND* returned_window = reinterpret_cast<HWND*>(param);
+    *returned_window = window;
+    return FALSE;
+  }
+  return TRUE;
+}
+
+HWND VersionUpdaterWin::GetElevationParent() {
+  if (window_ != NULL)

[Ben Goodger (Google), 2012/07/13 15:04:55]

it seems like you don't want this to be a member var. what if this window is
hidden or closed or otherwise becomes unsuitable? This function can just always
find the relevant window and return it. I don't think this gets called very
often.

[MAD, 2012/07/13 15:34:17]

Done.

+    return window_;
+  // Look for a visible window belonging to the UI thread.
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  EnumThreadWindows(GetCurrentThreadId(),
+                    WindowEnumeration,
+                    reinterpret_cast<LPARAM>(&window_));
+  DCHECK(window_ != NULL) << "Failed to find a valid window handle on thread: "
+                          << GetCurrentThreadId();
+  return window_;
+}
+
+}  // namespace
+
+VersionUpdater* VersionUpdater::Create() {
+  return new VersionUpdaterWin;
+}