Initial CL for Data Saver (Flywheel) prompt when cellular network detected

chrome/browser/chromeos/status/data_promo_notification.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 "chrome/browser/chromeos/status/data_promo_notification.h"
 
 #include "ash/system/system_notifier.h"
+#include "base/command_line.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/login/helper.h"
 #include "chrome/browser/chromeos/mobile_config.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/scoped_tabbed_browser_displayer.h"
 #include "chrome/browser/ui/singleton_tabs.h"
+#include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/grit/generated_resources.h"
+#include "chromeos/chromeos_switches.h"
 #include "chromeos/login/login_state.h"
 #include "chromeos/network/device_state.h"
 #include "chromeos/network/network_connection_handler.h"
 #include "chromeos/network/network_event_log.h"
 #include "chromeos/network/network_state.h"
 #include "chromeos/network/network_state_handler.h"
+#include "extensions/browser/extension_registry.h"
 #include "grit/ui_chromeos_resources.h"
 #include "third_party/cros_system_api/dbus/service_constants.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/chromeos/network/network_connect.h"
 #include "ui/chromeos/network/network_state_notifier.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/notification.h"
 #include "ui/views/view.h"
 #include "ui/views/widget/widget.h"
 
 namespace chromeos {
 
 namespace {
 
 const char kDataPromoNotificationId[] = "chrome://settings/internet/data_promo";
+const char kDataSaverExtensionUrl[] =
+    "https://chrome.google.com/webstore/detail/"
+    "pfmgfdlgomnbgkofeojodiodmgpgmkac";
 
 const int kNotificationCountPrefDefault = -1;
+const int kTimesToShowDataSaverPrompt = 2;
 
 bool GetBooleanPref(const char* pref_name) {
   Profile* profile = ProfileManager::GetPrimaryUserProfile();
   PrefService* prefs = profile->GetPrefs();
   return prefs->GetBoolean(pref_name);
 }
 
 int GetIntegerLocalPref(const char* pref_name) {
   PrefService* prefs = g_browser_process->local_state();
   return prefs->GetInteger(pref_name);
@@ skipping 22 matching lines @@
 // Returns prefs::kCarrierDealPromoShown which is number of times
 // carrier deal notification has been shown to users on this machine.
 int GetCarrierDealPromoShown() {
   return GetIntegerLocalPref(prefs::kCarrierDealPromoShown);
 }
 
 void SetCarrierDealPromoShown(int value) {
   SetIntegerLocalPref(prefs::kCarrierDealPromoShown, value);
 }
 
+// Returns number of times the Data Saver prompt has been displayed.
+int GetDataSaverPromptsShown() {
+  Profile* profile = ProfileManager::GetActiveUserProfile();
+  PrefService* prefs = profile->GetPrefs();

[stevenjb, 2015/03/25 17:57:08]

nit: I think these would actually be more clear without the extra locals, i.e.
ProfileManager::GetActiveUserProfile()->GetPrefs()->GetInteger

[Greg Levin, 2015/04/03 18:28:37]

Done.

+  return prefs->GetInteger(prefs::kDataSaverPromptsShown);
+}
+
+// Updates number of times the Data Saver prompt has been displayed.
+void SetDataSaverPromptsShown(int times_shown) {
+  Profile* profile = ProfileManager::GetActiveUserProfile();
+  PrefService* prefs = profile->GetPrefs();
+  prefs->SetInteger(prefs::kDataSaverPromptsShown, times_shown);
+}
+
 const chromeos::MobileConfig::Carrier* GetCarrier(
     const NetworkState* cellular) {
   const DeviceState* device = NetworkHandler::Get()->network_state_handler()->
       GetDeviceState(cellular->device_path());
   std::string carrier_id = device ? device->home_provider_id() : "";
   if (carrier_id.empty()) {
     NET_LOG_ERROR("Empty carrier ID for cellular network",
                   device ? device->path(): "No device");
     return NULL;
   }
@@ skipping 54 matching lines @@
 
 void DataPromoNotification::RegisterPrefs(PrefRegistrySimple* registry) {
   // Carrier deal notification shown count defaults to 0.
   registry->RegisterIntegerPref(prefs::kCarrierDealPromoShown, 0);
 }
 
 void DataPromoNotification::NetworkPropertiesUpdated(
     const NetworkState* network) {
   if (!network || network->type() != shill::kTypeCellular)
     return;
-  ShowOptionalMobileDataPromoNotification();
+
+  bool hotkey_debug = network->path() ==
+      NetworkStateHandlerObserver::kDebugDataSaverNetworkPath;
+  ShowOptionalMobileDataPromoNotification(hotkey_debug);
 }
 
 void DataPromoNotification::DefaultNetworkChanged(const NetworkState* network) {
   // Call NetworkPropertiesUpdated in case the Cellular network became the
   // default network.
   NetworkPropertiesUpdated(network);
 }
 
-void DataPromoNotification::ShowOptionalMobileDataPromoNotification() {
-  // Display a one-time notification for authenticated users on first use
-  // of Mobile Data connection or if there is a carrier deal defined
-  // show that even if user has already seen generic promo.
-  if (!check_for_promo_ || !LoginState::Get()->IsUserAuthenticated())
-    return;
+void DataPromoNotification::ShowOptionalMobileDataPromoNotification(
+    bool hotkey_debug) {
   const NetworkState* default_network =
       NetworkHandler::Get()->network_state_handler()->DefaultNetwork();
-  if (!default_network || default_network->type() != shill::kTypeCellular)
+  // Do not show notifications to unauthenticated users, or when requesting a
+  // network connection, or if no default_network.
+  if (!default_network || !LoginState::Get()->IsUserAuthenticated() ||
+      NetworkHandler::Get()
+          ->network_connection_handler()
+          ->HasPendingConnectRequest())
     return;
-  // When requesting a network connection, do not show the notification.
-  if (NetworkHandler::Get()->network_connection_handler()->
-      HasPendingConnectRequest())
+  // |hotkey_debug| is temporary code that lets us invoke this notification
+  // via CTRL+ALT+SHIFT+3.
+  if (!hotkey_debug && default_network->type() != shill::kTypeCellular)
     return;
 
+  bool data_saver_prompt_shown = ShowDataSaverNotification(hotkey_debug);
+
+  if (!check_for_promo_)
+    return;
+
+  // Display a one-time notification on first use of Mobile Data connection or
+  // if there is a carrier deal defined show that even if user has already seen
+  // generic promo.
   int carrier_deal_promo_pref = kNotificationCountPrefDefault;
   const MobileConfig::CarrierDeal* deal = NULL;
   const MobileConfig::Carrier* carrier = GetCarrier(default_network);
   if (carrier)
     deal = GetCarrierDeal(carrier);
 
   base::string16 message =
       l10n_util::GetStringUTF16(IDS_3G_NOTIFICATION_MESSAGE);
   std::string info_url;
   if (deal) {
     carrier_deal_promo_pref = GetCarrierDealPromoShown();
     const std::string locale = g_browser_process->GetApplicationLocale();
     std::string deal_text =
         deal->GetLocalizedString(locale, "notification_text");
     message = base::UTF8ToUTF16(deal_text + "\n\n") + message;
     info_url = deal->info_url();
     if (info_url.empty() && carrier)
       info_url = carrier->top_up_url();
-  } else if (!ShouldShow3gPromoNotification()) {
+  } else if (data_saver_prompt_shown || !ShouldShow3gPromoNotification()) {
     check_for_promo_ = false;
     return;
   }
 
   int icon_id;
   if (default_network->network_technology() == shill::kNetworkTechnologyLte)
     icon_id = IDR_AURA_UBER_TRAY_NOTIFICATION_LTE;
   else
     icon_id = IDR_AURA_UBER_TRAY_NOTIFICATION_3G;
   const gfx::Image& icon =
       ui::ResourceBundle::GetSharedInstance().GetImageNamed(icon_id);
 
   message_center::MessageCenter::Get()->AddNotification(
       message_center::Notification::CreateSystemNotification(
           kDataPromoNotificationId, base::string16() /* title */, message, icon,
           ui::NetworkStateNotifier::kNotifierNetwork,
           base::Bind(&NotificationClicked, default_network->path(), info_url)));
 
   check_for_promo_ = false;
   SetShow3gPromoNotification(false);
   if (carrier_deal_promo_pref != kNotificationCountPrefDefault)
     SetCarrierDealPromoShown(carrier_deal_promo_pref + 1);
 }
 
+bool DataPromoNotification::ShowDataSaverNotification(bool hotkey_debug) {
+  if (!base::CommandLine::ForCurrentProcess()->HasSwitch(
+          chromeos::switches::kEnableDataSaverPrompt))
+    return false;
+
+  int times_shown = GetDataSaverPromptsShown();
+  if (!hotkey_debug && times_shown >= kTimesToShowDataSaverPrompt)
+    return false;
+
+  Profile* profile = ProfileManager::GetActiveUserProfile();
+  if (extensions::ExtensionRegistry::Get(profile)
+          ->GetExtensionById(extension_misc::kDataSaverExtensionId,
+                             extensions::ExtensionRegistry::EVERYTHING)) {
+    // If extension is installed, disable future prompts.
+    SetDataSaverPromptsShown(kTimesToShowDataSaverPrompt);
+    return false;
+  }
+
+  const NetworkState* default_network =
+      NetworkHandler::Get()->network_state_handler()->DefaultNetwork();
+
+  base::string16 title = l10n_util::GetStringUTF16(IDS_3G_DATASAVER_TITLE);
+  base::string16 message = l10n_util::GetStringUTF16(IDS_3G_DATASAVER_MESSAGE);
+  const gfx::Image& icon =
+      ui::ResourceBundle::GetSharedInstance().GetImageNamed(
+          IDR_AURA_UBER_TRAY_NOTIFICATION_DATASAVER);
+
+  // Clears the cached notification; otherwise, future calls from this session
+  // won't display it.
+  if (hotkey_debug) {
+    message_center::MessageCenter::Get()->RemoveNotification(
+        kDataPromoNotificationId, false);
+  }
+
+  message_center::MessageCenter::Get()->AddNotification(
+      message_center::Notification::CreateSystemNotification(
+          kDataPromoNotificationId, title, message, icon,
+          ui::NetworkStateNotifier::kNotifierNetwork,
+          base::Bind(&NotificationClicked, default_network->path(),
+                     kDataSaverExtensionUrl)));
+
+  // Hotkey resets times shown count.
+  SetDataSaverPromptsShown(hotkey_debug ? 0 : times_shown + 1);
+  return true;
+}
+
 }  // namespace chromeos