mshtml: Added IHTMLScriptElement::put_text implementation.
[wine] / dlls / mshtml / htmlscript.c
1 /*
2  * Copyright 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 <stdarg.h>
20 #include <assert.h>
21
22 #define COBJMACROS
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winuser.h"
27 #include "ole2.h"
28
29 #include "wine/debug.h"
30
31 #include "mshtml_private.h"
32 #include "htmlscript.h"
33
34 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
35
36 static inline HTMLScriptElement *impl_from_IHTMLScriptElement(IHTMLScriptElement *iface)
37 {
38     return CONTAINING_RECORD(iface, HTMLScriptElement, IHTMLScriptElement_iface);
39 }
40
41 static HRESULT WINAPI HTMLScriptElement_QueryInterface(IHTMLScriptElement *iface,
42         REFIID riid, void **ppv)
43 {
44     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
45
46     return IHTMLDOMNode_QueryInterface(&This->element.node.IHTMLDOMNode_iface, riid, ppv);
47 }
48
49 static ULONG WINAPI HTMLScriptElement_AddRef(IHTMLScriptElement *iface)
50 {
51     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
52
53     return IHTMLDOMNode_AddRef(&This->element.node.IHTMLDOMNode_iface);
54 }
55
56 static ULONG WINAPI HTMLScriptElement_Release(IHTMLScriptElement *iface)
57 {
58     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
59
60     return IHTMLDOMNode_Release(&This->element.node.IHTMLDOMNode_iface);
61 }
62
63 static HRESULT WINAPI HTMLScriptElement_GetTypeInfoCount(IHTMLScriptElement *iface, UINT *pctinfo)
64 {
65     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
66     return IDispatchEx_GetTypeInfoCount(&This->element.node.dispex.IDispatchEx_iface, pctinfo);
67 }
68
69 static HRESULT WINAPI HTMLScriptElement_GetTypeInfo(IHTMLScriptElement *iface, UINT iTInfo,
70                                               LCID lcid, ITypeInfo **ppTInfo)
71 {
72     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
73     return IDispatchEx_GetTypeInfo(&This->element.node.dispex.IDispatchEx_iface, iTInfo, lcid,
74             ppTInfo);
75 }
76
77 static HRESULT WINAPI HTMLScriptElement_GetIDsOfNames(IHTMLScriptElement *iface, REFIID riid,
78                                                 LPOLESTR *rgszNames, UINT cNames,
79                                                 LCID lcid, DISPID *rgDispId)
80 {
81     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
82     return IDispatchEx_GetIDsOfNames(&This->element.node.dispex.IDispatchEx_iface, riid, rgszNames,
83             cNames, lcid, rgDispId);
84 }
85
86 static HRESULT WINAPI HTMLScriptElement_Invoke(IHTMLScriptElement *iface, DISPID dispIdMember,
87                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
88                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
89 {
90     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
91     return IDispatchEx_Invoke(&This->element.node.dispex.IDispatchEx_iface, dispIdMember, riid,
92             lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
93 }
94
95 static HRESULT WINAPI HTMLScriptElement_put_src(IHTMLScriptElement *iface, BSTR v)
96 {
97     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
98     HTMLInnerWindow *window;
99     nsIDOMNode *parent;
100     nsAString src_str;
101     nsresult nsres;
102
103     TRACE("(%p)->(%s)\n", This, debugstr_w(v));
104
105     if(!This->element.node.doc || !This->element.node.doc->window) {
106         WARN("no windoow\n");
107         return E_UNEXPECTED;
108     }
109
110     window = This->element.node.doc->window;
111
112     nsAString_InitDepend(&src_str, v);
113     nsres = nsIDOMHTMLScriptElement_SetSrc(This->nsscript, &src_str);
114     nsAString_Finish(&src_str);
115     if(NS_FAILED(nsres)) {
116         ERR("SetSrc failed: %08x\n", nsres);
117         return E_FAIL;
118     }
119
120     if(This->parsed) {
121         WARN("already parsed\n");
122         return S_OK;
123     }
124
125     if(window->parser_callback_cnt) {
126         script_queue_entry_t *queue;
127
128         queue = heap_alloc(sizeof(*queue));
129         if(!queue)
130             return E_OUTOFMEMORY;
131
132         IHTMLScriptElement_AddRef(&This->IHTMLScriptElement_iface);
133         queue->script = This;
134
135         list_add_tail(&window->script_queue, &queue->entry);
136         return S_OK;
137     }
138
139     nsres = nsIDOMHTMLScriptElement_GetParentNode(This->nsscript, &parent);
140     if(NS_FAILED(nsres) || !parent) {
141         TRACE("No parent, not executing\n");
142         This->parse_on_bind = TRUE;
143         return S_OK;
144     }
145
146     nsIDOMNode_Release(parent);
147     doc_insert_script(window, This);
148     return S_OK;
149 }
150
151 static HRESULT WINAPI HTMLScriptElement_get_src(IHTMLScriptElement *iface, BSTR *p)
152 {
153     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
154     nsAString src_str;
155     nsresult nsres;
156
157     TRACE("(%p)->(%p)\n", This, p);
158
159     nsAString_Init(&src_str, NULL);
160     nsres = nsIDOMHTMLScriptElement_GetSrc(This->nsscript, &src_str);
161     return return_nsstr(nsres, &src_str, p);
162 }
163
164 static HRESULT WINAPI HTMLScriptElement_put_htmlFor(IHTMLScriptElement *iface, BSTR v)
165 {
166     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
167     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
168     return E_NOTIMPL;
169 }
170
171 static HRESULT WINAPI HTMLScriptElement_get_htmlFor(IHTMLScriptElement *iface, BSTR *p)
172 {
173     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
174     FIXME("(%p)->(%p)\n", This, p);
175     return E_NOTIMPL;
176 }
177
178 static HRESULT WINAPI HTMLScriptElement_put_event(IHTMLScriptElement *iface, BSTR v)
179 {
180     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
181     FIXME("(%p)->(%s)\n", This, debugstr_w(v));
182     return E_NOTIMPL;
183 }
184
185 static HRESULT WINAPI HTMLScriptElement_get_event(IHTMLScriptElement *iface, BSTR *p)
186 {
187     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
188     FIXME("(%p)->(%p)\n", This, p);
189     return E_NOTIMPL;
190 }
191
192 static HRESULT WINAPI HTMLScriptElement_put_text(IHTMLScriptElement *iface, BSTR v)
193 {
194     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
195     HTMLInnerWindow *window;
196     nsIDOMNode *parent;
197     nsAString text_str;
198     nsresult nsres;
199
200     TRACE("(%p)->(%s)\n", This, debugstr_w(v));
201
202     if(!This->element.node.doc || !This->element.node.doc->window) {
203         WARN("no windoow\n");
204         return E_UNEXPECTED;
205     }
206
207     window = This->element.node.doc->window;
208
209     nsAString_InitDepend(&text_str, v);
210     nsres = nsIDOMHTMLScriptElement_SetText(This->nsscript, &text_str);
211     nsAString_Finish(&text_str);
212     if(NS_FAILED(nsres)) {
213         ERR("SetSrc failed: %08x\n", nsres);
214         return E_FAIL;
215     }
216
217     nsres = nsIDOMHTMLScriptElement_GetParentNode(This->nsscript, &parent);
218     if(NS_FAILED(nsres) || !parent) {
219         TRACE("No parent, not executing\n");
220         This->parse_on_bind = TRUE;
221         return S_OK;
222     }
223
224     nsIDOMNode_Release(parent);
225     doc_insert_script(window, This);
226     return S_OK;
227 }
228
229 static HRESULT WINAPI HTMLScriptElement_get_text(IHTMLScriptElement *iface, BSTR *p)
230 {
231     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
232     nsAString nsstr;
233     nsresult nsres;
234
235     TRACE("(%p)->(%p)\n", This, p);
236
237     nsAString_Init(&nsstr, NULL);
238     nsres = nsIDOMHTMLScriptElement_GetText(This->nsscript, &nsstr);
239     return return_nsstr(nsres, &nsstr, p);
240 }
241
242 static HRESULT WINAPI HTMLScriptElement_put_defer(IHTMLScriptElement *iface, VARIANT_BOOL v)
243 {
244     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
245     HRESULT hr = S_OK;
246     nsresult nsres;
247
248     TRACE("(%p)->(%x)\n", This, v);
249
250     nsres = nsIDOMHTMLScriptElement_SetDefer(This->nsscript, v != VARIANT_FALSE);
251     if(NS_FAILED(nsres))
252     {
253         hr = E_FAIL;
254     }
255
256     return hr;
257 }
258
259 static HRESULT WINAPI HTMLScriptElement_get_defer(IHTMLScriptElement *iface, VARIANT_BOOL *p)
260 {
261     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
262     cpp_bool defer = FALSE;
263     nsresult nsres;
264
265     TRACE("(%p)->(%p)\n", This, p);
266
267     if(!p)
268         return E_INVALIDARG;
269
270     nsres = nsIDOMHTMLScriptElement_GetDefer(This->nsscript, &defer);
271     if(NS_FAILED(nsres)) {
272         ERR("GetSrc failed: %08x\n", nsres);
273     }
274
275     *p = defer ? VARIANT_TRUE : VARIANT_FALSE;
276
277     TRACE("*p = %d\n", *p);
278     return S_OK;
279 }
280
281 static HRESULT WINAPI HTMLScriptElement_get_readyState(IHTMLScriptElement *iface, BSTR *p)
282 {
283     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
284     FIXME("(%p)->(%p)\n", This, p);
285     return E_NOTIMPL;
286 }
287
288 static HRESULT WINAPI HTMLScriptElement_put_onerror(IHTMLScriptElement *iface, VARIANT v)
289 {
290     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
291     FIXME("(%p)->(%s)\n", This, debugstr_variant(&v));
292     return E_NOTIMPL;
293 }
294
295 static HRESULT WINAPI HTMLScriptElement_get_onerror(IHTMLScriptElement *iface, VARIANT *p)
296 {
297     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
298     FIXME("(%p)->(%p)\n", This, p);
299     return E_NOTIMPL;
300 }
301
302 static HRESULT WINAPI HTMLScriptElement_put_type(IHTMLScriptElement *iface, BSTR v)
303 {
304     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
305     nsAString nstype_str;
306     nsresult nsres;
307
308     TRACE("(%p)->(%s)\n", This, debugstr_w(v));
309
310     nsAString_Init(&nstype_str, v);
311     nsres = nsIDOMHTMLScriptElement_SetType(This->nsscript, &nstype_str);
312     if (NS_FAILED(nsres))
313         ERR("SetType failed: %08x\n", nsres);
314     nsAString_Finish (&nstype_str);
315
316     return S_OK;
317 }
318
319 static HRESULT WINAPI HTMLScriptElement_get_type(IHTMLScriptElement *iface, BSTR *p)
320 {
321     HTMLScriptElement *This = impl_from_IHTMLScriptElement(iface);
322     nsAString nstype_str;
323     nsresult nsres;
324
325     TRACE("(%p)->(%p)\n", This, p);
326
327     nsAString_Init(&nstype_str, NULL);
328     nsres = nsIDOMHTMLScriptElement_GetType(This->nsscript, &nstype_str);
329     return return_nsstr(nsres, &nstype_str, p);
330 }
331
332 static const IHTMLScriptElementVtbl HTMLScriptElementVtbl = {
333     HTMLScriptElement_QueryInterface,
334     HTMLScriptElement_AddRef,
335     HTMLScriptElement_Release,
336     HTMLScriptElement_GetTypeInfoCount,
337     HTMLScriptElement_GetTypeInfo,
338     HTMLScriptElement_GetIDsOfNames,
339     HTMLScriptElement_Invoke,
340     HTMLScriptElement_put_src,
341     HTMLScriptElement_get_src,
342     HTMLScriptElement_put_htmlFor,
343     HTMLScriptElement_get_htmlFor,
344     HTMLScriptElement_put_event,
345     HTMLScriptElement_get_event,
346     HTMLScriptElement_put_text,
347     HTMLScriptElement_get_text,
348     HTMLScriptElement_put_defer,
349     HTMLScriptElement_get_defer,
350     HTMLScriptElement_get_readyState,
351     HTMLScriptElement_put_onerror,
352     HTMLScriptElement_get_onerror,
353     HTMLScriptElement_put_type,
354     HTMLScriptElement_get_type
355 };
356
357 static inline HTMLScriptElement *impl_from_HTMLDOMNode(HTMLDOMNode *iface)
358 {
359     return CONTAINING_RECORD(iface, HTMLScriptElement, element.node);
360 }
361
362 static HRESULT HTMLScriptElement_QI(HTMLDOMNode *iface, REFIID riid, void **ppv)
363 {
364     HTMLScriptElement *This = impl_from_HTMLDOMNode(iface);
365
366     *ppv = NULL;
367
368     if(IsEqualGUID(&IID_IUnknown, riid)) {
369         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
370         *ppv = &This->IHTMLScriptElement_iface;
371     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
372         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
373         *ppv = &This->IHTMLScriptElement_iface;
374     }else if(IsEqualGUID(&IID_IHTMLScriptElement, riid)) {
375         TRACE("(%p)->(IID_IHTMLScriptElement %p)\n", This, ppv);
376         *ppv = &This->IHTMLScriptElement_iface;
377     }
378
379     if(*ppv) {
380         IUnknown_AddRef((IUnknown*)*ppv);
381         return S_OK;
382     }
383
384     return HTMLElement_QI(&This->element.node, riid, ppv);
385 }
386
387 static HRESULT HTMLScriptElement_get_readystate(HTMLDOMNode *iface, BSTR *p)
388 {
389     HTMLScriptElement *This = impl_from_HTMLDOMNode(iface);
390
391     return IHTMLScriptElement_get_readyState(&This->IHTMLScriptElement_iface, p);
392 }
393
394 static const NodeImplVtbl HTMLScriptElementImplVtbl = {
395     HTMLScriptElement_QI,
396     HTMLElement_destructor,
397     HTMLElement_clone,
398     HTMLElement_handle_event,
399     HTMLElement_get_attr_col,
400     NULL,
401     NULL,
402     NULL,
403     NULL,
404     NULL,
405     HTMLScriptElement_get_readystate
406 };
407
408 HRESULT script_elem_from_nsscript(HTMLDocumentNode *doc, nsIDOMHTMLScriptElement *nsscript, HTMLScriptElement **ret)
409 {
410     HTMLDOMNode *node;
411     HRESULT hres;
412
413     hres = get_node(doc, (nsIDOMNode*)nsscript, TRUE, &node);
414     if(FAILED(hres))
415         return hres;
416
417     assert(node->vtbl == &HTMLScriptElementImplVtbl);
418     *ret = impl_from_HTMLDOMNode(node);
419     return S_OK;
420 }
421
422 static const tid_t HTMLScriptElement_iface_tids[] = {
423     HTMLELEMENT_TIDS,
424     IHTMLScriptElement_tid,
425     0
426 };
427
428 static dispex_static_data_t HTMLScriptElement_dispex = {
429     NULL,
430     DispHTMLScriptElement_tid,
431     NULL,
432     HTMLScriptElement_iface_tids
433 };
434
435 HRESULT HTMLScriptElement_Create(HTMLDocumentNode *doc, nsIDOMHTMLElement *nselem, HTMLElement **elem)
436 {
437     HTMLScriptElement *ret;
438     nsresult nsres;
439
440     ret = heap_alloc_zero(sizeof(HTMLScriptElement));
441     if(!ret)
442         return E_OUTOFMEMORY;
443
444     ret->IHTMLScriptElement_iface.lpVtbl = &HTMLScriptElementVtbl;
445     ret->element.node.vtbl = &HTMLScriptElementImplVtbl;
446
447     HTMLElement_Init(&ret->element, doc, nselem, &HTMLScriptElement_dispex);
448
449     nsres = nsIDOMHTMLElement_QueryInterface(nselem, &IID_nsIDOMHTMLScriptElement, (void**)&ret->nsscript);
450
451     /* Share nsscript reference with nsnode */
452     assert(nsres == NS_OK && (nsIDOMNode*)ret->nsscript == ret->element.node.nsnode);
453     nsIDOMNode_Release(ret->element.node.nsnode);
454
455     *elem = &ret->element;
456     return S_OK;
457 }