From e4d2368908e5d4549dc8978a4e45317fe3eb0d70 Mon Sep 17 00:00:00 2001
From: Cheng Zhao <zcbenz@gmail.com>
Date: Mon, 3 Jun 2013 10:34:42 +0800
Subject: [PATCH] Add a few more methods for AutoUpdater.

---
 browser/auto_updater.h      |  4 ++++
 browser/auto_updater_mac.mm | 11 +++++++++++
 2 files changed, 15 insertions(+)

diff --git a/browser/auto_updater.h b/browser/auto_updater.h
index 8426c03112..2166d49f71 100644
--- a/browser/auto_updater.h
+++ b/browser/auto_updater.h
@@ -5,6 +5,8 @@
 #ifndef ATOM_BROWSER_AUTO_UPDATER_H_
 #define ATOM_BROWSER_AUTO_UPDATER_H_
 
+#include <string>
+
 #include "base/basictypes.h"
 
 namespace auto_updater {
@@ -19,8 +21,10 @@ class AutoUpdater {
 
   static void Init();
 
+  static void SetFeedURL(const std::string& url);
   static void SetAutomaticallyChecksForUpdates(bool yes);
   static void SetAutomaticallyDownloadsUpdates(bool yes);
+  static void CheckForUpdates();
   static void CheckForUpdatesInBackground();
 
  private:
diff --git a/browser/auto_updater_mac.mm b/browser/auto_updater_mac.mm
index 3ac390da66..880e7967e4 100644
--- a/browser/auto_updater_mac.mm
+++ b/browser/auto_updater_mac.mm
@@ -71,6 +71,12 @@ void AutoUpdater::Init() {
   [[SUUpdater sharedUpdater] setDelegate:delegate];
 }
 
+// static
+void AutoUpdater::SetFeedURL(const std::string& url) {
+  NSString* url_str(base::SysUTF8ToNSString(url));
+  [[SUUpdater sharedUpdater] setFeedURL:[NSURL URLWithString:url_str]];
+}
+
 // static
 void AutoUpdater::SetAutomaticallyChecksForUpdates(bool yes) {
   [[SUUpdater sharedUpdater] setAutomaticallyChecksForUpdates:yes];
@@ -81,6 +87,11 @@ void AutoUpdater::SetAutomaticallyDownloadsUpdates(bool yes) {
   [[SUUpdater sharedUpdater] setAutomaticallyDownloadsUpdates:yes];
 }
 
+// static
+void AutoUpdater::CheckForUpdates() {
+  [[SUUpdater sharedUpdater] checkForUpdates:[SUUpdater sharedUpdater]];
+}
+
 // static
 void AutoUpdater::CheckForUpdatesInBackground() {
   [[SUUpdater sharedUpdater] checkForUpdatesInBackground];