[Win] Various DumpRenderTree Fixes.
[WebKit-https.git] / Tools / DumpRenderTree / win / DRTDesktopNotificationPresenter.cpp
1 /*
2  * Copyright (C) 2009 Google Inc. All rights reserved.
3  * Copyright (C) 2014 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 #include "DRTDesktopNotificationPresenter.h"
34
35 #include "DumpRenderTree.h"
36 #include "TestRunner.h"
37 #include <JavaScriptCore/JSStringRef.h>
38 #include <JavaScriptCore/JSStringRefBSTR.h>
39 #include <WebCore/NotificationClient.h>
40 #include <comutil.h>
41
42 DRTDesktopNotificationPresenter::DRTDesktopNotificationPresenter()
43     : m_refCount(1) {} 
44
45 HRESULT DRTDesktopNotificationPresenter::QueryInterface(REFIID riid, void** ppvObject)
46 {
47     *ppvObject = 0;
48     if (IsEqualGUID(riid, IID_IUnknown))
49         *ppvObject = static_cast<DRTDesktopNotificationPresenter*>(this);
50     else if (IsEqualGUID(riid, IID_IWebDesktopNotificationsDelegate))
51         *ppvObject = static_cast<DRTDesktopNotificationPresenter*>(this);
52     else
53         return E_NOINTERFACE;
54
55     AddRef();
56     return S_OK;
57 }
58
59 ULONG DRTDesktopNotificationPresenter::AddRef()
60 {
61     return ++m_refCount;
62 }
63
64 ULONG DRTDesktopNotificationPresenter::Release()
65 {
66     ULONG newRef = --m_refCount;
67     if (!newRef)
68         delete(this);
69
70     return newRef;
71 }
72
73 HRESULT DRTDesktopNotificationPresenter::showDesktopNotification(IWebDesktopNotification* notification)
74 {
75     _bstr_t title, text, url;
76     BOOL html;
77
78     if (!notification->isHTML(&html) && html) {
79         notification->contentsURL(&url.GetBSTR());    
80         printf("DESKTOP NOTIFICATION: contents at %S\n", static_cast<wchar_t*>(url));
81     } else {
82         notification->iconURL(&url.GetBSTR());
83         notification->title(&title.GetBSTR());
84         notification->text(&text.GetBSTR());
85         printf("DESKTOP NOTIFICATION: icon %S, title %S, text %S\n", static_cast<wchar_t*>(url), static_cast<wchar_t*>(title), static_cast<wchar_t*>(text));
86     }
87
88     // In this stub implementation, the notification is displayed immediately;
89     // we dispatch the display event to mimic that.
90     notification->notifyDisplay();
91
92     return S_OK;
93 }
94
95 HRESULT DRTDesktopNotificationPresenter::cancelDesktopNotification(IWebDesktopNotification* notification)
96 {
97     _bstr_t identifier;
98     BOOL html;
99     notification->isHTML(&html);
100     if (html)
101         notification->contentsURL(&identifier.GetBSTR());
102     else
103         notification->title(&identifier.GetBSTR());
104
105     printf("DESKTOP NOTIFICATION CLOSED: %S\n", static_cast<wchar_t*>(identifier));
106     notification->notifyClose(false);
107
108     return S_OK;
109 }
110
111 HRESULT DRTDesktopNotificationPresenter::notificationDestroyed(IWebDesktopNotification* /*notification*/)
112 {
113     // Since in these tests events happen immediately, we don't hold on to
114     // Notification pointers.  So there's no cleanup to do.
115     return S_OK;
116 }
117
118 HRESULT DRTDesktopNotificationPresenter::checkNotificationPermission(BSTR /*origin*/, int* /*result*/)
119 {
120 #if ENABLE(NOTIFICATIONS) || ENABLE(LEGACY_NOTIFICATIONS)
121     JSStringRef jsOrigin = JSStringCreateWithBSTR(origin);
122     bool allowed = ::gTestRunner->checkDesktopNotificationPermission(jsOrigin);
123
124     if (allowed)
125         *result = WebCore::NotificationClient::PermissionAllowed;
126     else
127         *result = WebCore::NotificationClient::PermissionDenied;
128
129     JSStringRelease(jsOrigin);
130 #endif
131     return S_OK;
132 }
133
134 HRESULT DRTDesktopNotificationPresenter::requestNotificationPermission(BSTR origin)
135 {
136     printf("DESKTOP NOTIFICATION PERMISSION REQUESTED: %S\n", origin ? origin : L"");
137     return S_OK;
138 }