Rename first/second to key/value in HashMap iterators
[WebKit-https.git] / Source / WebKit2 / WebProcess / Notifications / WebNotificationManager.cpp
1 /*
2  * Copyright (C) 2011, 2012 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "WebNotificationManager.h"
28
29 #include "WebPage.h"
30 #include "WebProcess.h"
31
32 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
33 #include "WebNotification.h"
34 #include "WebNotificationManagerProxyMessages.h"
35 #include "WebPageProxyMessages.h"
36 #include <WebCore/Document.h>
37 #include <WebCore/Notification.h>
38 #include <WebCore/Page.h>
39 #include <WebCore/ScriptExecutionContext.h>
40 #include <WebCore/SecurityOrigin.h>
41 #include <WebCore/Settings.h>
42 #include <WebCore/UserGestureIndicator.h>
43 #endif
44
45 using namespace WebCore;
46
47 namespace WebKit {
48
49 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
50 static uint64_t generateNotificationID()
51 {
52     static uint64_t uniqueNotificationID = 1;
53     return uniqueNotificationID++;
54 }
55 #endif
56
57 WebNotificationManager::WebNotificationManager(WebProcess* process)
58     : m_process(process)
59 {
60 }
61
62 WebNotificationManager::~WebNotificationManager()
63 {
64 }
65
66 void WebNotificationManager::didReceiveMessage(CoreIPC::Connection* connection, CoreIPC::MessageID messageID, CoreIPC::ArgumentDecoder* arguments)
67 {
68     didReceiveWebNotificationManagerMessage(connection, messageID, arguments);
69 }
70
71 void WebNotificationManager::initialize(const HashMap<String, bool>& permissions)
72 {
73 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
74     m_permissionsMap = permissions;
75 #else
76     UNUSED_PARAM(permissions);
77 #endif
78 }
79
80 void WebNotificationManager::didUpdateNotificationDecision(const String& originString, bool allowed)
81 {
82 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
83     m_permissionsMap.set(originString, allowed);
84 #else
85     UNUSED_PARAM(originString);
86     UNUSED_PARAM(allowed);
87 #endif
88 }
89
90 void WebNotificationManager::didRemoveNotificationDecisions(const Vector<String>& originStrings)
91 {
92 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
93     size_t count = originStrings.size();
94     for (size_t i = 0; i < count; ++i)
95         m_permissionsMap.remove(originStrings[i]);
96 #else
97     UNUSED_PARAM(originStrings);
98 #endif
99 }
100
101 NotificationClient::Permission WebNotificationManager::policyForOrigin(WebCore::SecurityOrigin *origin) const
102 {
103 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
104     if (!origin)
105         return NotificationClient::PermissionNotAllowed;
106
107     ASSERT(!origin->isUnique());
108     HashMap<String, bool>::const_iterator it = m_permissionsMap.find(origin->toRawString());
109     if (it != m_permissionsMap.end())
110         return it->value ? NotificationClient::PermissionAllowed : NotificationClient::PermissionDenied;
111 #else
112     UNUSED_PARAM(origin);
113 #endif
114     
115     return NotificationClient::PermissionNotAllowed;
116 }
117
118 void WebNotificationManager::removeAllPermissionsForTesting()
119 {
120 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
121     m_permissionsMap.clear();
122 #endif
123 }
124
125 uint64_t WebNotificationManager::notificationIDForTesting(Notification* notification)
126 {
127 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
128     if (!notification)
129         return 0;
130     return m_notificationMap.get(notification);
131 #else
132     UNUSED_PARAM(notification);
133     return 0;
134 #endif
135 }
136
137 bool WebNotificationManager::show(Notification* notification, WebPage* page)
138 {
139 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
140     if (!notification || !page->corePage()->settings()->notificationsEnabled())
141         return false;
142     
143     uint64_t notificationID = generateNotificationID();
144     m_notificationMap.set(notification, notificationID);
145     m_notificationIDMap.set(notificationID, notification);
146     
147     NotificationContextMap::iterator it = m_notificationContextMap.add(notification->scriptExecutionContext(), Vector<uint64_t>()).iterator;
148     it->value.append(notificationID);
149
150 #if ENABLE(NOTIFICATIONS)
151     m_process->connection()->send(Messages::WebPageProxy::ShowNotification(notification->title(), notification->body(), notification->iconURL().string(), notification->tag(), notification->lang(), notification->dir(), notification->scriptExecutionContext()->securityOrigin()->toString(), notificationID), page->pageID());
152 #else
153     m_process->connection()->send(Messages::WebPageProxy::ShowNotification(notification->title(), notification->body(), notification->iconURL().string(), notification->replaceId(), notification->lang(), notification->dir(), notification->scriptExecutionContext()->securityOrigin()->toString(), notificationID), page->pageID());
154 #endif
155     return true;
156 #else
157     UNUSED_PARAM(notification);
158     UNUSED_PARAM(page);
159     return false;
160 #endif
161 }
162
163 void WebNotificationManager::cancel(Notification* notification, WebPage* page)
164 {
165 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
166     if (!notification || !page->corePage()->settings()->notificationsEnabled())
167         return;
168     
169     uint64_t notificationID = m_notificationMap.get(notification);
170     if (!notificationID)
171         return;
172     
173     m_process->connection()->send(Messages::WebNotificationManagerProxy::Cancel(notificationID), page->pageID());
174 #else
175     UNUSED_PARAM(notification);
176     UNUSED_PARAM(page);
177 #endif
178 }
179
180 void WebNotificationManager::clearNotifications(WebCore::ScriptExecutionContext* context, WebPage* page)
181 {
182 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
183     NotificationContextMap::iterator it = m_notificationContextMap.find(context);
184     if (it == m_notificationContextMap.end())
185         return;
186     
187     Vector<uint64_t>& notificationIDs = it->value;
188     m_process->connection()->send(Messages::WebNotificationManagerProxy::ClearNotifications(notificationIDs), page->pageID());
189     size_t count = notificationIDs.size();
190     for (size_t i = 0; i < count; ++i) {
191         RefPtr<Notification> notification = m_notificationIDMap.take(notificationIDs[i]);
192         if (!notification)
193             continue;
194         notification->finalize();
195         m_notificationMap.remove(notification);
196     }
197     
198     m_notificationContextMap.remove(it);
199 #else
200     UNUSED_PARAM(context);
201     UNUSED_PARAM(page);
202 #endif
203 }
204
205 void WebNotificationManager::didDestroyNotification(Notification* notification, WebPage* page)
206 {
207 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
208     uint64_t notificationID = m_notificationMap.take(notification);
209     if (!notificationID)
210         return;
211
212     m_notificationIDMap.remove(notificationID);
213     removeNotificationFromContextMap(notificationID, notification);
214     m_process->connection()->send(Messages::WebNotificationManagerProxy::DidDestroyNotification(notificationID), page->pageID());
215 #else
216     UNUSED_PARAM(notification);
217     UNUSED_PARAM(page);
218 #endif
219 }
220
221 void WebNotificationManager::didShowNotification(uint64_t notificationID)
222 {
223 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
224     if (!isNotificationIDValid(notificationID))
225         return;
226     
227     RefPtr<Notification> notification = m_notificationIDMap.get(notificationID);
228     if (!notification)
229         return;
230
231     notification->dispatchShowEvent();
232 #else
233     UNUSED_PARAM(notificationID);
234 #endif
235 }
236
237 void WebNotificationManager::didClickNotification(uint64_t notificationID)
238 {
239 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
240     if (!isNotificationIDValid(notificationID))
241         return;
242
243     RefPtr<Notification> notification = m_notificationIDMap.get(notificationID);
244     if (!notification)
245         return;
246
247     // Indicate that this event is being dispatched in reaction to a user's interaction with a platform notification.
248     UserGestureIndicator indicator(DefinitelyProcessingUserGesture);
249     notification->dispatchClickEvent();
250 #else
251     UNUSED_PARAM(notificationID);
252 #endif
253 }
254
255 void WebNotificationManager::didCloseNotifications(const Vector<uint64_t>& notificationIDs)
256 {
257 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
258     size_t count = notificationIDs.size();
259     for (size_t i = 0; i < count; ++i) {
260         uint64_t notificationID = notificationIDs[i];
261         if (!isNotificationIDValid(notificationID))
262             continue;
263
264         RefPtr<Notification> notification = m_notificationIDMap.take(notificationID);
265         if (!notification)
266             continue;
267
268         m_notificationMap.remove(notification);
269         removeNotificationFromContextMap(notificationID, notification.get());
270
271         notification->dispatchCloseEvent();
272     }
273 #else
274     UNUSED_PARAM(notificationIDs);
275 #endif
276 }
277
278 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
279 void WebNotificationManager::removeNotificationFromContextMap(uint64_t notificationID, Notification* notification)
280 {
281     // This is a helper function for managing the hash maps.
282     NotificationContextMap::iterator it = m_notificationContextMap.find(notification->scriptExecutionContext());
283     ASSERT(it != m_notificationContextMap.end());
284     size_t index = it->value.find(notificationID);
285     ASSERT(index != notFound);
286     it->value.remove(index);
287     if (it->value.isEmpty())
288         m_notificationContextMap.remove(it);
289 }
290 #endif
291
292 } // namespace WebKit