#ifndef WebDataSource_H
#define WebDataSource_H
#include "WebKit.h"
#include <WebCore/COMPtr.h>
#include <wtf/RefPtr.h>
class WebDocumentLoader;
class WebMutableURLRequest;
extern const GUID IID_WebDataSource;
class WebDataSource : public IWebDataSource, public IWebDataSourcePrivate
{
public:
static WebDataSource* createInstance(WebDocumentLoader*);
protected:
WebDataSource(WebDocumentLoader*);
~WebDataSource();
public:
virtual HRESULT STDMETHODCALLTYPE QueryInterface(_In_ REFIID riid, _COM_Outptr_ void** ppvObject);
virtual ULONG STDMETHODCALLTYPE AddRef();
virtual ULONG STDMETHODCALLTYPE Release();
virtual HRESULT STDMETHODCALLTYPE initWithRequest(_In_opt_ IWebURLRequest*);
virtual HRESULT STDMETHODCALLTYPE data(_COM_Outptr_opt_ IStream**);
virtual HRESULT STDMETHODCALLTYPE representation(_COM_Outptr_opt_ IWebDocumentRepresentation**);
virtual HRESULT STDMETHODCALLTYPE webFrame(_COM_Outptr_opt_ IWebFrame**);
virtual HRESULT STDMETHODCALLTYPE initialRequest(_COM_Outptr_opt_ IWebURLRequest**);
virtual HRESULT STDMETHODCALLTYPE request(_COM_Outptr_opt_ IWebMutableURLRequest**);
virtual HRESULT STDMETHODCALLTYPE response(_COM_Outptr_opt_ IWebURLResponse**);
virtual HRESULT STDMETHODCALLTYPE textEncodingName(_Deref_opt_out_ BSTR*);
virtual HRESULT STDMETHODCALLTYPE isLoading(_Out_ BOOL*);
virtual HRESULT STDMETHODCALLTYPE pageTitle(_Deref_opt_out_ BSTR*);
virtual HRESULT STDMETHODCALLTYPE unreachableURL(_Deref_opt_out_ BSTR*);
virtual HRESULT STDMETHODCALLTYPE webArchive(_COM_Outptr_opt_ IWebArchive**);
virtual HRESULT STDMETHODCALLTYPE mainResource(_COM_Outptr_opt_ IWebResource**);
virtual HRESULT STDMETHODCALLTYPE subresources(_COM_Outptr_opt_ IEnumVARIANT** enumResources);
virtual HRESULT STDMETHODCALLTYPE subresourceForURL(_In_ BSTR url, _COM_Outptr_opt_ IWebResource**);
virtual HRESULT STDMETHODCALLTYPE addSubresource(_In_opt_ IWebResource*);
virtual HRESULT STDMETHODCALLTYPE overrideEncoding(_Deref_opt_out_ BSTR*);
virtual HRESULT STDMETHODCALLTYPE setOverrideEncoding(_In_ BSTR);
virtual HRESULT STDMETHODCALLTYPE mainDocumentError(_COM_Outptr_opt_ IWebError**);
virtual HRESULT STDMETHODCALLTYPE setDeferMainResourceDataLoad(BOOL);
WebDocumentLoader* documentLoader() const;
protected:
ULONG m_refCount { 0 };
RefPtr<WebDocumentLoader> m_loader;
COMPtr<IWebDocumentRepresentation> m_representation;
};
#endif