1// Copyright (c) 2012 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 "content/browser/download/save_item.h" 6 7#include "base/logging.h" 8#include "base/strings/string_util.h" 9#include "content/browser/download/save_file.h" 10#include "content/browser/download/save_file_manager.h" 11#include "content/browser/download/save_package.h" 12 13namespace content { 14 15// Constructor for SaveItem when creating each saving job. 16SaveItem::SaveItem(const GURL& url, 17 const Referrer& referrer, 18 SavePackage* package, 19 SaveFileCreateInfo::SaveFileSource save_source) 20 : save_id_(-1), 21 url_(url), 22 referrer_(referrer), 23 total_bytes_(0), 24 received_bytes_(0), 25 state_(WAIT_START), 26 has_final_name_(false), 27 is_success_(false), 28 save_source_(save_source), 29 package_(package) { 30 DCHECK(package); 31} 32 33SaveItem::~SaveItem() { 34} 35 36// Set start state for save item. 37void SaveItem::Start() { 38 DCHECK(state_ == WAIT_START); 39 state_ = IN_PROGRESS; 40} 41 42// If we've received more data than we were expecting (bad server info?), 43// revert to 'unknown size mode'. 44void SaveItem::UpdateSize(int64 bytes_so_far) { 45 received_bytes_ = bytes_so_far; 46 if (received_bytes_ >= total_bytes_) 47 total_bytes_ = 0; 48} 49 50// Updates from the file thread may have been posted while this saving job 51// was being canceled in the UI thread, so we'll accept them unless we're 52// complete. 53void SaveItem::Update(int64 bytes_so_far) { 54 if (state_ != IN_PROGRESS) { 55 NOTREACHED(); 56 return; 57 } 58 UpdateSize(bytes_so_far); 59} 60 61// Cancel this saving item job. If the job is not in progress, ignore 62// this command. The SavePackage will each in-progress SaveItem's cancel 63// when canceling whole saving page job. 64void SaveItem::Cancel() { 65 // If item is in WAIT_START mode, which means no request has been sent. 66 // So we need not to cancel it. 67 if (state_ != IN_PROGRESS) { 68 // Small downloads might be complete before method has a chance to run. 69 return; 70 } 71 state_ = CANCELED; 72 is_success_ = false; 73 Finish(received_bytes_, false); 74 package_->SaveCanceled(this); 75} 76 77// Set finish state for a save item 78void SaveItem::Finish(int64 size, bool is_success) { 79 // When this function is called, the SaveItem should be one of following 80 // three situations. 81 // a) The data of this SaveItem is finished saving. So it should have 82 // generated final name. 83 // b) Error happened before the start of saving process. So no |save_id_| is 84 // generated for this SaveItem and the |is_success_| should be false. 85 // c) Error happened in the start of saving process, the SaveItem has a save 86 // id, |is_success_| should be false, and the |size| should be 0. 87 DCHECK(has_final_name() || (save_id_ == -1 && !is_success_) || 88 (save_id_ != -1 && !is_success_ && !size)); 89 state_ = COMPLETE; 90 is_success_ = is_success; 91 UpdateSize(size); 92} 93 94// Calculate the percentage of the save item 95int SaveItem::PercentComplete() const { 96 switch (state_) { 97 case COMPLETE: 98 case CANCELED: 99 return 100; 100 case WAIT_START: 101 return 0; 102 case IN_PROGRESS: { 103 int percent = 0; 104 if (total_bytes_ > 0) 105 percent = static_cast<int>(received_bytes_ * 100.0 / total_bytes_); 106 return percent; 107 } 108 default: { 109 NOTREACHED(); 110 return -1; 111 } 112 } 113} 114 115// Rename the save item with new path. 116void SaveItem::Rename(const base::FilePath& full_path) { 117 DCHECK(!full_path.empty() && !has_final_name()); 118 full_path_ = full_path; 119 file_name_ = full_path_.BaseName(); 120 has_final_name_ = true; 121} 122 123void SaveItem::SetSaveId(int32 save_id) { 124 DCHECK_EQ(-1, save_id_); 125 save_id_ = save_id; 126} 127 128void SaveItem::SetTotalBytes(int64 total_bytes) { 129 DCHECK_EQ(0, total_bytes_); 130 total_bytes_ = total_bytes; 131} 132 133} // namespace content 134