OLD | NEW |
| (Empty) |
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 | |
33 ReadyModeWebBrowserAdapter::ReadyModeWebBrowserAdapter() { | |
34 } | |
35 | |
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 | |
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 | |
80 STDMETHODIMP_(void) ReadyModeWebBrowserAdapter::OnQuit() { | |
81 Uninitialize(); | |
82 } | |
83 | |
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 | |
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 } | |
OLD | NEW |