Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Unified Diff: src/plugin/WebBrowserEventsListener.cpp

Issue 6567422169448448: Issue 119 - Switch to injecting CSS for element hiding (Closed)
Patch Set: Created Dec. 10, 2014, 5:12 p.m.
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View side-by-side diff with in-line comments
Download patch
« src/plugin/PluginFilter.cpp ('K') | « src/plugin/WebBrowserEventsListener.h ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: src/plugin/WebBrowserEventsListener.cpp
diff --git a/src/plugin/WebBrowserEventsListener.cpp b/src/plugin/WebBrowserEventsListener.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..e856991006b03ed51e547ed2f593a5b1b74ae595
--- /dev/null
+++ b/src/plugin/WebBrowserEventsListener.cpp
@@ -0,0 +1,117 @@
+#include "PluginStdAfx.h"
+#include "WebBrowserEventsListener.h"
+#include "PluginClientBase.h" // for UnescapeUrl
+
+WebBrowserEventsListener::WebBrowserEventsListener()
+ : m_isDocumentEvents2Connected(false)
+ , m_state(State::FirstTimeLoading)
+{
+}
+
+WebBrowserEventsListener::~WebBrowserEventsListener()
+{
+ if (!!m_onDestroy)
+ {
+ m_onDestroy();
+ }
+}
+
+HRESULT STDMETHODCALLTYPE WebBrowserEventsListener::OnDocumentComplete(IDispatch* dispFrameBrowser, VARIANT* /*variantUrl*/)
+{
+ if (!dispFrameBrowser)
+ {
+ return E_POINTER;
+ }
+
+ // if it's a signal from another browser (sub-frame for-example) then ignore it.
+ if (!m_browser.IsEqualObject(dispFrameBrowser))
+ {
+ return S_OK;
+ }
+
+ if (!m_isDocumentEvents2Connected)
+ {
+ ATL::CComPtr<IDispatch> dispDocument;
+ ATL::CComQIPtr<IHTMLDocument2> htmlDocument2;
+ bool isHtmlDocument2 = SUCCEEDED(m_browser->get_Document(&dispDocument)) && (htmlDocument2 = dispDocument);
+ isHtmlDocument2 && (m_isDocumentEvents2Connected = SUCCEEDED(HTMLDocumentEvents2Listener::DispEventAdvise(htmlDocument2)));
+ }
+
+ if (m_state == State::FirstTimeLoading)
+ {
+ m_state = State::Loaded;
+ emitReloaded();
+ }
+ return S_OK;
+}
+
+void STDMETHODCALLTYPE WebBrowserEventsListener::OnReadyStateChange(IHTMLEventObj* /*pEvtObj*/)
+{
+ auto documentReadyState = [this]()->std::wstring
+ {
+ std::wstring notAvailableReadyState;
+ ATL::CComPtr<IDispatch> pDocDispatch;
+ m_browser->get_Document(&pDocDispatch);
+ ATL::CComQIPtr<IHTMLDocument2> htmlDocument2 = pDocDispatch;
+ if (!htmlDocument2)
+ {
+ assert(false && "htmlDocument2 in OnReadyStateChange should not be nullptr");
+ return notAvailableReadyState;
+ }
+ ATL::CComBSTR readyState;
+ if (FAILED(htmlDocument2->get_readyState(&readyState)) || !readyState)
+ {
+ assert(false && "cannot obtain document readyState in OnReadyStateChange");
+ return notAvailableReadyState;
+ }
+ return std::wstring(readyState, readyState.Length());
+ }();
+ if (documentReadyState == L"loading")
+ {
+ m_state = State::Loading;
+ }
+ else if (documentReadyState == L"interactive")
+ {
+ }
+ else if (documentReadyState == L"complete")
+ {
+ if (m_state == State::Loading)
+ {
+ m_state = State::Loaded;
+ emitReloaded();
+ }
+ else
+ {
+ assert(false);
+ }
+ }
+ else if (documentReadyState == L"uninitialized")
+ {
+ }
+ else
+ {
+ assert(false);
+ }
+}
+
+HRESULT WebBrowserEventsListener::Init(IWebBrowser2* webBrowser, const OnDestroy& onDestroy)
+{
+ if (!(m_browser = webBrowser))
+ {
+ return E_POINTER;
+ }
+ m_onDestroy = onDestroy;
+ if (FAILED(WebBrowserEvents2Listener::DispEventAdvise(m_browser, &DIID_DWebBrowserEvents2)))
+ {
+ return E_FAIL;
+ }
+ return S_OK;
+}
+
+void WebBrowserEventsListener::emitReloaded()
+{
+ if (reloaded)
+ {
+ reloaded();
+ }
+}
« src/plugin/PluginFilter.cpp ('K') | « src/plugin/WebBrowserEventsListener.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld