Handle interface to prerenders.

chrome/browser/prerender/prerender_handle.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 <algorithm>
+
+#include "chrome/browser/prerender/prerender_handle.h"
+
+#include "base/basictypes.h"
+#include "chrome/browser/prerender/prerender_contents.h"
+#include "chrome/browser/prerender/prerender_manager.h"
+
+namespace prerender {
+
+PrerenderHandle::~PrerenderHandle() {
+  DCHECK(CalledOnValidThread());
+  OnCancel();
+}
+
+void PrerenderHandle::OnNavigateAway() {
+  DCHECK(CalledOnValidThread());
+  // TODO(gavinp): Implement reasonable behaviour for navigation away from
+  // launcher.
+  prerender_data_.reset();
+}
+
+void PrerenderHandle::OnCancel() {
+  DCHECK(CalledOnValidThread());
+  if (!prerender_data_)
+    return;
+
+  if (--prerender_data_->instance_count == 0) {
+    if (IsPrerendering())
+      prerender_data_->contents->Destroy(FINAL_STATUS_CANCELLED);
+    else if (IsPending())
+      prerender_data_->manager->DestroyPendingPrerenderData(prerender_data_);
+    else
+      NOTREACHED();
+  }
+  prerender_data_.reset();
+}
+
+bool PrerenderHandle::IsPending() const {
+  DCHECK(CalledOnValidThread());
+  return prerender_data_ && !prerender_data_->contents;
+}
+
+bool PrerenderHandle::IsPrerendering() const {
+  DCHECK(CalledOnValidThread());
+  return prerender_data_ && prerender_data_->contents;
+}
+
+bool PrerenderHandle::DidFinishLoading() const {
+  DCHECK(CalledOnValidThread());
+  if (!prerender_data_ || IsPending())
+    return false;
+  return prerender_data_->contents->has_finished_loading();
+}
+
+PrerenderHandle::PrerenderData::PrerenderData(
+    PrerenderManager* manager) : manager(manager),
+                                 contents(NULL),
+                                 instance_count(1) {
+}
+
+PrerenderHandle::PrerenderData::PrerenderData(
+    PrerenderManager* manager,
+    PrerenderContents* contents) : manager(manager),
+                                   contents(contents),
+                                   instance_count(1) {
+}
+
+PrerenderHandle::PrerenderHandle() : weak_ptr_factory_(this) {
+}
+
+PrerenderHandle::PrerenderHandle(
+    base::WeakPtr<PrerenderData> prerender_data)
+    : prerender_data_(prerender_data),
+      weak_ptr_factory_(this) {
+}
+
+void PrerenderHandle::SwapPrerenderDataWith(
+    PrerenderHandle* other_prerender_handle) {
+  DCHECK(CalledOnValidThread());
+  std::swap(prerender_data_, other_prerender_handle->prerender_data_);
+}
+
+}  // namespace prerender