#include "wx/msw/missing.h"
#include "wx/filesys.h"
#include "wx/dynlib.h"
+#include <initguid.h>
+
+/* These GUID definitions are our own implementation to support interfaces
+ * normally in urlmon.h. See include/wx/msw/webview_ie.h
+ */
+
+namespace {
+
+DEFINE_GUID(wxIID_IInternetProtocolRoot,0x79eac9e3,0xbaf9,0x11ce,0x8c,0x82,0,0xaa,0,0x4b,0xa9,0xb);
+DEFINE_GUID(wxIID_IInternetProtocol,0x79eac9e4,0xbaf9,0x11ce,0x8c,0x82,0,0xaa,0,0x4b,0xa9,0xb);
+DEFINE_GUID(wxIID_IDocHostUIHandler, 0xbd3f23c0, 0xd43e, 0x11cf, 0x89, 0x3b, 0x00, 0xaa, 0x00, 0xbd, 0xce, 0x1a);
+DEFINE_GUID(wxIID_IHTMLElement2,0x3050f434,0x98b5,0x11cf,0xbb,0x82,0,0xaa,0,0xbd,0xce,0x0b);
+DEFINE_GUID(wxIID_IMarkupServices,0x3050f4a0,0x98b5,0x11cf,0xbb,0x82,0,0xaa,0,0xbd,0xce,0x0b);
+DEFINE_GUID(wxIID_IMarkupContainer,0x3050f5f9,0x98b5,0x11cf,0xbb,0x82,0,0xaa,0,0xbd,0xce,0x0b);
+
+enum //Internal find flags
+{
+ wxWEB_VIEW_FIND_ADD_POINTERS = 0x0001,
+ wxWEB_VIEW_FIND_REMOVE_HIGHLIGHT = 0x0002
+};
+
+}
wxIMPLEMENT_DYNAMIC_CLASS(wxWebViewIE, wxWebView);
m_historyEnabled = true;
m_historyPosition = -1;
m_zoomType = wxWEB_VIEW_ZOOM_TYPE_TEXT;
+ FindClear();
if (::CoCreateInstance(CLSID_WebBrowser, NULL,
CLSCTX_INPROC_SERVER, // CLSCTX_INPROC,
m_webBrowser->put_RegisterAsBrowser(VARIANT_TRUE);
m_webBrowser->put_RegisterAsDropTarget(VARIANT_TRUE);
- m_container = new wxActiveXContainer(this, IID_IWebBrowser2, m_webBrowser);
+ m_uiHandler = new DocHostUIHandler;
+
+ m_container = new wxIEContainer(this, IID_IWebBrowser2, m_webBrowser, m_uiHandler);
+
+ EnableControlFeature(21 /* FEATURE_DISABLE_NAVIGATION_SOUNDS */);
- SetBackgroundStyle(wxBG_STYLE_PAINT);
- SetDoubleBuffered(true);
LoadURL(url);
return true;
}
wxWebViewIE::~wxWebViewIE()
{
- for(unsigned int i = 0; i < m_factories.size(); i++)
+ wxDynamicLibrary urlMon(wxT("urlmon.dll"));
+ if(urlMon.HasSymbol(wxT("CoInternetGetSession")))
{
- m_factories[i]->Release();
+ typedef HRESULT (WINAPI *CoInternetGetSession_t)(DWORD,
+ wxIInternetSession**,
+ DWORD);
+ wxDYNLIB_FUNCTION(CoInternetGetSession_t, CoInternetGetSession, urlMon);
+
+ wxIInternetSession* session;
+ HRESULT res = (*pfnCoInternetGetSession)(0, &session, 0);
+ if(FAILED(res))
+ {
+ wxFAIL_MSG("Could not retrive internet session");
+ }
+
+ for(unsigned int i = 0; i < m_factories.size(); i++)
+ {
+ session->UnregisterNameSpace(m_factories[i],
+ (m_factories[i]->GetName()).wc_str());
+ m_factories[i]->Release();
+ }
}
+ FindClear();
}
void wxWebViewIE::LoadURL(const wxString& url)
{
- m_ie.CallMethod("Navigate", (BSTR) url.wc_str());
+ m_ie.CallMethod("Navigate", wxConvertStringToOle(url));
}
-void wxWebViewIE::SetPage(const wxString& html, const wxString& baseUrl)
+void wxWebViewIE::DoSetPage(const wxString& html, const wxString& baseUrl)
{
- BSTR bstr = SysAllocString(html.wc_str());
-
- // Creates a new one-dimensional array
+ BSTR bstr = SysAllocString(OLESTR(""));
SAFEARRAY *psaStrings = SafeArrayCreateVector(VT_VARIANT, 0, 1);
if (psaStrings != NULL)
{
VARIANT *param;
-
HRESULT hr = SafeArrayAccessData(psaStrings, (LPVOID*)¶m);
param->vt = VT_BSTR;
param->bstrVal = bstr;
+
hr = SafeArrayUnaccessData(psaStrings);
- IHTMLDocument2* document = GetDocument();
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(!document)
+ return;
+
document->write(psaStrings);
- document->Release();
+ document->close();
- // SafeArrayDestroy calls SysFreeString for each BSTR
SafeArrayDestroy(psaStrings);
- //We send the events when we are done to mimic webkit
- //Navigated event
- wxWebViewEvent event(wxEVT_COMMAND_WEB_VIEW_NAVIGATED,
- GetId(), baseUrl, "");
- event.SetEventObject(this);
- HandleWindowEvent(event);
-
- //Document complete event
- event.SetEventType(wxEVT_COMMAND_WEB_VIEW_LOADED);
- event.SetEventObject(this);
- HandleWindowEvent(event);
+ bstr = SysAllocString(html.wc_str());
+
+ // Creates a new one-dimensional array
+ psaStrings = SafeArrayCreateVector(VT_VARIANT, 0, 1);
+ if (psaStrings != NULL)
+ {
+ hr = SafeArrayAccessData(psaStrings, (LPVOID*)¶m);
+ param->vt = VT_BSTR;
+ param->bstrVal = bstr;
+ hr = SafeArrayUnaccessData(psaStrings);
+
+ document = GetDocument();
+
+ if(!document)
+ return;
+
+ document->write(psaStrings);
+
+ // SafeArrayDestroy calls SysFreeString for each BSTR
+ SafeArrayDestroy(psaStrings);
+
+ //We send the events when we are done to mimic webkit
+ //Navigated event
+ wxWebViewEvent event(wxEVT_COMMAND_WEB_VIEW_NAVIGATED,
+ GetId(), baseUrl, "");
+ event.SetEventObject(this);
+ HandleWindowEvent(event);
+
+ //Document complete event
+ event.SetEventType(wxEVT_COMMAND_WEB_VIEW_LOADED);
+ event.SetEventObject(this);
+ HandleWindowEvent(event);
+ }
+ else
+ {
+ wxLogError("wxWebViewIE::SetPage() : psaStrings is NULL");
+ }
}
else
{
- wxLogError("wxWebViewIE::SetPage() : psaStrings is NULL");
+ wxLogError("wxWebViewIE::SetPage() : psaStrings is NULL during clear");
}
-
}
wxString wxWebViewIE::GetPageSource() const
{
- IHTMLDocument2* document = GetDocument();
- IHTMLElement *bodyTag = NULL;
- IHTMLElement *htmlTag = NULL;
- wxString source;
- HRESULT hr = document->get_body(&bodyTag);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
{
- hr = bodyTag->get_parentElement(&htmlTag);
+ wxCOMPtr<IHTMLElement> bodyTag;
+ wxCOMPtr<IHTMLElement> htmlTag;
+ wxString source;
+ HRESULT hr = document->get_body(&bodyTag);
if(SUCCEEDED(hr))
{
- BSTR bstr;
- htmlTag->get_outerHTML(&bstr);
- source = wxString(bstr);
- htmlTag->Release();
+ hr = bodyTag->get_parentElement(&htmlTag);
+ if(SUCCEEDED(hr))
+ {
+ BSTR bstr;
+ htmlTag->get_outerHTML(&bstr);
+ source = wxString(bstr);
+ }
}
- bodyTag->Release();
+ return source;
+ }
+ else
+ {
+ return "";
}
-
- document->Release();
- return source;
}
wxWebViewZoom wxWebViewIE::GetZoom() const
}
bool wxWebViewIE::IsBusy() const
-{
+{
if (m_isBusy) return true;
wxVariant out = m_ie.GetProperty("Busy");
wxString wxWebViewIE::GetCurrentTitle() const
{
- IHTMLDocument2* document = GetDocument();
- BSTR title;
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
- document->get_nameProp(&title);
- document->Release();
- return wxString(title);
+ if(document)
+ {
+ BSTR title;
+ document->get_nameProp(&title);
+ return wxString(title);
+ }
+ else
+ {
+ return "";
+ }
}
bool wxWebViewIE::CanCut() const
{
return CanExecCommand("Copy");
}
+
bool wxWebViewIE::CanPaste() const
{
return CanExecCommand("Paste");
{
return CanExecCommand("Undo");
}
+
bool wxWebViewIE::CanRedo() const
{
return CanExecCommand("Redo");
ExecCommand("Redo");
}
+long wxWebViewIE::Find(const wxString& text, int flags)
+{
+ //If the text is empty then we clear.
+ if(text.IsEmpty())
+ {
+ ClearSelection();
+ if(m_findFlags & wxWEB_VIEW_FIND_HIGHLIGHT_RESULT)
+ {
+ FindInternal(m_findText, (m_findFlags &~ wxWEB_VIEW_FIND_HIGHLIGHT_RESULT), wxWEB_VIEW_FIND_REMOVE_HIGHLIGHT);
+ }
+ FindClear();
+ return wxNOT_FOUND;
+ }
+ //Have we done this search before?
+ if(m_findText == text)
+ {
+ //Just do a highlight?
+ if((flags & wxWEB_VIEW_FIND_HIGHLIGHT_RESULT) != (m_findFlags & wxWEB_VIEW_FIND_HIGHLIGHT_RESULT))
+ {
+ m_findFlags = flags;
+ if(!m_findPointers.empty())
+ {
+ FindInternal(m_findText, m_findFlags, ((flags & wxWEB_VIEW_FIND_HIGHLIGHT_RESULT) == 0 ? wxWEB_VIEW_FIND_REMOVE_HIGHLIGHT : 0));
+ }
+ return m_findPosition;
+ }
+ else if(((m_findFlags & wxWEB_VIEW_FIND_ENTIRE_WORD) == (flags & wxWEB_VIEW_FIND_ENTIRE_WORD)) && ((m_findFlags & wxWEB_VIEW_FIND_MATCH_CASE) == (flags&wxWEB_VIEW_FIND_MATCH_CASE)))
+ {
+ m_findFlags = flags;
+ return FindNext(((flags & wxWEB_VIEW_FIND_BACKWARDS) ? -1 : 1));
+ }
+ }
+ //Remove old highlight if any.
+ if(m_findFlags & wxWEB_VIEW_FIND_HIGHLIGHT_RESULT)
+ {
+ FindInternal(m_findText, (m_findFlags &~ wxWEB_VIEW_FIND_HIGHLIGHT_RESULT), wxWEB_VIEW_FIND_REMOVE_HIGHLIGHT);
+ }
+ //Reset find variables.
+ FindClear();
+ ClearSelection();
+ m_findText = text;
+ m_findFlags = flags;
+ //find the text and return count.
+ FindInternal(text, flags, wxWEB_VIEW_FIND_ADD_POINTERS);
+ return m_findPointers.empty() ? wxNOT_FOUND : m_findPointers.size();
+}
+
void wxWebViewIE::SetEditable(bool enable)
{
- IHTMLDocument2* document = GetDocument();
- if( enable )
- document->put_designMode(SysAllocString(L"On"));
- else
- document->put_designMode(SysAllocString(L"Off"));
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
- document->Release();
+ if(document)
+ {
+ if( enable )
+ document->put_designMode(SysAllocString(L"On"));
+ else
+ document->put_designMode(SysAllocString(L"Off"));
+
+ }
}
bool wxWebViewIE::IsEditable() const
{
- IHTMLDocument2* document = GetDocument();
- BSTR mode;
- document->get_designMode(&mode);
- document->Release();
- if(wxString(mode) == "On")
- return true;
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ BSTR mode;
+ document->get_designMode(&mode);
+ if(wxString(mode) == "On")
+ return true;
+ else
+ return false;
+ }
else
+ {
return false;
+ }
}
void wxWebViewIE::SelectAll()
bool wxWebViewIE::HasSelection() const
{
- IHTMLDocument2* document = GetDocument();
- IHTMLSelectionObject* selection;
- wxString sel;
- HRESULT hr = document->get_selection(&selection);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
{
- BSTR type;
- selection->get_type(&type);
- sel = wxString(type);
- selection->Release();
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString sel;
+ HRESULT hr = document->get_selection(&selection);
+ if(SUCCEEDED(hr))
+ {
+ BSTR type;
+ selection->get_type(&type);
+ sel = wxString(type);
+ }
+ return sel != "None";
+ }
+ else
+ {
+ return false;
}
- document->Release();
- return sel != "None";
}
void wxWebViewIE::DeleteSelection()
wxString wxWebViewIE::GetSelectedText() const
{
- IHTMLDocument2* document = GetDocument();
- IHTMLSelectionObject* selection;
- wxString selected;
- HRESULT hr = document->get_selection(&selection);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
{
- IDispatch* disrange;
- hr = selection->createRange(&disrange);
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString selected;
+ HRESULT hr = document->get_selection(&selection);
if(SUCCEEDED(hr))
{
- IHTMLTxtRange* range;
- hr = disrange->QueryInterface(IID_IHTMLTxtRange, (void**)&range);
+ wxCOMPtr<IDispatch> disrange;
+ hr = selection->createRange(&disrange);
if(SUCCEEDED(hr))
{
- BSTR text;
- range->get_text(&text);
- selected = wxString(text);
- range->Release();
+ wxCOMPtr<IHTMLTxtRange> range;
+ hr = disrange->QueryInterface(IID_IHTMLTxtRange, (void**)&range);
+ if(SUCCEEDED(hr))
+ {
+ BSTR text;
+ range->get_text(&text);
+ selected = wxString(text);
+ }
}
- disrange->Release();
}
- selection->Release();
+ return selected;
+ }
+ else
+ {
+ return "";
}
- document->Release();
- return selected;
}
wxString wxWebViewIE::GetSelectedSource() const
{
- IHTMLDocument2* document = GetDocument();
- IHTMLSelectionObject* selection;
- wxString selected;
- HRESULT hr = document->get_selection(&selection);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
{
- IDispatch* disrange;
- hr = selection->createRange(&disrange);
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString selected;
+ HRESULT hr = document->get_selection(&selection);
if(SUCCEEDED(hr))
{
- IHTMLTxtRange* range;
- hr = disrange->QueryInterface(IID_IHTMLTxtRange, (void**)&range);
+ wxCOMPtr<IDispatch> disrange;
+ hr = selection->createRange(&disrange);
if(SUCCEEDED(hr))
{
- BSTR text;
- range->get_htmlText(&text);
- selected = wxString(text);
- range->Release();
+ wxCOMPtr<IHTMLTxtRange> range;
+ hr = disrange->QueryInterface(IID_IHTMLTxtRange, (void**)&range);
+ if(SUCCEEDED(hr))
+ {
+ BSTR text;
+ range->get_htmlText(&text);
+ selected = wxString(text);
+ }
}
- disrange->Release();
}
- selection->Release();
+ return selected;
+ }
+ else
+ {
+ return "";
}
- document->Release();
- return selected;
}
void wxWebViewIE::ClearSelection()
{
- IHTMLDocument2* document = GetDocument();
- IHTMLSelectionObject* selection;
- wxString selected;
- HRESULT hr = document->get_selection(&selection);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
{
- selection->empty();
- selection->Release();
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString selected;
+ HRESULT hr = document->get_selection(&selection);
+ if(SUCCEEDED(hr))
+ {
+ selection->empty();
+ }
}
- document->Release();
}
wxString wxWebViewIE::GetPageText() const
{
- IHTMLDocument2* document = GetDocument();
- wxString text;
- IHTMLElement* body;
- HRESULT hr = document->get_body(&body);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ wxString text;
+ wxCOMPtr<IHTMLElement> body;
+ HRESULT hr = document->get_body(&body);
+ if(SUCCEEDED(hr))
+ {
+ BSTR out;
+ body->get_innerText(&out);
+ text = wxString(out);
+ }
+ return text;
+ }
+ else
{
- BSTR out;
- body->get_innerText(&out);
- text = wxString(out);
- body->Release();
+ return "";
}
- document->Release();
- return text;
}
void wxWebViewIE::RunScript(const wxString& javascript)
{
- IHTMLDocument2* document = GetDocument();
- IHTMLWindow2* window;
- wxString language = "javascript";
- HRESULT hr = document->get_parentWindow(&window);
- if(SUCCEEDED(hr))
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
{
- VARIANT level;
- VariantInit(&level);
- V_VT(&level) = VT_EMPTY;
- window->execScript(SysAllocString(javascript), SysAllocString(language), &level);
+ wxCOMPtr<IHTMLWindow2> window;
+ wxString language = "javascript";
+ HRESULT hr = document->get_parentWindow(&window);
+ if(SUCCEEDED(hr))
+ {
+ VARIANT level;
+ VariantInit(&level);
+ V_VT(&level) = VT_EMPTY;
+ window->execScript(SysAllocString(javascript.wc_str()),
+ SysAllocString(language.wc_str()),
+ &level);
+ }
}
- document->Release();
}
void wxWebViewIE::RegisterHandler(wxSharedPtr<wxWebViewHandler> handler)
wxDynamicLibrary urlMon(wxT("urlmon.dll"));
if(urlMon.HasSymbol(wxT("CoInternetGetSession")))
{
- typedef HRESULT (WINAPI *CoInternetGetSession_t)(DWORD, IInternetSession**, DWORD);
+ typedef HRESULT (WINAPI *CoInternetGetSession_t)(DWORD, wxIInternetSession**, DWORD);
wxDYNLIB_FUNCTION(CoInternetGetSession_t, CoInternetGetSession, urlMon);
ClassFactory* cf = new ClassFactory(handler);
- IInternetSession* session;
+ wxIInternetSession* session;
HRESULT res = (*pfnCoInternetGetSession)(0, &session, 0);
if(FAILED(res))
{
wxFAIL_MSG("Could not retrive internet session");
}
- HRESULT hr = session->RegisterNameSpace(cf, CLSID_FileProtocol, handler->GetName(), 0, NULL, 0);
+ HRESULT hr = session->RegisterNameSpace(cf, CLSID_FileProtocol,
+ handler->GetName().wc_str(),
+ 0, NULL, 0);
if(FAILED(hr))
{
wxFAIL_MSG("Could not register protocol");
bool wxWebViewIE::CanExecCommand(wxString command) const
{
- IHTMLDocument2* document = GetDocument();
- VARIANT_BOOL enabled;
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
- document->queryCommandEnabled(SysAllocString(command.wc_str()), &enabled);
- document->Release();
+ if(document)
+ {
+ VARIANT_BOOL enabled;
+
+ document->queryCommandEnabled(SysAllocString(command.wc_str()), &enabled);
+
+ return (enabled == VARIANT_TRUE);
+ }
+ else
+ {
+ return false;
+ }
- return (enabled == VARIANT_TRUE);
}
void wxWebViewIE::ExecCommand(wxString command)
{
- IHTMLDocument2* document = GetDocument();
- document->execCommand(SysAllocString(command.wc_str()), VARIANT_FALSE, VARIANT(), NULL);
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ document->execCommand(SysAllocString(command.wc_str()), VARIANT_FALSE, VARIANT(), NULL);
+ }
+}
+
+wxCOMPtr<IHTMLDocument2> wxWebViewIE::GetDocument() const
+{
+ wxCOMPtr<IDispatch> dispatch;
+ wxCOMPtr<IHTMLDocument2> document;
+ HRESULT result = m_webBrowser->get_Document(&dispatch);
+ if(dispatch && SUCCEEDED(result))
+ {
+ //document is set to null automatically if the interface isn't supported
+ dispatch->QueryInterface(IID_IHTMLDocument2, (void**)&document);
+ }
+ return document;
+}
+
+bool wxWebViewIE::IsElementVisible(IHTMLElement* elm)
+{
+ wxIHTMLCurrentStyle* style;
+ IHTMLElement *elm1 = elm;
+ wxIHTMLElement2 *elm2;
+ BSTR tmp_bstr;
+ bool is_visible = true;
+ //This method is not perfect but it does discover most of the hidden elements.
+ //so if a better solution is found, then please do improve.
+ while(elm1)
+ {
+ if(SUCCEEDED(elm1->QueryInterface(wxIID_IHTMLElement2, (void**) &elm2)))
+ {
+ if(SUCCEEDED(elm2->get_currentStyle(&style)))
+ {
+ //Check if the object has the style display:none.
+ if((style->get_display(&tmp_bstr) != S_OK) ||
+ (tmp_bstr != NULL && (_wcsicmp(tmp_bstr, L"none") == 0)))
+ {
+ is_visible = false;
+ }
+ //Check if the object has the style visibility:hidden.
+ if(is_visible && (style->get_visibility(&tmp_bstr) != S_OK) ||
+ (tmp_bstr != NULL && _wcsicmp(tmp_bstr, L"hidden") == 0))
+ {
+ is_visible = false;
+ }
+ style->Release();
+ }
+ elm2->Release();
+ }
+
+ //Lets check the object's parent element.
+ IHTMLElement* parent;
+ if(is_visible && SUCCEEDED(elm1->get_parentElement(&parent)))
+ {
+ elm1->Release();
+ elm1 = parent;
+ }
+ else
+ {
+ elm1->Release();
+ break;
+ }
+ }
+ return is_visible;
+}
+
+void wxWebViewIE::FindInternal(const wxString& text, int flags, int internal_flag)
+{
+ wxIMarkupServices *pIMS;
+ wxIMarkupContainer *pIMC;
+ wxIMarkupPointer *ptrBegin, *ptrEnd;
+ IHTMLElement* elm;
+ long find_flag = 0;
+ IHTMLDocument2 *document = GetDocument();
+ //This function does the acutal work.
+ if(SUCCEEDED(document->QueryInterface(wxIID_IMarkupServices, (void **)&pIMS)))
+ {
+ if(SUCCEEDED(document->QueryInterface(wxIID_IMarkupContainer, (void **)&pIMC)))
+ {
+ BSTR attr_bstr = SysAllocString(L"style=\"background-color:#ffff00\"");
+ BSTR text_bstr = SysAllocString(text.wc_str());
+ pIMS->CreateMarkupPointer(&ptrBegin);
+ pIMS->CreateMarkupPointer(&ptrEnd);
+
+ ptrBegin->SetGravity(POINTER_GRAVITY_Right);
+ ptrBegin->MoveToContainer(pIMC, TRUE);
+ //Create the find flag from the wx one.
+ if(flags & wxWEB_VIEW_FIND_ENTIRE_WORD)
+ {
+ find_flag |= FINDTEXT_WHOLEWORD;
+ }
+ if(flags & wxWEB_VIEW_FIND_MATCH_CASE)
+ {
+ find_flag |= FINDTEXT_MATCHCASE;
+ }
+
+ //A little speed-up to avoid to re-alloc in the positions vector.
+ if(text.Len() < 3 && m_findPointers.capacity() < 500)
+ {
+ m_findPointers.reserve(text.Len() == 1 ? 1000 : 500);
+ }
+
+ while(ptrBegin->FindText(text_bstr, find_flag, ptrEnd, NULL) == S_OK)
+ {
+ if(ptrBegin->CurrentScope(&elm) == S_OK)
+ {
+ if(IsElementVisible(elm))
+ {
+ //Highlight the word if the flag was set.
+ if(flags & wxWEB_VIEW_FIND_HIGHLIGHT_RESULT)
+ {
+ IHTMLElement* pFontEl;
+ pIMS->CreateElement(TAGID_FONT, attr_bstr, &pFontEl);
+ pIMS->InsertElement(pFontEl, ptrBegin, ptrEnd);
+ }
+ if(internal_flag & wxWEB_VIEW_FIND_REMOVE_HIGHLIGHT)
+ {
+ IHTMLElement* pFontEl;
+ ptrBegin->CurrentScope(&pFontEl);
+ pIMS->RemoveElement(pFontEl);
+ pFontEl->Release();
+ }
+ if(internal_flag & wxWEB_VIEW_FIND_ADD_POINTERS)
+ {
+ wxIMarkupPointer *cptrBegin, *cptrEnd;
+ pIMS->CreateMarkupPointer(&cptrBegin);
+ pIMS->CreateMarkupPointer(&cptrEnd);
+ cptrBegin->MoveToPointer(ptrBegin);
+ cptrEnd->MoveToPointer(ptrEnd);
+ m_findPointers.push_back(wxFindPointers(cptrBegin,cptrEnd));
+ }
+ }
+ elm->Release();
+ }
+ ptrBegin->MoveToPointer(ptrEnd);
+ }
+ //Clean up.
+ SysFreeString(text_bstr);
+ SysFreeString(attr_bstr);
+ pIMC->Release();
+ ptrBegin->Release();
+ ptrEnd->Release();
+ }
+ pIMS->Release();
+ }
+ document->Release();
+}
+
+long wxWebViewIE::FindNext(int direction)
+{
+ //Don't bother if we have no pointers set.
+ if(m_findPointers.empty())
+ {
+ return wxNOT_FOUND;
+ }
+ //Manage the find position and do some checks.
+ if(direction > 0)
+ {
+ m_findPosition++;
+ }
+ else
+ {
+ m_findPosition--;
+ }
+
+ if(m_findPosition >= (signed)m_findPointers.size())
+ {
+ if(m_findFlags & wxWEB_VIEW_FIND_WRAP)
+ {
+ m_findPosition = 0;
+ }
+ else
+ {
+ m_findPosition--;
+ return wxNOT_FOUND;
+ }
+ }
+ else if(m_findPosition < 0)
+ {
+ if(m_findFlags & wxWEB_VIEW_FIND_WRAP)
+ {
+ m_findPosition = m_findPointers.size()-1;
+ }
+ else
+ {
+ m_findPosition++;
+ return wxNOT_FOUND;
+ }
+ }
+ //some variables to use later on.
+ IHTMLElement *body_element;
+ IHTMLBodyElement *body;
+ wxIHTMLTxtRange *range = NULL;
+ wxIMarkupServices *pIMS;
+ IHTMLDocument2 *document = GetDocument();
+ long ret = -1;
+ //Now try to create a range from the body.
+ if(SUCCEEDED(document->get_body(&body_element)))
+ {
+ if(SUCCEEDED(body_element->QueryInterface(IID_IHTMLBodyElement,(void**)&body)))
+ {
+ if(SUCCEEDED(body->createTextRange((IHTMLTxtRange**)(&range))))
+ {
+ //So far so good, now we try to position our find pointers.
+ if(SUCCEEDED(document->QueryInterface(wxIID_IMarkupServices,(void **)&pIMS)))
+ {
+ wxIMarkupPointer *begin = m_findPointers[m_findPosition].begin, *end = m_findPointers[m_findPosition].end;
+ if(pIMS->MoveRangeToPointers(begin,end,range) == S_OK && range->select() == S_OK)
+ {
+ ret = m_findPosition;
+ }
+ pIMS->Release();
+ }
+ range->Release();
+ }
+ body->Release();
+ }
+ body_element->Release();
+ }
document->Release();
+ return ret;
}
-IHTMLDocument2* wxWebViewIE::GetDocument() const
+void wxWebViewIE::FindClear()
{
- wxVariant variant = m_ie.GetProperty("Document");
- IHTMLDocument2* document = (IHTMLDocument2*)variant.GetVoidPtr();
+ //Reset find variables.
+ m_findText.Empty();
+ m_findFlags = wxWEB_VIEW_FIND_DEFAULT;
+ m_findPosition = -1;
- wxASSERT(document);
+ //The m_findPointers contains pointers for the found text.
+ //Since it uses ref counting we call release on the pointers first
+ //before we remove them from the vector. In other words do not just
+ //remove elements from m_findPointers without calling release first
+ //or you will get a memory leak.
+ size_t count = m_findPointers.size();
+ for(size_t i = 0; i < count; i++)
+ {
+ m_findPointers[i].begin->Release();
+ m_findPointers[i].end->Release();
+ }
+ m_findPointers.clear();
+}
- return document;
+bool wxWebViewIE::EnableControlFeature(long flag, bool enable)
+{
+#if wxUSE_DYNLIB_CLASS
+
+ wxDynamicLibrary urlMon(wxT("urlmon.dll"));
+ if( urlMon.IsLoaded() &&
+ urlMon.HasSymbol("CoInternetSetFeatureEnabled") &&
+ urlMon.HasSymbol("CoInternetIsFeatureEnabled"))
+ {
+ typedef HRESULT (WINAPI *CoInternetSetFeatureEnabled_t)(DWORD, DWORD, BOOL);
+ typedef HRESULT (WINAPI *CoInternetIsFeatureEnabled_t)(DWORD, DWORD);
+
+ wxDYNLIB_FUNCTION(CoInternetSetFeatureEnabled_t, CoInternetSetFeatureEnabled, urlMon);
+ wxDYNLIB_FUNCTION(CoInternetIsFeatureEnabled_t, CoInternetIsFeatureEnabled, urlMon);
+
+ HRESULT hr = (*pfnCoInternetIsFeatureEnabled)(flag,
+ 0x2 /* SET_FEATURE_ON_PROCESS */);
+ if((hr == S_OK && enable) || (hr == S_FALSE && !enable))
+ return true;
+
+ hr = (*pfnCoInternetSetFeatureEnabled)(flag,
+ 0x2/* SET_FEATURE_ON_PROCESS */,
+ (enable ? TRUE : FALSE));
+ if ( FAILED(hr) )
+ {
+ wxLogApiError(wxT("CoInternetSetFeatureEnabled"), hr);
+ return false;
+ }
+ return true;
+ }
+ return false;
+#else
+ wxUnusedVar(flag);
+ wxUnusedVar(enable);
+ return false;
+#endif // wxUSE_DYNLIB_CLASS/!wxUSE_DYNLIB_CLASS
}
void wxWebViewIE::onActiveXEvent(wxActiveXEvent& evt)
//As we are complete we also add to the history list, but not if the
//page is not the main page, ie it is a subframe
- //We also have to check if we are loading a file:// url, if so we
+ //We also have to check if we are loading a file:// url, if so we
//need to change the comparison as ie passes back a different style
//of url
- if(m_historyEnabled && !m_historyLoadingFromList &&
- (url == GetCurrentURL() ||
- (GetCurrentURL().substr(0, 4) == "file" &&
+ if(m_historyEnabled && !m_historyLoadingFromList &&
+ (url == GetCurrentURL() ||
+ (GetCurrentURL().substr(0, 4) == "file" &&
wxFileSystem::URLToFileName(GetCurrentURL()).GetFullPath() == url)))
{
//If we are not at the end of the list, then erase everything
}
//Reset as we are done now
m_historyLoadingFromList = false;
+ //Reset the find values.
+ FindClear();
// TODO: set target parameter if possible
wxString target = wxEmptyString;
wxWebViewEvent event(wxEVT_COMMAND_WEB_VIEW_LOADED, GetId(),
VirtualProtocol::VirtualProtocol(wxSharedPtr<wxWebViewHandler> handler)
{
- m_refCount = 0;
m_file = NULL;
m_handler = handler;
}
-VirtualProtocol::~VirtualProtocol()
-{
-}
+BEGIN_IID_TABLE(VirtualProtocol)
+ ADD_IID(Unknown)
+ ADD_RAW_IID(wxIID_IInternetProtocolRoot)
+ ADD_RAW_IID(wxIID_IInternetProtocol)
+END_IID_TABLE;
-ULONG VirtualProtocol::AddRef()
-{
- m_refCount++;
- return m_refCount;
-}
+IMPLEMENT_IUNKNOWN_METHODS(VirtualProtocol)
-HRESULT VirtualProtocol::QueryInterface(REFIID riid, void **ppvObject)
-{
- if(riid == IID_IUnknown || riid == IID_IInternetProtocolRoot ||
- riid == IID_IInternetProtocol)
- {
- *ppvObject = (IInternetProtocol*)this;
- AddRef();
- return S_OK;
- }
- else
- {
- *ppvObject = NULL;
- return E_POINTER;
- }
-}
-
-ULONG VirtualProtocol::Release()
-{
- m_refCount--;
- if (m_refCount > 0)
- {
- return m_refCount;
- }
- else
- {
- delete this;
- return 0;
- }
-}
-
-HRESULT VirtualProtocol::Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink,
- IInternetBindInfo *pOIBindInfo, DWORD grfPI,
+HRESULT STDMETHODCALLTYPE VirtualProtocol::Start(LPCWSTR szUrl, wxIInternetProtocolSink *pOIProtSink,
+ wxIInternetBindInfo *pOIBindInfo, DWORD grfPI,
HANDLE_PTR dwReserved)
{
wxUnusedVar(szUrl);
wxUnusedVar(grfPI);
wxUnusedVar(dwReserved);
m_protocolSink = pOIProtSink;
-
+
//We get the file itself from the protocol handler
m_file = m_handler->GetFile(szUrl);
//We return the stream length for current and total size as we can always
//read the whole file from the stream
wxFileOffset length = m_file->GetStream()->GetLength();
- m_protocolSink->ReportData(BSCF_FIRSTDATANOTIFICATION |
- BSCF_DATAFULLYAVAILABLE |
- BSCF_LASTDATANOTIFICATION,
+ m_protocolSink->ReportData(wxBSCF_FIRSTDATANOTIFICATION |
+ wxBSCF_DATAFULLYAVAILABLE |
+ wxBSCF_LASTDATANOTIFICATION,
length, length);
- return S_OK;
+ return S_OK;
}
-HRESULT VirtualProtocol::Read(void *pv, ULONG cb, ULONG *pcbRead)
+HRESULT STDMETHODCALLTYPE VirtualProtocol::Read(void *pv, ULONG cb, ULONG *pcbRead)
{
//If the file is null we return false to indicte it is finished
- if(!m_file)
+ if(!m_file)
return S_FALSE;
wxStreamError err = m_file->GetStream()->Read(pv, cb).GetLastError();
}
}
-HRESULT ClassFactory::CreateInstance(IUnknown* pUnkOuter, REFIID riid,
+BEGIN_IID_TABLE(ClassFactory)
+ ADD_IID(Unknown)
+ ADD_IID(ClassFactory)
+END_IID_TABLE;
+
+IMPLEMENT_IUNKNOWN_METHODS(ClassFactory)
+
+HRESULT STDMETHODCALLTYPE ClassFactory::CreateInstance(IUnknown* pUnkOuter, REFIID riid,
void ** ppvObject)
{
- if (pUnkOuter)
+ if (pUnkOuter)
return CLASS_E_NOAGGREGATION;
VirtualProtocol* vp = new VirtualProtocol(m_handler);
vp->AddRef();
vp->Release();
return hr;
-}
+}
STDMETHODIMP ClassFactory::LockServer(BOOL fLock)
{
return S_OK;
}
-ULONG ClassFactory::AddRef(void)
+wxIEContainer::wxIEContainer(wxWindow *parent, REFIID iid, IUnknown *pUnk,
+ DocHostUIHandler* uiHandler) :
+ wxActiveXContainer(parent,iid,pUnk)
{
- m_refCount++;
- return m_refCount;
+ m_uiHandler = uiHandler;
}
-HRESULT ClassFactory::QueryInterface(REFIID riid, void **ppvObject)
+wxIEContainer::~wxIEContainer()
{
- if ((riid == IID_IUnknown) || (riid == IID_IClassFactory))
- {
- *ppvObject = this;
- AddRef();
- return S_OK;
- }
- else
+}
+
+bool wxIEContainer::QueryClientSiteInterface(REFIID iid, void **_interface,
+ const char *&desc)
+{
+ if (m_uiHandler && IsEqualIID(iid, wxIID_IDocHostUIHandler))
{
- *ppvObject = NULL;
- return E_POINTER;
+ *_interface = (IUnknown *) (wxIDocHostUIHandler *) m_uiHandler;
+ desc = "IDocHostUIHandler";
+ return true;
}
+ return false;
+}
+HRESULT wxSTDCALL DocHostUIHandler::ShowContextMenu(DWORD dwID, POINT *ppt,
+ IUnknown *pcmdtReserved,
+ IDispatch *pdispReserved)
+{
+ wxUnusedVar(dwID);
+ wxUnusedVar(ppt);
+ wxUnusedVar(pcmdtReserved);
+ wxUnusedVar(pdispReserved);
+ return E_NOTIMPL;
}
-ULONG ClassFactory::Release(void)
+HRESULT wxSTDCALL DocHostUIHandler::GetHostInfo(DOCHOSTUIINFO *pInfo)
{
- m_refCount--;
- if (m_refCount > 0)
- {
- return m_refCount;
- }
- else
+ //don't show 3d border and enable themes.
+ pInfo->dwFlags = pInfo->dwFlags | DOCHOSTUIFLAG_NO3DBORDER | DOCHOSTUIFLAG_THEME;
+ return S_OK;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::ShowUI(DWORD dwID,
+ IOleInPlaceActiveObject *pActiveObject,
+ IOleCommandTarget *pCommandTarget,
+ IOleInPlaceFrame *pFrame,
+ IOleInPlaceUIWindow *pDoc)
+{
+ wxUnusedVar(dwID);
+ wxUnusedVar(pActiveObject);
+ wxUnusedVar(pCommandTarget);
+ wxUnusedVar(pFrame);
+ wxUnusedVar(pDoc);
+ return S_FALSE;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::HideUI(void)
+{
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::UpdateUI(void)
+{
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::EnableModeless(BOOL fEnable)
+{
+ wxUnusedVar(fEnable);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::OnDocWindowActivate(BOOL fActivate)
+{
+ wxUnusedVar(fActivate);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::OnFrameWindowActivate(BOOL fActivate)
+{
+ wxUnusedVar(fActivate);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::ResizeBorder(LPCRECT prcBorder,
+ IOleInPlaceUIWindow *pUIWindow,
+ BOOL fFrameWindow)
+{
+ wxUnusedVar(prcBorder);
+ wxUnusedVar(pUIWindow);
+ wxUnusedVar(fFrameWindow);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::TranslateAccelerator(LPMSG lpMsg,
+ const GUID *pguidCmdGroup,
+ DWORD nCmdID)
+{
+ if(lpMsg && lpMsg->message == WM_KEYDOWN)
{
- delete this;
- return 0;
+ //control is down?
+ if((GetKeyState(VK_CONTROL) & 0x8000 ))
+ {
+ //skip the accelerators used by the control
+ switch(lpMsg->wParam)
+ {
+ case 'F':
+ case 'L':
+ case 'N':
+ case 'O':
+ case 'P':
+ return S_OK;
+ }
+ }
+ //skip F5
+ if(lpMsg->wParam == VK_F5)
+ {
+ return S_OK;
+ }
}
-}
+ wxUnusedVar(pguidCmdGroup);
+ wxUnusedVar(nCmdID);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::GetOptionKeyPath(LPOLESTR *pchKey,DWORD dw)
+{
+ wxUnusedVar(pchKey);
+ wxUnusedVar(dw);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::GetDropTarget(IDropTarget *pDropTarget,
+ IDropTarget **ppDropTarget)
+{
+ wxUnusedVar(pDropTarget);
+ wxUnusedVar(ppDropTarget);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::GetExternal(IDispatch **ppDispatch)
+{
+ wxUnusedVar(ppDispatch);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::TranslateUrl(DWORD dwTranslate,
+ OLECHAR *pchURLIn,
+ OLECHAR **ppchURLOut)
+{
+ wxUnusedVar(dwTranslate);
+ wxUnusedVar(pchURLIn);
+ wxUnusedVar(ppchURLOut);
+ return E_NOTIMPL;
+}
+
+HRESULT wxSTDCALL DocHostUIHandler::FilterDataObject(IDataObject *pDO, IDataObject **ppDORet)
+{
+ wxUnusedVar(pDO);
+ wxUnusedVar(ppDORet);
+ return E_NOTIMPL;
+}
+
+BEGIN_IID_TABLE(DocHostUIHandler)
+ ADD_IID(Unknown)
+ ADD_RAW_IID(wxIID_IDocHostUIHandler)
+END_IID_TABLE;
+
+IMPLEMENT_IUNKNOWN_METHODS(DocHostUIHandler)
#endif // wxUSE_WEBVIEW && wxUSE_WEBVIEW_IE
projects / wxWidgets.git / blobdiff