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#ifndef NotificationPermissionRequestManager_h 27#define NotificationPermissionRequestManager_h 28 29#include <WebCore/NotificationClient.h> 30#include <WebCore/NotificationPermissionCallback.h> 31#include <WebCore/SecurityOriginHash.h> 32#include <WebCore/VoidCallback.h> 33#include <wtf/HashMap.h> 34#include <wtf/RefCounted.h> 35#include <wtf/RefPtr.h> 36#include <wtf/text/WTFString.h> 37 38namespace WebCore { 39class Notification; 40class SecurityOrigin; 41} 42 43namespace WebKit { 44 45class WebPage; 46 47/// FIXME: Need to keep a queue of pending notifications which permission is still being requested. 48class NotificationPermissionRequestManager : public RefCounted<NotificationPermissionRequestManager> { 49public: 50 static PassRefPtr<NotificationPermissionRequestManager> create(WebPage*); 51 52#if ENABLE(NOTIFICATIONS) 53 void startRequest(WebCore::SecurityOrigin*, PassRefPtr<WebCore::NotificationPermissionCallback>); 54#endif 55#if ENABLE(LEGACY_NOTIFICATIONS) 56 void startRequest(WebCore::SecurityOrigin*, PassRefPtr<WebCore::VoidCallback>); 57#endif 58 void cancelRequest(WebCore::SecurityOrigin*); 59 60 WebCore::NotificationClient::Permission permissionLevel(WebCore::SecurityOrigin*); 61 62 // For testing purposes only. 63 void setPermissionLevelForTesting(const String& originString, bool allowed); 64 void removeAllPermissionsForTesting(); 65 66 void didReceiveNotificationPermissionDecision(uint64_t notificationID, bool allowed); 67 68private: 69 NotificationPermissionRequestManager(WebPage*); 70 71#if ENABLE(NOTIFICATIONS) 72 HashMap<uint64_t, RefPtr<WebCore::NotificationPermissionCallback>> m_idToCallbackMap; 73#endif 74#if ENABLE(LEGACY_NOTIFICATIONS) 75 HashMap<uint64_t, RefPtr<WebCore::VoidCallback>> m_idToVoidCallbackMap; 76#endif 77 HashMap<RefPtr<WebCore::SecurityOrigin>, uint64_t> m_originToIDMap; 78 HashMap<uint64_t, RefPtr<WebCore::SecurityOrigin>> m_idToOriginMap; 79 80#if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS) 81 WebPage* m_page; 82#endif 83}; 84 85inline bool isRequestIDValid(uint64_t id) 86{ 87 // This check makes sure that the ID is not equal to values needed by 88 // HashMap for bucketing. 89 return id && id != static_cast<uint64_t>(-1); 90} 91 92} // namespace WebKit 93 94#endif // NotificationPermissionRequestManager_h 95