2 * Copyright 2006-2008 Jacek Caban for CodeWeavers
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.
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.
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
28 #include "wine/debug.h"
30 #include "mshtml_private.h"
32 WINE_DEFAULT_DEBUG_CHANNEL(mshtml
);
36 const IHTMLElementCollectionVtbl
*lpHTMLElementCollectionVtbl
;
43 } HTMLElementCollection
;
45 #define HTMLELEMCOL(x) ((IHTMLElementCollection*) &(x)->lpHTMLElementCollectionVtbl)
53 static IHTMLElementCollection
*HTMLElementCollection_Create(IUnknown
*ref_unk
,
54 HTMLElement
**elems
, DWORD len
);
56 static void elem_vector_add(elem_vector_t
*buf
, HTMLElement
*elem
)
58 if(buf
->len
== buf
->size
) {
60 buf
->buf
= heap_realloc(buf
->buf
, buf
->size
*sizeof(HTMLElement
**));
63 buf
->buf
[buf
->len
++] = elem
;
66 static void elem_vector_normalize(elem_vector_t
*buf
)
71 }else if(buf
->size
> buf
->len
) {
72 buf
->buf
= heap_realloc(buf
->buf
, buf
->len
*sizeof(HTMLElement
**));
78 static inline BOOL
is_elem_node(nsIDOMNode
*node
)
82 nsIDOMNode_GetNodeType(node
, &type
);
84 return type
== ELEMENT_NODE
|| type
== COMMENT_NODE
;
87 #define ELEMCOL_THIS(iface) DEFINE_THIS(HTMLElementCollection, HTMLElementCollection, iface)
88 #define HTMLELEM_NODE_THIS(iface) DEFINE_THIS2(HTMLElement, node, iface)
90 static HRESULT WINAPI
HTMLElementCollection_QueryInterface(IHTMLElementCollection
*iface
,
91 REFIID riid
, void **ppv
)
93 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
97 if(IsEqualGUID(&IID_IUnknown
, riid
)) {
98 TRACE("(%p)->(IID_IUnknown %p)\n", This
, ppv
);
99 *ppv
= HTMLELEMCOL(This
);
100 }else if(IsEqualGUID(&IID_IHTMLElementCollection
, riid
)) {
101 TRACE("(%p)->(IID_IHTMLElementCollection %p)\n", This
, ppv
);
102 *ppv
= HTMLELEMCOL(This
);
103 }else if(dispex_query_interface(&This
->dispex
, riid
, ppv
)) {
104 return *ppv
? S_OK
: E_NOINTERFACE
;
108 IHTMLElementCollection_AddRef(HTMLELEMCOL(This
));
112 FIXME("(%p)->(%s %p)\n", This
, debugstr_guid(riid
), ppv
);
113 return E_NOINTERFACE
;
116 static ULONG WINAPI
HTMLElementCollection_AddRef(IHTMLElementCollection
*iface
)
118 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
119 LONG ref
= InterlockedIncrement(&This
->ref
);
121 TRACE("(%p) ref=%d\n", This
, ref
);
126 static ULONG WINAPI
HTMLElementCollection_Release(IHTMLElementCollection
*iface
)
128 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
129 LONG ref
= InterlockedDecrement(&This
->ref
);
131 TRACE("(%p) ref=%d\n", This
, ref
);
134 IUnknown_Release(This
->ref_unk
);
135 heap_free(This
->elems
);
142 static HRESULT WINAPI
HTMLElementCollection_GetTypeInfoCount(IHTMLElementCollection
*iface
,
145 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
146 return IDispatchEx_GetTypeInfoCount(DISPATCHEX(&This
->dispex
), pctinfo
);
149 static HRESULT WINAPI
HTMLElementCollection_GetTypeInfo(IHTMLElementCollection
*iface
,
150 UINT iTInfo
, LCID lcid
, ITypeInfo
**ppTInfo
)
152 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
153 return IDispatchEx_GetTypeInfo(DISPATCHEX(&This
->dispex
), iTInfo
, lcid
, ppTInfo
);
156 static HRESULT WINAPI
HTMLElementCollection_GetIDsOfNames(IHTMLElementCollection
*iface
,
157 REFIID riid
, LPOLESTR
*rgszNames
, UINT cNames
, LCID lcid
, DISPID
*rgDispId
)
159 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
160 return IDispatchEx_GetIDsOfNames(DISPATCHEX(&This
->dispex
), riid
, rgszNames
, cNames
, lcid
, rgDispId
);
163 static HRESULT WINAPI
HTMLElementCollection_Invoke(IHTMLElementCollection
*iface
,
164 DISPID dispIdMember
, REFIID riid
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pDispParams
,
165 VARIANT
*pVarResult
, EXCEPINFO
*pExcepInfo
, UINT
*puArgErr
)
167 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
168 return IDispatchEx_Invoke(DISPATCHEX(&This
->dispex
), dispIdMember
, riid
, lcid
,
169 wFlags
, pDispParams
, pVarResult
, pExcepInfo
, puArgErr
);
172 static HRESULT WINAPI
HTMLElementCollection_toString(IHTMLElementCollection
*iface
,
175 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
176 FIXME("(%p)->(%p)\n", This
, String
);
180 static HRESULT WINAPI
HTMLElementCollection_put_length(IHTMLElementCollection
*iface
,
183 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
184 FIXME("(%p)->(%ld)\n", This
, v
);
188 static HRESULT WINAPI
HTMLElementCollection_get_length(IHTMLElementCollection
*iface
,
191 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
193 TRACE("(%p)->(%p)\n", This
, p
);
199 static HRESULT WINAPI
HTMLElementCollection_get__newEnum(IHTMLElementCollection
*iface
,
202 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
203 FIXME("(%p)->(%p)\n", This
, p
);
207 static BOOL
is_elem_name(HTMLElement
*elem
, LPCWSTR name
)
209 const PRUnichar
*str
;
210 nsAString nsstr
, nsname
;
214 static const PRUnichar nameW
[] = {'n','a','m','e',0};
219 nsAString_Init(&nsstr
, NULL
);
220 nsIDOMHTMLElement_GetId(elem
->nselem
, &nsstr
);
221 nsAString_GetData(&nsstr
, &str
);
222 if(!strcmpiW(str
, name
)) {
223 nsAString_Finish(&nsstr
);
227 nsAString_Init(&nsname
, nameW
);
228 nsres
= nsIDOMHTMLElement_GetAttribute(elem
->nselem
, &nsname
, &nsstr
);
229 nsAString_Finish(&nsname
);
230 if(NS_SUCCEEDED(nsres
)) {
231 nsAString_GetData(&nsstr
, &str
);
232 ret
= !strcmpiW(str
, name
);
235 nsAString_Finish(&nsstr
);
239 static HRESULT WINAPI
HTMLElementCollection_item(IHTMLElementCollection
*iface
,
240 VARIANT name
, VARIANT index
, IDispatch
**pdisp
)
242 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
244 TRACE("(%p)->(v(%d) v(%d) %p)\n", This
, V_VT(&name
), V_VT(&index
), pdisp
);
248 if(V_VT(&name
) == VT_I4
) {
249 TRACE("name is VT_I4: %d\n", V_I4(&name
));
253 if(V_I4(&name
) >= This
->len
)
256 *pdisp
= (IDispatch
*)This
->elems
[V_I4(&name
)];
257 IDispatch_AddRef(*pdisp
);
258 TRACE("Returning pdisp=%p\n", pdisp
);
262 if(V_VT(&name
) == VT_BSTR
) {
265 TRACE("name is VT_BSTR: %s\n", debugstr_w(V_BSTR(&name
)));
267 if(V_VT(&index
) == VT_I4
) {
268 LONG idx
= V_I4(&index
);
270 TRACE("index = %d\n", idx
);
275 for(i
=0; i
<This
->len
; i
++) {
276 if(is_elem_name(This
->elems
[i
], V_BSTR(&name
)) && !idx
--)
281 *pdisp
= (IDispatch
*)HTMLELEM(This
->elems
[i
]);
282 IDispatch_AddRef(*pdisp
);
287 elem_vector_t buf
= {NULL
, 0, 8};
289 buf
.buf
= heap_alloc(buf
.size
*sizeof(HTMLElement
*));
291 for(i
=0; i
<This
->len
; i
++) {
292 if(is_elem_name(This
->elems
[i
], V_BSTR(&name
)))
293 elem_vector_add(&buf
, This
->elems
[i
]);
297 elem_vector_normalize(&buf
);
298 *pdisp
= (IDispatch
*)HTMLElementCollection_Create(This
->ref_unk
, buf
.buf
, buf
.len
);
301 *pdisp
= (IDispatch
*)HTMLELEM(buf
.buf
[0]);
302 IDispatch_AddRef(*pdisp
);
312 FIXME("unsupported arguments\n");
316 static HRESULT WINAPI
HTMLElementCollection_tags(IHTMLElementCollection
*iface
,
317 VARIANT tagName
, IDispatch
**pdisp
)
319 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
322 const PRUnichar
*tag
;
323 elem_vector_t buf
= {NULL
, 0, 8};
325 if(V_VT(&tagName
) != VT_BSTR
) {
326 WARN("Invalid arg\n");
327 return DISP_E_MEMBERNOTFOUND
;
330 TRACE("(%p)->(%s %p)\n", This
, debugstr_w(V_BSTR(&tagName
)), pdisp
);
332 buf
.buf
= heap_alloc(buf
.size
*sizeof(HTMLElement
*));
334 nsAString_Init(&tag_str
, NULL
);
336 for(i
=0; i
<This
->len
; i
++) {
337 if(!This
->elems
[i
]->nselem
)
340 nsIDOMElement_GetTagName(This
->elems
[i
]->nselem
, &tag_str
);
341 nsAString_GetData(&tag_str
, &tag
);
343 if(CompareStringW(LOCALE_SYSTEM_DEFAULT
, NORM_IGNORECASE
, tag
, -1,
344 V_BSTR(&tagName
), -1) == CSTR_EQUAL
)
345 elem_vector_add(&buf
, This
->elems
[i
]);
348 nsAString_Finish(&tag_str
);
349 elem_vector_normalize(&buf
);
351 TRACE("fount %d tags\n", buf
.len
);
353 *pdisp
= (IDispatch
*)HTMLElementCollection_Create(This
->ref_unk
, buf
.buf
, buf
.len
);
357 #define DISPID_ELEMCOL_0 MSHTML_DISPID_CUSTOM_MIN
359 static HRESULT
HTMLElementCollection_get_dispid(IUnknown
*iface
, BSTR name
, DWORD flags
, DISPID
*dispid
)
361 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
366 return DISP_E_UNKNOWNNAME
;
368 for(ptr
= name
; *ptr
&& isdigitW(*ptr
); ptr
++)
369 idx
= idx
*10 + (*ptr
-'0');
371 if(*ptr
|| idx
>= This
->len
)
372 return DISP_E_UNKNOWNNAME
;
374 *dispid
= DISPID_ELEMCOL_0
+ idx
;
375 TRACE("ret %x\n", *dispid
);
379 static HRESULT
HTMLElementCollection_invoke(IUnknown
*iface
, DISPID id
, LCID lcid
, WORD flags
, DISPPARAMS
*params
,
380 VARIANT
*res
, EXCEPINFO
*ei
, IServiceProvider
*caller
)
382 HTMLElementCollection
*This
= ELEMCOL_THIS(iface
);
385 TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This
, id
, lcid
, flags
, params
, res
, ei
, caller
);
387 idx
= id
- DISPID_ELEMCOL_0
;
389 return DISP_E_UNKNOWNNAME
;
392 case INVOKE_PROPERTYGET
:
393 V_VT(res
) = VT_DISPATCH
;
394 V_DISPATCH(res
) = (IDispatch
*)HTMLELEM(This
->elems
[idx
]);
395 IHTMLElement_AddRef(HTMLELEM(This
->elems
[idx
]));
398 FIXME("unimplemented flags %x\n", flags
);
407 static const IHTMLElementCollectionVtbl HTMLElementCollectionVtbl
= {
408 HTMLElementCollection_QueryInterface
,
409 HTMLElementCollection_AddRef
,
410 HTMLElementCollection_Release
,
411 HTMLElementCollection_GetTypeInfoCount
,
412 HTMLElementCollection_GetTypeInfo
,
413 HTMLElementCollection_GetIDsOfNames
,
414 HTMLElementCollection_Invoke
,
415 HTMLElementCollection_toString
,
416 HTMLElementCollection_put_length
,
417 HTMLElementCollection_get_length
,
418 HTMLElementCollection_get__newEnum
,
419 HTMLElementCollection_item
,
420 HTMLElementCollection_tags
423 static const dispex_static_data_vtbl_t HTMLElementColection_dispex_vtbl
= {
424 HTMLElementCollection_get_dispid
,
425 HTMLElementCollection_invoke
428 static const tid_t HTMLElementCollection_iface_tids
[] = {
429 IHTMLElementCollection_tid
,
432 static dispex_static_data_t HTMLElementCollection_dispex
= {
433 &HTMLElementColection_dispex_vtbl
,
434 DispHTMLElementCollection_tid
,
436 HTMLElementCollection_iface_tids
439 static void create_all_list(HTMLDocument
*doc
, HTMLDOMNode
*elem
, elem_vector_t
*buf
)
441 nsIDOMNodeList
*nsnode_list
;
443 PRUint32 list_len
= 0, i
;
446 nsres
= nsIDOMNode_GetChildNodes(elem
->nsnode
, &nsnode_list
);
447 if(NS_FAILED(nsres
)) {
448 ERR("GetChildNodes failed: %08x\n", nsres
);
452 nsIDOMNodeList_GetLength(nsnode_list
, &list_len
);
456 for(i
=0; i
<list_len
; i
++) {
457 nsres
= nsIDOMNodeList_Item(nsnode_list
, i
, &iter
);
458 if(NS_FAILED(nsres
)) {
459 ERR("Item failed: %08x\n", nsres
);
463 if(is_elem_node(iter
)) {
464 HTMLDOMNode
*node
= get_node(doc
, iter
, TRUE
);
466 elem_vector_add(buf
, HTMLELEM_NODE_THIS(node
));
467 create_all_list(doc
, node
, buf
);
472 IHTMLElementCollection
*create_all_collection(HTMLDOMNode
*node
, BOOL include_root
)
474 elem_vector_t buf
= {NULL
, 0, 8};
476 buf
.buf
= heap_alloc(buf
.size
*sizeof(HTMLElement
**));
479 elem_vector_add(&buf
, HTMLELEM_NODE_THIS(node
));
480 create_all_list(node
->doc
, node
, &buf
);
481 elem_vector_normalize(&buf
);
483 return HTMLElementCollection_Create((IUnknown
*)HTMLDOMNODE(node
), buf
.buf
, buf
.len
);
486 IHTMLElementCollection
*create_collection_from_nodelist(HTMLDocument
*doc
, IUnknown
*unk
, nsIDOMNodeList
*nslist
)
488 PRUint32 length
= 0, i
;
491 nsIDOMNodeList_GetLength(nslist
, &length
);
498 buf
.buf
= heap_alloc(buf
.size
*sizeof(HTMLElement
*));
500 for(i
=0; i
<length
; i
++) {
501 nsIDOMNodeList_Item(nslist
, i
, &nsnode
);
502 if(is_elem_node(nsnode
))
503 buf
.buf
[buf
.len
++] = HTMLELEM_NODE_THIS(get_node(doc
, nsnode
, TRUE
));
504 nsIDOMNode_Release(nsnode
);
507 elem_vector_normalize(&buf
);
512 return HTMLElementCollection_Create(unk
, buf
.buf
, buf
.len
);
515 IHTMLElementCollection
*create_collection_from_htmlcol(HTMLDocument
*doc
, IUnknown
*unk
, nsIDOMHTMLCollection
*nscol
)
517 PRUint32 length
= 0, i
;
520 nsIDOMHTMLCollection_GetLength(nscol
, &length
);
522 buf
.len
= buf
.size
= length
;
526 buf
.buf
= heap_alloc(buf
.size
*sizeof(HTMLElement
*));
528 for(i
=0; i
<length
; i
++) {
529 nsIDOMHTMLCollection_Item(nscol
, i
, &nsnode
);
530 buf
.buf
[i
] = HTMLELEM_NODE_THIS(get_node(doc
, nsnode
, TRUE
));
531 nsIDOMNode_Release(nsnode
);
537 return HTMLElementCollection_Create(unk
, buf
.buf
, buf
.len
);
540 static IHTMLElementCollection
*HTMLElementCollection_Create(IUnknown
*ref_unk
,
541 HTMLElement
**elems
, DWORD len
)
543 HTMLElementCollection
*ret
= heap_alloc_zero(sizeof(HTMLElementCollection
));
545 ret
->lpHTMLElementCollectionVtbl
= &HTMLElementCollectionVtbl
;
550 init_dispex(&ret
->dispex
, (IUnknown
*)HTMLELEMCOL(ret
), &HTMLElementCollection_dispex
);
552 IUnknown_AddRef(ref_unk
);
553 ret
->ref_unk
= ref_unk
;
555 TRACE("ret=%p len=%d\n", ret
, len
);
557 return HTMLELEMCOL(ret
);