chrome/renderer/prerender/prerenderer_client.cc

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/renderer/prerender/prerenderer_client.h"

#include "base/logging.h"
#include "chrome/renderer/prerender/prerender_extra_data.h"
#include "chrome/renderer/prerender/prerender_helper.h"
#include "content/public/renderer/render_view.h"
#include "third_party/WebKit/public/web/WebView.h"

namespace {
static int s_last_prerender_id = 0;
}

namespace prerender {

PrerendererClient::PrerendererClient(content::RenderView* render_view)
    : content::RenderViewObserver(render_view) {
  DCHECK(render_view);
  DVLOG(5) << "PrerendererClient::PrerendererClient()";
  render_view->GetWebView()->setPrerendererClient(this);
}

PrerendererClient::~PrerendererClient() {
}

void PrerendererClient::willAddPrerender(
    blink::WebPrerender* prerender) {
  DVLOG(3) << "PrerendererClient::willAddPrerender url = "
           << prerender->url().string().utf8();
  prerender->setExtraData(new PrerenderExtraData(++s_last_prerender_id,
                                                 routing_id(),
                                                 render_view()->GetSize()));
}

bool PrerendererClient::isPrefetchOnly() {
  return PrerenderHelper::GetPrerenderMode(
             render_view()->GetMainRenderFrame()) == PREFETCH_ONLY;
}

void PrerendererClient::OnDestruct() {
  delete this;
}

}  // namespace prerender