dmloader: Simplify the module refcount handling.
[wine] / dlls / mshtml / nsevents.c
1 /*
2  * Copyright 2007-2008 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "config.h"
20
21 #include <stdarg.h>
22
23 #define COBJMACROS
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winuser.h"
28 #include "ole2.h"
29 #include "mshtmcid.h"
30 #include "shlguid.h"
31
32 #include "wine/debug.h"
33 #include "wine/unicode.h"
34
35 #include "mshtml_private.h"
36 #include "htmlevent.h"
37 #include "resource.h"
38
39 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
40
41 typedef struct {
42     nsIDOMEventListener nsIDOMEventListener_iface;
43     nsDocumentEventListener *This;
44 } nsEventListener;
45
46 struct nsDocumentEventListener {
47     nsEventListener blur_listener;
48     nsEventListener focus_listener;
49     nsEventListener keypress_listener;
50     nsEventListener load_listener;
51     nsEventListener htmlevent_listener;
52
53     LONG ref;
54
55     HTMLDocumentNode *doc;
56 };
57
58 static LONG release_listener(nsDocumentEventListener *This)
59 {
60     LONG ref = InterlockedDecrement(&This->ref);
61
62     TRACE("(%p) ref=%d\n", This, ref);
63
64     if(!ref)
65         heap_free(This);
66
67     return ref;
68 }
69
70 static inline nsEventListener *impl_from_nsIDOMEventListener(nsIDOMEventListener *iface)
71 {
72     return CONTAINING_RECORD(iface, nsEventListener, nsIDOMEventListener_iface);
73 }
74
75 static nsresult NSAPI nsDOMEventListener_QueryInterface(nsIDOMEventListener *iface,
76         nsIIDRef riid, void **result)
77 {
78     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
79
80     *result = NULL;
81
82     if(IsEqualGUID(&IID_nsISupports, riid)) {
83         TRACE("(%p)->(IID_nsISupports, %p)\n", This, result);
84         *result = &This->nsIDOMEventListener_iface;
85     }else if(IsEqualGUID(&IID_nsIDOMEventListener, riid)) {
86         TRACE("(%p)->(IID_nsIDOMEventListener %p)\n", This, result);
87         *result = &This->nsIDOMEventListener_iface;
88     }
89
90     if(*result) {
91         nsIWebBrowserChrome_AddRef(&This->nsIDOMEventListener_iface);
92         return NS_OK;
93     }
94
95     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), result);
96     return NS_NOINTERFACE;
97 }
98
99 static nsrefcnt NSAPI nsDOMEventListener_AddRef(nsIDOMEventListener *iface)
100 {
101     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
102     LONG ref = InterlockedIncrement(&This->This->ref);
103
104     TRACE("(%p) ref=%d\n", This->This, ref);
105
106     return ref;
107 }
108
109 static nsrefcnt NSAPI nsDOMEventListener_Release(nsIDOMEventListener *iface)
110 {
111     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
112
113     return release_listener(This->This);
114 }
115
116 static BOOL is_doc_child_focus(NSContainer *nscontainer)
117 {
118     HWND hwnd;
119
120     for(hwnd = GetFocus(); hwnd && hwnd != nscontainer->hwnd; hwnd = GetParent(hwnd));
121
122     return hwnd != NULL;
123 }
124
125 static nsresult NSAPI handle_blur(nsIDOMEventListener *iface, nsIDOMEvent *event)
126 {
127     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
128     HTMLDocumentNode *doc = This->This->doc;
129     HTMLDocumentObj *doc_obj;
130
131     TRACE("(%p)\n", doc);
132
133     if(!doc || !doc->basedoc.doc_obj)
134         return NS_ERROR_FAILURE;
135     doc_obj = doc->basedoc.doc_obj;
136
137     if(doc_obj->focus && !is_doc_child_focus(doc_obj->nscontainer)) {
138         doc_obj->focus = FALSE;
139         notif_focus(doc_obj);
140     }
141
142     return NS_OK;
143 }
144
145 static nsresult NSAPI handle_focus(nsIDOMEventListener *iface, nsIDOMEvent *event)
146 {
147     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
148     HTMLDocumentNode *doc = This->This->doc;
149     HTMLDocumentObj *doc_obj;
150
151     TRACE("(%p)\n", doc);
152
153     if(!doc)
154         return NS_ERROR_FAILURE;
155     doc_obj = doc->basedoc.doc_obj;
156
157     if(!doc_obj->focus) {
158         doc_obj->focus = TRUE;
159         notif_focus(doc_obj);
160     }
161
162     return NS_OK;
163 }
164
165 static nsresult NSAPI handle_keypress(nsIDOMEventListener *iface,
166         nsIDOMEvent *event)
167 {
168     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
169     HTMLDocumentNode *doc = This->This->doc;
170     HTMLDocumentObj *doc_obj;
171
172     if(!doc)
173         return NS_ERROR_FAILURE;
174     doc_obj = doc->basedoc.doc_obj;
175
176     TRACE("(%p)->(%p)\n", doc, event);
177
178     update_doc(&doc_obj->basedoc, UPDATE_UI);
179     if(doc_obj->usermode == EDITMODE)
180         handle_edit_event(&doc_obj->basedoc, event);
181
182     return NS_OK;
183 }
184
185 static void handle_docobj_load(HTMLDocumentObj *doc)
186 {
187     IOleCommandTarget *olecmd = NULL;
188     HRESULT hres;
189
190     if(doc->nscontainer->editor_controller) {
191         nsIController_Release(doc->nscontainer->editor_controller);
192         doc->nscontainer->editor_controller = NULL;
193     }
194
195     if(doc->usermode == EDITMODE)
196         handle_edit_load(&doc->basedoc);
197
198     if(doc->client) {
199         hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
200         if(FAILED(hres))
201             olecmd = NULL;
202     }
203
204     if(doc->download_state) {
205         if(olecmd) {
206             VARIANT progress;
207
208             V_VT(&progress) = VT_I4;
209             V_I4(&progress) = 0;
210             IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS,
211                     OLECMDEXECOPT_DONTPROMPTUSER, &progress, NULL);
212         }
213
214         set_download_state(doc, 0);
215     }
216
217     if(olecmd) {
218         IOleCommandTarget_Exec(olecmd, &CGID_ShellDocView, 103, 0, NULL, NULL);
219         IOleCommandTarget_Exec(olecmd, &CGID_MSHTML, IDM_PARSECOMPLETE, 0, NULL, NULL);
220         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_HTTPEQUIV_DONE, 0, NULL, NULL);
221
222         IOleCommandTarget_Release(olecmd);
223     }
224 }
225
226 static nsresult NSAPI handle_load(nsIDOMEventListener *iface, nsIDOMEvent *event)
227 {
228     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
229     HTMLDocumentNode *doc = This->This->doc;
230     nsIDOMHTMLElement *nsbody = NULL;
231     HTMLDocumentObj *doc_obj = NULL;
232
233     TRACE("(%p)\n", doc);
234
235     if(!doc || !doc->basedoc.window)
236         return NS_ERROR_FAILURE;
237     if(doc->basedoc.doc_obj && doc->basedoc.doc_obj->basedoc.doc_node == doc)
238         doc_obj = doc->basedoc.doc_obj;
239
240     connect_scripts(doc->basedoc.window);
241
242     if(doc_obj)
243         handle_docobj_load(doc_obj);
244
245     set_ready_state(doc->basedoc.window, READYSTATE_COMPLETE);
246
247     if(doc_obj) {
248         if(doc_obj->view_sink)
249             IAdviseSink_OnViewChange(doc_obj->view_sink, DVASPECT_CONTENT, -1);
250
251         set_statustext(doc_obj, IDS_STATUS_DONE, NULL);
252
253         update_title(doc_obj);
254     }
255
256     if(doc_obj && doc_obj->usermode!=EDITMODE && doc_obj->doc_object_service)
257         IDocObjectService_FireDocumentComplete(doc_obj->doc_object_service,
258                 &doc->basedoc.window->IHTMLWindow2_iface, 0);
259
260     if(!doc->nsdoc) {
261         ERR("NULL nsdoc\n");
262         return NS_ERROR_FAILURE;
263     }
264
265     nsIDOMHTMLDocument_GetBody(doc->nsdoc, &nsbody);
266     if(nsbody) {
267         fire_event(doc, EVENTID_LOAD, TRUE, (nsIDOMNode*)nsbody, event);
268         nsIDOMHTMLElement_Release(nsbody);
269     }
270
271     return NS_OK;
272 }
273
274 static nsresult NSAPI handle_htmlevent(nsIDOMEventListener *iface, nsIDOMEvent *event)
275 {
276     nsEventListener *This = impl_from_nsIDOMEventListener(iface);
277     HTMLDocumentNode *doc = This->This->doc;
278     const PRUnichar *type;
279     nsIDOMEventTarget *event_target;
280     nsIDOMNode *nsnode;
281     nsAString type_str;
282     eventid_t eid;
283     nsresult nsres;
284
285     TRACE("\n");
286
287     nsAString_Init(&type_str, NULL);
288     nsIDOMEvent_GetType(event, &type_str);
289     nsAString_GetData(&type_str, &type);
290     eid = str_to_eid(type);
291     nsAString_Finish(&type_str);
292
293     nsres = nsIDOMEvent_GetTarget(event, &event_target);
294     if(NS_FAILED(nsres) || !event_target) {
295         ERR("GetEventTarget failed: %08x\n", nsres);
296         return NS_OK;
297     }
298
299     nsres = nsIDOMEventTarget_QueryInterface(event_target, &IID_nsIDOMNode, (void**)&nsnode);
300     nsIDOMEventTarget_Release(event_target);
301     if(NS_FAILED(nsres)) {
302         ERR("Could not get nsIDOMNode: %08x\n", nsres);
303         return NS_OK;
304     }
305
306     fire_event(doc, eid, TRUE, nsnode, event);
307
308     nsIDOMNode_Release(nsnode);
309
310     return NS_OK;
311 }
312
313 #define EVENTLISTENER_VTBL(handler) \
314     { \
315         nsDOMEventListener_QueryInterface, \
316         nsDOMEventListener_AddRef, \
317         nsDOMEventListener_Release, \
318         handler, \
319     }
320
321 static const nsIDOMEventListenerVtbl blur_vtbl =      EVENTLISTENER_VTBL(handle_blur);
322 static const nsIDOMEventListenerVtbl focus_vtbl =     EVENTLISTENER_VTBL(handle_focus);
323 static const nsIDOMEventListenerVtbl keypress_vtbl =  EVENTLISTENER_VTBL(handle_keypress);
324 static const nsIDOMEventListenerVtbl load_vtbl =      EVENTLISTENER_VTBL(handle_load);
325 static const nsIDOMEventListenerVtbl htmlevent_vtbl = EVENTLISTENER_VTBL(handle_htmlevent);
326
327 static void init_event(nsIDOMEventTarget *target, const PRUnichar *type,
328         nsIDOMEventListener *listener, BOOL capture)
329 {
330     nsAString type_str;
331     nsresult nsres;
332
333     nsAString_InitDepend(&type_str, type);
334     nsres = nsIDOMEventTarget_AddEventListener(target, &type_str, listener, capture);
335     nsAString_Finish(&type_str);
336     if(NS_FAILED(nsres))
337         ERR("AddEventTarget failed: %08x\n", nsres);
338
339 }
340
341 static void init_listener(nsEventListener *This, nsDocumentEventListener *listener,
342         const nsIDOMEventListenerVtbl *vtbl)
343 {
344     This->nsIDOMEventListener_iface.lpVtbl = vtbl;
345     This->This = listener;
346 }
347
348 void add_nsevent_listener(HTMLDocumentNode *doc, nsIDOMNode *nsnode, LPCWSTR type)
349 {
350     nsIDOMEventTarget *target;
351     nsresult nsres;
352
353     if(nsnode)
354         nsres = nsIDOMNode_QueryInterface(nsnode, &IID_nsIDOMEventTarget, (void**)&target);
355     else
356         nsres = nsIDOMWindow_QueryInterface(doc->basedoc.window->nswindow, &IID_nsIDOMEventTarget, (void**)&target);
357     if(NS_FAILED(nsres)) {
358         ERR("Could not get nsIDOMEventTarget interface: %08x\n", nsres);
359         return;
360     }
361
362     init_event(target, type, &doc->nsevent_listener->htmlevent_listener.nsIDOMEventListener_iface,
363             TRUE);
364     nsIDOMEventTarget_Release(target);
365 }
366
367 void release_nsevents(HTMLDocumentNode *doc)
368 {
369     if(doc->nsevent_listener) {
370         doc->nsevent_listener->doc = NULL;
371         release_listener(doc->nsevent_listener);
372         doc->nsevent_listener = NULL;
373     }
374 }
375
376 void init_nsevents(HTMLDocumentNode *doc)
377 {
378     nsDocumentEventListener *listener;
379     nsIDOMEventTarget *target;
380     nsresult nsres;
381
382     static const PRUnichar wsz_blur[]      = {'b','l','u','r',0};
383     static const PRUnichar wsz_focus[]     = {'f','o','c','u','s',0};
384     static const PRUnichar wsz_keypress[]  = {'k','e','y','p','r','e','s','s',0};
385     static const PRUnichar wsz_load[]      = {'l','o','a','d',0};
386
387     listener = heap_alloc(sizeof(nsDocumentEventListener));
388     if(!listener)
389         return;
390
391     listener->ref = 1;
392     listener->doc = doc;
393
394     init_listener(&listener->blur_listener,        listener, &blur_vtbl);
395     init_listener(&listener->focus_listener,       listener, &focus_vtbl);
396     init_listener(&listener->keypress_listener,    listener, &keypress_vtbl);
397     init_listener(&listener->load_listener,        listener, &load_vtbl);
398     init_listener(&listener->htmlevent_listener,   listener, &htmlevent_vtbl);
399
400     doc->nsevent_listener = listener;
401
402     nsres = nsIDOMWindow_QueryInterface(doc->basedoc.window->nswindow, &IID_nsIDOMEventTarget, (void**)&target);
403     if(NS_FAILED(nsres)) {
404         ERR("Could not get nsIDOMEventTarget interface: %08x\n", nsres);
405         return;
406     }
407
408     init_event(target, wsz_blur,     &listener->blur_listener.nsIDOMEventListener_iface,     TRUE);
409     init_event(target, wsz_focus,    &listener->focus_listener.nsIDOMEventListener_iface,    TRUE);
410     init_event(target, wsz_keypress, &listener->keypress_listener.nsIDOMEventListener_iface, FALSE);
411     init_event(target, wsz_load,     &listener->load_listener.nsIDOMEventListener_iface,     TRUE);
412
413     nsIDOMEventTarget_Release(target);
414 }