diff -r 000000000000 -r 4f2f89ce4247 WebKit/win/WebElementPropertyBag.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/WebKit/win/WebElementPropertyBag.cpp Fri Sep 17 09:02:29 2010 +0300 @@ -0,0 +1,203 @@ +/* + * Copyright (C) 2006, 2007 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR + * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY + * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "config.h" +#include "WebKitDLL.h" +#include "WebElementPropertyBag.h" + +#include "MarshallingHelpers.h" +#include "DOMCoreClasses.h" +#include "WebFrame.h" +#pragma warning(push, 0) +#include +#include +#include +#include +#include +#include +#include +#pragma warning(pop) + +using namespace WebCore; + +// WebElementPropertyBag ----------------------------------------------- +WebElementPropertyBag::WebElementPropertyBag(const HitTestResult& result) + : m_result(new HitTestResult(result)) + , m_refCount(0) +{ + gClassCount++; + gClassNameCount.add("WebElementPropertyBag"); +} + +WebElementPropertyBag::~WebElementPropertyBag() +{ + gClassCount--; + gClassNameCount.remove("WebElementPropertyBag"); +} + +WebElementPropertyBag* WebElementPropertyBag::createInstance(const HitTestResult& result) +{ + WebElementPropertyBag* instance = new WebElementPropertyBag(result); + instance->AddRef(); + + return instance; +} + +// IUnknown ------------------------------------------------------------------- + +HRESULT STDMETHODCALLTYPE WebElementPropertyBag::QueryInterface(REFIID riid, void** ppvObject) +{ + *ppvObject = 0; + if (IsEqualGUID(riid, IID_IUnknown)) + *ppvObject = static_cast(this); + else if (IsEqualGUID(riid, IID_IPropertyBag)) + *ppvObject = static_cast(this); + else + return E_NOINTERFACE; + + AddRef(); + return S_OK; +} + +ULONG STDMETHODCALLTYPE WebElementPropertyBag::AddRef(void) +{ + return ++m_refCount; +} + +ULONG STDMETHODCALLTYPE WebElementPropertyBag::Release(void) +{ + ULONG newRef = --m_refCount; + if (!newRef) + delete this; + + return newRef; +} + +static bool isEqual(LPCWSTR s1, LPCWSTR s2) +{ + return !wcscmp(s1, s2); +} + +static HRESULT convertStringToVariant(VARIANT* pVar, const String& string) +{ + V_VT(pVar) = VT_BSTR; + V_BSTR(pVar) = SysAllocStringLen(string.characters(), string.length()); + if (string.length() && !V_BSTR(pVar)) + return E_OUTOFMEMORY; + + return S_OK; +} + + +HRESULT STDMETHODCALLTYPE WebElementPropertyBag::Read(LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog * /*pErrorLog*/) +{ + if (!pszPropName) + return E_POINTER; + + if (!m_result) + return E_FAIL; + + BSTR key = (BSTR)pszPropName; + VariantClear(pVar); + if (isEqual(WebElementDOMNodeKey, key)) { + IDOMNode* node = DOMNode::createInstance(m_result->innerNonSharedNode()); + V_VT(pVar) = VT_UNKNOWN; + V_UNKNOWN(pVar) = node; + return S_OK; + } else if (isEqual(WebElementFrameKey, key)) { + if (!(m_result->innerNonSharedNode() && m_result->innerNonSharedNode()->document() + && m_result->innerNonSharedNode()->document()->frame())) + return E_FAIL; + Frame* coreFrame = m_result->innerNonSharedNode()->document()->frame(); + WebFrame* webFrame = static_cast(coreFrame->loader()->client()); + IWebFrame* iWebFrame; + if (FAILED(webFrame->QueryInterface(IID_IWebFrame, (void**)&iWebFrame))) + return E_FAIL; + V_VT(pVar) = VT_UNKNOWN; + V_UNKNOWN(pVar) = iWebFrame; + return S_OK; + } else if (isEqual(WebElementImageAltStringKey, key)) + return convertStringToVariant(pVar, m_result->altDisplayString()); + else if (isEqual(WebElementImageKey, key)) { + V_VT(pVar) = VT_BYREF; + V_BYREF(pVar) = m_result->image(); + return S_OK; + } else if (isEqual(WebElementImageRectKey, key)) { + V_VT(pVar) = VT_ARRAY; + IntRect boundingBox = m_result->innerNonSharedNode() && m_result->innerNonSharedNode()->renderer() ? + m_result->innerNonSharedNode()->renderer()->absoluteBoundingBoxRect(true) : IntRect(); + V_ARRAY(pVar) = MarshallingHelpers::intRectToSafeArray(boundingBox); + return S_OK; + } else if (isEqual(WebElementImageURLKey, key)) + return convertStringToVariant(pVar, m_result->absoluteImageURL().string()); + else if (isEqual(WebElementIsSelectedKey, key)) { + V_VT(pVar) = VT_BOOL; + if (m_result->isSelected()) + V_BOOL(pVar) = VARIANT_TRUE; + else + V_BOOL(pVar) = VARIANT_FALSE; + return S_OK; + } else if (isEqual(WebElementSpellingToolTipKey, key)) { + TextDirection dir; + return convertStringToVariant(pVar, m_result->spellingToolTip(dir)); + } else if (isEqual(WebElementTitleKey, key)) { + TextDirection dir; + return convertStringToVariant(pVar, m_result->title(dir)); + } + else if (isEqual(WebElementLinkURLKey, key)) + return convertStringToVariant(pVar, m_result->absoluteLinkURL().string()); + else if (isEqual(WebElementLinkTargetFrameKey, key)) { + if (!m_result->targetFrame()) + return E_FAIL; + WebFrame* webFrame = kit(m_result->targetFrame()); + IWebFrame* iWebFrame; + if (FAILED(webFrame->QueryInterface(IID_IWebFrame, (void**)&iWebFrame))) + return E_FAIL; + V_VT(pVar) = VT_UNKNOWN; + V_UNKNOWN(pVar) = iWebFrame; + return S_OK; + } else if (isEqual(WebElementLinkTitleKey, key)) + return convertStringToVariant(pVar, m_result->titleDisplayString()); + else if (isEqual(WebElementLinkLabelKey, key)) + return convertStringToVariant(pVar, m_result->textContent()); + else if (isEqual(WebElementIsContentEditableKey, key)) { + V_VT(pVar) = VT_BOOL; + if (m_result->isContentEditable()) + V_BOOL(pVar) = VARIANT_TRUE; + else + V_BOOL(pVar) = VARIANT_FALSE; + return S_OK; + } + + return E_INVALIDARG; +} + +HRESULT STDMETHODCALLTYPE WebElementPropertyBag::Write(LPCOLESTR pszPropName, VARIANT* pVar) +{ + if (!pszPropName || !pVar) + return E_POINTER; + + return E_FAIL; +}