[protector] Support for collapsing multiple changes into a single one.

chrome/browser/protector/session_startup_change.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 "base/basictypes.h"
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "base/metrics/histogram.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/prefs/session_startup_pref.h"
@@ skipping 25 matching lines @@
   virtual void Apply(Browser* browser) OVERRIDE;
   virtual void Discard(Browser* browser) OVERRIDE;
   virtual void Timeout() OVERRIDE;
   virtual int GetBadgeIconID() const OVERRIDE;
   virtual int GetMenuItemIconID() const OVERRIDE;
   virtual int GetBubbleIconID() const OVERRIDE;
   virtual string16 GetBubbleTitle() const OVERRIDE;
   virtual string16 GetBubbleMessage() const OVERRIDE;
   virtual string16 GetApplyButtonText() const OVERRIDE;
   virtual string16 GetDiscardButtonText() const OVERRIDE;
+  virtual DisplayName GetApplyDisplayName() const OVERRIDE;
+  virtual GURL GetNewSettingURL() const OVERRIDE;
 
  private:
   virtual ~SessionStartupChange();
 
+  // Returns the first URL among new startup pages. Returns an empty URL if
+  // there are no new startup pages.
+  GURL GetFirstNewURL() const;
+
   // Opens all tabs in |tabs| and makes them pinned.
   void OpenPinnedTabs(Browser* browser, const PinnedTabCodec::Tabs& tabs);
 
   SessionStartupPref new_startup_pref_;
   SessionStartupPref backup_startup_pref_;
   PinnedTabCodec::Tabs new_pinned_tabs_;
   PinnedTabCodec::Tabs backup_pinned_tabs_;
 
   DISALLOW_COPY_AND_ASSIGN(SessionStartupChange);
 };
@@ skipping 72 matching lines @@
 }
 
 string16 SessionStartupChange::GetBubbleMessage() const {
   return l10n_util::GetStringUTF16(IDS_STARTUP_SETTINGS_CHANGE_BUBBLE_MESSAGE);
 }
 
 string16 SessionStartupChange::GetApplyButtonText() const {
   if (new_startup_pref_.type == SessionStartupPref::LAST)
       return l10n_util::GetStringUTF16(IDS_CHANGE_STARTUP_SETTINGS_RESTORE);
 
-  string16 first_url;
-  if (new_startup_pref_.type == SessionStartupPref::URLS &&
-      !new_startup_pref_.urls.empty()) {
-    // Display the domain name of the first statrup URL.
-    first_url = UTF8ToUTF16(new_startup_pref_.urls[0].host());
-  } else if (!new_pinned_tabs_.empty()) {
-    // Start with NTP or no URLs (basically the same): display the domain name
-    // of the first pinned tab, if any.
-    first_url = UTF8ToUTF16(new_pinned_tabs_[0].url.host());
-  }
-  return first_url.empty() ?
+  // Display the domain name of the first startup/pinned URL.
+  string16 first_domain = UTF8ToUTF16(GetFirstNewURL().host());
+  return first_domain.empty() ?
       l10n_util::GetStringUTF16(IDS_CHANGE_STARTUP_SETTINGS_NTP) :
       l10n_util::GetStringFUTF16(IDS_CHANGE_STARTUP_SETTINGS_URLS,
-                                 first_url);
+                                 first_domain);
 }
 
 string16 SessionStartupChange::GetDiscardButtonText() const {
   return l10n_util::GetStringUTF16(IDS_KEEP_SETTING);
 }
 
+BaseSettingChange::DisplayName
+SessionStartupChange::GetApplyDisplayName() const {
+  string16 first_domain = UTF8ToUTF16(GetFirstNewURL().host());
+  return first_domain.empty() ?
+      BaseSettingChange::GetApplyDisplayName() :
+      DisplayName(kSessionStartupChangeNamePriority, first_domain);
+}
+
+GURL SessionStartupChange::GetNewSettingURL() const {
+  VLOG(1) << "HP URL: " << GetFirstNewURL().spec();
+  return GetFirstNewURL();
+}
+
+GURL SessionStartupChange::GetFirstNewURL() const {
+  // TODO(ivankr): this actually should return the first URL present in new
+  // startup prefs AND not in old.
+  if (new_startup_pref_.type == SessionStartupPref::URLS &&
+      !new_startup_pref_.urls.empty()) {
+    return new_startup_pref_.urls[0];
+  }
+  if (!new_pinned_tabs_.empty())
+    return new_pinned_tabs_[0].url;
+  return GURL();
+}
+
 void SessionStartupChange::OpenPinnedTabs(
     Browser* browser,
     const PinnedTabCodec::Tabs& tabs) {
   for (size_t i = 0; i < tabs.size(); ++i) {
     browser::NavigateParams params(browser, tabs[i].url,
                                    content::PAGE_TRANSITION_START_PAGE);
     params.disposition = NEW_BACKGROUND_TAB;
     params.tabstrip_index = -1;
     params.tabstrip_add_types = TabStripModel::ADD_PINNED;
     params.extension_app_id = tabs[i].app_id;
     browser::Navigate(&params);
   }
 }
 
 BaseSettingChange* CreateSessionStartupChange(
     const SessionStartupPref& actual_startup_pref,
     const PinnedTabCodec::Tabs& actual_pinned_tabs,
     const SessionStartupPref& backup_startup_pref,
     const PinnedTabCodec::Tabs& backup_pinned_tabs) {
   return new SessionStartupChange(actual_startup_pref, actual_pinned_tabs,
                                   backup_startup_pref, backup_pinned_tabs);
 }
 
 }  // namespace protector