Source/WebKit/win/WebHTMLRepresentation.cpp

   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 "WebHTMLRepresentation.h"
  29
  30 #include "WebKit.h"
  31 #include "WebFrame.h"
  32 #include "WebKitStatisticsPrivate.h"
  33 #include <WebCore/BString.h>
  34 #include <WebCore/Frame.h>
  35 #include <WebCore/HTMLInputElement.h>
  36 #include <WebCore/TextResourceDecoder.h>
  37
  38 using namespace WebCore;
  39
  40 // WebHTMLRepresentation ------------------------------------------------------
  41
  42 WebHTMLRepresentation::WebHTMLRepresentation()
  43     : m_refCount(0)
  44     , m_frame(0)
  45 {
  46     WebHTMLRepresentationCount++;
  47     gClassCount++;
  48     gClassNameCount.add("WebHTMLRepresentation");
  49 }
  50
  51 WebHTMLRepresentation::~WebHTMLRepresentation()
  52 {
  53     if (m_frame) {
  54         m_frame->Release();
  55         m_frame = 0;
  56     }
  57
  58     WebHTMLRepresentationCount--;
  59     gClassCount--;
  60     gClassNameCount.remove("WebHTMLRepresentation");
  61 }
  62
  63 WebHTMLRepresentation* WebHTMLRepresentation::createInstance(WebFrame* frame)
  64 {
  65     WebHTMLRepresentation* instance = new WebHTMLRepresentation();
  66     instance->m_frame = frame;
  67     frame->AddRef();
  68     instance->AddRef();
  69     return instance;
  70 }
  71
  72 // IUnknown -------------------------------------------------------------------
  73
  74 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::QueryInterface(REFIID riid, void** ppvObject)
  75 {
  76     *ppvObject = 0;
  77     if (IsEqualGUID(riid, IID_IUnknown))
  78         *ppvObject = static_cast<IWebHTMLRepresentation*>(this);
  79     else if (IsEqualGUID(riid, IID_IWebHTMLRepresentation))
  80         *ppvObject = static_cast<IWebHTMLRepresentation*>(this);
  81     else if (IsEqualGUID(riid, IID_IWebDocumentRepresentation))
  82         *ppvObject = static_cast<IWebDocumentRepresentation*>(this);
  83     else
  84         return E_NOINTERFACE;
  85
  86     AddRef();
  87     return S_OK;
  88 }
  89
  90 ULONG STDMETHODCALLTYPE WebHTMLRepresentation::AddRef()
  91 {
  92     return ++m_refCount;
  93 }
  94
  95 ULONG STDMETHODCALLTYPE WebHTMLRepresentation::Release()
  96 {
  97     ULONG newRef = --m_refCount;
  98     if (!newRef)
  99         delete(this);
 100
 101     return newRef;
 102 }
 103
 104 // IWebHTMLRepresentation --------------------------------------------------------------------
 105
 106 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedMIMETypes(
 107         /* [out][in] */ BSTR* /*types*/,
 108         /* [out][in] */ int* /*cTypes*/)
 109 {
 110     ASSERT_NOT_REACHED();
 111     return E_NOTIMPL;
 112 }
 113
 114 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedNonImageMIMETypes(
 115         /* [out][in] */ BSTR* /*types*/,
 116         /* [out][in] */ int* /*cTypes*/)
 117 {
 118     ASSERT_NOT_REACHED();
 119     return E_NOTIMPL;
 120 }
 121
 122 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedImageMIMETypes(
 123         /* [out][in] */ BSTR* /*types*/,
 124         /* [out][in] */ int* /*cTypes*/)
 125 {
 126     ASSERT_NOT_REACHED();
 127     return E_NOTIMPL;
 128 }
 129
 130 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::attributedStringFromDOMNodes(
 131         /* [in] */ IDOMNode* /*startNode*/,
 132         /* [in] */ int /*startOffset*/,
 133         /* [in] */ IDOMNode* /*endNode*/,
 134         /* [in] */ int /*endOffset*/,
 135         /* [retval][out] */ IDataObject** /*attributedString*/)
 136 {
 137     ASSERT_NOT_REACHED();
 138     return E_NOTIMPL;
 139 }
 140
 141 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementWithName(
 142         /* [in] */ BSTR name,
 143         /* [in] */ IDOMElement* form,
 144         /* [retval][out] */ IDOMElement** element)
 145 {
 146     if (!m_frame)
 147         return E_FAIL;
 148
 149     return m_frame->elementWithName(name, form, element);
 150 }
 151
 152 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementDoesAutoComplete(
 153         /* [in] */ IDOMElement* element,
 154         /* [retval][out] */ BOOL* result)
 155 {
 156     BOOL doesAutoComplete;
 157     HRESULT hr = m_frame->elementDoesAutoComplete(element, &doesAutoComplete);
 158     *result = doesAutoComplete ? TRUE : FALSE;
 159     return hr;
 160 }
 161
 162 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementIsPassword(
 163         /* [in] */ IDOMElement* element,
 164         /* [retval][out] */ BOOL* result)
 165 {
 166     bool isPassword;
 167     HRESULT hr = m_frame->elementIsPassword(element, &isPassword);
 168     *result = isPassword ?  TRUE : FALSE;
 169     return hr;
 170 }
 171
 172 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::formForElement(
 173         /* [in] */ IDOMElement* element,
 174         /* [retval][out] */ IDOMElement** form)
 175 {
 176     if (!m_frame)
 177         return E_FAIL;
 178
 179     return m_frame->formForElement(element, form);
 180 }
 181
 182 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::currentForm(
 183         /* [retval][out] */ IDOMElement** form)
 184 {
 185     if (!m_frame)
 186         return E_FAIL;
 187
 188     return m_frame->currentForm(form);
 189 }
 190
 191 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::controlsInForm(
 192         /* [in] */ IDOMElement* form,
 193         /* [out][in] */ IDOMElement** controls,
 194         /* [out][in] */ int* cControls)
 195 {
 196     return m_frame->controlsInForm(form, controls, cControls);
 197 }
 198
 199 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::deprecatedSearchForLabels(
 200         /* [size_is][in] */ BSTR* labels,
 201         /* [in] */ int cLabels,
 202         /* [in] */ IDOMElement* beforeElement,
 203         /* [retval][out] */ BSTR* result)
 204 {
 205     return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, 0, 0, result);
 206 }
 207
 208 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::matchLabels(
 209         /* [size_is][in] */ BSTR* labels,
 210         /* [in] */ int cLabels,
 211         /* [in] */ IDOMElement* againstElement,
 212         /* [retval][out] */ BSTR* result)
 213 {
 214     return m_frame->matchLabelsAgainstElement(labels, cLabels, againstElement, result);
 215 }
 216
 217 HRESULT WebHTMLRepresentation::searchForLabels(BSTR* labels, unsigned cLabels, IDOMElement* beforeElement, unsigned* resultDistance, BOOL* resultIsInCellAbove, BSTR* result)
 218 {
 219     return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, resultDistance, resultIsInCellAbove, result);
 220 }
 221
 222 // IWebDocumentRepresentation ----------------------------------------------------------------
 223
 224 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::setDataSource(
 225         /* [in] */ IWebDataSource* /*dataSource*/)
 226 {
 227     ASSERT_NOT_REACHED();
 228     return E_NOTIMPL;
 229 }
 230
 231 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedData(
 232         /* [in] */ IStream* /*data*/,
 233         /* [in] */ IWebDataSource* /*dataSource*/)
 234 {
 235     ASSERT_NOT_REACHED();
 236     return E_NOTIMPL;
 237 }
 238
 239 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedError(
 240         /* [in] */ IWebError* /*error*/,
 241         /* [in] */ IWebDataSource* /*dataSource*/)
 242 {
 243     ASSERT_NOT_REACHED();
 244     return E_NOTIMPL;
 245 }
 246
 247 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::finishedLoadingWithDataSource(
 248         /* [in] */ IWebDataSource* /*dataSource*/)
 249 {
 250     ASSERT_NOT_REACHED();
 251     return E_NOTIMPL;
 252 }
 253
 254 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::canProvideDocumentSource(
 255         /* [retval][out] */ BOOL* result)
 256 {
 257     bool canProvideSource;
 258     HRESULT hr = this->m_frame->canProvideDocumentSource(&canProvideSource);
 259     *result = canProvideSource ? TRUE : FALSE;
 260     return hr;
 261 }
 262
 263 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::documentSource(
 264         /* [retval][out] */ BSTR* source)
 265 {
 266     if (!source)
 267         return E_FAIL;
 268
 269     *source = 0;
 270
 271     HRESULT hr = S_OK;
 272
 273     COMPtr<IWebDataSource> dataSource;
 274     hr = m_frame->dataSource(&dataSource);
 275     if (FAILED(hr))
 276         return hr;
 277
 278     COMPtr<IStream> data;
 279     hr = dataSource->data(&data);
 280     if (FAILED(hr))
 281         return hr;
 282
 283     STATSTG stat;
 284     hr = data->Stat(&stat, STATFLAG_NONAME);
 285     if (FAILED(hr))
 286         return hr;
 287
 288     if (stat.cbSize.HighPart || !stat.cbSize.LowPart)
 289         return E_FAIL;
 290
 291     Vector<char> dataBuffer(stat.cbSize.LowPart);
 292     ULONG read;
 293
 294     hr = data->Read(dataBuffer.data(), static_cast<ULONG>(dataBuffer.size()), &read);
 295     if (FAILED(hr))
 296         return hr;
 297
 298     WebCore::Frame* frame = core(m_frame);
 299     if (!frame)
 300         return E_FAIL;
 301
 302     WebCore::Document* doc = frame->document();
 303     if (!doc)
 304         return E_FAIL;
 305
 306     WebCore::TextResourceDecoder* decoder = doc->decoder();
 307     if (!decoder)
 308         return E_FAIL;
 309
 310     *source = WebCore::BString(decoder->encoding().decode(dataBuffer.data(), dataBuffer.size())).release();
 311     return S_OK;
 312 }
 313
 314 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::title(
 315         /* [retval][out] */ BSTR* /*docTitle*/)
 316 {
 317     ASSERT_NOT_REACHED();
 318     return E_NOTIMPL;
 319 }