1 // Copyright (c) 2012 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 #include "chrome/browser/ui/startup/autolaunch_prompt.h"
6
7 #include "base/command_line.h"
8 #include "base/message_loop/message_loop.h"
9 #include "base/prefs/pref_service.h"
10 #include "base/strings/utf_string_conversions.h"
11 #include "chrome/browser/auto_launch_trial.h"
12 #include "chrome/browser/first_run/first_run.h"
13 #include "chrome/browser/infobars/infobar_service.h"
14 #include "chrome/browser/profiles/profile.h"
15 #include "chrome/browser/ui/browser.h"
16 #include "chrome/browser/ui/tabs/tab_strip_model.h"
17 #include "chrome/common/chrome_constants.h"
18 #include "chrome/common/chrome_switches.h"
19 #include "chrome/common/pref_names.h"
20 #include "chrome/installer/util/auto_launch_util.h"
21 #include "components/infobars/core/confirm_infobar_delegate.h"
22 #include "components/infobars/core/infobar.h"
23 #include "components/pref_registry/pref_registry_syncable.h"
24 #include "content/public/browser/browser_thread.h"
25 #include "content/public/browser/web_contents.h"
26 #include "grit/chromium_strings.h"
27 #include "grit/generated_resources.h"
28 #include "grit/theme_resources.h"
29 #include "ui/base/l10n/l10n_util.h"
30
31
32 // AutolaunchInfoBarDelegate --------------------------------------------------
33
34 namespace {
35
36 // The delegate for the infobar shown when Chrome is auto-launched.
37 class AutolaunchInfoBarDelegate : public ConfirmInfoBarDelegate {
38 public:
39 // Creates an autolaunch infobar and delegate and adds the infobar to
40 // |infobar_service|.
41 static void Create(InfoBarService* infobar_service, Profile* profile);
42
43 private:
44 explicit AutolaunchInfoBarDelegate(Profile* profile);
45 virtual ~AutolaunchInfoBarDelegate();
46
set_should_expire()47 void set_should_expire() { should_expire_ = true; }
48
49 // ConfirmInfoBarDelegate:
50 virtual int GetIconID() const OVERRIDE;
51 virtual base::string16 GetMessageText() const OVERRIDE;
52 virtual base::string16 GetButtonLabel(InfoBarButton button) const OVERRIDE;
53 virtual bool Accept() OVERRIDE;
54 virtual bool Cancel() OVERRIDE;
55 virtual bool ShouldExpireInternal(
56 const NavigationDetails& details) const OVERRIDE;
57
58 // Weak pointer to the profile, not owned by us.
59 Profile* profile_;
60
61 // Whether the info-bar should be dismissed on the next navigation.
62 bool should_expire_;
63
64 // Used to delay the expiration of the info-bar.
65 base::WeakPtrFactory<AutolaunchInfoBarDelegate> weak_factory_;
66
67 DISALLOW_COPY_AND_ASSIGN(AutolaunchInfoBarDelegate);
68 };
69
70 // static
Create(InfoBarService * infobar_service,Profile * profile)71 void AutolaunchInfoBarDelegate::Create(InfoBarService* infobar_service,
72 Profile* profile) {
73 infobar_service->AddInfoBar(ConfirmInfoBarDelegate::CreateInfoBar(
74 scoped_ptr<ConfirmInfoBarDelegate>(
75 new AutolaunchInfoBarDelegate(profile))));
76 }
77
AutolaunchInfoBarDelegate(Profile * profile)78 AutolaunchInfoBarDelegate::AutolaunchInfoBarDelegate(
79 Profile* profile)
80 : ConfirmInfoBarDelegate(),
81 profile_(profile),
82 should_expire_(false),
83 weak_factory_(this) {
84 PrefService* prefs = profile->GetPrefs();
85 prefs->SetInteger(prefs::kShownAutoLaunchInfobar,
86 prefs->GetInteger(prefs::kShownAutoLaunchInfobar) + 1);
87
88 // We want the info-bar to stick-around for a few seconds and then be hidden
89 // on the next navigation after that.
90 base::MessageLoop::current()->PostDelayedTask(
91 FROM_HERE,
92 base::Bind(&AutolaunchInfoBarDelegate::set_should_expire,
93 weak_factory_.GetWeakPtr()),
94 base::TimeDelta::FromSeconds(8));
95 }
96
~AutolaunchInfoBarDelegate()97 AutolaunchInfoBarDelegate::~AutolaunchInfoBarDelegate() {
98 }
99
GetIconID() const100 int AutolaunchInfoBarDelegate::GetIconID() const {
101 return IDR_PRODUCT_LOGO_32;
102 }
103
GetMessageText() const104 base::string16 AutolaunchInfoBarDelegate::GetMessageText() const {
105 return l10n_util::GetStringUTF16(IDS_AUTO_LAUNCH_INFOBAR_TEXT);
106 }
107
GetButtonLabel(InfoBarButton button) const108 base::string16 AutolaunchInfoBarDelegate::GetButtonLabel(
109 InfoBarButton button) const {
110 return l10n_util::GetStringUTF16((button == BUTTON_OK) ?
111 IDS_AUTO_LAUNCH_OK : IDS_AUTO_LAUNCH_REVERT);
112 }
113
Accept()114 bool AutolaunchInfoBarDelegate::Accept() {
115 return true;
116 }
117
Cancel()118 bool AutolaunchInfoBarDelegate::Cancel() {
119 content::BrowserThread::PostTask(
120 content::BrowserThread::FILE, FROM_HERE,
121 base::Bind(&auto_launch_util::DisableForegroundStartAtLogin,
122 profile_->GetPath().BaseName().value()));
123 return true;
124 }
125
ShouldExpireInternal(const NavigationDetails & details) const126 bool AutolaunchInfoBarDelegate::ShouldExpireInternal(
127 const NavigationDetails& details) const {
128 return should_expire_;
129 }
130
131 } // namespace
132
133
134 // Functions ------------------------------------------------------------------
135
136 namespace chrome {
137
ShowAutolaunchPrompt(Browser * browser)138 bool ShowAutolaunchPrompt(Browser* browser) {
139 if (!auto_launch_trial::IsInAutoLaunchGroup())
140 return false;
141
142 // Only supported on the main profile for now.
143 Profile* profile = browser->profile();
144 if (profile->GetPath().BaseName() !=
145 base::FilePath(base::ASCIIToUTF16(chrome::kInitialProfile))) {
146 return false;
147 }
148
149 int infobar_shown =
150 profile->GetPrefs()->GetInteger(prefs::kShownAutoLaunchInfobar);
151 const int kMaxTimesToShowInfoBar = 5;
152 if (infobar_shown >= kMaxTimesToShowInfoBar)
153 return false;
154
155 const CommandLine& command_line = *CommandLine::ForCurrentProcess();
156 if (!command_line.HasSwitch(switches::kAutoLaunchAtStartup) &&
157 !first_run::IsChromeFirstRun()) {
158 return false;
159 }
160
161 content::WebContents* web_contents =
162 browser->tab_strip_model()->GetActiveWebContents();
163 AutolaunchInfoBarDelegate::Create(
164 InfoBarService::FromWebContents(web_contents),
165 Profile::FromBrowserContext(web_contents->GetBrowserContext()));
166 return true;
167 }
168
RegisterAutolaunchUserPrefs(user_prefs::PrefRegistrySyncable * registry)169 void RegisterAutolaunchUserPrefs(user_prefs::PrefRegistrySyncable* registry) {
170 registry->RegisterIntegerPref(
171 prefs::kShownAutoLaunchInfobar, 0,
172 user_prefs::PrefRegistrySyncable::UNSYNCABLE_PREF);
173 }
174
175 } // namespace chrome
176