1 /*

     2  * Copyright (C) 2006, 2007 Apple Inc.  All rights reserved.

     3  *

     4  * Redistribution and use in source and binary forms, with or without

     5  * modification, are permitted provided that the following conditions

     6  * are met:

     7  * 1. Redistributions of source code must retain the above copyright

     8  *    notice, this list of conditions and the following disclaimer.

     9  * 2. Redistributions in binary form must reproduce the above copyright

    10  *    notice, this list of conditions and the following disclaimer in the

    11  *    documentation and/or other materials provided with the distribution.

    12  *

    13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

    14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

    17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

    18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

    19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

    20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

    21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

    24  */

    25 

    26 #include "config.h"

    27 #include "WebKitDLL.h"

    28 #include "WebElementPropertyBag.h"

    29 

    30 #include "MarshallingHelpers.h"

    31 #include "DOMCoreClasses.h"

    32 #include "WebFrame.h"

    33 #pragma warning(push, 0)

    34 #include <WebCore/Document.h>

    35 #include <WebCore/Frame.h>

    36 #include <WebCore/HitTestResult.h>

    37 #include <WebCore/FrameLoader.h>

    38 #include <WebCore/Image.h>

    39 #include <WebCore/KURL.h>

    40 #include <WebCore/RenderObject.h>

    41 #pragma warning(pop)

    42 

    43 using namespace WebCore;

    44 

    45 // WebElementPropertyBag -----------------------------------------------

    46 WebElementPropertyBag::WebElementPropertyBag(const HitTestResult& result)

    47     : m_result(new HitTestResult(result))

    48     , m_refCount(0)

    49 {

    50     gClassCount++;

    51     gClassNameCount.add("WebElementPropertyBag");

    52 }

    53 

    54 WebElementPropertyBag::~WebElementPropertyBag()

    55 {

    56     gClassCount--;

    57     gClassNameCount.remove("WebElementPropertyBag");

    58 }

    59 

    60 WebElementPropertyBag* WebElementPropertyBag::createInstance(const HitTestResult& result)

    61 {

    62     WebElementPropertyBag* instance = new WebElementPropertyBag(result); 

    63     instance->AddRef();

    64 

    65     return instance;

    66 }

    67 

    68 // IUnknown -------------------------------------------------------------------

    69 

    70 HRESULT STDMETHODCALLTYPE WebElementPropertyBag::QueryInterface(REFIID riid, void** ppvObject)

    71 {

    72     *ppvObject = 0;

    73     if (IsEqualGUID(riid, IID_IUnknown))

    74         *ppvObject = static_cast<IPropertyBag*>(this);

    75     else if (IsEqualGUID(riid, IID_IPropertyBag))

    76         *ppvObject = static_cast<IPropertyBag*>(this);

    77     else

    78         return E_NOINTERFACE;

    79 

    80     AddRef();

    81     return S_OK;

    82 }

    83 

    84 ULONG STDMETHODCALLTYPE WebElementPropertyBag::AddRef(void)

    85 {

    86     return ++m_refCount;

    87 }

    88 

    89 ULONG STDMETHODCALLTYPE WebElementPropertyBag::Release(void)

    90 {

    91     ULONG newRef = --m_refCount;

    92     if (!newRef)

    93         delete this;

    94 

    95     return newRef;

    96 }

    97 

    98 static bool isEqual(LPCWSTR s1, LPCWSTR s2)

    99 {

   100     return !wcscmp(s1, s2);

   101 }

   102 

   103 static HRESULT convertStringToVariant(VARIANT* pVar, const String& string)

   104 {

   105     V_VT(pVar) = VT_BSTR;

   106     V_BSTR(pVar) = SysAllocStringLen(string.characters(), string.length());

   107     if (string.length() && !V_BSTR(pVar))

   108         return E_OUTOFMEMORY;

   109 

   110     return S_OK;

   111 }

   112 

   113 

   114 HRESULT STDMETHODCALLTYPE WebElementPropertyBag::Read(LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog * /*pErrorLog*/)

   115 {

   116     if (!pszPropName)

   117         return E_POINTER;

   118 

   119     if (!m_result)

   120         return E_FAIL;

   121 

   122     BSTR key = (BSTR)pszPropName;

   123     VariantClear(pVar);

   124     if (isEqual(WebElementDOMNodeKey, key)) {

   125         IDOMNode* node = DOMNode::createInstance(m_result->innerNonSharedNode());

   126         V_VT(pVar) = VT_UNKNOWN;

   127         V_UNKNOWN(pVar) = node;

   128         return S_OK;

   129     } else if (isEqual(WebElementFrameKey, key)) {

   130         if (!(m_result->innerNonSharedNode() && m_result->innerNonSharedNode()->document()

   131            && m_result->innerNonSharedNode()->document()->frame()))

   132             return E_FAIL;

   133         Frame* coreFrame = m_result->innerNonSharedNode()->document()->frame();

   134         WebFrame* webFrame = static_cast<WebFrame*>(coreFrame->loader()->client());

   135         IWebFrame* iWebFrame;

   136         if (FAILED(webFrame->QueryInterface(IID_IWebFrame, (void**)&iWebFrame)))

   137             return E_FAIL;

   138         V_VT(pVar) = VT_UNKNOWN;

   139         V_UNKNOWN(pVar) = iWebFrame;

   140         return S_OK;

   141     } else if (isEqual(WebElementImageAltStringKey, key))

   142         return convertStringToVariant(pVar, m_result->altDisplayString());

   143     else if (isEqual(WebElementImageKey, key)) {

   144         V_VT(pVar) = VT_BYREF;

   145         V_BYREF(pVar) = m_result->image();

   146         return S_OK;

   147     } else if (isEqual(WebElementImageRectKey, key)) {

   148         V_VT(pVar) = VT_ARRAY;

   149         IntRect boundingBox = m_result->innerNonSharedNode() && m_result->innerNonSharedNode()->renderer() ?

   150                                 m_result->innerNonSharedNode()->renderer()->absoluteBoundingBoxRect(true) : IntRect();

   151         V_ARRAY(pVar) = MarshallingHelpers::intRectToSafeArray(boundingBox);

   152         return S_OK;

   153     } else if (isEqual(WebElementImageURLKey, key))

   154         return convertStringToVariant(pVar, m_result->absoluteImageURL().string());

   155     else if (isEqual(WebElementIsSelectedKey, key)) {

   156         V_VT(pVar) = VT_BOOL;

   157         if (m_result->isSelected())

   158             V_BOOL(pVar) = VARIANT_TRUE;

   159         else

   160             V_BOOL(pVar) = VARIANT_FALSE;

   161         return S_OK;

   162     } else if (isEqual(WebElementSpellingToolTipKey, key)) {

   163         TextDirection dir;

   164         return convertStringToVariant(pVar, m_result->spellingToolTip(dir));

   165     } else if (isEqual(WebElementTitleKey, key)) {

   166         TextDirection dir;

   167         return convertStringToVariant(pVar, m_result->title(dir));

   168     }

   169     else if (isEqual(WebElementLinkURLKey, key))

   170         return convertStringToVariant(pVar, m_result->absoluteLinkURL().string());

   171     else if (isEqual(WebElementLinkTargetFrameKey, key)) {

   172         if (!m_result->targetFrame())

   173             return E_FAIL;

   174         WebFrame* webFrame = kit(m_result->targetFrame());

   175         IWebFrame* iWebFrame;

   176         if (FAILED(webFrame->QueryInterface(IID_IWebFrame, (void**)&iWebFrame)))

   177             return E_FAIL;

   178         V_VT(pVar) = VT_UNKNOWN;

   179         V_UNKNOWN(pVar) = iWebFrame;

   180         return S_OK;

   181     } else if (isEqual(WebElementLinkTitleKey, key))

   182         return convertStringToVariant(pVar, m_result->titleDisplayString());

   183     else if (isEqual(WebElementLinkLabelKey, key))

   184         return convertStringToVariant(pVar, m_result->textContent());

   185     else if (isEqual(WebElementIsContentEditableKey, key)) {

   186         V_VT(pVar) = VT_BOOL;

   187         if (m_result->isContentEditable())

   188             V_BOOL(pVar) = VARIANT_TRUE;

   189         else

   190             V_BOOL(pVar) = VARIANT_FALSE;

   191         return S_OK;

   192     }

   193 

   194     return E_INVALIDARG;

   195 }

   196 

   197 HRESULT STDMETHODCALLTYPE WebElementPropertyBag::Write(LPCOLESTR pszPropName, VARIANT* pVar)

   198 {

   199     if (!pszPropName || !pVar)

   200         return E_POINTER;

   201 

   202     return E_FAIL;

   203 }