/* Anakrino - .NET Reflection Browser
 * Copyright (C) 2001-2002  Jay Freeman (saurik)
*/

/*
 *        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.
 * 3. The name of the author may not be used to endorse
 *    or promote products derived from this software
 *    without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 THE AUTHOR 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.
*/

#ifndef ANAKRINO_IEBROWSER_H
#define ANAKRINO_IEBROWSER_H

#if _MSC_VER >= 1000
#pragma once
#endif // _MSC_VER >= 1000

namespace Anakrino {

static _ATL_FUNC_INFO OnNavigateComplete2Info = {CC_STDCALL, VT_ERROR, 1, {VT_DISPATCH, VT_BYREF | VT_VARIANT}};
static _ATL_FUNC_INFO OnDocumentCompleteInfo = {CC_STDCALL, VT_ERROR, 2, {VT_DISPATCH, VT_BYREF | VT_VARIANT}};

template<typename base>
class IEBrowser :
	public CWindowImpl<base, CAxWindow>,
	public IDispEventSimpleImpl<31337, IEBrowser, &__uuidof(SHDocVw::DWebBrowserEvents2)>
{
  protected:
	SHDocVw::IWebBrowserPtr browser;
	MSHTML::IHTMLDocumentPtr document;
	MSHTML::IHTMLDocument2Ptr document2;
	MSHTML::IHTMLDocument3Ptr document3;

  public:
	BEGIN_SINK_MAP(IEBrowser)
		SINK_ENTRY_INFO(31337, __uuidof(SHDocVw::DWebBrowserEvents2), 0x103, OnDocumentComplete, &OnDocumentCompleteInfo)
	END_SINK_MAP()

	BOOL PreTranslateMessage(MSG* pMsg) {
		if ((pMsg->message >= WM_KEYFIRST	&& pMsg->message <= WM_KEYLAST) &&
			(pMsg->message >= WM_MOUSEFIRST	&& pMsg->message <= WM_MOUSELAST))
			return (BOOL) SendMessage(WM_FORWARDMSG, 0, (LPARAM)pMsg);
		return FALSE;
	}

	virtual void Select(const Metallurgy::Token *token) = 0;

	HRESULT __stdcall OnDocumentComplete(IDispatch *pDisp, VARIANT *URL) {
		document = browser->GetDocument();
		document3 = document2 = document;
		Select(NULL);
		return S_OK;
	}
};

}

#endif//ANAKRINO_IEBROWSER_H