components/dom_distiller/content/web_contents_main_frame_observer.cc

   1 // Copyright 2014 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 "components/dom_distiller/content/web_contents_main_frame_observer.h"
   6
   7 #include "content/public/browser/navigation_details.h"
   8 #include "content/public/browser/render_frame_host.h"
   9 #include "content/public/browser/web_contents.h"
  10 #include "content/public/browser/web_contents_observer.h"
  11 #include "content/public/browser/web_contents_user_data.h"
  12
  13 DEFINE_WEB_CONTENTS_USER_DATA_KEY(dom_distiller::WebContentsMainFrameObserver);
  14
  15 namespace dom_distiller {
  16
  17 WebContentsMainFrameObserver::WebContentsMainFrameObserver(
  18     content::WebContents* web_contents)
  19     : is_document_loaded_in_main_frame_(false),
  20       is_initialized_(false),
  21       web_contents_(web_contents) {
  22   content::WebContentsObserver::Observe(web_contents);
  23 }
  24
  25 WebContentsMainFrameObserver::~WebContentsMainFrameObserver() {
  26   CleanUp();
  27 }
  28
  29 void WebContentsMainFrameObserver::DocumentLoadedInFrame(
  30     int64 frame_id,
  31     content::RenderViewHost* render_view_host) {
  32   if (web_contents_ &&
  33       frame_id == web_contents_->GetMainFrame()->GetRoutingID()) {
  34     is_document_loaded_in_main_frame_ = true;
  35   }
  36 }
  37
  38 void WebContentsMainFrameObserver::DidNavigateMainFrame(
  39     const content::LoadCommittedDetails& details,
  40     const content::FrameNavigateParams& params) {
  41   if (details.is_main_frame) {
  42     is_document_loaded_in_main_frame_ = false;
  43     is_initialized_ = true;
  44   }
  45 }
  46
  47 void WebContentsMainFrameObserver::RenderProcessGone(
  48     base::TerminationStatus status) {
  49   CleanUp();
  50 }
  51
  52 void WebContentsMainFrameObserver::WebContentsDestroyed() {
  53   CleanUp();
  54 }
  55
  56 void WebContentsMainFrameObserver::CleanUp() {
  57   content::WebContentsObserver::Observe(NULL);
  58   web_contents_ = NULL;
  59 }
  60
  61 }  // namespace dom_distiller