1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome_frame/ready_mode/internal/ready_mode_web_browser_adapter.h"
6
7 #include "base/logging.h"
8 #include "base/win/win_util.h"
9 #include "chrome_frame/chrome_tab.h"
10
11 _ATL_FUNC_INFO ReadyModeWebBrowserAdapter::kBeforeNavigate2Info = {
12 CC_STDCALL, VT_EMPTY, 7, {
13 VT_DISPATCH,
14 VT_VARIANT | VT_BYREF,
15 VT_VARIANT | VT_BYREF,
16 VT_VARIANT | VT_BYREF,
17 VT_VARIANT | VT_BYREF,
18 VT_VARIANT | VT_BYREF,
19 VT_BOOL | VT_BYREF
20 }
21 };
22
23 _ATL_FUNC_INFO ReadyModeWebBrowserAdapter::kDocumentCompleteInfo = {
24 CC_STDCALL, VT_EMPTY, 2, {
25 VT_DISPATCH,
26 VT_VARIANT | VT_BYREF
27 }
28 };
29
30 _ATL_FUNC_INFO ReadyModeWebBrowserAdapter::kOnQuitInfo = {
31 CC_STDCALL, VT_EMPTY, 0, {NULL}};
32
ReadyModeWebBrowserAdapter()33 ReadyModeWebBrowserAdapter::ReadyModeWebBrowserAdapter() {
34 }
35
Initialize(IWebBrowser2 * web_browser,Observer * observer)36 bool ReadyModeWebBrowserAdapter::Initialize(IWebBrowser2* web_browser,
37 Observer* observer) {
38 base::win::ScopedComPtr<ReadyModeWebBrowserAdapter, NULL> self(this);
39
40 DCHECK(web_browser != NULL);
41 DCHECK(web_browser_ == NULL);
42 DCHECK(observer != NULL);
43 DCHECK(observer_ == NULL);
44
45 observer_.reset(observer);
46
47 web_browser->AddRef();
48 web_browser_.Attach(web_browser);
49
50 HRESULT hr = DispEventAdvise(web_browser_, &DIID_DWebBrowserEvents2);
51 if (FAILED(hr)) {
52 observer_.reset();
53 web_browser_.Release();
54 DLOG(ERROR) << "DispEventAdvise failed. Error: " << hr;
55 }
56
57 return SUCCEEDED(hr);
58 }
59
Uninitialize()60 void ReadyModeWebBrowserAdapter::Uninitialize() {
61 // DispEventUnadvise will free the WebBrowser's reference to us. In case
62 // that's the last reference, take a temporary reference in this function's
63 // scope to allow us to finish the cleanup we would otherwise like to do.
64 base::win::ScopedComPtr<ReadyModeWebBrowserAdapter, NULL> self(this);
65
66 observer_.reset();
67
68 DCHECK(web_browser_ != NULL);
69 if (web_browser_ == NULL)
70 return;
71
72 HRESULT hr = DispEventUnadvise(web_browser_, &DIID_DWebBrowserEvents2);
73 if (FAILED(hr)) {
74 DLOG(ERROR) << "DispEventUnadvise failed. Error: " << hr;
75 } else {
76 web_browser_.Release();
77 }
78 }
79
STDMETHODIMP_(void)80 STDMETHODIMP_(void) ReadyModeWebBrowserAdapter::OnQuit() {
81 Uninitialize();
82 }
83
BeforeNavigate2(IDispatch *,VARIANT * url,VARIANT *,VARIANT *,VARIANT *,VARIANT *,VARIANT_BOOL *)84 STDMETHODIMP ReadyModeWebBrowserAdapter::BeforeNavigate2(
85 IDispatch* /*dispatch*/, VARIANT* url, VARIANT* /*flags*/,
86 VARIANT* /*target_frame_name*/, VARIANT* /*post_data*/,
87 VARIANT* /*headers*/, VARIANT_BOOL* /*cancel*/) {
88 if (observer_ != NULL)
89 observer_->OnNavigateTo(url->bstrVal);
90
91 return S_OK;
92 }
93
STDMETHODIMP_(void)94 STDMETHODIMP_(void) ReadyModeWebBrowserAdapter::DocumentComplete(
95 IDispatch* /*dispatch*/, VARIANT* url) {
96 if (!url || V_VT(url) != VT_BSTR || url->bstrVal == NULL)
97 return;
98
99 if (observer_ == NULL)
100 return;
101
102 base::win::ScopedComPtr<IDispatch> doc_disp;
103 web_browser_->get_Document(doc_disp.Receive());
104 if (!doc_disp)
105 return;
106
107 base::win::ScopedComPtr<IChromeFrame> chrome_frame_doc;
108 chrome_frame_doc.QueryFrom(doc_disp);
109
110 if (chrome_frame_doc)
111 observer_->OnRenderInChromeFrame(url->bstrVal);
112 else
113 observer_->OnRenderInHost(url->bstrVal);
114 }
115