1 /*

     2  * Copyright (C) 2010 Google 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 are

     6  * met:

     7  *

     8  *     * Redistributions of source code must retain the above copyright

     9  * notice, this list of conditions and the following disclaimer.

    10  *     * Redistributions in binary form must reproduce the above

    11  * copyright notice, this list of conditions and the following disclaimer

    12  * in the documentation and/or other materials provided with the

    13  * distribution.

    14  *     * Neither the name of Google Inc. nor the names of its

    15  * contributors may be used to endorse or promote products derived from

    16  * this software without specific prior written permission.

    17  *

    18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

    19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

    20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

    21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT

    22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

    23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

    24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

    25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

    26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

    29  */

    30 

    31 #ifndef NotificationPresenter_h

    32 #define NotificationPresenter_h

    33 

    34 #include "public/WebNotificationPresenter.h"

    35 #include <wtf/HashMap.h>

    36 #include <wtf/HashSet.h>

    37 #include <wtf/text/StringHash.h>

    38 #include <wtf/text/WTFString.h>

    39 

    40 class TestShell;

    41 

    42 // A class that implements WebNotificationPresenter for DRT.

    43 class NotificationPresenter : public WebKit::WebNotificationPresenter {

    44 public:

    45     explicit NotificationPresenter(TestShell* shell) : m_shell(shell) {}

    46 

    47     // Called by the LayoutTestController to simulate a user granting permission.

    48     void grantPermission(const WebKit::WebString& origin);

    49 

    50     // WebKit::WebNotificationPresenter interface

    51     virtual bool show(const WebKit::WebNotification&);

    52     virtual void cancel(const WebKit::WebNotification&);

    53     virtual void objectDestroyed(const WebKit::WebNotification&);

    54     virtual Permission checkPermission(const WebKit::WebURL&);

    55     virtual void requestPermission(const WebKit::WebSecurityOrigin&, WebKit::WebNotificationPermissionCallback*);

    56 

    57     void reset() { m_allowedOrigins.clear(); }

    58 

    59 private:

    60     // Non-owned pointer. The NotificationPresenter is owned by the test shell.

    61     TestShell* m_shell;

    62 

    63     // Set of allowed origins.

    64     HashSet<WebCore::String> m_allowedOrigins;

    65 

    66     // Map of active replacement IDs to the titles of those notifications

    67     HashMap<WebCore::String, WebCore::String> m_replacements;

    68 };

    69 

    70 #endif // NotificationPresenter_h