mirror of
https://github.com/reactos/reactos.git
synced 2024-08-31 07:38:48 +00:00
828 lines
23 KiB
C
828 lines
23 KiB
C
/*
|
|
* Copyright 2006-2008 Jacek Caban for CodeWeavers
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "mshtml_private.h"
|
|
|
|
typedef struct {
|
|
DispatchEx dispex;
|
|
IHTMLElementCollection IHTMLElementCollection_iface;
|
|
|
|
HTMLElement **elems;
|
|
DWORD len;
|
|
|
|
LONG ref;
|
|
} HTMLElementCollection;
|
|
|
|
typedef struct {
|
|
IEnumVARIANT IEnumVARIANT_iface;
|
|
|
|
LONG ref;
|
|
|
|
ULONG iter;
|
|
HTMLElementCollection *col;
|
|
} HTMLElementCollectionEnum;
|
|
|
|
typedef struct {
|
|
HTMLElement **buf;
|
|
DWORD len;
|
|
DWORD size;
|
|
} elem_vector_t;
|
|
|
|
/* FIXME: Handle it better way */
|
|
static inline HTMLElement *elem_from_HTMLDOMNode(HTMLDOMNode *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, HTMLElement, node);
|
|
}
|
|
|
|
static IHTMLElementCollection *HTMLElementCollection_Create(HTMLElement **elems, DWORD len);
|
|
|
|
static void elem_vector_add(elem_vector_t *buf, HTMLElement *elem)
|
|
{
|
|
if(buf->len == buf->size) {
|
|
buf->size <<= 1;
|
|
buf->buf = heap_realloc(buf->buf, buf->size*sizeof(HTMLElement*));
|
|
}
|
|
|
|
buf->buf[buf->len++] = elem;
|
|
}
|
|
|
|
static void elem_vector_normalize(elem_vector_t *buf)
|
|
{
|
|
if(!buf->len) {
|
|
heap_free(buf->buf);
|
|
buf->buf = NULL;
|
|
}else if(buf->size > buf->len) {
|
|
buf->buf = heap_realloc(buf->buf, buf->len*sizeof(HTMLElement*));
|
|
}
|
|
|
|
buf->size = buf->len;
|
|
}
|
|
|
|
static inline BOOL is_elem_node(nsIDOMNode *node)
|
|
{
|
|
UINT16 type=0;
|
|
|
|
nsIDOMNode_GetNodeType(node, &type);
|
|
|
|
return type == ELEMENT_NODE || type == COMMENT_NODE;
|
|
}
|
|
|
|
static inline HTMLElementCollectionEnum *impl_from_IEnumVARIANT(IEnumVARIANT *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, HTMLElementCollectionEnum, IEnumVARIANT_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollectionEnum_QueryInterface(IEnumVARIANT *iface, REFIID riid, void **ppv)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv);
|
|
|
|
if(IsEqualGUID(riid, &IID_IUnknown)) {
|
|
*ppv = &This->IEnumVARIANT_iface;
|
|
}else if(IsEqualGUID(riid, &IID_IEnumVARIANT)) {
|
|
*ppv = &This->IEnumVARIANT_iface;
|
|
}else {
|
|
FIXME("Unsupported iface %s\n", debugstr_mshtml_guid(riid));
|
|
*ppv = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IUnknown_AddRef((IUnknown*)*ppv);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI HTMLElementCollectionEnum_AddRef(IEnumVARIANT *iface)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
LONG ref = InterlockedIncrement(&This->ref);
|
|
|
|
TRACE("(%p) ref=%d\n", This, ref);
|
|
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI HTMLElementCollectionEnum_Release(IEnumVARIANT *iface)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
LONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p) ref=%d\n", This, ref);
|
|
|
|
if(!ref) {
|
|
IHTMLElementCollection_Release(&This->col->IHTMLElementCollection_iface);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollectionEnum_Next(IEnumVARIANT *iface, ULONG celt, VARIANT *rgVar, ULONG *pCeltFetched)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
ULONG fetched = 0;
|
|
|
|
TRACE("(%p)->(%d %p %p)\n", This, celt, rgVar, pCeltFetched);
|
|
|
|
while(This->iter+fetched < This->col->len && fetched < celt) {
|
|
V_VT(rgVar+fetched) = VT_DISPATCH;
|
|
V_DISPATCH(rgVar+fetched) = (IDispatch*)&This->col->elems[This->iter+fetched]->IHTMLElement_iface;
|
|
IDispatch_AddRef(V_DISPATCH(rgVar+fetched));
|
|
fetched++;
|
|
}
|
|
|
|
This->iter += fetched;
|
|
if(pCeltFetched)
|
|
*pCeltFetched = fetched;
|
|
return fetched == celt ? S_OK : S_FALSE;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollectionEnum_Skip(IEnumVARIANT *iface, ULONG celt)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)->(%d)\n", This, celt);
|
|
|
|
if(This->iter + celt > This->col->len) {
|
|
This->iter = This->col->len;
|
|
return S_FALSE;
|
|
}
|
|
|
|
This->iter += celt;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollectionEnum_Reset(IEnumVARIANT *iface)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)->()\n", This);
|
|
|
|
This->iter = 0;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollectionEnum_Clone(IEnumVARIANT *iface, IEnumVARIANT **ppEnum)
|
|
{
|
|
HTMLElementCollectionEnum *This = impl_from_IEnumVARIANT(iface);
|
|
FIXME("(%p)->(%p)\n", This, ppEnum);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IEnumVARIANTVtbl HTMLElementCollectionEnumVtbl = {
|
|
HTMLElementCollectionEnum_QueryInterface,
|
|
HTMLElementCollectionEnum_AddRef,
|
|
HTMLElementCollectionEnum_Release,
|
|
HTMLElementCollectionEnum_Next,
|
|
HTMLElementCollectionEnum_Skip,
|
|
HTMLElementCollectionEnum_Reset,
|
|
HTMLElementCollectionEnum_Clone
|
|
};
|
|
|
|
static inline HTMLElementCollection *impl_from_IHTMLElementCollection(IHTMLElementCollection *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, HTMLElementCollection, IHTMLElementCollection_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_QueryInterface(IHTMLElementCollection *iface,
|
|
REFIID riid, void **ppv)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv);
|
|
|
|
if(IsEqualGUID(&IID_IUnknown, riid)) {
|
|
*ppv = &This->IHTMLElementCollection_iface;
|
|
}else if(IsEqualGUID(&IID_IHTMLElementCollection, riid)) {
|
|
*ppv = &This->IHTMLElementCollection_iface;
|
|
}else if(dispex_query_interface(&This->dispex, riid, ppv)) {
|
|
return *ppv ? S_OK : E_NOINTERFACE;
|
|
}else {
|
|
*ppv = NULL;
|
|
FIXME("Unsupported iface %s\n", debugstr_mshtml_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IHTMLElementCollection_AddRef(&This->IHTMLElementCollection_iface);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI HTMLElementCollection_AddRef(IHTMLElementCollection *iface)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
LONG ref = InterlockedIncrement(&This->ref);
|
|
|
|
TRACE("(%p) ref=%d\n", This, ref);
|
|
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI HTMLElementCollection_Release(IHTMLElementCollection *iface)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
LONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p) ref=%d\n", This, ref);
|
|
|
|
if(!ref) {
|
|
unsigned i;
|
|
|
|
for(i=0; i < This->len; i++)
|
|
node_release(&This->elems[i]->node);
|
|
heap_free(This->elems);
|
|
|
|
release_dispex(&This->dispex);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_GetTypeInfoCount(IHTMLElementCollection *iface,
|
|
UINT *pctinfo)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo);
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_GetTypeInfo(IHTMLElementCollection *iface,
|
|
UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_GetIDsOfNames(IHTMLElementCollection *iface,
|
|
REFIID riid, LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames, cNames,
|
|
lcid, rgDispId);
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_Invoke(IHTMLElementCollection *iface,
|
|
DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
|
|
VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid, lcid,
|
|
wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_toString(IHTMLElementCollection *iface,
|
|
BSTR *String)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
FIXME("(%p)->(%p)\n", This, String);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_put_length(IHTMLElementCollection *iface,
|
|
LONG v)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
FIXME("(%p)->(%d)\n", This, v);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_get_length(IHTMLElementCollection *iface,
|
|
LONG *p)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, p);
|
|
|
|
*p = This->len;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_get__newEnum(IHTMLElementCollection *iface,
|
|
IUnknown **p)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
HTMLElementCollectionEnum *ret;
|
|
|
|
TRACE("(%p)->(%p)\n", This, p);
|
|
|
|
ret = heap_alloc(sizeof(*ret));
|
|
if(!ret)
|
|
return E_OUTOFMEMORY;
|
|
|
|
ret->IEnumVARIANT_iface.lpVtbl = &HTMLElementCollectionEnumVtbl;
|
|
ret->ref = 1;
|
|
ret->iter = 0;
|
|
|
|
IHTMLElementCollection_AddRef(&This->IHTMLElementCollection_iface);
|
|
ret->col = This;
|
|
|
|
*p = (IUnknown*)&ret->IEnumVARIANT_iface;
|
|
return S_OK;
|
|
}
|
|
|
|
static BOOL is_elem_id(HTMLElement *elem, LPCWSTR name)
|
|
{
|
|
BSTR elem_id;
|
|
HRESULT hres;
|
|
|
|
hres = IHTMLElement_get_id(&elem->IHTMLElement_iface, &elem_id);
|
|
if(FAILED(hres)){
|
|
WARN("IHTMLElement_get_id failed: 0x%08x\n", hres);
|
|
return FALSE;
|
|
}
|
|
|
|
if(elem_id && !strcmpW(elem_id, name)) {
|
|
SysFreeString(elem_id);
|
|
return TRUE;
|
|
}
|
|
|
|
SysFreeString(elem_id);
|
|
return FALSE;
|
|
}
|
|
|
|
static BOOL is_elem_name(HTMLElement *elem, LPCWSTR name)
|
|
{
|
|
const PRUnichar *str;
|
|
nsAString nsstr;
|
|
BOOL ret = FALSE;
|
|
nsresult nsres;
|
|
|
|
static const PRUnichar nameW[] = {'n','a','m','e',0};
|
|
|
|
if(!elem->nselem)
|
|
return FALSE;
|
|
|
|
nsAString_Init(&nsstr, NULL);
|
|
nsIDOMHTMLElement_GetId(elem->nselem, &nsstr);
|
|
nsAString_GetData(&nsstr, &str);
|
|
if(!strcmpiW(str, name)) {
|
|
nsAString_Finish(&nsstr);
|
|
return TRUE;
|
|
}
|
|
|
|
nsres = get_elem_attr_value(elem->nselem, nameW, &nsstr, &str);
|
|
if(NS_SUCCEEDED(nsres)) {
|
|
ret = !strcmpiW(str, name);
|
|
nsAString_Finish(&nsstr);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static HRESULT get_item_idx(HTMLElementCollection *This, UINT idx, IDispatch **ret)
|
|
{
|
|
if(idx < This->len) {
|
|
*ret = (IDispatch*)This->elems[idx];
|
|
IDispatch_AddRef(*ret);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_item(IHTMLElementCollection *iface,
|
|
VARIANT name, VARIANT index, IDispatch **pdisp)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
HRESULT hres = S_OK;
|
|
|
|
TRACE("(%p)->(%s %s %p)\n", This, debugstr_variant(&name), debugstr_variant(&index), pdisp);
|
|
|
|
*pdisp = NULL;
|
|
|
|
switch(V_VT(&name)) {
|
|
case VT_I4:
|
|
case VT_INT:
|
|
if(V_I4(&name) < 0)
|
|
return E_INVALIDARG;
|
|
hres = get_item_idx(This, V_I4(&name), pdisp);
|
|
break;
|
|
|
|
case VT_UI4:
|
|
case VT_UINT:
|
|
hres = get_item_idx(This, V_UINT(&name), pdisp);
|
|
break;
|
|
|
|
case VT_BSTR: {
|
|
DWORD i;
|
|
|
|
if(V_VT(&index) == VT_I4) {
|
|
LONG idx = V_I4(&index);
|
|
|
|
if(idx < 0)
|
|
return E_INVALIDARG;
|
|
|
|
for(i=0; i<This->len; i++) {
|
|
if(is_elem_name(This->elems[i], V_BSTR(&name)) && !idx--)
|
|
break;
|
|
}
|
|
|
|
if(i != This->len) {
|
|
*pdisp = (IDispatch*)&This->elems[i]->IHTMLElement_iface;
|
|
IDispatch_AddRef(*pdisp);
|
|
}
|
|
}else {
|
|
elem_vector_t buf = {NULL, 0, 8};
|
|
|
|
buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
|
|
|
|
for(i=0; i<This->len; i++) {
|
|
if(is_elem_name(This->elems[i], V_BSTR(&name))) {
|
|
node_addref(&This->elems[i]->node);
|
|
elem_vector_add(&buf, This->elems[i]);
|
|
}
|
|
}
|
|
|
|
if(buf.len > 1) {
|
|
elem_vector_normalize(&buf);
|
|
*pdisp = (IDispatch*)HTMLElementCollection_Create(buf.buf, buf.len);
|
|
}else {
|
|
if(buf.len == 1) {
|
|
/* Already AddRef-ed */
|
|
*pdisp = (IDispatch*)&buf.buf[0]->IHTMLElement_iface;
|
|
}
|
|
|
|
heap_free(buf.buf);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
default:
|
|
FIXME("Unsupported name %s\n", debugstr_variant(&name));
|
|
hres = E_NOTIMPL;
|
|
}
|
|
|
|
if(SUCCEEDED(hres))
|
|
TRACE("returning %p\n", *pdisp);
|
|
return hres;
|
|
}
|
|
|
|
static HRESULT WINAPI HTMLElementCollection_tags(IHTMLElementCollection *iface,
|
|
VARIANT tagName, IDispatch **pdisp)
|
|
{
|
|
HTMLElementCollection *This = impl_from_IHTMLElementCollection(iface);
|
|
DWORD i;
|
|
nsAString tag_str;
|
|
const PRUnichar *tag;
|
|
elem_vector_t buf = {NULL, 0, 8};
|
|
|
|
if(V_VT(&tagName) != VT_BSTR) {
|
|
WARN("Invalid arg\n");
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_w(V_BSTR(&tagName)), pdisp);
|
|
|
|
buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
|
|
|
|
nsAString_Init(&tag_str, NULL);
|
|
|
|
for(i=0; i<This->len; i++) {
|
|
if(!This->elems[i]->nselem)
|
|
continue;
|
|
|
|
nsIDOMHTMLElement_GetTagName(This->elems[i]->nselem, &tag_str);
|
|
nsAString_GetData(&tag_str, &tag);
|
|
|
|
if(CompareStringW(LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE, tag, -1,
|
|
V_BSTR(&tagName), -1) == CSTR_EQUAL) {
|
|
node_addref(&This->elems[i]->node);
|
|
elem_vector_add(&buf, This->elems[i]);
|
|
}
|
|
}
|
|
|
|
nsAString_Finish(&tag_str);
|
|
elem_vector_normalize(&buf);
|
|
|
|
TRACE("fount %d tags\n", buf.len);
|
|
|
|
*pdisp = (IDispatch*)HTMLElementCollection_Create(buf.buf, buf.len);
|
|
return S_OK;
|
|
}
|
|
|
|
static const IHTMLElementCollectionVtbl HTMLElementCollectionVtbl = {
|
|
HTMLElementCollection_QueryInterface,
|
|
HTMLElementCollection_AddRef,
|
|
HTMLElementCollection_Release,
|
|
HTMLElementCollection_GetTypeInfoCount,
|
|
HTMLElementCollection_GetTypeInfo,
|
|
HTMLElementCollection_GetIDsOfNames,
|
|
HTMLElementCollection_Invoke,
|
|
HTMLElementCollection_toString,
|
|
HTMLElementCollection_put_length,
|
|
HTMLElementCollection_get_length,
|
|
HTMLElementCollection_get__newEnum,
|
|
HTMLElementCollection_item,
|
|
HTMLElementCollection_tags
|
|
};
|
|
|
|
static inline HTMLElementCollection *impl_from_DispatchEx(DispatchEx *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, HTMLElementCollection, dispex);
|
|
}
|
|
|
|
#define DISPID_ELEMCOL_0 MSHTML_DISPID_CUSTOM_MIN
|
|
|
|
static HRESULT HTMLElementCollection_get_dispid(DispatchEx *dispex, BSTR name, DWORD flags, DISPID *dispid)
|
|
{
|
|
HTMLElementCollection *This = impl_from_DispatchEx(dispex);
|
|
WCHAR *ptr;
|
|
DWORD idx=0;
|
|
|
|
if(!*name)
|
|
return DISP_E_UNKNOWNNAME;
|
|
|
|
for(ptr = name; *ptr && isdigitW(*ptr); ptr++)
|
|
idx = idx*10 + (*ptr-'0');
|
|
|
|
if(*ptr) {
|
|
/* the name contains alpha characters, so search by name & id */
|
|
for(idx = 0; idx < This->len; ++idx) {
|
|
if(is_elem_id(This->elems[idx], name) ||
|
|
is_elem_name(This->elems[idx], name))
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(idx >= This->len)
|
|
return DISP_E_UNKNOWNNAME;
|
|
|
|
*dispid = DISPID_ELEMCOL_0 + idx;
|
|
TRACE("ret %x\n", *dispid);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT HTMLElementCollection_invoke(DispatchEx *dispex, DISPID id, LCID lcid, WORD flags, DISPPARAMS *params,
|
|
VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
|
|
{
|
|
HTMLElementCollection *This = impl_from_DispatchEx(dispex);
|
|
DWORD idx;
|
|
|
|
TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, flags, params, res, ei, caller);
|
|
|
|
idx = id - DISPID_ELEMCOL_0;
|
|
if(idx >= This->len)
|
|
return DISP_E_UNKNOWNNAME;
|
|
|
|
switch(flags) {
|
|
case DISPATCH_PROPERTYGET:
|
|
V_VT(res) = VT_DISPATCH;
|
|
V_DISPATCH(res) = (IDispatch*)&This->elems[idx]->IHTMLElement_iface;
|
|
IHTMLElement_AddRef(&This->elems[idx]->IHTMLElement_iface);
|
|
break;
|
|
default:
|
|
FIXME("unimplemented flags %x\n", flags);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static const dispex_static_data_vtbl_t HTMLElementColection_dispex_vtbl = {
|
|
NULL,
|
|
HTMLElementCollection_get_dispid,
|
|
HTMLElementCollection_invoke,
|
|
NULL
|
|
};
|
|
|
|
static const tid_t HTMLElementCollection_iface_tids[] = {
|
|
IHTMLElementCollection_tid,
|
|
0
|
|
};
|
|
|
|
static dispex_static_data_t HTMLElementCollection_dispex = {
|
|
&HTMLElementColection_dispex_vtbl,
|
|
DispHTMLElementCollection_tid,
|
|
NULL,
|
|
HTMLElementCollection_iface_tids
|
|
};
|
|
|
|
static void create_all_list(HTMLDocumentNode *doc, HTMLDOMNode *elem, elem_vector_t *buf)
|
|
{
|
|
nsIDOMNodeList *nsnode_list;
|
|
nsIDOMNode *iter;
|
|
UINT32 list_len = 0, i;
|
|
nsresult nsres;
|
|
HRESULT hres;
|
|
|
|
nsres = nsIDOMNode_GetChildNodes(elem->nsnode, &nsnode_list);
|
|
if(NS_FAILED(nsres)) {
|
|
ERR("GetChildNodes failed: %08x\n", nsres);
|
|
return;
|
|
}
|
|
|
|
nsIDOMNodeList_GetLength(nsnode_list, &list_len);
|
|
if(!list_len)
|
|
return;
|
|
|
|
for(i=0; i<list_len; i++) {
|
|
nsres = nsIDOMNodeList_Item(nsnode_list, i, &iter);
|
|
if(NS_FAILED(nsres)) {
|
|
ERR("Item failed: %08x\n", nsres);
|
|
continue;
|
|
}
|
|
|
|
if(is_elem_node(iter)) {
|
|
HTMLDOMNode *node;
|
|
|
|
hres = get_node(doc, iter, TRUE, &node);
|
|
if(FAILED(hres)) {
|
|
FIXME("get_node failed: %08x\n", hres);
|
|
continue;
|
|
}
|
|
|
|
elem_vector_add(buf, elem_from_HTMLDOMNode(node));
|
|
create_all_list(doc, node, buf);
|
|
}
|
|
}
|
|
}
|
|
|
|
IHTMLElementCollection *create_all_collection(HTMLDOMNode *node, BOOL include_root)
|
|
{
|
|
elem_vector_t buf = {NULL, 0, 8};
|
|
|
|
buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
|
|
|
|
if(include_root) {
|
|
node_addref(node);
|
|
elem_vector_add(&buf, elem_from_HTMLDOMNode(node));
|
|
}
|
|
create_all_list(node->doc, node, &buf);
|
|
elem_vector_normalize(&buf);
|
|
|
|
return HTMLElementCollection_Create(buf.buf, buf.len);
|
|
}
|
|
|
|
IHTMLElementCollection *create_collection_from_nodelist(HTMLDocumentNode *doc, nsIDOMNodeList *nslist)
|
|
{
|
|
UINT32 length = 0, i;
|
|
HTMLDOMNode *node;
|
|
elem_vector_t buf;
|
|
HRESULT hres;
|
|
|
|
nsIDOMNodeList_GetLength(nslist, &length);
|
|
|
|
buf.len = 0;
|
|
buf.size = length;
|
|
if(length) {
|
|
nsIDOMNode *nsnode;
|
|
|
|
buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
|
|
|
|
for(i=0; i<length; i++) {
|
|
nsIDOMNodeList_Item(nslist, i, &nsnode);
|
|
if(is_elem_node(nsnode)) {
|
|
hres = get_node(doc, nsnode, TRUE, &node);
|
|
if(FAILED(hres))
|
|
continue;
|
|
buf.buf[buf.len++] = elem_from_HTMLDOMNode(node);
|
|
}
|
|
nsIDOMNode_Release(nsnode);
|
|
}
|
|
|
|
elem_vector_normalize(&buf);
|
|
}else {
|
|
buf.buf = NULL;
|
|
}
|
|
|
|
return HTMLElementCollection_Create(buf.buf, buf.len);
|
|
}
|
|
|
|
IHTMLElementCollection *create_collection_from_htmlcol(HTMLDocumentNode *doc, nsIDOMHTMLCollection *nscol)
|
|
{
|
|
UINT32 length = 0, i;
|
|
elem_vector_t buf;
|
|
HTMLDOMNode *node;
|
|
HRESULT hres = S_OK;
|
|
|
|
nsIDOMHTMLCollection_GetLength(nscol, &length);
|
|
|
|
buf.len = buf.size = length;
|
|
if(buf.len) {
|
|
nsIDOMNode *nsnode;
|
|
|
|
buf.buf = heap_alloc(buf.size*sizeof(HTMLElement*));
|
|
|
|
for(i=0; i<length; i++) {
|
|
nsIDOMHTMLCollection_Item(nscol, i, &nsnode);
|
|
hres = get_node(doc, nsnode, TRUE, &node);
|
|
nsIDOMNode_Release(nsnode);
|
|
if(FAILED(hres))
|
|
break;
|
|
buf.buf[i] = elem_from_HTMLDOMNode(node);
|
|
}
|
|
}else {
|
|
buf.buf = NULL;
|
|
}
|
|
|
|
if(FAILED(hres)) {
|
|
heap_free(buf.buf);
|
|
return NULL;
|
|
}
|
|
|
|
return HTMLElementCollection_Create(buf.buf, buf.len);
|
|
}
|
|
|
|
HRESULT get_elem_source_index(HTMLElement *elem, LONG *ret)
|
|
{
|
|
elem_vector_t buf = {NULL, 0, 8};
|
|
nsIDOMNode *parent_node, *iter;
|
|
UINT16 parent_type;
|
|
HTMLDOMNode *node;
|
|
int i;
|
|
nsresult nsres;
|
|
HRESULT hres;
|
|
|
|
iter = elem->node.nsnode;
|
|
nsIDOMNode_AddRef(iter);
|
|
|
|
/* Find document or document fragment parent. */
|
|
while(1) {
|
|
nsres = nsIDOMNode_GetParentNode(iter, &parent_node);
|
|
nsIDOMNode_Release(iter);
|
|
assert(nsres == NS_OK);
|
|
if(!parent_node)
|
|
break;
|
|
|
|
nsres = nsIDOMNode_GetNodeType(parent_node, &parent_type);
|
|
assert(nsres == NS_OK);
|
|
|
|
if(parent_type != ELEMENT_NODE) {
|
|
if(parent_type != DOCUMENT_NODE && parent_type != DOCUMENT_FRAGMENT_NODE)
|
|
FIXME("Unexpected parent_type %d\n", parent_type);
|
|
break;
|
|
}
|
|
|
|
iter = parent_node;
|
|
}
|
|
|
|
if(!parent_node) {
|
|
*ret = -1;
|
|
return S_OK;
|
|
}
|
|
|
|
hres = get_node(elem->node.doc, parent_node, TRUE, &node);
|
|
nsIDOMNode_Release(parent_node);
|
|
if(FAILED(hres))
|
|
return hres;
|
|
|
|
|
|
/* Create all children collection and find the element in it.
|
|
* This could be optimized if we ever find the reason. */
|
|
buf.buf = heap_alloc(buf.size*sizeof(*buf.buf));
|
|
if(!buf.buf) {
|
|
IHTMLDOMNode_Release(&node->IHTMLDOMNode_iface);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
create_all_list(elem->node.doc, node, &buf);
|
|
|
|
for(i=0; i < buf.len; i++) {
|
|
if(buf.buf[i] == elem)
|
|
break;
|
|
}
|
|
IHTMLDOMNode_Release(&node->IHTMLDOMNode_iface);
|
|
heap_free(buf.buf);
|
|
if(i == buf.len) {
|
|
FIXME("The element is not in parent's child list?\n");
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
*ret = i;
|
|
return S_OK;
|
|
}
|
|
|
|
static IHTMLElementCollection *HTMLElementCollection_Create(HTMLElement **elems, DWORD len)
|
|
{
|
|
HTMLElementCollection *ret = heap_alloc_zero(sizeof(HTMLElementCollection));
|
|
|
|
if (!ret)
|
|
return NULL;
|
|
|
|
ret->IHTMLElementCollection_iface.lpVtbl = &HTMLElementCollectionVtbl;
|
|
ret->ref = 1;
|
|
ret->elems = elems;
|
|
ret->len = len;
|
|
|
|
init_dispex(&ret->dispex, (IUnknown*)&ret->IHTMLElementCollection_iface,
|
|
&HTMLElementCollection_dispex);
|
|
|
|
TRACE("ret=%p len=%d\n", ret, len);
|
|
|
|
return &ret->IHTMLElementCollection_iface;
|
|
}
|