| 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 |