+ if(document)
+ {
+ BSTR title;
+ document->get_nameProp(&title);
+ return wxString(title);
+ }
+ else
+ {
+ return "";
+ }
+}
+
+bool wxWebViewIE::CanCut() const
+{
+ return CanExecCommand("Cut");
+}
+
+bool wxWebViewIE::CanCopy() const
+{
+ return CanExecCommand("Copy");
+}
+
+bool wxWebViewIE::CanPaste() const
+{
+ return CanExecCommand("Paste");
+}
+
+void wxWebViewIE::Cut()
+{
+ ExecCommand("Cut");
+}
+
+void wxWebViewIE::Copy()
+{
+ ExecCommand("Copy");
+}
+
+void wxWebViewIE::Paste()
+{
+ ExecCommand("Paste");
+}
+
+bool wxWebViewIE::CanUndo() const
+{
+ return CanExecCommand("Undo");
+}
+
+bool wxWebViewIE::CanRedo() const
+{
+ return CanExecCommand("Redo");
+}
+
+void wxWebViewIE::Undo()
+{
+ ExecCommand("Undo");
+}
+
+void wxWebViewIE::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)
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ if( enable )
+ document->put_designMode(SysAllocString(L"On"));
+ else
+ document->put_designMode(SysAllocString(L"Off"));
+
+ }
+}
+
+bool wxWebViewIE::IsEditable() const
+{
+ 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()
+{
+ ExecCommand("SelectAll");
+}
+
+bool wxWebViewIE::HasSelection() const
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ 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;
+ }
+}
+
+void wxWebViewIE::DeleteSelection()
+{
+ ExecCommand("Delete");
+}
+
+wxString wxWebViewIE::GetSelectedText() const
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString selected;
+ HRESULT hr = document->get_selection(&selection);
+ if(SUCCEEDED(hr))
+ {
+ wxCOMPtr<IDispatch> disrange;
+ hr = selection->createRange(&disrange);
+ if(SUCCEEDED(hr))
+ {
+ wxCOMPtr<IHTMLTxtRange> range;
+ hr = disrange->QueryInterface(IID_IHTMLTxtRange, (void**)&range);
+ if(SUCCEEDED(hr))
+ {
+ BSTR text;
+ range->get_text(&text);
+ selected = wxString(text);
+ }
+ }
+ }
+ return selected;
+ }
+ else
+ {
+ return "";
+ }
+}
+
+wxString wxWebViewIE::GetSelectedSource() const
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString selected;
+ HRESULT hr = document->get_selection(&selection);
+ if(SUCCEEDED(hr))
+ {
+ wxCOMPtr<IDispatch> disrange;
+ hr = selection->createRange(&disrange);
+ if(SUCCEEDED(hr))
+ {
+ wxCOMPtr<IHTMLTxtRange> range;
+ hr = disrange->QueryInterface(IID_IHTMLTxtRange, (void**)&range);
+ if(SUCCEEDED(hr))
+ {
+ BSTR text;
+ range->get_htmlText(&text);
+ selected = wxString(text);
+ }
+ }
+ }
+ return selected;
+ }
+ else
+ {
+ return "";
+ }
+}
+
+void wxWebViewIE::ClearSelection()
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ wxCOMPtr<IHTMLSelectionObject> selection;
+ wxString selected;
+ HRESULT hr = document->get_selection(&selection);
+ if(SUCCEEDED(hr))
+ {
+ selection->empty();
+ }
+ }
+}
+
+wxString wxWebViewIE::GetPageText() const
+{
+ 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
+ {
+ return "";
+ }
+}
+
+void wxWebViewIE::RunScript(const wxString& javascript)
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ 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);
+ }
+ }
+}
+
+void wxWebViewIE::RegisterHandler(wxSharedPtr<wxWebViewHandler> handler)
+{
+ wxDynamicLibrary urlMon(wxT("urlmon.dll"));
+ if(urlMon.HasSymbol(wxT("CoInternetGetSession")))
+ {
+ typedef HRESULT (WINAPI *CoInternetGetSession_t)(DWORD, wxIInternetSession**, DWORD);
+ wxDYNLIB_FUNCTION(CoInternetGetSession_t, CoInternetGetSession, urlMon);
+
+ ClassFactory* cf = new ClassFactory(handler);
+ 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().wc_str(),
+ 0, NULL, 0);
+ if(FAILED(hr))
+ {
+ wxFAIL_MSG("Could not register protocol");
+ }
+ m_factories.push_back(cf);
+ }
+ else
+ {
+ wxFAIL_MSG("urlmon does not contain CoInternetGetSession");
+ }
+}
+
+bool wxWebViewIE::CanExecCommand(wxString command) const
+{
+ wxCOMPtr<IHTMLDocument2> document(GetDocument());
+
+ if(document)
+ {
+ VARIANT_BOOL enabled;
+
+ document->queryCommandEnabled(SysAllocString(command.wc_str()), &enabled);
+
+ return (enabled == VARIANT_TRUE);
+ }
+ else
+ {
+ return false;
+ }
+
+}
+
+void wxWebViewIE::ExecCommand(wxString command)
+{
+ 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;
+}
+
+void wxWebViewIE::FindClear()
+{
+ //Reset find variables.
+ m_findText.Empty();
+ m_findFlags = wxWEB_VIEW_FIND_DEFAULT;
+ m_findPosition = -1;
+
+ //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();
+}
+
+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
projects / wxWidgets.git / blobdiff