#ifndef WEB_VIEW_WINDOW_H

#define WEB_VIEW_WINDOW_H

#include "stdafx.h"

#include "HTMLMoniker.h"

#include "WebResourceResolver.h"

#include "LinkParams.h"

#include "LinkParamCollection.h"

#include "HTMLMoniker.h"

#include "util/Logging.h"

#include <GL/gl.h>

#include <vector>

#include <tlogstg.h>

extern const wchar_t *DEFAULT_BASE_URL;

const int DEFAULT_MIN_CONTENT_WIDTH = 300;

const int DEFAULT_MIN_CONTENT_HEIGHT = 100;

const UINT WM_SET_HTML = WM_APP + 0;

const UINT WM_GO_BACK = WM_APP + 2;

const UINT WM_GO_FORWARD = WM_APP + 3;

const UINT WM_WEBVIEW_CREATE = WM_APP + 4;

const UINT WM_WEBVIEW_DESTROY = WM_APP + 5;

const UINT WM_WEBVIEW_UPDATE = WM_APP + 6;

const UINT WM_SIM_MOUSEWHEEL = WM_APP + 8;

const UINT WM_WEBVIEW_ACTIVATE = WM_APP + 9;

const UINT WM_WEBVIEW_SET_BACKGROUND_COLOR = WM_APP + 10;

const UINT WM_WEBVIEW_SET_RESOURCE_RESOLVER = WM_APP + 11;

const UINT WM_WEBVIEW_SET_ADVISE = WM_APP + 12;

const UINT WM_WEBVIEW_SET_MIN_CONTENT_SIZE = WM_APP + 13;

const UINT WM_WEBVIEW_HIPRIORITY_LAST = WM_WEBVIEW_SET_ADVISE;

const UINT WM_WEBVIEW_CAPTURE = WM_APP + 14;

class WebViewTravelLogEntry

};

class ATL_NO_VTABLE WebViewWindow :

public CComObjectRootEx<CComSingleThreadModel>,

public CWindowImpl<WebViewWindow, CAxWindow>,

public IDispatch,

public IAdviseSink

