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