15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/prerender/prerender_handle.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <algorithm> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/logging.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/prerender/prerender_contents.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace prerender { 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)PrerenderHandle::Observer::Observer() { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)PrerenderHandle::Observer::~Observer() { 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)PrerenderHandle::~PrerenderHandle() { 21868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (prerender_data_.get()) { 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_->contents()->RemoveObserver(this); 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::SetObserver(Observer* observer) { 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_NE(static_cast<Observer*>(NULL), observer); 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) observer_ = observer; 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::OnNavigateAway() { 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 33868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (prerender_data_.get()) 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_->OnHandleNavigatedAway(this); 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::OnCancel() { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(CalledOnValidThread()); 39868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (prerender_data_.get()) 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_->OnHandleCanceled(this); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool PrerenderHandle::IsPrerendering() const { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(CalledOnValidThread()); 45868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return prerender_data_.get() != NULL; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool PrerenderHandle::IsFinishedLoading() const { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(CalledOnValidThread()); 50868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!prerender_data_.get()) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return prerender_data_->contents()->has_finished_loading(); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles)bool PrerenderHandle::Matches( 56a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) const GURL& url, 57a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) const content::SessionStorageNamespace* session_storage_namespace) const { 58a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) DCHECK(CalledOnValidThread()); 59868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!prerender_data_.get()) 60a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) return false; 61a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) return prerender_data_->contents()->Matches(url, session_storage_namespace); 62a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles)} 63a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)PrerenderHandle::PrerenderHandle( 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrerenderManager::PrerenderData* prerender_data) 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : observer_(NULL), 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) weak_ptr_factory_(this) { 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (prerender_data) { 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_ = prerender_data->AsWeakPtr(); 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data->OnHandleCreated(this); 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::AdoptPrerenderDataFrom(PrerenderHandle* other_handle) { 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(static_cast<PrerenderManager::PrerenderData*>(NULL), 76868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) prerender_data_.get()); 77868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (other_handle->prerender_data_.get() && 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) other_handle->prerender_data_->contents()) { 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) other_handle->prerender_data_->contents()->RemoveObserver(other_handle); 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_ = other_handle->prerender_data_; 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) other_handle->prerender_data_.reset(); 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 85868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (prerender_data_.get()) { 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_NE(static_cast<PrerenderContents*>(NULL), 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_->contents()); 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_data_->contents()->AddObserver(this); 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We are joining a prerender that has already started so we fire off an 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // extra start event at ourselves. 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) OnPrerenderStart(prerender_data_->contents()); 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::OnPrerenderStart(PrerenderContents* prerender_contents) { 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 97868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DCHECK(prerender_data_.get()); 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(prerender_data_->contents(), prerender_contents); 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (observer_) 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) observer_->OnPrerenderStart(this); 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::OnPrerenderStopLoading( 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) PrerenderContents* prerender_contents) { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(CalledOnValidThread()); 106868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DCHECK(prerender_data_.get()); 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(prerender_data_->contents(), prerender_contents); 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (observer_) 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) observer_->OnPrerenderStopLoading(this); 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::OnPrerenderStop(PrerenderContents* prerender_contents) { 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (observer_) 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) observer_->OnPrerenderStop(this); 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void PrerenderHandle::OnPrerenderCreatedMatchCompleteReplacement( 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) PrerenderContents* contents, PrerenderContents* replacement) { 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(CalledOnValidThread()); 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // This should occur in the middle of the surgery on the PrerenderData, and 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // so we expect to not have our new contents in our PrerenderData yet. The 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // switch occurs in 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // PrerenderManager::PrerenderData::MakeIntoMatchCompleteReplacement, so 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // this method only needs to switch observing. 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) contents->RemoveObserver(this); 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) replacement->AddObserver(this); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1327d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)bool PrerenderHandle::RepresentingSamePrerenderAs( 1337d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) PrerenderHandle* other) const { 1347d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) return other && other->prerender_data_.get() && prerender_data_.get() 1357d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) && prerender_data_.get() == other->prerender_data_.get(); 1367d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)} 1377d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace prerender 139