/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* ***** BEGIN LICENSE BLOCK ***** * Version: NPL 1.1/GPL 2.0/LGPL 2.1 * * The contents of this file are subject to the Netscape Public License * Version 1.1 (the "License"); you may not use this file except in * compliance with the License. You may obtain a copy of the License at * http://www.mozilla.org/NPL/ * * Software distributed under the License is distributed on an "AS IS" basis, * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License * for the specific language governing rights and limitations under the * License. * * The Original Code is mozilla.org code. * * The Initial Developer of the Original Code is * Netscape Communications Corporation. * Portions created by the Initial Developer are Copyright (C) 1998 * the Initial Developer. All Rights Reserved. * * Contributor(s): * * Adam Lock * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the NPL, indicate your * decision by deleting the provisions above and replace them with the notice * and other provisions required by the GPL or the LGPL. If you do not delete * the provisions above, a recipient may use your version of this file under * the terms of any one of the NPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ #include "stdafx.h" #include "nsCOMPtr.h" #include "nsIDOMElement.h" #include "nsString.h" #include "IEHtmlElement.h" #include "IEHtmlElementCollection.h" #include "nsIDOMNSHTMLElement.h" #include "nsIDOMDocumentRange.h" #include "nsIDOMRange.h" #include "nsIDOMNSRange.h" #include "nsIDOMDocumentFragment.h" #include "nsIDocumentEncoder.h" CIEHtmlElement::CIEHtmlElement() { } CIEHtmlElement::~CIEHtmlElement() { } HRESULT CIEHtmlElement::GetChildren(CIEHtmlElementCollectionInstance **ppCollection) { // Validate parameters if (ppCollection == NULL) { return E_INVALIDARG; } *ppCollection = NULL; // Create a collection representing the children of this node CIEHtmlElementCollectionInstance *pCollection = NULL; CIEHtmlElementCollection::CreateFromParentNode(this, FALSE, (CIEHtmlElementCollection **) &pCollection); if (pCollection) { pCollection->AddRef(); *ppCollection = pCollection; } return S_OK; } /////////////////////////////////////////////////////////////////////////////// // IHTMLElement implementation HRESULT STDMETHODCALLTYPE CIEHtmlElement::setAttribute(BSTR strAttributeName, VARIANT AttributeValue, LONG lFlags) { if (strAttributeName == NULL) { return E_INVALIDARG; } nsCOMPtr element = do_QueryInterface(mDOMNode); if (!element) { return E_UNEXPECTED; } // Get the name from the BSTR USES_CONVERSION; nsAutoString name(OLE2W(strAttributeName)); // Get the value from the variant CComVariant vValue; if (FAILED(vValue.ChangeType(VT_BSTR, &AttributeValue))) { return E_INVALIDARG; } // Set the attribute nsAutoString value(OLE2W(vValue.bstrVal)); element->SetAttribute(name, value); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::getAttribute(BSTR strAttributeName, LONG lFlags, VARIANT __RPC_FAR *AttributeValue) { if (strAttributeName == NULL) { return E_INVALIDARG; } if (AttributeValue == NULL) { return E_INVALIDARG; } VariantInit(AttributeValue); // Get the name from the BSTR USES_CONVERSION; nsAutoString name(OLE2W(strAttributeName)); nsCOMPtr element = do_QueryInterface(mDOMNode); if (!element) { return E_UNEXPECTED; } BOOL bCaseSensitive = (lFlags == VARIANT_TRUE) ? TRUE : FALSE; // Get the attribute nsAutoString value; nsresult rv = element->GetAttribute(name, value); if (NS_SUCCEEDED(rv)) { USES_CONVERSION; AttributeValue->vt = VT_BSTR; AttributeValue->bstrVal = SysAllocString(W2COLE(value.get())); return S_OK; } else { return S_FALSE; } return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::removeAttribute(BSTR strAttributeName, LONG lFlags, VARIANT_BOOL __RPC_FAR *pfSuccess) { if (strAttributeName == NULL) { return E_INVALIDARG; } nsCOMPtr element = do_QueryInterface(mDOMNode); if (!element) { return E_UNEXPECTED; } BOOL bCaseSensitive = (lFlags == VARIANT_TRUE) ? TRUE : FALSE; // Get the name from the BSTR USES_CONVERSION; nsAutoString name(OLE2W(strAttributeName)); // Remove the attribute nsresult nr = element->RemoveAttribute(name); BOOL bRemoved = (nr == NS_OK) ? TRUE : FALSE; if (pfSuccess) { *pfSuccess = (bRemoved) ? VARIANT_TRUE : VARIANT_FALSE; } return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_className(BSTR v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_className(BSTR __RPC_FAR *p) { if (p == NULL) { return E_INVALIDARG; } VARIANT vValue; VariantInit(&vValue); BSTR bstrName = SysAllocString(OLESTR("class")); getAttribute(bstrName, FALSE, &vValue); SysFreeString(bstrName); *p = vValue.bstrVal; return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_id(BSTR v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_id(BSTR __RPC_FAR *p) { if (p == NULL) { return E_INVALIDARG; } VARIANT vValue; VariantInit(&vValue); BSTR bstrName = SysAllocString(OLESTR("id")); getAttribute(bstrName, FALSE, &vValue); SysFreeString(bstrName); *p = vValue.bstrVal; return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_tagName(BSTR __RPC_FAR *p) { if (p == NULL) { return E_INVALIDARG; } nsCOMPtr element = do_QueryInterface(mDOMNode); if (!element) { return E_UNEXPECTED; } nsAutoString tagName; element->GetTagName(tagName); USES_CONVERSION; *p = SysAllocString(W2COLE(tagName.get())); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_parentElement(IHTMLElement __RPC_FAR *__RPC_FAR *p) { if (p == NULL) { return E_INVALIDARG; } *p = NULL; if (mParent) { IDispatch *pDisp = reinterpret_cast(mParent); pDisp->QueryInterface(IID_IHTMLElement, (void **) p); } else { nsCOMPtr parentNode; mDOMNode->GetParentNode(getter_AddRefs(parentNode)); nsCOMPtr domElement = do_QueryInterface(parentNode); if (domElement) { CIEHtmlNode *pHtmlNode = NULL; CIEHtmlElementInstance *pHtmlElement = NULL; CIEHtmlElementInstance::FindFromDOMNode(parentNode, &pHtmlNode); if (!pHtmlNode) { CIEHtmlElementInstance::CreateInstance(&pHtmlElement); if (!pHtmlElement) { NS_ASSERTION(0, "Could not create element"); return E_OUTOFMEMORY; } pHtmlElement->SetDOMNode(parentNode); } else { pHtmlElement = (CIEHtmlElementInstance *) pHtmlNode; } pHtmlElement->QueryInterface(IID_IHTMLElement, (void **) p); } } return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_style(IHTMLStyle __RPC_FAR *__RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onhelp(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onhelp(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onclick(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onclick(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_ondblclick(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_ondblclick(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onkeydown(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onkeydown(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onkeyup(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onkeyup(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onkeypress(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onkeypress(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onmouseout(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onmouseout(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onmouseover(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onmouseover(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onmousemove(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onmousemove(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onmousedown(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onmousedown(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onmouseup(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onmouseup(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_document(IDispatch __RPC_FAR *__RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_title(BSTR v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_title(BSTR __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_language(BSTR v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_language(BSTR __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onselectstart(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onselectstart(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::scrollIntoView(VARIANT varargStart) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::contains(IHTMLElement __RPC_FAR *pChild, VARIANT_BOOL __RPC_FAR *pfResult) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_sourceIndex(long __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_recordNumber(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_lang(BSTR v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_lang(BSTR __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_offsetLeft(long __RPC_FAR *p) { nsCOMPtr nodeAsHTMLElement = do_QueryInterface(mDOMNode); if (!nodeAsHTMLElement) { return E_NOINTERFACE; } PRInt32 nData; nodeAsHTMLElement->GetOffsetLeft(&nData); *p = nData; return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_offsetTop(long __RPC_FAR *p) { nsCOMPtr nodeAsHTMLElement = do_QueryInterface(mDOMNode); if (!nodeAsHTMLElement) { return E_NOINTERFACE; } PRInt32 nData; nodeAsHTMLElement->GetOffsetTop(&nData); *p = nData; return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_offsetWidth(long __RPC_FAR *p) { nsCOMPtr nodeAsHTMLElement = do_QueryInterface(mDOMNode); if (!nodeAsHTMLElement) { return E_NOINTERFACE; } PRInt32 nData; nodeAsHTMLElement->GetOffsetWidth(&nData); *p = nData; return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_offsetHeight(long __RPC_FAR *p) { nsCOMPtr nodeAsHTMLElement = do_QueryInterface(mDOMNode); if (!nodeAsHTMLElement) { return E_NOINTERFACE; } PRInt32 nData; nodeAsHTMLElement->GetOffsetHeight(&nData); *p = nData; return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_offsetParent(IHTMLElement __RPC_FAR *__RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_innerHTML(BSTR v) { nsCOMPtr elementHTML = do_QueryInterface(mDOMNode); if (!elementHTML) { return E_UNEXPECTED; } USES_CONVERSION; nsAutoString innerHTML(OLE2W(v)); elementHTML->SetInnerHTML(innerHTML); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_innerHTML(BSTR __RPC_FAR *p) { nsCOMPtr elementHTML = do_QueryInterface(mDOMNode); if (!elementHTML) { return E_UNEXPECTED; } nsAutoString innerHTML; elementHTML->GetInnerHTML(innerHTML); USES_CONVERSION; *p = SysAllocString(W2COLE(innerHTML.get())); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_innerText(BSTR v) { nsCOMPtr elementHTML = do_QueryInterface(mDOMNode); if (!elementHTML) { return E_UNEXPECTED; } USES_CONVERSION; nsAutoString innerHTML(OLE2W(v)); elementHTML->SetInnerHTML(innerHTML); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_innerText(BSTR __RPC_FAR *p) { return get_innerHTML(p); } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_outerHTML(BSTR v) { nsresult rv; nsCOMPtr domDoc; nsCOMPtr domRange; nsCOMPtr domDocFragment; mDOMNode->GetOwnerDocument(getter_AddRefs(domDoc)); nsCOMPtr domDocRange = do_QueryInterface(domDoc); if (!domDocRange) return E_FAIL; domDocRange->CreateRange(getter_AddRefs(domRange)); if (!domRange) return E_FAIL; if (domRange->SetStartBefore(mDOMNode)) return E_FAIL; if (domRange->DeleteContents()) return E_FAIL; nsAutoString outerHTML(OLE2W(v)); nsCOMPtr domNSRange = do_QueryInterface(domRange); rv = domNSRange->CreateContextualFragment(outerHTML, getter_AddRefs(domDocFragment)); if (!domDocFragment) return E_FAIL; nsCOMPtr parentNode; mDOMNode->GetParentNode(getter_AddRefs(parentNode)); nsCOMPtr domNode; parentNode->ReplaceChild(domDocFragment, mDOMNode, getter_AddRefs(domNode)); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_outerHTML(BSTR __RPC_FAR *p) { if (p == NULL) { return E_INVALIDARG; } *p = NULL; nsresult rv; nsAutoString outerHTML; nsCOMPtr domDoc; nsCOMPtr docEncoder; nsCOMPtr domRange; mDOMNode->GetOwnerDocument(getter_AddRefs(domDoc)); nsCOMPtr doc = do_QueryInterface(domDoc); if (!doc) return E_FAIL; docEncoder = do_CreateInstance(NS_DOC_ENCODER_CONTRACTID_BASE "text/html"); NS_ENSURE_TRUE(docEncoder, NS_ERROR_FAILURE); docEncoder->Init(doc, NS_LITERAL_STRING("text/html"), nsIDocumentEncoder::OutputEncodeBasicEntities); nsCOMPtr domDocRange = do_QueryInterface(domDoc); if (!domDocRange) return E_FAIL; domDocRange->CreateRange(getter_AddRefs(domRange)); if (!domRange) return E_FAIL; rv = domRange->SelectNode(mDOMNode); NS_ENSURE_SUCCESS(rv, rv); docEncoder->SetRange(domRange); docEncoder->EncodeToString(outerHTML); USES_CONVERSION; *p = SysAllocString(W2COLE(outerHTML.get())); return S_OK; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_outerText(BSTR v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_outerText(BSTR __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::insertAdjacentHTML(BSTR where, BSTR html) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::insertAdjacentText(BSTR where, BSTR text) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_parentTextEdit(IHTMLElement __RPC_FAR *__RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_isTextEdit(VARIANT_BOOL __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::click(void) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_filters(IHTMLFiltersCollection __RPC_FAR *__RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_ondragstart(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_ondragstart(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::toString(BSTR __RPC_FAR *String) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onbeforeupdate(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onbeforeupdate(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onafterupdate(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onafterupdate(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onerrorupdate(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onerrorupdate(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onrowexit(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onrowexit(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onrowenter(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onrowenter(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_ondatasetchanged(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_ondatasetchanged(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_ondataavailable(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_ondataavailable(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_ondatasetcomplete(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_ondatasetcomplete(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::put_onfilterchange(VARIANT v) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_onfilterchange(VARIANT __RPC_FAR *p) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_children(IDispatch __RPC_FAR *__RPC_FAR *p) { // Validate parameters if (p == NULL) { return E_INVALIDARG; } *p = NULL; // Create a collection representing the children of this node CIEHtmlElementCollectionInstance *pCollection = NULL; HRESULT hr = GetChildren(&pCollection); if (SUCCEEDED(hr)) { *p = pCollection; } return hr; } HRESULT STDMETHODCALLTYPE CIEHtmlElement::get_all(IDispatch __RPC_FAR *__RPC_FAR *p) { // Validate parameters if (p == NULL) { return E_INVALIDARG; } *p = NULL; // TODO get ALL contained elements, not just the immediate children CIEHtmlElementCollectionInstance *pCollection = NULL; CIEHtmlElementCollection::CreateFromParentNode(this, TRUE, (CIEHtmlElementCollection **) &pCollection); if (pCollection) { pCollection->AddRef(); *p = pCollection; } return S_OK; }