WebKit/win/WebHTMLRepresentation.cpp
changeset 0 4f2f89ce4247
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebKit/win/WebHTMLRepresentation.cpp	Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,319 @@
+/*
+ * 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 "WebHTMLRepresentation.h"
+
+#include "WebKit.h"
+#include "WebFrame.h"
+#include "WebKitStatisticsPrivate.h"
+#include <WebCore/BString.h>
+#include <WebCore/Frame.h>
+#include <WebCore/HTMLInputElement.h>
+#include <WebCore/TextResourceDecoder.h>
+
+using namespace WebCore;
+
+// WebHTMLRepresentation ------------------------------------------------------
+
+WebHTMLRepresentation::WebHTMLRepresentation()
+    : m_refCount(0)
+    , m_frame(0)
+{
+    WebHTMLRepresentationCount++;
+    gClassCount++;
+    gClassNameCount.add("WebHTMLRepresentation");
+}
+
+WebHTMLRepresentation::~WebHTMLRepresentation()
+{
+    if (m_frame) {
+        m_frame->Release();
+        m_frame = 0;
+    }
+
+    WebHTMLRepresentationCount--;
+    gClassCount--;
+    gClassNameCount.remove("WebHTMLRepresentation");
+}
+
+WebHTMLRepresentation* WebHTMLRepresentation::createInstance(WebFrame* frame)
+{
+    WebHTMLRepresentation* instance = new WebHTMLRepresentation();
+    instance->m_frame = frame;
+    frame->AddRef();
+    instance->AddRef();
+    return instance;
+}
+
+// IUnknown -------------------------------------------------------------------
+
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::QueryInterface(REFIID riid, void** ppvObject)
+{
+    *ppvObject = 0;
+    if (IsEqualGUID(riid, IID_IUnknown))
+        *ppvObject = static_cast<IWebHTMLRepresentation*>(this);
+    else if (IsEqualGUID(riid, IID_IWebHTMLRepresentation))
+        *ppvObject = static_cast<IWebHTMLRepresentation*>(this);
+    else if (IsEqualGUID(riid, IID_IWebDocumentRepresentation))
+        *ppvObject = static_cast<IWebDocumentRepresentation*>(this);
+    else
+        return E_NOINTERFACE;
+
+    AddRef();
+    return S_OK;
+}
+
+ULONG STDMETHODCALLTYPE WebHTMLRepresentation::AddRef()
+{
+    return ++m_refCount;
+}
+
+ULONG STDMETHODCALLTYPE WebHTMLRepresentation::Release()
+{
+    ULONG newRef = --m_refCount;
+    if (!newRef)
+        delete(this);
+
+    return newRef;
+}
+
+// IWebHTMLRepresentation --------------------------------------------------------------------
+
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedMIMETypes(
+        /* [out][in] */ BSTR* /*types*/,
+        /* [out][in] */ int* /*cTypes*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedNonImageMIMETypes(
+        /* [out][in] */ BSTR* /*types*/,
+        /* [out][in] */ int* /*cTypes*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedImageMIMETypes(
+        /* [out][in] */ BSTR* /*types*/,
+        /* [out][in] */ int* /*cTypes*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::attributedStringFromDOMNodes(
+        /* [in] */ IDOMNode* /*startNode*/,
+        /* [in] */ int /*startOffset*/,
+        /* [in] */ IDOMNode* /*endNode*/,
+        /* [in] */ int /*endOffset*/,
+        /* [retval][out] */ IDataObject** /*attributedString*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementWithName(
+        /* [in] */ BSTR name,
+        /* [in] */ IDOMElement* form,
+        /* [retval][out] */ IDOMElement** element)
+{
+    if (!m_frame)
+        return E_FAIL;
+
+    return m_frame->elementWithName(name, form, element);
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementDoesAutoComplete(
+        /* [in] */ IDOMElement* element,
+        /* [retval][out] */ BOOL* result)
+{
+    BOOL doesAutoComplete;
+    HRESULT hr = m_frame->elementDoesAutoComplete(element, &doesAutoComplete);
+    *result = doesAutoComplete ? TRUE : FALSE;
+    return hr;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementIsPassword(
+        /* [in] */ IDOMElement* element,
+        /* [retval][out] */ BOOL* result)
+{
+    bool isPassword;
+    HRESULT hr = m_frame->elementIsPassword(element, &isPassword);
+    *result = isPassword ?  TRUE : FALSE;
+    return hr;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::formForElement(
+        /* [in] */ IDOMElement* element,
+        /* [retval][out] */ IDOMElement** form)
+{
+    if (!m_frame)
+        return E_FAIL;
+
+    return m_frame->formForElement(element, form);
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::currentForm(
+        /* [retval][out] */ IDOMElement** form)
+{
+    if (!m_frame)
+        return E_FAIL;
+
+    return m_frame->currentForm(form);
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::controlsInForm(
+        /* [in] */ IDOMElement* form,
+        /* [out][in] */ IDOMElement** controls,
+        /* [out][in] */ int* cControls)
+{
+    return m_frame->controlsInForm(form, controls, cControls);
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::deprecatedSearchForLabels(
+        /* [size_is][in] */ BSTR* labels,
+        /* [in] */ int cLabels,
+        /* [in] */ IDOMElement* beforeElement,
+        /* [retval][out] */ BSTR* result)
+{
+    return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, 0, 0, result);
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::matchLabels(
+        /* [size_is][in] */ BSTR* labels,
+        /* [in] */ int cLabels,
+        /* [in] */ IDOMElement* againstElement,
+        /* [retval][out] */ BSTR* result)
+{
+    return m_frame->matchLabelsAgainstElement(labels, cLabels, againstElement, result);
+}
+
+HRESULT WebHTMLRepresentation::searchForLabels(BSTR* labels, unsigned cLabels, IDOMElement* beforeElement, unsigned* resultDistance, BOOL* resultIsInCellAbove, BSTR* result)
+{
+    return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, resultDistance, resultIsInCellAbove, result);
+}
+
+// IWebDocumentRepresentation ----------------------------------------------------------------
+
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::setDataSource(
+        /* [in] */ IWebDataSource* /*dataSource*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedData(
+        /* [in] */ IStream* /*data*/,
+        /* [in] */ IWebDataSource* /*dataSource*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedError(
+        /* [in] */ IWebError* /*error*/,
+        /* [in] */ IWebDataSource* /*dataSource*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::finishedLoadingWithDataSource(
+        /* [in] */ IWebDataSource* /*dataSource*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::canProvideDocumentSource(
+        /* [retval][out] */ BOOL* result)
+{
+    bool canProvideSource;
+    HRESULT hr = this->m_frame->canProvideDocumentSource(&canProvideSource);
+    *result = canProvideSource ? TRUE : FALSE;
+    return hr;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::documentSource(
+        /* [retval][out] */ BSTR* source)
+{
+    if (!source)
+        return E_FAIL;
+
+    *source = 0;
+
+    HRESULT hr = S_OK;
+
+    COMPtr<IWebDataSource> dataSource;
+    hr = m_frame->dataSource(&dataSource);
+    if (FAILED(hr))
+        return hr;
+
+    COMPtr<IStream> data;
+    hr = dataSource->data(&data);
+    if (FAILED(hr))
+        return hr;
+
+    STATSTG stat;
+    hr = data->Stat(&stat, STATFLAG_NONAME);
+    if (FAILED(hr))
+        return hr;
+
+    if (stat.cbSize.HighPart || !stat.cbSize.LowPart)
+        return E_FAIL;
+
+    Vector<char> dataBuffer(stat.cbSize.LowPart);
+    ULONG read;
+    
+    hr = data->Read(dataBuffer.data(), static_cast<ULONG>(dataBuffer.size()), &read);
+    if (FAILED(hr))
+        return hr;
+
+    WebCore::Frame* frame = core(m_frame);
+    if (!frame)
+        return E_FAIL;
+
+    WebCore::Document* doc = frame->document();
+    if (!doc)
+        return E_FAIL;
+
+    WebCore::TextResourceDecoder* decoder = doc->decoder();
+    if (!decoder)
+        return E_FAIL;
+
+    *source = WebCore::BString(decoder->encoding().decode(dataBuffer.data(), dataBuffer.size())).release();
+    return S_OK;
+}
+    
+HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::title(
+        /* [retval][out] */ BSTR* /*docTitle*/)
+{
+    ASSERT_NOT_REACHED();
+    return E_NOTIMPL;
+}