src/content/shell/renderer/test_runner/notification_presenter.cc

   1 // Copyright 2014 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 "content/shell/renderer/test_runner/notification_presenter.h"
   6
   7 #include "content/shell/renderer/test_runner/WebTestDelegate.h"
   8 #include "third_party/WebKit/public/platform/Platform.h"
   9 #include "third_party/WebKit/public/platform/WebString.h"
  10 #include "third_party/WebKit/public/platform/WebURL.h"
  11 #include "third_party/WebKit/public/web/WebKit.h"
  12 #include "third_party/WebKit/public/web/WebNotificationPermissionCallback.h"
  13 #include "third_party/WebKit/public/web/WebSecurityOrigin.h"
  14 #include "url/gurl.h"
  15
  16 using blink::Platform;
  17 using blink::WebNotification;
  18 using blink::WebNotificationPermissionCallback;
  19 using blink::WebNotificationPresenter;
  20 using blink::WebSecurityOrigin;
  21 using blink::WebTextDirectionRightToLeft;
  22
  23 namespace content {
  24
  25 namespace {
  26
  27 void DeferredDisplayDispatch(void* context) {
  28   WebNotification* notification = static_cast<WebNotification*>(context);
  29   notification->dispatchDisplayEvent();
  30
  31   delete notification;
  32 }
  33
  34 }  // namespace
  35
  36 NotificationPresenter::NotificationPresenter() : delegate_(0) {}
  37
  38 NotificationPresenter::~NotificationPresenter() {}
  39
  40 void NotificationPresenter::GrantPermission(const std::string& origin,
  41                                             bool permission_granted) {
  42   known_origins_[origin] = permission_granted;
  43 }
  44
  45 bool NotificationPresenter::SimulateClick(const std::string& title) {
  46   ActiveNotificationMap::iterator iter = active_notifications_.find(title);
  47   if (iter == active_notifications_.end())
  48     return false;
  49
  50   const WebNotification& notification = iter->second;
  51
  52   WebNotification event_target(notification);
  53   event_target.dispatchClickEvent();
  54
  55   return true;
  56 }
  57
  58 void NotificationPresenter::Reset() {
  59   while (!active_notifications_.empty()) {
  60     const WebNotification& notification = active_notifications_.begin()->second;
  61     cancel(notification);
  62   }
  63
  64   known_origins_.clear();
  65   replacements_.clear();
  66 }
  67
  68 bool NotificationPresenter::show(const WebNotification& notification) {
  69   if (!notification.replaceId().isEmpty()) {
  70     std::string replaceId(notification.replaceId().utf8());
  71     if (replacements_.find(replaceId) != replacements_.end()) {
  72       delegate_->printMessage(std::string("REPLACING NOTIFICATION ") +
  73                               replacements_.find(replaceId)->second + "\n");
  74     }
  75     replacements_[replaceId] = notification.title().utf8();
  76   }
  77
  78   delegate_->printMessage("DESKTOP NOTIFICATION SHOWN: ");
  79   if (!notification.title().isEmpty())
  80     delegate_->printMessage(notification.title().utf8().data());
  81
  82   if (notification.direction() == WebTextDirectionRightToLeft)
  83     delegate_->printMessage(", RTL");
  84
  85   // TODO(beverloo): WebNotification should expose the "lang" attribute's value.
  86
  87   if (!notification.body().isEmpty()) {
  88     delegate_->printMessage(std::string(", body: ") +
  89                             notification.body().utf8().data());
  90   }
  91
  92   if (!notification.replaceId().isEmpty()) {
  93     delegate_->printMessage(std::string(", tag: ") +
  94                             notification.replaceId().utf8().data());
  95   }
  96
  97   if (!notification.iconURL().isEmpty()) {
  98     delegate_->printMessage(std::string(", icon: ") +
  99                             notification.iconURL().spec().data());
 100   }
 101
 102   delegate_->printMessage("\n");
 103
 104   std::string title = notification.title().utf8();
 105   active_notifications_[title] = notification;
 106
 107   Platform::current()->callOnMainThread(DeferredDisplayDispatch,
 108                                         new WebNotification(notification));
 109
 110   return true;
 111 }
 112
 113 void NotificationPresenter::cancel(const WebNotification& notification) {
 114   std::string title = notification.title().utf8();
 115
 116   delegate_->printMessage(std::string("DESKTOP NOTIFICATION CLOSED: ") + title +
 117                           "\n");
 118
 119   WebNotification event_target(notification);
 120   event_target.dispatchCloseEvent(false);
 121
 122   active_notifications_.erase(title);
 123 }
 124
 125 void NotificationPresenter::objectDestroyed(
 126     const WebNotification& notification) {
 127   std::string title = notification.title().utf8();
 128   active_notifications_.erase(title);
 129 }
 130
 131 WebNotificationPresenter::Permission NotificationPresenter::checkPermission(
 132     const WebSecurityOrigin& security_origin) {
 133   const std::string origin = security_origin.toString().utf8();
 134   const KnownOriginMap::iterator it = known_origins_.find(origin);
 135   if (it == known_origins_.end())
 136     return WebNotificationPresenter::PermissionNotAllowed;
 137
 138   // Values in |known_origins_| indicate whether permission has been granted.
 139   if (it->second)
 140     return WebNotificationPresenter::PermissionAllowed;
 141
 142   return WebNotificationPresenter::PermissionDenied;
 143 }
 144
 145 void NotificationPresenter::requestPermission(
 146     const WebSecurityOrigin& security_origin,
 147     WebNotificationPermissionCallback* callback) {
 148   std::string origin = security_origin.toString().utf8();
 149   delegate_->printMessage("DESKTOP NOTIFICATION PERMISSION REQUESTED: " +
 150                           origin + "\n");
 151   callback->permissionRequestComplete();
 152 }
 153
 154 }  // namespace content