NotificationPromo

chrome/browser/ui/webui/ntp/new_tab_page_handler.cc

 // Copyright (c) 2011 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/ui/webui/ntp/new_tab_page_handler.h"
 
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/ui/webui/ntp/new_tab_ui.h"
+#include "chrome/browser/web_resource/notification_promo.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
 #include "content/common/notification_service.h"
 #include "grit/chromium_strings.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 static const int kIntroDisplayMax = 10;
 
 // The URL of a knowledge-base article about the new NTP.
 static const char kNTP4IntroURL[] =
   "http://www.google.com/support/chrome/bin/answer.py?answer=95451";
 
 void NewTabPageHandler::RegisterMessages() {
-  web_ui_->RegisterMessageCallback("closePromo", NewCallback(
-      this, &NewTabPageHandler::HandleClosePromo));
+  web_ui_->RegisterMessageCallback("closeNotificationPromo", NewCallback(
+      this, &NewTabPageHandler::HandleCloseNotificationPromo));
+  web_ui_->RegisterMessageCallback("notificationPromoViewed", NewCallback(
+      this, &NewTabPageHandler::HandleNotificationPromoViewed));
   web_ui_->RegisterMessageCallback("pageSelected", NewCallback(
       this, &NewTabPageHandler::HandlePageSelected));
   web_ui_->RegisterMessageCallback("introMessageDismissed", NewCallback(
       this, &NewTabPageHandler::HandleIntroMessageDismissed));
   web_ui_->RegisterMessageCallback("introMessageSeen", NewCallback(
       this, &NewTabPageHandler::HandleIntroMessageSeen));
 }
 
-void NewTabPageHandler::HandleClosePromo(const ListValue* args) {
-  Profile::FromWebUI(web_ui_)->GetPrefs()->SetBoolean(prefs::kNTPPromoClosed,
-                                                      true);
-  NotificationService* service = NotificationService::current();
-  service->Notify(chrome::NOTIFICATION_PROMO_RESOURCE_STATE_CHANGED,
-                  Source<NewTabPageHandler>(this),
-                  NotificationService::NoDetails());
+void NewTabPageHandler::HandleCloseNotificationPromo(const ListValue* args) {
+  NotificationPromo notification_promo(
+      Profile::FromWebUI(web_ui_)->GetPrefs(), NULL);
+  notification_promo.HandleClosed();
+  NotifyPromoResourceChanged();
+}
+
+void NewTabPageHandler::HandleNotificationPromoViewed(const ListValue* args) {
+  NotificationPromo notification_promo(
+      Profile::FromWebUI(web_ui_)->GetPrefs(), NULL);
+  if (notification_promo.HandleViewed()) {
+    NotifyPromoResourceChanged();
+  }
 }
 
 void NewTabPageHandler::HandlePageSelected(const ListValue* args) {
   double page_id_double;
   CHECK(args->GetDouble(0, &page_id_double));
   int page_id = static_cast<int>(page_id_double);
 
   double index_double;
   CHECK(args->GetDouble(1, &index_double));
   int index = static_cast<int>(index_double);
@@ skipping 44 matching lines @@
     values->SetString("ntp4_intro_url", kNTP4IntroURL);
     values->SetString("learn_more",
                       l10n_util::GetStringUTF16(IDS_LEARN_MORE));
   }
 }
 
 // static
 void NewTabPageHandler::DismissIntroMessage(PrefService* prefs) {
   prefs->SetInteger(prefs::kNTP4IntroDisplayCount, kIntroDisplayMax + 1);
 }
+
+void NewTabPageHandler::NotifyPromoResourceChanged() {
+  NotificationService* service = NotificationService::current();
+  service->Notify(chrome::NOTIFICATION_PROMO_RESOURCE_STATE_CHANGED,
+                  Source<NewTabPageHandler>(this),
+                  NotificationService::NoDetails());
+}