chrome_frame/ready_mode/internal/ready_mode_web_browser_adapter.cc

   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 }