{

WebViewWindow();

void FinalRelease();

BEGIN_COM_MAP(WebViewWindow)

COM_INTERFACE_ENTRY(IDispatch)

COM_INTERFACE_ENTRY(IAdviseSink)

END_COM_MAP()

DECLARE_NOT_AGGREGATABLE(WebViewWindow)

BEGIN_MSG_MAP(WebViewWindow)

MESSAGE_HANDLER(WM_WEBVIEW_CAPTURE, OnCapture)

MESSAGE_HANDLER(WM_WEBVIEW_ACTIVATE, OnActivate)

MESSAGE_HANDLER(WM_SET_HTML, OnSetHTML)

MESSAGE_HANDLER(WM_TIMER, OnTimer)

MESSAGE_HANDLER(WM_GO_BACK, OnGoBack)

MESSAGE_HANDLER(WM_GO_FORWARD, OnGoForward)

MESSAGE_HANDLER(WM_WEBVIEW_SET_BACKGROUND_COLOR, OnSetBackgroundColor)

MESSAGE_HANDLER(WM_WEBVIEW_SET_RESOURCE_RESOLVER, OnSetResourceResolver)

MESSAGE_HANDLER(WM_WEBVIEW_SET_ADVISE, OnSetAdvise)

MESSAGE_HANDLER(WM_WEBVIEW_SET_MIN_CONTENT_SIZE, OnSetMinContentSize)

MESSAGE_HANDLER(WM_SIZE, OnFrameSizeChanged);

// The following events will be forwarded to the web browser control

MESSAGE_HANDLER(WM_KEYDOWN, OnSimulateInput);

MESSAGE_HANDLER(WM_KEYUP, OnSimulateInput);

MESSAGE_HANDLER(WM_MOUSEMOVE, OnSimulateInput);

MESSAGE_HANDLER(WM_LBUTTONDOWN, OnSimulateInput);

MESSAGE_HANDLER(WM_LBUTTONUP, OnSimulateInput);

MESSAGE_HANDLER(WM_RBUTTONDOWN, OnSimulateInput);

MESSAGE_HANDLER(WM_RBUTTONUP, OnSimulateInput);

MESSAGE_HANDLER(WM_MBUTTONDOWN, OnSimulateInput);

MESSAGE_HANDLER(WM_MBUTTONUP, OnSimulateInput);

MESSAGE_HANDLER(WM_LBUTTONDBLCLK, OnSimulateInput);

MESSAGE_HANDLER(WM_RBUTTONDBLCLK, OnSimulateInput);

MESSAGE_HANDLER(WM_MBUTTONDBLCLK, OnSimulateInput);

MESSAGE_HANDLER(WM_SIM_MOUSEWHEEL, OnSimulateInput);

END_MSG_MAP()

/**

static WebViewWindow* WebViewWindow::FindWebView(LONG_PTR webViewId);

/** Create the WebBrowser control. This method must be called on the thread will own this window. */

HRESULT CreateWebBrowser();

/**

LONG_PTR GetWebViewId() const;

/**

void GetResourceResolver(WebResourceResolver **resolver) const;

/**

BOOL CaptureToGLTexture(GLenum target);

/**

DWORD GetUpdateTime() const { return this->updateTime; }

/**

void GetLinks(LinkParamCollection **linkCollection) const;

/** Get the size of the scrollable content. */

void GetContentSize(int *width, int *height) const;

/**

void GetMinContentSize(int *width, int *height) const;

/**

BSTR GetContentURL() const;

/**

void ScheduleCapture();

/**

HWND GetControlWindow() const { return this->controlWnd; }

/**

void SetControlWindow(HWND wnd) { this->controlWnd = wnd; }

// IDispatch

STDMETHODIMP GetTypeInfoCount(UINT *pctinfo);

STDMETHODIMP GetTypeInfo(UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo);

STDMETHODIMP GetIDsOfNames(REFIID riid, LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId);

STDMETHODIMP Invoke(DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags,

DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo,

UINT *puArgErr);

// IAdviseSink

void STDMETHODCALLTYPE OnDataChange(FORMATETC *pFormatetc, STGMEDIUM *pStgmed) { };

void STDMETHODCALLTYPE OnViewChange(DWORD dwAspect, LONG lindex);

void STDMETHODCALLTYPE OnRename(IMoniker *pmk) { }

void STDMETHODCALLTYPE OnSave() { }

void STDMETHODCALLTYPE OnClose() { }

/**

static BOOL protocolInitialized;

/**

void InitializeWebViewProtocol();

/** Handle to the message-only control window that manages this WebView. */

HWND controlWnd;

/** The WebBrowser control. */

IWebBrowser2 *browser;

/** Cookie needed for IDispatch. */

DWORD dispatchCookie;

/**

HWND hWndBrowser;

/** Handle to the hidden window that handles scrollbar input. */

HWND hWndScrollControl;

/**

HWND hWndUnderCursor;

/** Indicates that the rendered content has changed, and needs to be captured. */

BOOL needToCapture;

/** Indicates that the WebViewWindow has user input focus. The mouse is within the window. */

BOOL active;

/**

BOOL alwaysCapture;

/** Content string that was originally loaded into browser. */

HTMLMoniker *htmlContent;

/**

BOOL browserInitialized;

/** Mutex to synchronize access to the bitmap. */

HANDLE bitmapMutex;

/** Handle to Device Independent Bitmap. */

HBITMAP hCaptureDIB;

/** Width of the captured image. */

int captureWidth;

/** Height of the captured image. */

int captureHeight;

/** Bitmap contents. */

char *captureBits;

/** Indicates that a message has been sent that will trigger a WebView capture some time in the future. */

BOOL captureScheduled;

/** Time at which the rendered page was last updated. */

DWORD updateTime;

/** Mutex to synchronize access fields read by the WebView thread and Java threads. */

HANDLE mutex;

/** Links in the current content. */

LinkParamCollection *links;

/** Listener that will be notified when the rendered page changes. */

IAdviseSink *adviseSink;

/** Object to resolver relative references. */

WebResourceResolver *resourceResolver;

/** Indicates if the WebView is scrolling. */

BOOL isScrolling;

/** Indicates which scroll bar thumb is active. NULL if no thumb is active. */

const char *activeScroller;

/** The mouse point at which a scroll operation started. */

LPARAM scrollRefPoint;

/** The element that is scrolling. NULL if no element is scrolling. */

IHTMLElement2 *scrollElement;

/**

long scrollRefPosition;

/** Last point to receive mouse input. This is used to fake mouse hover events. */

LPARAM lastInputPoint;

/** Background color (as a Hex string) of the WebView. */

wchar_t *backgroundColor;

/**

DWORD contentMetadataUpdateID;

/** ID that is incremented when the new content is loaded. */

DWORD contentLoadID;

/** Width of the WebView content. This is equal to the scroll width of the page, as reported by the WebBrowser. */

int contentWidth;

/** Height of the WebView content. This is equal to the scroll width of the page, as reported by the WebBrowser. */

int contentHeight;

/** Minimum width of the WebView content. */

int minContentWidth;

/** Minimum height of the WebView content. */

int minContentHeight;

/** Indicates whether or not the original WebView content is loaded. */

BOOL originalContentLoaded;

/** URL of the currently loaded content, or NULL if the content was loaded from an HTML moniker. */

BSTR contentURL;

/**

std::vector<WebViewTravelLogEntry> savedTravelLog;

/**

BOOL mustClearTravelLog;

/** Load the original HTML content from the htmlContent field. */

HRESULT SetHTML();

/**

BOOL CaptureWebView();

/** Assign an ID number to this WebView. */

void AssignWebViewId();

/** Create a bitmap representation of the current WebView contents. */

int CaptureBitMap(IViewObject *pViewObject);

/**

void DrawCaret(HDC hDC) const;

/**

BOOL OnDocumentComplete();

/**

void MoveWindowToCursor(int clientX, int clientY);

/**

BSTR GetComponentAtPoint(int x, int y) const;

/**

HRESULT ApplyBackgroundColor();

/**

BOOL ContainsEmbeddedContent() const;

/** Enable or disable scroll bars depending on the size of the content. */

void DetermineScrollBars();

/** Determine the size of the scrollable content. */

HRESULT DetermineContentSize();

/** Determine the size of the scrollable content in compatibility mode (quirks mode). */

HRESULT DetermineContentSizeCompatibilityMode(IHTMLDocument2 *pDoc);

/** Determine the size of the scrollable content in standards mode (strict mode). */

HRESULT DetermineContentSizeStandardsMode(IHTMLDocument2 *pDoc);

/** Determine the URL of the content currently loaded in the WebView. */

void DetermineContentURL();

/**

BSTR GetLinkUrl(IHTMLAnchorElement *anchor) const;

/**

void TrackScrollThumb(int x, int y);

/**

void AutoScroll();

/** Remove all entries from the brower's back/forward navigation list (TravelLog). */

void ClearTravelLog();

protected: // Link finding methods

/**

HRESULT FindLinks();

/**

LinkParams *GetLinkParams(IHTMLAnchorElement *anchor, RECT *viewport) const;

/**

BOOL IsVisible(IHTMLElement *element) const;

/**

void AddLineBoxRects(LinkParams *linkParams, IHTMLRectCollection *rects, RECT *viewport) const;

/**

void AddImageRects(LinkParams *linkParams, IHTMLElement2 *element2, IHTMLCurrentStyle *anchorStyle, RECT *linkRect, RECT *viewport) const;

/**

void ComputeRectIntersect(RECT *intersection, const RECT *r1, const RECT *r2) const;

/**

void ComputeRectUnion(RECT *unionRect, const RECT *r1, const RECT *r2) const;

/**

BOOL IsEmptyRect(const RECT *rect) const;

/**

BOOL RectsIntersect(const RECT *r1, const RECT *r2) const;

// Message handlers

/** Handler for WM_SIZE message. */

LRESULT OnFrameSizeChanged(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_TIMER message. */

LRESULT OnTimer(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_WEBVIEW_ACTIVATE message. */

LRESULT OnActivate(UINT uMsg, WPARAM active, LPARAM unused, BOOL bHandled);

/** Send a simulated input message to the web browser. */

LRESULT OnSimulateInput(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_SET_HTML message. */

LRESULT OnSetHTML(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_GO_BACK message. */

LRESULT OnGoBack(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_GO_FORWARD message. */

LRESULT OnGoForward(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_WEBVIEW_SET_BACKGROUND_COLOR message. */

LRESULT OnSetBackgroundColor(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_WEBVIEW_SET_RESOURCE_RESOLVER message. */

LRESULT OnSetResourceResolver(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_WEBVIEW_SET_ADVISE message. */

LRESULT OnSetAdvise(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

/** Handler for WM_WEBVIEW_SET_MIN_CONTENT_SIZE message. */

LRESULT OnSetMinContentSize(UINT uMsg, WPARAM width, LPARAM height, BOOL bHandled);

/** Handler for WM_WEBVIEW_CAPTURE message. */

LRESULT OnCapture(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL bHandled);

private: // Debugging support

/** Save the current bitmap capture to a file. */

void WriteBitmapToFile(wchar_t *fileName) const;

/** Write the contents of the TravelLog to the debugger console. */

void DumpTravelLog() const;

};

#endif