Replace WTF::move with WTFMove
[WebKit-https.git] / Source / WebCore / Modules / notifications / NotificationCenter.cpp
1 /*
2  * Copyright (C) 2009 Google Inc. All rights reserved.
3  * Copyright (C) 2012, 2015 Apple Inc. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are
7  * met:
8  *
9  *     * Redistributions of source code must retain the above copyright
10  * notice, this list of conditions and the following disclaimer.
11  *     * Redistributions in binary form must reproduce the above
12  * copyright notice, this list of conditions and the following disclaimer
13  * in the documentation and/or other materials provided with the
14  * distribution.
15  *     * Neither the name of Google Inc. nor the names of its
16  * contributors may be used to endorse or promote products derived from
17  * this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30  */
31
32 #include "config.h"
33
34 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
35
36 #include "NotificationCenter.h"
37
38 #include "Notification.h"
39 #include "ScriptExecutionContext.h"
40 #include "SecurityOrigin.h"
41
42 namespace WebCore {
43
44 Ref<NotificationCenter> NotificationCenter::create(ScriptExecutionContext* context, NotificationClient* client)
45 {
46     auto notificationCenter = adoptRef(*new NotificationCenter(context, client));
47     notificationCenter->suspendIfNeeded();
48     return notificationCenter;
49 }
50
51 NotificationCenter::NotificationCenter(ScriptExecutionContext* context, NotificationClient* client)
52     : ActiveDOMObject(context)
53     , m_client(client)
54     , m_timer([this]() { timerFired(); })
55 {
56 }
57
58 #if ENABLE(LEGACY_NOTIFICATIONS)
59
60 RefPtr<Notification> NotificationCenter::createNotification(const String& iconURI, const String& title, const String& body, ExceptionCode& ec)
61 {
62     if (!m_client) {
63         ec = INVALID_STATE_ERR;
64         return nullptr;
65     }
66     return Notification::create(title, body, iconURI, scriptExecutionContext(), ec, this);
67 }
68
69 int NotificationCenter::checkPermission()
70 {
71     if (!m_client || !scriptExecutionContext())
72         return NotificationClient::PermissionDenied;
73
74     switch (scriptExecutionContext()->securityOrigin()->canShowNotifications()) {
75     case SecurityOrigin::AlwaysAllow:
76         return NotificationClient::PermissionAllowed;
77     case SecurityOrigin::AlwaysDeny:
78         return NotificationClient::PermissionDenied;
79     case SecurityOrigin::Ask:
80         return m_client->checkPermission(scriptExecutionContext());
81     }
82
83     ASSERT_NOT_REACHED();
84     return m_client->checkPermission(scriptExecutionContext());
85 }
86
87 void NotificationCenter::requestPermission(const RefPtr<VoidCallback>& callback)
88 {
89     if (!m_client || !scriptExecutionContext())
90         return;
91
92     switch (scriptExecutionContext()->securityOrigin()->canShowNotifications()) {
93     case SecurityOrigin::AlwaysAllow:
94     case SecurityOrigin::AlwaysDeny:
95         if (m_callbacks.isEmpty()) {
96             ref(); // Balanced by the derefs in NotificationCenter::stop and NotificationCenter::timerFired.
97             m_timer.startOneShot(0);
98         }
99         m_callbacks.append([callback]() {
100             if (callback)
101                 callback->handleEvent();
102         });
103         return;
104     case SecurityOrigin::Ask:
105         m_client->requestPermission(scriptExecutionContext(), callback.get());
106         return;
107     }
108
109     ASSERT_NOT_REACHED();
110     m_client->requestPermission(scriptExecutionContext(), callback.get());
111 }
112
113 #endif
114
115 void NotificationCenter::stop()
116 {
117     if (!m_client)
118         return;
119
120     // Clear m_client immediately to guarantee reentrant calls to NotificationCenter do nothing.
121     // Also protect |this| so it's not indirectly destroyed under us by work done by the client.
122     auto& client = *std::exchange(m_client, nullptr);
123     Ref<NotificationCenter> protector(*this);
124
125     if (!m_callbacks.isEmpty())
126         deref(); // Balanced by the ref in NotificationCenter::requestPermission.
127
128     m_timer.stop();
129     m_callbacks.clear();
130
131     client.cancelRequestsForPermission(scriptExecutionContext());
132     client.clearNotifications(scriptExecutionContext());
133 }
134
135 const char* NotificationCenter::activeDOMObjectName() const
136 {
137     return "NotificationCenter";
138 }
139
140 bool NotificationCenter::canSuspendForDocumentSuspension() const
141 {
142     // We don't need to worry about Notifications because those are ActiveDOMObject too.
143     // The NotificationCenter can safely be suspended if there are no pending permission requests.
144     return m_callbacks.isEmpty() && (!m_client || !m_client->hasPendingPermissionRequests(scriptExecutionContext()));
145 }
146
147 void NotificationCenter::timerFired()
148 {
149     ASSERT(m_client);
150     auto callbacks = WTFMove(m_callbacks);
151     for (auto& callback : callbacks)
152         callback();
153     deref(); // Balanced by the ref in NotificationCenter::requestPermission.
154 }
155
156 } // namespace WebCore
157
158 #endif // ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)