Reset homepage and startup pages

chrome/browser/profile_resetter/profile_resetter.cc

 // Copyright (c) 2013 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/profile_resetter/profile_resetter.h"
 
+#include "base/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/common/pref_names.h"
 #include "content/public/browser/browser_thread.h"
 
 ProfileResetter::ProfileResetter(Profile* profile)
     : profile_(profile),
       pending_reset_flags_(0) {
   DCHECK(CalledOnValidThread());
 }
 
 ProfileResetter::~ProfileResetter() {}
 
@@ skipping 33 matching lines @@
   if (resettable_flags & COOKIES_AND_SITE_DATA) {
     reset_triggered_for_flags |= COOKIES_AND_SITE_DATA;
     ResetCookiesAndSiteData();
   }
 
   if (resettable_flags & EXTENSIONS) {
     reset_triggered_for_flags |= EXTENSIONS;
     ResetExtensions(extension_handling);
   }
 
+  if (resettable_flags & STARTUP_PAGE) {
+    reset_triggered_for_flags |= STARTUP_PAGE;
+    ResetStartPage();
+  }
+
   DCHECK_EQ(resettable_flags, reset_triggered_for_flags);
 }
 
 void ProfileResetter::MarkAsDone(Resettable resettable) {
   DCHECK(CalledOnValidThread());
 
   // Check that we are never called twice or unexpectedly.
   CHECK(pending_reset_flags_ & resettable);
 
   pending_reset_flags_ &= ~resettable;
 
   if (!pending_reset_flags_)
-    callback_.Run();
+    content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE,
+                                     callback_);
 }
 
 void ProfileResetter::ResetDefaultSearchEngine() {
   DCHECK(CalledOnValidThread());
   NOTIMPLEMENTED();
   // TODO(battre/vabr): Implement
   MarkAsDone(DEFAULT_SEARCH_ENGINE);
 }
 
 void ProfileResetter::ResetHomepage() {
   DCHECK(CalledOnValidThread());
-  NOTIMPLEMENTED();
-  // TODO(battre/vabr): Implement
+  PrefService* prefs = profile_->GetPrefs();
+  DCHECK(prefs);
+  prefs->ClearPref(prefs::kHomePageIsNewTabPage);
+  prefs->ClearPref(prefs::kHomePage);
+  prefs->ClearPref(prefs::kShowHomeButton);
   MarkAsDone(HOMEPAGE);
 }
 
 void ProfileResetter::ResetContentSettings() {
   DCHECK(CalledOnValidThread());
   NOTIMPLEMENTED();
   // TODO(battre/vabr): Implement
   MarkAsDone(CONTENT_SETTINGS);
 }
 
 void ProfileResetter::ResetCookiesAndSiteData() {
   DCHECK(CalledOnValidThread());
   NOTIMPLEMENTED();
   // TODO(battre/vabr): Implement
   MarkAsDone(COOKIES_AND_SITE_DATA);
 }
 
 void ProfileResetter::ResetExtensions(ExtensionHandling extension_handling) {
   DCHECK(CalledOnValidThread());
   NOTIMPLEMENTED();
   // TODO(battre/vabr): Implement
   MarkAsDone(EXTENSIONS);
 }
+
+void ProfileResetter::ResetStartPage() {
+  DCHECK(CalledOnValidThread());
+  PrefService* prefs = profile_->GetPrefs();
+  DCHECK(prefs);
+  prefs->ClearPref(prefs::kRestoreOnStartup);
+  prefs->ClearPref(prefs::kURLsToRestoreOnStartup);
+  prefs->SetBoolean(prefs::kRestoreOnStartupMigrated, true);
+  MarkAsDone(STARTUP_PAGE);
+}