1// Copyright (c) 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef CHROME_BROWSER_PROFILE_RESETTER_PROFILE_RESETTER_H_
6#define CHROME_BROWSER_PROFILE_RESETTER_PROFILE_RESETTER_H_
7
8#include <utility>
9#include <vector>
10
11#include "base/basictypes.h"
12#include "base/callback.h"
13#include "base/files/file_path.h"
14#include "base/memory/ref_counted.h"
15#include "base/memory/scoped_ptr.h"
16#include "base/memory/weak_ptr.h"
17#include "base/strings/string16.h"
18#include "base/threading/non_thread_safe.h"
19#include "chrome/browser/browsing_data/browsing_data_remover.h"
20#include "chrome/browser/profile_resetter/brandcoded_default_settings.h"
21#include "components/search_engines/template_url_service.h"
22
23class Profile;
24
25namespace base {
26class CancellationFlag;
27}
28
29// This class allows resetting certain aspects of a profile to default values.
30// It is used in case the profile has been damaged due to malware or bad user
31// settings.
32class ProfileResetter : public base::NonThreadSafe,
33                        public BrowsingDataRemover::Observer {
34 public:
35  // Flags indicating what aspects of a profile shall be reset.
36  enum Resettable {
37    DEFAULT_SEARCH_ENGINE = 1 << 0,
38    HOMEPAGE = 1 << 1,
39    CONTENT_SETTINGS = 1 << 2,
40    COOKIES_AND_SITE_DATA = 1 << 3,
41    EXTENSIONS = 1 << 4,
42    STARTUP_PAGES = 1 << 5,
43    PINNED_TABS = 1 << 6,
44    SHORTCUTS = 1 << 7,
45    // Update ALL if you add new values and check whether the type of
46    // ResettableFlags needs to be enlarged.
47    ALL = DEFAULT_SEARCH_ENGINE | HOMEPAGE | CONTENT_SETTINGS |
48          COOKIES_AND_SITE_DATA | EXTENSIONS | STARTUP_PAGES | PINNED_TABS |
49          SHORTCUTS
50  };
51
52  // Bit vector for Resettable enum.
53  typedef uint32 ResettableFlags;
54
55  COMPILE_ASSERT(sizeof(ResettableFlags) == sizeof(Resettable),
56                 type_ResettableFlags_doesnt_match_Resettable);
57
58  explicit ProfileResetter(Profile* profile);
59  virtual ~ProfileResetter();
60
61  // Resets |resettable_flags| and calls |callback| on the UI thread on
62  // completion. |default_settings| allows the caller to specify some default
63  // settings. |default_settings| shouldn't be NULL. |accepted_send_feedback|
64  // identifies whether the user accepted to send feedback or not.
65  void Reset(ResettableFlags resettable_flags,
66             scoped_ptr<BrandcodedDefaultSettings> master_settings,
67             bool accepted_send_feedback,
68             const base::Closure& callback);
69
70  bool IsActive() const;
71
72 private:
73  // Marks |resettable| as done and triggers |callback_| if all pending jobs
74  // have completed.
75  void MarkAsDone(Resettable resettable);
76
77  void ResetDefaultSearchEngine();
78  void ResetHomepage();
79  void ResetContentSettings();
80  void ResetCookiesAndSiteData();
81  void ResetExtensions();
82  void ResetStartupPages();
83  void ResetPinnedTabs();
84  void ResetShortcuts();
85
86  // BrowsingDataRemover::Observer:
87  virtual void OnBrowsingDataRemoverDone() OVERRIDE;
88
89  // Callback for when TemplateURLService has loaded.
90  void OnTemplateURLServiceLoaded();
91
92  Profile* const profile_;
93  scoped_ptr<BrandcodedDefaultSettings> master_settings_;
94  TemplateURLService* template_url_service_;
95
96  // Flags of a Resetable indicating which reset operations we are still waiting
97  // for.
98  ResettableFlags pending_reset_flags_;
99
100  // Called on UI thread when reset has been completed.
101  base::Closure callback_;
102
103  // If non-null it means removal is in progress. BrowsingDataRemover takes care
104  // of deleting itself when done.
105  BrowsingDataRemover* cookies_remover_;
106
107  scoped_ptr<TemplateURLService::Subscription> template_url_service_sub_;
108
109  base::WeakPtrFactory<ProfileResetter> weak_ptr_factory_;
110
111  DISALLOW_COPY_AND_ASSIGN(ProfileResetter);
112};
113
114// Path to shortcut and command line arguments.
115typedef std::pair<base::FilePath, base::string16> ShortcutCommand;
116
117typedef base::RefCountedData<base::CancellationFlag> SharedCancellationFlag;
118
119// On Windows returns all the shortcuts which launch Chrome and corresponding
120// arguments. |cancel| can be passed to abort the operation earlier.
121// Call on FILE thread.
122std::vector<ShortcutCommand> GetChromeLaunchShortcuts(
123    const scoped_refptr<SharedCancellationFlag>& cancel);
124
125#endif  // CHROME_BROWSER_PROFILE_RESETTER_PROFILE_RESETTER_H_
126