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